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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
use std::{cell::RefCell, path::PathBuf, rc::Rc};
use crate::{
builtins::global_builtins,
errors::{Error, ErrorKind, EvalResult},
observer::{DisassemblingObserver, NoOpObserver, TracingObserver},
value::Value,
SourceCode,
};
/// Runtime options for the Tvix interpreter
#[derive(Debug, Clone, Copy, Default)]
#[cfg_attr(feature = "repl", derive(clap::Parser))]
pub struct Options {
/// Dump the raw AST to stdout before interpreting
#[cfg_attr(feature = "repl", clap(long, env = "TVIX_DISPLAY_AST"))]
display_ast: bool,
/// Dump the bytecode to stdout before evaluating
#[cfg_attr(feature = "repl", clap(long, env = "TVIX_DUMP_BYTECODE"))]
dump_bytecode: bool,
/// Trace the runtime of the VM
#[cfg_attr(feature = "repl", clap(long, env = "TVIX_TRACE_RUNTIME"))]
trace_runtime: bool,
}
pub fn interpret(code: &str, location: Option<PathBuf>, options: Options) -> EvalResult<Value> {
let source = SourceCode::new();
let file = source.add_file(
location
.as_ref()
.map(|p| p.to_string_lossy().to_string())
.unwrap_or_else(|| "[tvix-repl]".into()),
code.into(),
);
let parsed = rnix::ast::Root::parse(code);
let errors = parsed.errors();
if !errors.is_empty() {
let err = Error {
kind: ErrorKind::ParseErrors(errors.to_vec()),
span: file.span,
};
err.fancy_format_stderr(&source);
return Err(err);
}
// If we've reached this point, there are no errors.
let root_expr = parsed
.tree()
.expr()
.expect("expression should exist if no errors occured");
if options.display_ast {
println!("{:?}", root_expr);
}
// TODO: encapsulate this import weirdness in builtins
let builtins = Rc::new(RefCell::new(global_builtins()));
#[cfg(feature = "impure")]
{
// We need to insert import into the builtins, but the
// builtins passed to import must have import *in it*.
let import = Value::Builtin(crate::builtins::impure::builtins_import(
builtins.clone(),
source.clone(),
));
builtins.borrow_mut().insert("import", import);
// TODO: also add it into the inner builtins set
};
let result = if options.dump_bytecode {
crate::compiler::compile(
&root_expr,
location,
file.clone(),
builtins,
&mut DisassemblingObserver::new(source.clone(), std::io::stderr()),
)
} else {
crate::compiler::compile(
&root_expr,
location,
file.clone(),
builtins,
&mut NoOpObserver::default(),
)
}?;
for warning in result.warnings {
warning.fancy_format_stderr(&source);
}
for error in &result.errors {
error.fancy_format_stderr(&source);
}
if let Some(err) = result.errors.last() {
return Err(err.clone());
}
let result = if options.trace_runtime {
crate::vm::run_lambda(&mut TracingObserver::new(std::io::stderr()), result.lambda)
} else {
crate::vm::run_lambda(&mut NoOpObserver::default(), result.lambda)
};
if let Err(err) = &result {
err.fancy_format_stderr(&source);
}
result.map(|r| {
for warning in r.warnings {
warning.fancy_format_stderr(&source);
}
r.value
})
}
|
