pub mod abstract_tree;
pub mod context;
pub mod error;
pub mod lexer;
pub mod parser;
pub mod value;

use abstract_tree::{AbstractTree, Action};
use context::Context;
use error::Error;
use lexer::lex;
pub use parser::{parse, parser, DustParser};
pub use value::Value;

pub fn interpret(source: &str) -> Result<Option<Value>, Vec<Error>> {
    let context = Context::new();
    let mut interpreter = Interpreter::new(context);

    interpreter.run(source)
}

pub struct Interpreter {
    context: Context,
}

impl Interpreter {
    pub fn new(context: Context) -> Self {
        Interpreter { context }
    }

    pub fn run(&mut self, source: &str) -> Result<Option<Value>, Vec<Error>> {
        let tokens = lex(source)?;
        let statements = parse(&tokens)?;
        let errors = statements
            .iter()
            .filter_map(|(statement, span)| {
                statement
                    .validate(&self.context)
                    .map_err(|validation_error| Error::Validation {
                        error: validation_error,
                        span: span.clone(),
                    })
                    .err()
            })
            .collect::<Vec<Error>>();

        if !errors.is_empty() {
            return Err(errors);
        }

        let mut value = None;

        for (statement, _span) in statements {
            value = match statement.run(&self.context) {
                Ok(action) => match action {
                    Action::Break(value) => Some(value),
                    Action::Return(value) => Some(value),
                    Action::None => continue,
                },
                Err(runtime_error) => return Err(vec![Error::Runtime(runtime_error)]),
            }
        }

        Ok(value)
    }
}