cs420/bin/kecc.rs

#[macro_use]
extern crate clap;
use clap::{crate_authors, crate_description, crate_version, App};

#[macro_use]
extern crate kecc;

use kecc::{Codegen, Irgen, Optimize, Parse, Translate, O1};

fn main() {
    let yaml = load_yaml!("kecc_cli.yml");
    #[allow(deprecated)]
    let matches = App::from_yaml(yaml)
        .version(crate_version!())
        .about(crate_description!())
        .author(crate_authors!(", "))
        .get_matches();

    let input = matches.value_of("INPUT").unwrap();
    let unit = ok_or_exit!(Parse::default().translate(&input), 1);

    let output = matches.value_of("output").unwrap_or_else(|| "-");
    let mut output: Box<dyn ::std::io::Write> = if output == "-" {
        Box::new(::std::io::stdout())
    } else {
        Box::new(ok_or_exit!(::std::fs::File::open(output), 1))
    };

    if matches.is_present("print") {
        kecc::write_c(&unit, &mut output).unwrap();
        return;
    }

    let mut ir = match Irgen::default().translate(&unit) {
        Ok(ir) => ir,
        Err(irgen_error) => {
            println!("{}", irgen_error);
            return;
        }
    };
    if matches.is_present("irgen") {
        kecc::write_ir(&ir, &mut output).unwrap();
        return;
    }

    if matches.is_present("optimize") {
        O1::default().optimize(&mut ir);
    }

    let asm = ok_or_exit!(Codegen::default().translate(&ir), 1);
    kecc::write_asm(&asm, &mut output);
}