tvix/eval/src/main.rs


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

use std::{
    env, fs,
    io::{self, Write},
    mem, process,
};

mod chunk;
mod errors;
mod eval;
mod opcode;
mod value;

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),
    }
}