dust/src/abstract_tree/loop.rs

use crate::{
    context::Context,
    error::{RuntimeError, ValidationError},
};

use super::{AbstractTree, Action, Statement, Type};

#[derive(Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
pub struct Loop {
    statements: Vec<Statement>,
}

impl Loop {
    pub fn new(statements: Vec<Statement>) -> Self {
        Self { statements }
    }
}

impl AbstractTree for Loop {
    fn expected_type(&self, _context: &Context) -> Result<Type, ValidationError> {
        Ok(Type::None)
    }

    fn validate(&self, _context: &Context) -> Result<(), ValidationError> {
        for statement in &self.statements {
            statement.validate(_context)?;
        }

        Ok(())
    }

    fn run(self, _context: &Context) -> Result<Action, RuntimeError> {
        let mut index = 0;

        loop {
            if index == self.statements.len() - 1 {
                index = 0;
            } else {
                index += 1;
            }

            let statement = self.statements[index].clone();
            let action = statement.run(_context)?;

            match action {
                Action::Return(_) => {}
                Action::None => {}
                r#break => return Ok(r#break),
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::{
        abstract_tree::{Expression, ValueNode},
        Value,
    };

    use super::*;

    #[test]
    fn basic_loop() {
        let result = Loop {
            statements: vec![Statement::Break(Some(Expression::Value(
                ValueNode::Integer(42),
            )))],
        }
        .run(&Context::new());

        assert_eq!(result, Ok(Action::Break(Some(Value::integer(42)))))
    }
}