#![allow(non_snake_case, unused_variables)]

extern crate marpa;

use marpa::grammar::Grammar;
use marpa::lexer::byte_scanner::*;
use marpa::parser::*;
use marpa::result::Result;
use marpa::stack::*;
use marpa::tree_builder::*;

use std::io::Cursor;

#[test]
fn main() {
    real_main().unwrap();
}

fn real_main() -> Result<()> {
    let mut g = Grammar::new()?;

    let ws_char = g.string_set(None, "\t\n\r ")?;
    let ws = g.star(None, ws_char)?;
    let sep = g.literal_string(None, "::=")?;
    let term = g.literal_string(None, ";")?;

    let dquote = g.literal_string(None, "\"")?;
    let squote = g.literal_string(None, "'")?;
    let escape = g.literal_string(None, "\\")?;
    let not_dquote = g.inverse_string_set(None, "\"")?;
    let not_squote = g.inverse_string_set(None, "'")?;
    let dquote_escape = g.rule(None, &[escape, dquote])?;
    let squote_escape = g.rule(None, &[escape, squote])?;
    let str_chars = g.alternative(None, &[not_dquote, dquote_escape])?;
    let str_chars_star = g.star(None, str_chars)?;
    let string = g.rule(None, &[dquote, str_chars_star, dquote])?;
    let char_chars = g.alternative(None, &[not_squote, squote_escape])?;

    let lower = g.char_range(None, 'a', 'z')?;
    let upper = g.char_range(None, 'A', 'Z')?;
    let digit = g.char_range(None, '0', '9')?;

    let alpha_num = g.alternative(None, &[lower, upper, digit])?;

    let ident = g.plus(None, alpha_num)?;

    let rule = g.rule(None, &[ident, ws, sep, ws, string, ws, term])?;
    let rules = g.sequence(None, rule, ws_char, false, false)?;

    let start = rules;

    g.set_start(start)?;

    let mut b = TreeBuilder::new();

    for r in [start, rules].iter().map(|x| x.rule()) {
        b.rule(r);
    }

    for t in [ident, string, sep, term].iter().map(|x| x.rule()) {
        b.token(t);
    }

    for d in [ws_char].iter().map(|x| x.rule()) {
        b.discard(d);
    }

    let mut p = Parser::with_grammar(g.unwrap());

    let mut t = p.run_recognizer(ByteScanner::new(Cursor::new("a ::= \"test\";")))?;
    let v = t.next().unwrap();

    println!("{}", proc_value(b, v));
    Ok(())
}