cambridge-asm 0.8.1

// Copyright (c) 2021 Saadi Save
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#![warn(clippy::pedantic)]

use cambridge_asm::parse;
use clap::{load_yaml, App};
use std::path::PathBuf;

fn main() {
    let yaml = load_yaml!("cli.yml");
    let opts = App::from_yaml(yaml).get_matches();

    let input = opts.value_of_os("INPUT").unwrap();

    let verbosity = opts.occurrences_of("verbose");

    set_log_level(verbosity);

    #[cfg(not(debug_assertions))]
    std::panic::set_hook(Box::new(handle_panic));

    env_logger::builder()
        .format_timestamp(None)
        .format_indent(None)
        .format_module_path(false)
        .init();

    let mut x = opts.is_present("perf").then(std::time::Instant::now);

    let path = PathBuf::from(input);

    #[cfg(feature = "cambridge")]
    let mut exec = parse::from_file(&path, parse::get_fn);

    #[cfg(not(feature = "cambridge"))]
    let mut exec = parse::from_file(&path, parse::get_fn_ext);

    x.is_some().then(|| {
        println!("Total parse time: {:?}", x.unwrap().elapsed());
        x = Some(std::time::Instant::now());
    });

    (x.is_some() || verbosity > 0).then(|| println!("Execution starts on next line"));

    exec.exec();

    x.is_some()
        .then(|| println!("Execution done\nExecution time: {:?}", x.unwrap().elapsed()));
}

fn set_log_level(v: u64) {
    use std::env;
    match v {
        0 => env::set_var("RUST_LOG", "off"),
        1 => env::set_var("RUST_LOG", "warn"),
        2 => env::set_var("RUST_LOG", "info"),
        3 => env::set_var("RUST_LOG", "debug"),
        _ => env::set_var("RUST_LOG", "trace"),
    }
}

#[cfg(not(debug_assertions))]
fn handle_panic(info: &std::panic::PanicInfo) {
    if let Some(l) = info.location() {
        println!(
            "Program panicked (crashed). Panic occurred at {}:{} -",
            l.file(),
            l.line()
        );
    } else {
        println!("Program panicked (crashed). Unable to locate the source of the panic -");
    }

    if let Some(msg) = info.payload().downcast_ref::<String>() {
        println!("{}\n\nTo debug, try increasing the verbosity by passing -v flags if the error message is unclear.\nOpen an issue on github if the panic appears to be an internal error.", msg);
    }
}