diff options
Diffstat (limited to 'users/tazjin/rlox/src/main.rs')
-rw-r--r-- | users/tazjin/rlox/src/main.rs | 48 |
1 files changed, 4 insertions, 44 deletions
diff --git a/users/tazjin/rlox/src/main.rs b/users/tazjin/rlox/src/main.rs index 1c1dd6f42ff6..13a5748187ad 100644 --- a/users/tazjin/rlox/src/main.rs +++ b/users/tazjin/rlox/src/main.rs @@ -4,56 +4,16 @@ use std::io; use std::io::Write; use std::process; +mod bytecode; mod errors; mod parser; mod scanner; mod treewalk; fn main() { - let mut args = env::args(); - - if args.len() > 2 { - println!("Usage: rlox [script]"); - process::exit(1); - } else if let Some(file) = args.nth(1) { - run_file(&file); - } else { - run_prompt(); - } -} - -// Run Lox code from a file and print results to stdout -fn run_file(file: &str) { - let contents = fs::read_to_string(file).expect("failed to read the input file"); - let mut lox = treewalk::interpreter::Interpreter::create(); - run(&mut lox, &contents); -} - -// Evaluate Lox code interactively in a shitty REPL. -fn run_prompt() { - let mut line = String::new(); - let mut lox = treewalk::interpreter::Interpreter::create(); - - loop { - print!("> "); - io::stdout().flush().unwrap(); - io::stdin() - .read_line(&mut line) - .expect("failed to read user input"); - run(&mut lox, &line); - line.clear(); - } -} - -fn run(lox: &mut treewalk::interpreter::Interpreter, code: &str) { - let chars: Vec<char> = code.chars().collect(); - - let result = scanner::scan(&chars) - .and_then(|tokens| parser::parse(tokens)) - .and_then(|program| lox.interpret(program).map_err(|e| vec![e])); - - if let Err(errors) = result { - report_errors(errors); + match env::var("LOX_INTERPRETER").as_ref().map(String::as_str) { + Ok("treewalk") => treewalk::main(), + _ => bytecode::main(), } } |