This creates a REPL which outputs compiled bytecode, constants, and VM results for code snippets. Change-Id: If63f79a961456afd6a4cdf59b994107ff7ab8b47 Reviewed-on: https://cl.tvl.fyi/c/depot/+/6072 Tested-by: BuildkiteCI Reviewed-by: grfn <grfn@gws.fyi> Autosubmit: tazjin <tazjin@tvl.su>
54 lines
988 B
Rust
54 lines
988 B
Rust
use std::{
|
|
env, fs,
|
|
io::{self, Write},
|
|
mem, process,
|
|
};
|
|
|
|
mod chunk;
|
|
mod compiler;
|
|
mod errors;
|
|
mod eval;
|
|
mod opcode;
|
|
mod value;
|
|
mod vm;
|
|
|
|
fn main() {
|
|
let mut args = env::args();
|
|
if args.len() > 2 {
|
|
println!("Usage: tvix-eval [script]");
|
|
process::exit(1);
|
|
}
|
|
|
|
if let Some(file) = args.nth(1) {
|
|
run_file(&file);
|
|
} else {
|
|
run_prompt();
|
|
}
|
|
}
|
|
|
|
fn run_file(file: &str) {
|
|
let contents = fs::read_to_string(file).expect("failed to read the input file");
|
|
|
|
run(contents);
|
|
}
|
|
|
|
fn run_prompt() {
|
|
let mut line = String::new();
|
|
|
|
loop {
|
|
print!("> ");
|
|
io::stdout().flush().unwrap();
|
|
io::stdin()
|
|
.read_line(&mut line)
|
|
.expect("failed to read user input");
|
|
run(mem::take(&mut line));
|
|
line.clear();
|
|
}
|
|
}
|
|
|
|
fn run(code: String) {
|
|
match eval::interpret(code) {
|
|
Ok(result) => println!("=> {}", result),
|
|
Err(err) => eprintln!("{}", err),
|
|
}
|
|
}
|