schala/src/parser.rs

use std::iter::Peekable;
use std::slice::Iter;
use std::vec::IntoIter;

use tokenizer::Token;

#[derive(Debug)]
pub enum AST {
    BinOp(Box<AST>, Box<AST>, Box<AST>),
    Number(f64),
    Name(String),
}

#[derive(Debug)]
pub struct ParseError {
    err: String
}

pub type ParseResult<T> = Result<T, ParseError>;

/* grammar

   expr : term ((PLUS|MIMUS) term)*
   term : factor ((MUL | DIV) factor)*
   factor  : NUM | LPAREN expr RPAREN

*/

struct Parser {
    tokens: Peekable<IntoIter<Token>>
}

impl Parser {
    fn next(&mut self) -> Option<Token> {
        self.tokens.next()
    }

    fn lookahead(&mut self) -> Option<&Token> {
        self.tokens.peek()
    }
}

impl Parser {

    fn expect(&mut self, expected: Token) -> ParseResult<()> {
        match self.next() {
            Some(ref next) if *next == expected => Ok(()),
            Some(next) => {
                let err = format!("Expected {:?} but got {:?}", expected, next);
                return Err(ParseError { err: err });
            },
            None => {
                let err = format!("Expected {:?} but got end of input", expected);
                return Err(ParseError { err: err });
            }
        }
    }

    fn expect_identifier(&mut self, identifier_str: &str) -> ParseResult<()> {
        use tokenizer::Token::*;
        match self.next() {
            Some(Identifier(ref s)) if s == identifier_str => Ok(()),
            Some(next) => {
                let err = format!("Expected identifier `{}` but got {:?}", identifier_str, next);
                Err(ParseError { err: err })
            }
            None => {
                let err = format!("Expected identifier `{}` but got end of input", identifier_str);
                Err(ParseError { err: err })
            }
        }
    }

    fn parse(&mut self) -> ParseResult<AST> {
        let r = self.expr();
        try!(self.expect(Token::Separator));
        try!(self.expect(Token::EOF));
        r
    }

    fn expr(&mut self) -> ParseResult<AST> {
        self.next();
        return Ok(AST::Number(5.0));
    }
}

pub fn parse(input: Vec<Token>) -> ParseResult<AST> {
    let mut iter = input.into_iter().peekable();
    let mut parser = Parser { tokens: iter };
    return parser.parse();
}