diff --git a/Cargo.lock b/Cargo.lock index bb318c06..6a984ff9 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -142,6 +142,7 @@ dependencies = [ name = "kind2" version = "0.1.0" dependencies = [ + "clap", "hvm", ] diff --git a/Cargo.toml b/Cargo.toml index f0851fc5..9f8b90ff 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -7,3 +7,4 @@ edition = "2021" [dependencies] hvm = "0.1.40" +clap = { version = "3.1.8", features = ["derive"] } \ No newline at end of file diff --git a/src/cli.rs b/src/cli.rs new file mode 100644 index 00000000..1d5f5dac --- /dev/null +++ b/src/cli.rs @@ -0,0 +1,20 @@ +pub use clap::{Parser, Subcommand}; + +#[derive(Parser)] +#[clap(author, version, about, long_about = None)] +#[clap(propagate_version = true)] +pub struct Cli { + #[clap(subcommand)] + pub command: Command, +} + +#[derive(Subcommand)] +pub enum Command { + /// Run a file interpreted + #[clap(aliases = &["r"])] + Run { file: String, params: Vec }, + + /// Check a file + #[clap(aliases = &["c"])] + Check { file: String, params: Vec }, +} diff --git a/src/main.rs b/src/main.rs index bcb3f86f..b34ab417 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,27 +1,42 @@ #![allow(dead_code)] #![allow(unused_variables)] +mod cli; mod language; +use cli::{Cli, Parser, Command}; use language::{*}; use hvm::parser as parser; const CHECKER_HVM: &str = include_str!("checker.hvm"); -fn main() -> Result<(), String> { - //gen_debug_file(); +fn main() { + match run_cli() { + Ok(..) => {} + Err(err) => { + eprintln!("{}", err); + } + }; - let args: Vec = std::env::args().collect(); +// ------------------------------------------------------------ +// ------------------------------------------------------------ +// ------------------------------------------------------------ - if args.len() <= 2 || args[1] != "check" && args[1] != "run" { - println!("Usage:"); - println!("$ kind2 check file.kind"); - println!("$ kind2 run file.kind"); - return Ok(()); +fn run_cli() -> Result<(), String> { + let cli_matches = Cli::parse(); + + match cli_matches.command { + Command::Run { file: path, params } => { + kind2(&path, "Api.run_main") + } + + Command::Check { file: path, params } => { + kind2(&path, "Api.check_all") + } } +} - let path = &args[2]; - +fn kind2(path: &str, main_function: &str) -> Result<(), String> { // Reads definitions from file let file = match std::fs::read_to_string(path) { Ok(code) => read_file(&code), @@ -47,7 +62,6 @@ fn main() -> Result<(), String> { //println!("[{}]\n{}\n", name, show_entry(entry)); //} - // let code = compile_file(&file); let mut checker = (&CHECKER_HVM[0 .. CHECKER_HVM.find("////INJECT////").unwrap()]).to_string(); checker.push_str(&code); @@ -57,14 +71,14 @@ fn main() -> Result<(), String> { // Runs with the interpreter let mut rt = hvm::Runtime::from_code(&checker)?; - let main = rt.alloc_code(if args[1] == "check" { "Api.check_all" } else { "Api.run_main" })?; + let main = rt.alloc_code(main_function)?; rt.normalize(main); println!("{}", readback_string(&rt, main)); // TODO: optimize by deserializing term into text directly // Display stats println!("Rewrites: {}", rt.get_rewrites()); - return Ok(()); + Ok(()) } // ------------------------------------------------------------