blob: c9cc96d2e6a995836d3a04b636c08149fd6e7ecc (
plain) (
blame)
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
|
use std::env;
use std::fs;
use std::io;
use std::io::Write;
use std::process;
mod bytecode;
mod treewalk;
/// Trait for making the different interpreters callable in the same
/// way.
pub trait Lox {
type Value;
type Error: std::fmt::Display;
fn create() -> Self;
fn interpret(&mut self, source: String) -> Result<Self::Value, Vec<Self::Error>>;
}
fn main() {
let mut args = env::args();
if args.len() > 2 {
println!("Usage: rlox [script]");
process::exit(1);
}
match env::var("LOX_INTERPRETER").as_ref().map(String::as_str) {
Ok("treewalk") => pick::<treewalk::interpreter::Interpreter>(args.nth(1)),
_ => pick::<bytecode::Interpreter>(args.nth(1)),
}
}
fn pick<I: Lox>(file_arg: Option<String>) {
if let Some(file) = file_arg {
run_file::<I>(&file);
} else {
run_prompt::<I>();
}
}
// Run Lox code from a file and print results to stdout
fn run_file<I: Lox>(file: &str) {
let contents = fs::read_to_string(file).expect("failed to read the input file");
let mut lox = I::create();
run(&mut lox, contents);
}
// Evaluate Lox code interactively in a shitty REPL.
fn run_prompt<I: Lox>() {
let mut line = String::new();
let mut lox = I::create();
loop {
print!("> ");
io::stdout().flush().unwrap();
io::stdin()
.read_line(&mut line)
.expect("failed to read user input");
run(&mut lox, std::mem::take(&mut line));
line.clear();
}
}
fn run<I: Lox>(lox: &mut I, code: String) {
if let Err(errors) = lox.interpret(code) {
for error in errors {
eprintln!("{}", error);
}
}
}
|
