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
71
72
73
74
75
76
|
use std::{env, fs, path::PathBuf, process};
use rustyline::{error::ReadlineError, Editor};
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");
match tvix_eval::interpret(&contents) {
Ok(result) => println!("=> {} :: {}", result, result.type_of()),
Err(err) => eprintln!("{}", err),
}
}
fn state_dir() -> Option<PathBuf> {
let mut path = dirs::data_dir();
path.as_mut().map(|p| p.push("tvix"));
path
}
fn run_prompt() {
let mut rl = Editor::<()>::new().expect("should be able to launch rustyline");
let history_path = match state_dir() {
Some(mut path) => {
path.push("history.txt");
rl.load_history(&path).ok();
Some(path)
}
None => None,
};
loop {
let readline = rl.readline("tvix-repl> ");
match readline {
Ok(line) => {
if line.is_empty() {
continue;
}
match tvix_eval::interpret(&line) {
Ok(result) => {
println!("=> {} :: {}", result, result.type_of());
rl.add_history_entry(line);
}
Err(err) => println!("{}", err),
}
}
Err(ReadlineError::Interrupted) | Err(ReadlineError::Eof) => break,
Err(err) => {
eprintln!("error: {}", err);
break;
}
}
}
if let Some(path) = history_path {
rl.save_history(&path).unwrap();
}
}
|