mers/mers_lib/examples/01_user_scripts.rs

58 lines
2.0 KiB
Rust
Raw Normal View History

2024-02-17 13:23:50 +01:00
use std::sync::Arc;
use mers_lib::{
data::{self, Data, MersType, Type},
errors::CheckError,
prelude_compile::{parse, CompInfo, Config, Source},
};
fn main() -> Result<(), CheckError> {
let (_, func) = parse_compile_check_run(
// The `[(String -> String)]` type annotation ensures that decorate.mers returns a `String -> String` function.
"[(String -> String)] #include \"examples/decorate.mers\"".to_owned(),
)?;
// We can unwrap the downcasts because mers has type-checked that `func` is a `(String -> String)`.
let func = func.get();
let func = func
.as_any()
.downcast_ref::<data::function::Function>()
.unwrap();
// use the function to decorate these 3 test strings
for input in ["my test string", "Main Menu", "O.o"] {
let result = func.run(Data::new(data::string::String(input.to_owned())));
let result = result.get();
let result = &result
.as_any()
.downcast_ref::<data::string::String>()
.unwrap()
.0;
eprintln!("{result}");
}
Ok(())
}
fn parse_compile_check_run(src: String) -> Result<(Type, Data), CheckError> {
// prepare the string for parsing
let mut source = Source::new_from_string(src);
// this is used for error messages
let srca = Arc::new(source.clone());
// parse the code
let parsed = parse(&mut source, &srca)?;
// get infos
let (mut i1, mut i2, mut i3) = Config::new().bundle_std().infos();
// compile
let compiled = parsed.compile(&mut i1, CompInfo::default())?;
// check (this step is optional, but if it is skipped when it would have returned an error, `run` will likely panic)
let output_type = compiled.check(&mut i3, None)?;
// run
let output_value = compiled.run(&mut i2);
// check that the predicted output type was correct
assert!(output_value.get().as_type().is_included_in(&output_type));
// return the produced value
Ok((output_type, output_value))
}