schala/src/main.rs

#![feature(advanced_slice_patterns, slice_patterns, box_patterns)]
extern crate getopts;
extern crate linefeed;

use std::path::Path;
use std::fs::File;
use std::io::Read;
use std::process;

mod tokenizer;
mod parser;

use eval::Evaluator;
mod eval;

use language::{ProgrammingLanguage, ParseError, TokenError};
mod language;

use compilation::{compilation_sequence, compile_ast};
mod compilation;
mod llvm_wrap;

fn main() {
    let option_matches =
        match program_options().parse(std::env::args()) {
            Ok(o) => o,
            Err(e) => {
                println!("{:?}", e);
                std::process::exit(1);
            }
        };
    let trace = option_matches.opt_present("t");
    let show_llvm = option_matches.opt_present("l");
    match option_matches.free[..] {
        [] | [_] => {
            let mut repl = Repl::new(trace, show_llvm);
            repl.run();
        }
        [_, ref filename, _..] => {
            let language = Schala { };
            run_noninteractive(filename, !option_matches.opt_present("i"), trace, &language);
        }
    };
}

fn program_options() -> getopts::Options {
    let mut options = getopts::Options::new();
    options.optflag("i",
                    "interpret",
                    "Interpret source file instead of compiling");
    options.optflag("t",
                    "trace-evaluation",
                    "Print out trace of evaluation");
    options.optflag("l",
                    "llvm-in-repl",
                    "Show LLVM IR in REPL");
    options
}

fn run_noninteractive<'a, T: ProgrammingLanguage<eval::Evaluator<'a>>>(filename: &str, compile: bool, trace_evaluation: bool, language: &T) {
    let mut source_file = File::open(&Path::new(filename)).unwrap();
    let mut buffer = String::new();
    source_file.read_to_string(&mut buffer).unwrap();

    let tokens = match T::tokenize(&buffer) {
        Ok(t) => t,
        Err(e) => {
            println!("Tokenization error: {}", e.msg);
            std::process::exit(1)
        }
    };

    let ast = match T::parse(tokens) {
        Ok(ast) => ast,
        Err(err) => {
            println!("Parse error: {:?}", err.msg);
            /*println!("Remaining tokens: {:?}", err.remaining_tokens);*/
            std::process::exit(1)
        }
    };

    if compile {
        compilation_sequence(T::compile(ast), filename);
    } else {
        let mut evaluator = Evaluator::new_with_opts(None, trace_evaluation);
        let results = T::evaluate(ast, &mut evaluator);
        for result in results.iter() {
            println!("{}", result);
        }
    }
}

type LineReader = linefeed::Reader<linefeed::terminal::DefaultTerminal>;
struct Repl<'a> {
    show_tokens: bool,
    show_parse: bool,
    show_llvm_ir: bool,
    evaluator: Evaluator<'a>,
    interpreter_directive_sigil: char,
    reader: LineReader,
}

impl<'a> Repl<'a> {
    fn new(trace_evaluation: bool, show_llvm: bool) -> Repl<'a> {
        let mut reader: linefeed::Reader<_> = linefeed::Reader::new("Schala").unwrap();
        reader.set_prompt(">> ");
        Repl {
            show_tokens: false,
            show_parse: false,
            show_llvm_ir: show_llvm,
            evaluator: Evaluator::new_with_opts(None, trace_evaluation),
            interpreter_directive_sigil: '.',
            reader: reader,
        }
    }
    fn run(&mut self) {
        use linefeed::ReadResult::*;
        println!("Schala v 0.02");
        loop {
            match self.reader.read_line() {
                Err(e) => {
                    println!("Terminal read error: {}", e);
                },
                Ok(Eof) => {
                    break;
                }
                Ok(Input(ref input)) => {
                    self.reader.add_history(input.clone());
                    if self.handle_interpreter_directive(input) {
                        continue;
                    }
                    let output = self.input_handler(input);
                    println!("=> {}", output);
                }
                _ => (),
            }
        }
        println!("Exiting...");
    }

    fn input_handler(&mut self, input: &str) -> String {
        let schala = Schala { };
        self.input_handler_new(input, schala)
    }

    fn input_handler_new<T: ProgrammingLanguage<eval::Evaluator<'a>>>(&mut self, input: &str, language: T) -> String {
        let mut output = String::new();

        let tokens = match T::tokenize(input) {
            Ok(tokens) => tokens,
            Err(err) => {
                output.push_str(&format!("Tokenization error: {}\n", err.msg));
                return output;
            }
        };

        if self.show_tokens {
            output.push_str(&format!("Tokens: {:?}\n", tokens));
        }

        let ast = match T::parse(tokens) {
            Ok(ast) => ast,
            Err(err) => {
                output.push_str(&format!("Parse error: {:?}\n", err.msg));
                return output;
            }
        };

        if self.show_parse {
            output.push_str(&format!("AST: {:?}\n", ast));
        }

        if self.show_llvm_ir {
            let s = T::compile(ast);
            output.push_str(&s);
        } else {
            // for now only handle last output
            let mut full_output: Vec<String> = T::evaluate(ast, &mut self.evaluator);
            output.push_str(&full_output.pop().unwrap_or("".to_string()));
        }
        output
    }

    fn handle_interpreter_directive(&mut self, input: &str) -> bool {
        match input.chars().nth(0) {
            Some(ch) if ch == self.interpreter_directive_sigil => (),
            _ => return false
        }

        let mut iter = input.chars();
        iter.next();
        let trimmed_sigil: &str = iter.as_str();

        let commands: Vec<&str> = trimmed_sigil
            .split_whitespace()
            .collect();

        let cmd: &str = match commands.get(0).clone() {
            None => return true,
            Some(s) => s
        };

        match cmd {
            "exit" | "quit"  => process::exit(0),
            "history"  => {
                for item in self.reader.history() {
                    println!("{}", item);
                }
            },
            "set" => {
                let show = match commands[1] {
                    "show" => true,
                    "hide" => false,
                    e => {
                        println!("Bad `set` argument: {}", e);
                        return true;
                    }
                };
                match commands[2] {
                    "tokens" => self.show_tokens = show,
                    "parse" => self.show_parse = show,
                    "eval" => self.evaluator.trace_evaluation = show,
                    "llvm" => self.show_llvm_ir = show,
                    e => {
                        println!("Bad `show`/`hide` argument: {}", e);
                        return true;
                    }
                }
            },
            e => println!("Unknown command: {}", e)
        }
        return true;
    }
}

struct Schala { }

impl<'a> ProgrammingLanguage<eval::Evaluator<'a>> for Schala {
    type Token = tokenizer::Token;
    type AST = parser::AST;

    fn tokenize(input: &str) -> Result<Vec<Self::Token>, TokenError> {
        tokenizer::tokenize(input).map_err(|x| TokenError { msg: x.msg })
    }

    fn parse(input: Vec<Self::Token>) -> Result<Self::AST, ParseError> {
        parser::parse(&input, &[]).map_err(|x| ParseError { msg: x.msg })
    }
    fn evaluate(ast: Self::AST, evaluator: &mut Evaluator) -> Vec<String> {
        evaluator.run(ast)
    }
    fn compile(ast: Self::AST) -> String {
        compile_ast(ast)
    }
}