1use super::*;
2
3impl Interpreter {
4 pub fn exec_items(&mut self, items: &[TopLevel]) -> Result<Value, RuntimeError> {
5 for item in items {
6 match item {
7 TopLevel::FnDef(fd) => self.exec_fn_def(fd)?,
8 TopLevel::Module(_) => {}
9 TopLevel::Verify(_) => {}
10 TopLevel::Decision(_) => {}
11 TopLevel::TypeDef(td) => self.register_type_def(td),
12 TopLevel::EffectSet { .. } => {}
13 TopLevel::Stmt(s) => {
14 self.exec_stmt(s)?;
15 }
16 }
17 }
18 Ok(Value::Unit)
19 }
20
21 pub fn register_type_def(&mut self, td: &TypeDef) {
23 match td {
24 TypeDef::Sum {
25 name: type_name,
26 variants,
27 ..
28 } => {
29 let mut members = HashMap::new();
30 for variant in variants {
31 if variant.fields.is_empty() {
32 members.insert(
34 variant.name.clone(),
35 Value::Variant {
36 type_name: type_name.clone(),
37 variant: variant.name.clone(),
38 fields: vec![],
39 },
40 );
41 } else {
42 members.insert(
44 variant.name.clone(),
45 Value::Builtin(format!("__ctor:{}:{}", type_name, variant.name)),
46 );
47 }
48 }
49 self.define(
50 type_name.clone(),
51 Value::Namespace {
52 name: type_name.clone(),
53 members,
54 },
55 );
56 }
57 TypeDef::Product { name, fields, .. } => {
58 let schema = fields.iter().map(|(field, _)| field.clone()).collect();
62 self.record_schemas.insert(name.clone(), schema);
63 }
64 }
65 }
66
67 pub fn exec_fn_def(&mut self, fd: &FnDef) -> Result<(), RuntimeError> {
68 let val = Value::Fn {
69 name: fd.name.clone(),
70 params: fd.params.clone(),
71 return_type: fd.return_type.clone(),
72 effects: fd.effects.clone(),
73 body: Rc::clone(&fd.body),
74 resolution: fd.resolution.clone(),
75 memo_eligible: self.memo_fns.contains(&fd.name),
76 home_globals: None,
77 };
78 self.define(fd.name.clone(), val);
79 Ok(())
80 }
81
82 pub fn exec_stmt(&mut self, stmt: &Stmt) -> Result<Value, RuntimeError> {
83 match stmt {
84 Stmt::Binding(name, _, expr) => {
85 let val = self.eval_expr(expr)?;
86 self.define(name.clone(), val);
87 Ok(Value::Unit)
88 }
89 Stmt::Expr(expr) => self.eval_expr(expr),
90 }
91 }
92
93 pub fn exec_body(&mut self, stmts: &[Stmt]) -> Result<Value, RuntimeError> {
94 let mut last = Value::Unit;
95 for stmt in stmts {
96 last = self.exec_stmt(stmt)?;
97 }
98 Ok(last)
99 }
100
101 pub fn run_file(&mut self, source: &str) -> Result<Value, RuntimeError> {
102 let mut items = parse_source(source).map_err(RuntimeError::Error)?;
103 crate::resolver::resolve_program(&mut items);
104
105 for item in &items {
107 match item {
108 TopLevel::FnDef(fd) => self.exec_fn_def(fd)?,
109 TopLevel::Stmt(s) => {
110 self.exec_stmt(s)?;
111 }
112 _ => {}
113 }
114 }
115
116 let main_fn = self.lookup("main");
118 if let Ok(fn_val) = main_fn {
119 let allowed = Self::callable_declared_effects(&fn_val);
120 self.call_value_with_effects_pub(fn_val, vec![], "<main>", allowed)?;
121 }
122
123 Ok(Value::Unit)
124 }
125}