write generated assembly to file

diff --git a/rel-lang/relc/src/main.rs b/rel-lang/relc/src/main.rs
index 0c6f5f004cd2e9bda30c770edf7762df0c9d9739..3171c43d3c679cd3bd06f0512a6f025e88abe9b3 100644 (file)
--- a/rel-lang/relc/src/main.rs
+++ b/rel-lang/relc/src/main.rs
@@ -7,12 +7,19 @@ use rel_interpret::Interpreter;
 
 use anyhow::Result;
 
+use std::fs::File;
+use std::io::Write;
+use std::path::PathBuf;
+
 fn main() -> Result<()> {
     let filename = std::env::args().nth(1).expect("must provide a file to compile");
     let module = ir::compile(&filename)?;
     // let mut interp = Interpreter::new(&module)?;
     // interp.run()?;
     let mut generator = riscv_asm_codegen::CodeGen::new(&module, 4096);
-    println!("{}", generator.assembly()?);
+    let mut asm_path = PathBuf::from(filename);
+    asm_path.set_extension("asm");
+    let mut output = File::create(asm_path)?;
+    write!(output, "{}\n", generator.assembly()?)?;
     Ok(())
 }