[driver,filer] implement bitcode file writing and dump REPL command

Author: jauhien

Cargo.lock

@@ -111,6 +111,28 @@ pub fn main_loop(stage: Stage) {
                 Err(err) => print!("Error occured during loading: {}\n", err)
             };
             continue;
+        } else if &input[0..5] == ".dump" {
+            let mut path = input[6..].to_string();
+            match path.pop() {
+                Some(_) => (),
+                None => {
+                    print!("Error occured during dumping: empty path\n");
+                    continue;
+                }
+            };
+            match filer::dump_bitcode(&path, ir_container.get_module_provider()) {
+                Ok(_) => (),
+                Err(_) => print!("Error occured during dumping\n")
+            };
+            continue;
+        } else if input.as_str() == ".help\n" {
+            print!("Enter Kaleidoscope expressions or special commands.\n");
+            print!("Special commands are:\n");
+            print!(".quit -- quit\n");
+            print!(".load <path> -- load .ks file\n");
+            print!(".dump <path> -- dump bitcode of currently open module\n");
+            print!(".help -- show this help message\n");
+            continue;
 //< ch-0 ch-1 ch-2 ch-3 parser-driver
         }
 

src/driver.rs

@@ -8,6 +8,8 @@ use std::path::Path;
 
 use llvm_sys::prelude::LLVMValueRef;
 
+use iron_llvm::bitcode::write_bitcode_to_file;
+
 use builder;
 use builder::IRBuilder;
 use lexer;
@@ -85,3 +87,7 @@ pub fn load_stdlib(parser_settings: &mut parser::ParserSettings,
 
     Ok(try!(load_ks(path, parser_settings, context, module_provider)))
 }
+
+pub fn dump_bitcode(path: &str, module_provider: &mut builder::ModuleProvider) -> Result<(), ()> {
+    write_bitcode_to_file(module_provider.get_module(), path)
+}