schala/schala-lang/language/src/lib.rs

172 lines
5.9 KiB
Rust
Raw Normal View History

2018-07-16 03:40:35 -07:00
#![feature(trace_macros)]
#![feature(custom_attribute)]
#![feature(unrestricted_attribute_tokens)]
2018-04-03 23:24:13 -07:00
#![feature(slice_patterns, box_patterns, box_syntax)]
2018-11-11 18:04:44 -08:00
//! `schala-lang` is where the Schala programming language is actually implemented.
//! It defines the `Schala` type, which contains the state for a Schala REPL, and implements
//! `ProgrammingLanguageInterface` and the chain of compiler passes for it.
2018-03-23 18:43:43 -07:00
extern crate itertools;
#[macro_use]
extern crate lazy_static;
#[macro_use]
extern crate maplit;
extern crate schala_repl;
2018-05-02 02:14:36 -07:00
#[macro_use]
extern crate schala_repl_codegen;
#[macro_use]
2018-10-19 09:57:35 -07:00
extern crate schala_lang_codegen;
2018-03-23 18:43:43 -07:00
use std::cell::RefCell;
use std::rc::Rc;
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;
2018-06-04 19:25:40 -07:00
mod ast;
mod parsing;
mod symbol_table;
2018-02-21 02:31:28 -08:00
mod typechecking;
2018-06-04 19:12:48 -07:00
mod reduced_ast;
mod eval;
2018-07-16 19:36:55 -07:00
//trace_macros!(true);
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-07-16 19:38:13 -07:00
#[PipelineSteps(tokenizing, parsing(compact,expanded,trace), symbol_table, typechecking, ast_reducing, eval)]
#[DocMethod = get_doc]
#[HandleCustomInterpreterDirectives = handle_custom_interpreter_directives]
2018-11-11 18:04:44 -08:00
/// All bits of state necessary to parse and execute a Schala program are stored in this struct
/// `state` represents the execution state for the AST-walking interpreter, the other fields
/// should be self-explanatory.
pub struct Schala {
state: eval::State<'static>,
symbol_table: Rc<RefCell<symbol_table::SymbolTable>>,
2018-11-08 20:30:17 -08:00
type_context: typechecking::TypeContext<'static>,
2018-10-20 14:27:00 -07:00
active_parser: Option<parsing::Parser>,
}
2018-09-22 00:24:27 -07:00
impl Schala {
fn get_doc(&self, commands: &Vec<&str>) -> Option<String> {
Some(format!("Documentation on commands: {:?}", commands))
2018-09-22 00:24:27 -07:00
}
fn handle_custom_interpreter_directives(&mut self, commands: &Vec<&str>) -> Option<String> {
2018-10-20 00:55:37 -07:00
Some(format!("Schala-lang command: {:?} not supported", commands.get(0)))
}
2018-09-22 00:24:27 -07:00
}
2018-07-16 03:40:35 -07:00
impl Schala {
2018-11-11 18:04:44 -08:00
/// Creates a new Schala environment *without* any prelude.
2018-06-03 18:43:31 -07:00
fn new_blank_env() -> Schala {
let symbols = Rc::new(RefCell::new(symbol_table::SymbolTable::new()));
Schala {
symbol_table: symbols.clone(),
state: eval::State::new(symbols),
2018-11-06 13:44:52 -08:00
type_context: typechecking::TypeContext::new(),
2018-10-20 14:27:00 -07:00
active_parser: None,
}
}
2018-06-03 18:43:31 -07:00
2018-11-11 18:04:44 -08:00
/// Creates a new Schala environment with the standard prelude, which is defined as ordinary
/// Schala code in the file `prelude.schala`
2018-06-03 18:43:31 -07:00
pub fn new() -> Schala {
2018-11-05 20:55:03 -08:00
let prelude = include_str!("prelude.schala");
2018-06-03 18:43:31 -07:00
let mut s = Schala::new_blank_env();
s.execute_pipeline(prelude, &EvalOptions::default());
s
}
}
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-10-20 14:27:00 -07:00
fn parsing(handle: &mut Schala, input: Vec<tokenizing::Token>, comp: Option<&mut UnfinishedComputation>) -> Result<ast::AST, String> {
use parsing::Parser;
let mut parser = match handle.active_parser.take() {
None => Parser::new(input),
Some(parser) => parser
};
let ast = parser.parse();
2018-10-20 14:27:00 -07:00
let trace = parser.format_parse_trace();
comp.map(|comp| {
//TODO need to control which of these debug stages get added
2018-07-04 19:46:36 -07:00
let opt = comp.cur_debug_options.get(0).map(|s| s.clone());
match opt {
None => comp.add_artifact(TraceArtifact::new("ast", format!("{:?}", ast))),
Some(ref s) if s == "compact" => comp.add_artifact(TraceArtifact::new("ast", format!("{:?}", ast))),
Some(ref s) if s == "expanded" => comp.add_artifact(TraceArtifact::new("ast", format!("{:#?}", ast))),
Some(ref s) if s == "trace" => comp.add_artifact(TraceArtifact::new_parse_trace(trace)),
2018-07-06 03:17:37 -07:00
Some(ref x) => println!("Bad parsing debug option: {}", x),
2018-07-04 19:46:36 -07:00
};
});
2018-05-13 13:40:11 -07:00
ast.map_err(|err| err.msg)
}
2018-06-04 19:25:40 -07:00
fn symbol_table(handle: &mut Schala, input: ast::AST, comp: Option<&mut UnfinishedComputation>) -> Result<ast::AST, String> {
let add = handle.symbol_table.borrow_mut().add_top_level_symbols(&input);
match add {
2018-04-29 22:51:01 -07:00
Ok(()) => {
let artifact = TraceArtifact::new("symbol_table", handle.symbol_table.borrow().debug_symbol_table());
2018-05-05 12:14:19 -07:00
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-11-07 13:44:28 -08:00
fn typechecking(handle: &mut Schala, input: ast::AST, comp: Option<&mut UnfinishedComputation>) -> Result<ast::AST, String> {
2018-11-09 02:02:08 -08:00
let result = handle.type_context.typecheck(&input);
2018-11-06 13:44:52 -08:00
2018-11-09 02:02:08 -08:00
comp.map(|comp| {
let artifact = TraceArtifact::new("type", format!("{:?}", result));
comp.add_artifact(artifact);
});
2018-11-07 13:44:28 -08:00
2018-11-09 02:02:08 -08:00
Ok(input)
2018-04-29 03:45:31 -07:00
}
fn ast_reducing(handle: &mut Schala, input: ast::AST, comp: Option<&mut UnfinishedComputation>) -> Result<reduced_ast::ReducedAST, String> {
let ref symbol_table = handle.symbol_table.borrow();
let output = input.reduce(symbol_table);
2018-05-11 02:58:14 -07:00
comp.map(|comp| comp.add_artifact(TraceArtifact::new("ast_reducing", format!("{:?}", output))));
2018-05-12 02:20:50 -07:00
Ok(output)
2018-05-09 02:02:17 -07:00
}
2018-06-04 19:12:48 -07:00
fn eval(handle: &mut Schala, input: reduced_ast::ReducedAST, comp: Option<&mut UnfinishedComputation>) -> Result<String, String> {
2018-05-11 02:33:19 -07:00
comp.map(|comp| comp.add_artifact(TraceArtifact::new("value_state", handle.state.debug_print())));
2018-05-12 02:20:50 -07:00
let evaluation_outputs = handle.state.evaluate(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