rhasm 0.2.0

A Rust implementation of the hasm assembler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use std::{ borrow::BorrowMut, fs::File, io::{ self, Write as _ }, path::PathBuf };
use rhasm::{ Assembler, Disassembler };
use clap::{ Parser, ArgAction };

#[derive(Parser, Debug)]
#[command(
    name = "rhasm",
    version = "0.1.1",
    about = "A simple assembler/disassembler for the Hack computer from the Nand2Tetris course",
    author = "Muaaz Bhyat muu794@gmail.com"
)]
struct Cli {
    /// The input file to read from
    /// Is required and does not have an option switch
    #[arg(required = true)]
    in_file_path: PathBuf,

    /// The output file to write
    /// Can be specified with the -o or --output option
    #[arg(short, long)]
    output: Option<PathBuf>,

    /// Disassemble the input file
    #[arg(short, long, action = ArgAction::SetTrue)]
    disassemble: bool,
}

fn main() -> io::Result<()> {
    let args = Cli::parse();

    let disassemble = args.disassemble;
    let in_file_path = args.in_file_path;
    let out_file_path = match args.output.as_ref() {
        Some(filename) => filename.clone(),
        None => {
            let mut out_file = in_file_path.clone();
            match disassemble {
                true => {
                    out_file.set_extension("asm");
                }
                false => {
                    out_file.set_extension("hack");
                }
            }
            out_file
        }
    };

    let mut in_file = std::fs::File::open(&in_file_path)?;

    let out_file_create_result = std::fs::File::create_new(&out_file_path);
    let mut out_file = out_file_create_result.unwrap_or_else(|_| {
        eprint!(
            "Could not create output file, file {} already exists
            Would you like to overwrite the file? (y/n)",
            out_file_path.display()
        );
        io::stdout().flush().unwrap();
        let mut input = String::new();
        io::stdin().read_line(&mut input).unwrap();
        if input.trim().to_lowercase() == "y" {
            std::fs::File::create(&out_file_path).unwrap()
        } else {
            std::process::exit(1);
        }
    });

    let reader = &mut in_file;
    let writer = Some(out_file.borrow_mut());
    let mut label_file = File::create_new(in_file_path.with_extension("labels")).unwrap();
    let label_table = Some(&mut label_file);

    if disassemble {
        let args = rhasm::DisassemblerConfig {
            reader,
            writer,
        };
        let mut disassembler = Disassembler::new(args);
        disassembler.write_to_end()?;
        
    } else {
        let assembler = Assembler::build(&mut in_file, &mut out_file, label_table);
        assembler.unwrap().advance_to_end();
    }
    Ok(())
}