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
|
use rebel_lang::{
func::{Func, FuncDef, FuncType},
scope::Context,
typing::{ArrayLen, Type, TypeFamily},
value::{EvalError, Result, Value},
};
use rebel_parse::{recipe, tokenize};
fn intrinsic_array_len(params: &[Value]) -> Result<Value> {
assert!(params.len() == 1);
let Value::Array(array) = ¶ms[0] else {
panic!();
};
Ok(Value::Integer(array.len().try_into().or(Err(EvalError))?))
}
fn intrinsic_string_len(params: &[Value]) -> Result<Value> {
assert!(params.len() == 1);
let Value::Str(string) = ¶ms[0] else {
panic!();
};
Ok(Value::Integer(
string.chars().count().try_into().or(Err(EvalError))?,
))
}
fn main() -> rustyline::Result<()> {
let mut ctx = Context::default();
ctx.methods.entry(TypeFamily::Array).or_default().insert(
"len",
Func {
typ: FuncType {
params: vec![Type::Array(Box::new(Type::Free), ArrayLen::Dynamic)],
ret: Type::Int,
},
def: Some(FuncDef::Intrinsic(intrinsic_array_len)),
},
);
ctx.methods.entry(TypeFamily::Str).or_default().insert(
"len",
Func {
typ: FuncType {
params: vec![Type::Str],
ret: Type::Int,
},
def: Some(FuncDef::Intrinsic(intrinsic_string_len)),
},
);
let mut rl = rustyline::DefaultEditor::new()?;
while let Ok(line) = rl.readline("> ") {
rl.add_history_entry(line.as_str())?;
let tokens = match tokenize::token_stream(&line) {
Ok(value) => value,
Err(err) => {
println!("Tokenize error: {err}");
continue;
}
};
let expr = match recipe::expr(&tokens) {
Ok(value) => value,
Err(err) => {
println!("Parse error: {err}");
continue;
}
};
let value = match Value::eval(&ctx, &expr) {
Ok(value) => value,
Err(err) => {
println!("Eval error: {err:?}");
continue;
}
};
println!("{value}");
}
Ok(())
}
|