schala/schala-lang/src/lib.rs

115 lines
3.5 KiB
Rust
Raw Normal View History

2018-04-03 23:24:13 -07:00
#![feature(slice_patterns, box_patterns, box_syntax)]
#![feature(proc_macro)]
2018-03-23 18:43:43 -07:00
extern crate itertools;
#[macro_use]
extern crate lazy_static;
#[macro_use]
extern crate maplit;
#[macro_use]
2018-03-23 18:43:43 -07:00
extern crate schala_repl;
2018-05-02 02:14:36 -07:00
#[macro_use]
extern crate schala_codegen;
2018-03-23 18:43:43 -07:00
use itertools::Itertools;
2018-03-20 21:17:46 -07:00
use schala_repl::{ProgrammingLanguageInterface, EvalOptions, TraceArtifact, UnfinishedComputation, FinishedComputation};
macro_rules! bx {
($e:expr) => { Box::new($e) }
}
mod util;
mod builtin;
mod tokenizing;
mod parsing;
2018-02-21 02:31:28 -08:00
mod typechecking;
2018-05-09 02:02:17 -07:00
mod ast_reducing;
mod eval;
2018-05-02 02:14:36 -07:00
#[derive(ProgrammingLanguageInterface)]
2018-05-02 03:53:38 -07:00
#[LanguageName = "Schala"]
2018-05-02 20:43:05 -07:00
#[SourceFileExtension = "schala"]
2018-05-09 02:02:17 -07:00
#[PipelineSteps(tokenizing, parsing, symbol_table, typechecking, ast_reducing, eval)]
pub struct Schala {
state: eval::State<'static>,
2018-05-05 02:06:08 -07:00
type_context: typechecking::TypeContext
}
impl Schala {
pub fn new() -> Schala {
Schala {
state: eval::State::new(),
2018-05-05 02:06:08 -07:00
type_context: typechecking::TypeContext::new(),
}
}
}
2018-05-05 02:06:08 -07:00
fn tokenizing(_handle: &mut Schala, input: &str, comp: Option<&mut UnfinishedComputation>) -> Result<Vec<tokenizing::Token>, String> {
let tokens = tokenizing::tokenize(input);
comp.map(|comp| {
let token_string = tokens.iter().map(|t| format!("{:?}<L:{},C:{}>", t.token_type, t.offset.0, t.offset.1)).join(", ");
comp.add_artifact(TraceArtifact::new("tokens", token_string));
});
2018-05-02 00:27:58 -07:00
2018-05-02 01:14:46 -07:00
let errors: Vec<String> = tokens.iter().filter_map(|t| t.get_error()).collect();
if errors.len() == 0 {
Ok(tokens)
} else {
Err(format!("{:?}", errors))
}
}
2018-05-05 02:06:08 -07:00
fn parsing(_handle: &mut Schala, input: Vec<tokenizing::Token>, comp: Option<&mut UnfinishedComputation>) -> Result<parsing::AST, parsing::ParseError> {
let (ast, trace) = parsing::parse(input);
comp.map(|comp| {
//TODO need to control which of these debug stages get added
comp.add_artifact(TraceArtifact::new_parse_trace(trace));
comp.add_artifact(TraceArtifact::new("ast", format!("{:#?}", ast)));
});
ast
}
2018-05-05 02:06:08 -07:00
fn symbol_table(handle: &mut Schala, input: parsing::AST, comp: Option<&mut UnfinishedComputation>) -> Result<parsing::AST, String> {
2018-04-29 03:45:31 -07:00
match handle.type_context.add_top_level_types(&input) {
2018-04-29 22:51:01 -07:00
Ok(()) => {
2018-05-05 12:14:19 -07:00
let artifact = TraceArtifact::new("symbol_table", handle.type_context.debug_symbol_table());
comp.map(|comp| comp.add_artifact(artifact));
2018-04-29 22:51:01 -07:00
Ok(input)
},
2018-04-29 03:45:31 -07:00
Err(msg) => Err(msg)
}
}
2018-05-05 02:06:08 -07:00
fn typechecking(handle: &mut Schala, input: parsing::AST, comp: Option<&mut UnfinishedComputation>) -> Result<parsing::AST, String> {
2018-04-29 03:45:31 -07:00
match handle.type_context.type_check_ast(&input) {
Ok(ty) => {
2018-04-29 22:51:01 -07:00
comp.map(|comp| comp.add_artifact(TraceArtifact::new("type_check", format!("{:?}", ty))));
2018-04-29 03:45:31 -07:00
Ok(input)
},
Err(msg) => {
2018-05-05 12:14:19 -07:00
comp.map(|comp| comp.add_artifact(TraceArtifact::new("type_check", format!("Type error: {:?}", msg))));
Ok(input)
}
2018-04-29 03:45:31 -07:00
}
}
2018-05-09 02:02:17 -07:00
type TempASTReduction = (ast_reducing::ReducedAST, parsing::AST);
fn ast_reducing(handle: &mut Schala, input: parsing::AST, comp: Option<&mut UnfinishedComputation>) -> Result<TempASTReduction, String> {
2018-05-11 00:45:32 -07:00
let output = input.reduce();
2018-05-09 02:27:57 -07:00
Ok((output, input))
2018-05-09 02:02:17 -07:00
}
fn eval(handle: &mut Schala, input: TempASTReduction, _comp: Option<&mut UnfinishedComputation>) -> Result<String, String> {
2018-05-09 02:27:57 -07:00
let new_input = input.0;
let evaluation_outputs = handle.state.evaluate_new(new_input, true);
2018-04-29 03:45:31 -07:00
let text_output: Result<Vec<String>, String> = evaluation_outputs
.into_iter()
.collect();
let eval_output: Result<String, String> = text_output
.map(|v| { v.into_iter().intersperse(format!("\n")).collect() });
eval_output
}
2018-05-09 02:02:17 -07:00