dust/dust-lang/src/parser.rs

//! Parsing tools.
//!
//! This module provides two parsing options:
//! - `parse` convenience function
//! - `Parser` struct, which parses the input a statement at a time
use std::{
    collections::VecDeque,
    fmt::{self, Display, Formatter},
    num::{ParseFloatError, ParseIntError},
    str::ParseBoolError,
};

use crate::{
    ast::*, Context, DustError, Identifier, LexError, Lexer, Token, TokenKind, TokenOwned, Type,
};

/// Parses the input into an abstract syntax tree.
///
/// # Examples
/// ```
/// # use dust_lang::*;
/// let source = "42.to_string()";
///
/// assert_eq!(
///     parse(source),
///     Ok(AbstractSyntaxTree::with_statements([
///         Statement::Expression(Expression::call(
///             Expression::field_access(
///                 Expression::literal(42, (0, 2)),
///                 Node::new(Identifier::new("to_string"), (3, 12)),
///                 (0, 12)
///             ),
///             vec![],
///             (0, 14)
///         ))
///     ]))
/// );
/// ```
pub fn parse(source: &str) -> Result<AbstractSyntaxTree, DustError> {
    let mut tree = AbstractSyntaxTree::new();

    parse_into(source, &mut tree)?;

    Ok(tree)
}

pub fn parse_into<'src>(
    source: &'src str,
    tree: &mut AbstractSyntaxTree,
) -> Result<(), DustError<'src>> {
    let lexer = Lexer::new(source);
    let mut parser = Parser::new(lexer);

    loop {
        let node = parser
            .parse_statement()
            .map_err(|parse_error| DustError::Parse {
                parse_error,
                source,
            })?;

        tree.statements.push_back(node);

        if let Token::Eof = parser.current_token {
            break;
        }
    }

    Ok(())
}

/// Low-level tool for parsing the input a statement at a time.
///
/// # Examples
/// ```
/// # use std::collections::VecDeque;
/// # use dust_lang::*;
/// let source = "x = 42";
/// let lexer = Lexer::new(source);
/// let mut parser = Parser::new(lexer);
/// let mut statements = VecDeque::new();
///
/// loop {
///     let statement = parser.parse_statement().unwrap();
///
///     statements.push_back(statement);
///
///     if parser.is_eof() {
///         break;
///     }
/// }
///
/// let tree = AbstractSyntaxTree { statements };
///
/// ```
pub struct Parser<'src> {
    lexer: Lexer<'src>,
    current_token: Token<'src>,
    current_position: Span,
    mode: ParserMode,
    context: Context,
}

impl<'src> Parser<'src> {
    pub fn new(mut lexer: Lexer<'src>) -> Self {
        let (current_token, current_position) = lexer.next_token().unwrap_or((Token::Eof, (0, 0)));

        Parser {
            lexer,
            current_token,
            current_position,
            mode: ParserMode::Normal,
            context: Context::new(),
        }
    }

    pub fn is_eof(&self) -> bool {
        matches!(self.current_token, Token::Eof)
    }

    pub fn parse_statement(&mut self) -> Result<Statement, ParseError> {
        let start_position = self.current_position;

        if let Token::Let = self.current_token {
            log::trace!("Parsing let statement");

            self.next_token()?;

            let is_mutable = if let Token::Mut = self.current_token {
                self.next_token()?;

                true
            } else {
                false
            };

            let identifier = self.parse_identifier()?;

            if let Token::Equal = self.current_token {
                self.next_token()?;
            } else {
                return Err(ParseError::ExpectedToken {
                    expected: TokenKind::Equal,
                    actual: self.current_token.to_owned(),
                    position: self.current_position,
                });
            }

            let value = self.parse_expression(0)?;

            let end = if let Token::Semicolon = self.current_token {
                let end = self.current_position.1;

                self.next_token()?;

                end
            } else {
                value.position().1
            };
            let r#let = if is_mutable {
                LetStatement::LetMut { identifier, value }
            } else {
                LetStatement::Let { identifier, value }
            };
            let position = (start_position.0, end);

            return Ok(Statement::Let(Node::new(r#let, position)));
        }

        if let Token::Struct = self.current_token {
            log::trace!("Parsing struct definition");

            self.next_token()?;

            let name = if let Token::Identifier(_) = self.current_token {
                self.parse_identifier()?
            } else {
                return Err(ParseError::ExpectedToken {
                    expected: TokenKind::Identifier,
                    actual: self.current_token.to_owned(),
                    position: self.current_position,
                });
            };

            if let Token::Semicolon = self.current_token {
                let end = self.current_position.1;

                self.next_token()?;

                return Ok(Statement::struct_definition(
                    StructDefinition::Unit { name },
                    (start_position.0, end),
                ));
            }

            if let Token::LeftParenthesis = self.current_token {
                self.next_token()?;

                let mut types = Vec::new();

                loop {
                    if let Token::RightParenthesis = self.current_token {
                        self.next_token()?;

                        if let Token::Semicolon = self.current_token {
                            self.next_token()?;
                        }

                        break;
                    }

                    let type_node = self.parse_type()?;

                    types.push(type_node);

                    if let Token::Comma = self.current_token {
                        self.next_token()?;

                        continue;
                    }
                }

                let position = (start_position.0, self.current_position.1);

                return if types.is_empty() {
                    Ok(Statement::struct_definition(
                        StructDefinition::Unit { name },
                        position,
                    ))
                } else {
                    Ok(Statement::struct_definition(
                        StructDefinition::Tuple { name, items: types },
                        position,
                    ))
                };
            }

            if let Token::LeftCurlyBrace = self.current_token {
                self.next_token()?;

                let mut fields = Vec::new();

                loop {
                    if let Token::RightCurlyBrace = self.current_token {
                        self.next_token()?;

                        if let Token::Semicolon = self.current_token {
                            self.next_token()?;
                        }

                        break;
                    }

                    let field_name = self.parse_identifier()?;

                    if let Token::Colon = self.current_token {
                        self.next_token()?;
                    } else {
                        return Err(ParseError::ExpectedToken {
                            expected: TokenKind::Colon,
                            actual: self.current_token.to_owned(),
                            position: self.current_position,
                        });
                    }

                    let field_type = self.parse_type()?;

                    fields.push((field_name, field_type));

                    if let Token::Comma = self.current_token {
                        self.next_token()?;

                        continue;
                    }
                }

                let position = (start_position.0, self.current_position.1);

                return if fields.is_empty() {
                    Ok(Statement::struct_definition(
                        StructDefinition::Unit { name },
                        position,
                    ))
                } else {
                    Ok(Statement::struct_definition(
                        StructDefinition::Fields { name, fields },
                        position,
                    ))
                };
            }

            return Err(ParseError::ExpectedTokenMultiple {
                expected: vec![
                    TokenKind::LeftParenthesis,
                    TokenKind::LeftCurlyBrace,
                    TokenKind::Semicolon,
                ],
                actual: self.current_token.to_owned(),
                position: self.current_position,
            });
        }

        let expression = self.parse_expression(0)?;

        if let Token::Semicolon = self.current_token {
            let position = (start_position.0, self.current_position.1);

            self.next_token()?;

            Ok(Statement::ExpressionNullified(Node::new(
                expression, position,
            )))
        } else {
            Ok(Statement::Expression(expression))
        }
    }

    fn next_token(&mut self) -> Result<(), ParseError> {
        let (token, position) = self.lexer.next_token()?;

        self.current_token = token;
        self.current_position = position;

        Ok(())
    }

    fn parse_expression(&mut self, mut precedence: u8) -> Result<Expression, ParseError> {
        // Parse a statement starting from the current node.
        let mut left = if self.current_token.is_prefix() {
            self.parse_prefix()?
        } else {
            self.parse_primary()?
        };

        // While the current token has a higher precedence than the given precedence
        while precedence < self.current_token.precedence() {
            // Give precedence to postfix operations
            left = if self.current_token.is_postfix() {
                let statement = self.parse_postfix(left)?;

                precedence = self.current_token.precedence();

                // Replace the left-hand side with the postfix operation
                statement
            } else {
                // Replace the left-hand side with the infix operation
                self.parse_infix(left)?
            };
        }

        Ok(left)
    }

    fn parse_prefix(&mut self) -> Result<Expression, ParseError> {
        log::trace!("Parsing {} as prefix operator", self.current_token);

        let operator_start = self.current_position.0;

        match self.current_token {
            Token::Bang => {
                self.next_token()?;

                let operand = self.parse_expression(0)?;
                let position = (operator_start, self.current_position.1);

                Ok(Expression::not(operand, position))
            }
            Token::Minus => {
                self.next_token()?;

                let operand = self.parse_expression(0)?;
                let position = (operator_start, self.current_position.1);

                Ok(Expression::negation(operand, position))
            }
            Token::Star => {
                self.next_token()?;

                let operand = self.parse_expression(0)?;
                let position = (operator_start, self.current_position.1);

                Ok(Expression::dereference(operand, position))
            }
            _ => Err(ParseError::ExpectedTokenMultiple {
                expected: vec![TokenKind::Bang, TokenKind::Minus, TokenKind::Star],
                actual: self.current_token.to_owned(),
                position: self.current_position,
            }),
        }
    }

    fn parse_primary(&mut self) -> Result<Expression, ParseError> {
        log::trace!("Parsing {} as primary", self.current_token);

        let start_position = self.current_position;

        match self.current_token {
            Token::Async => {
                let block = self.parse_block()?;
                let position = (start_position.0, block.position.1);

                Ok(Expression::block(block.inner, position))
            }
            Token::Boolean(text) => {
                self.next_token()?;

                let boolean = text.parse::<bool>().map_err(|error| ParseError::Boolean {
                    error,
                    position: start_position,
                })?;
                let expression = Expression::literal(boolean, start_position);

                Ok(expression)
            }
            Token::Break => {
                let break_end = self.current_position.1;

                self.next_token()?;

                let (expression_option, end) = if let Token::Semicolon = self.current_token {
                    // Do not consume the semicolon, allowing it to nullify the expression

                    (None, break_end)
                } else if let Ok(expression) = self.parse_expression(0) {
                    let end = expression.position().1;

                    (Some(expression), end)
                } else {
                    return Err(ParseError::ExpectedToken {
                        expected: TokenKind::Semicolon,
                        actual: self.current_token.to_owned(),
                        position: self.current_position,
                    });
                };
                let position = (start_position.0, end);

                Ok(Expression::r#break(expression_option, position))
            }
            Token::Character(character) => {
                self.next_token()?;

                let expression = Expression::literal(character, start_position);

                Ok(expression)
            }
            Token::Float(text) => {
                self.next_token()?;

                let float = text.parse::<f64>().map_err(|error| ParseError::Float {
                    error,
                    position: start_position,
                })?;

                Ok(Expression::literal(float, start_position))
            }
            Token::Identifier(text) => {
                self.next_token()?;

                let identifier = Identifier::new(text);

                if let ParserMode::Condition = self.mode {
                    return Ok(Expression::identifier(identifier, start_position));
                }

                if let Token::LeftCurlyBrace = self.current_token {
                    self.next_token()?;

                    let name = Node::new(identifier, start_position);
                    let mut fields = Vec::new();

                    loop {
                        if let Token::RightCurlyBrace = self.current_token {
                            self.next_token()?;

                            break;
                        }

                        let field_name = self.parse_identifier()?;

                        if let Token::Colon = self.current_token {
                            self.next_token()?;
                        } else {
                            return Err(ParseError::ExpectedToken {
                                expected: TokenKind::Colon,
                                actual: self.current_token.to_owned(),
                                position: self.current_position,
                            });
                        }

                        let field_value = self.parse_expression(0)?;

                        fields.push((field_name, field_value));

                        if let Token::Comma = self.current_token {
                            self.next_token()?;
                        }
                    }

                    let position = (start_position.0, self.current_position.1);

                    return Ok(Expression::r#struct(
                        StructExpression::Fields { name, fields },
                        position,
                    ));
                }

                Ok(Expression::identifier(identifier, start_position))
            }
            Token::Integer(text) => {
                self.next_token()?;

                let integer = text.parse::<i64>().map_err(|error| ParseError::Integer {
                    error,
                    position: start_position,
                })?;

                Ok(Expression::literal(integer, start_position))
            }
            Token::If => {
                self.next_token()?;

                let r#if = self.parse_if()?;
                let position = (start_position.0, self.current_position.1);

                Ok(Expression::r#if(r#if, position))
            }
            Token::String(text) => {
                self.next_token()?;

                Ok(Expression::literal(text.to_string(), start_position))
            }
            Token::LeftCurlyBrace => {
                let block_node = self.parse_block()?;

                Ok(Expression::block(block_node.inner, block_node.position))
            }
            Token::LeftParenthesis => {
                self.next_token()?;

                let node = self.parse_expression(0)?;

                if let Token::RightParenthesis = self.current_token {
                    let position = (start_position.0, self.current_position.1);

                    self.next_token()?;

                    Ok(Expression::grouped(node, position))
                } else {
                    Err(ParseError::ExpectedToken {
                        expected: TokenKind::RightParenthesis,
                        actual: self.current_token.to_owned(),
                        position: self.current_position,
                    })
                }
            }
            Token::LeftSquareBrace => {
                self.next_token()?;

                if let Token::RightSquareBrace = self.current_token {
                    let position = (start_position.0, self.current_position.1);

                    self.next_token()?;

                    return Ok(Expression::list(Vec::new(), position));
                }

                let first_expression = self.parse_expression(0)?;

                if let Token::Semicolon = self.current_token {
                    self.next_token()?;

                    let repeat_operand = self.parse_expression(0)?;

                    if let Token::RightSquareBrace = self.current_token {
                        let position = (start_position.0, self.current_position.1);

                        self.next_token()?;

                        return Ok(Expression::auto_fill_list(
                            first_expression,
                            repeat_operand,
                            position,
                        ));
                    } else {
                        return Err(ParseError::ExpectedToken {
                            expected: TokenKind::RightSquareBrace,
                            actual: self.current_token.to_owned(),
                            position: self.current_position,
                        });
                    }
                }

                let mut expressions = vec![first_expression];

                loop {
                    if let Token::RightSquareBrace = self.current_token {
                        let position = (start_position.0, self.current_position.1);

                        self.next_token()?;

                        return Ok(Expression::list(expressions, position));
                    }

                    if let Token::Comma = self.current_token {
                        self.next_token()?;

                        continue;
                    }

                    let expression = self.parse_expression(0)?;

                    expressions.push(expression);
                }
            }
            Token::Loop => {
                self.next_token()?;

                let block = self.parse_block()?;
                let position = (start_position.0, block.position.1);

                Ok(Expression::infinite_loop(block, position))
            }
            Token::Map => {
                self.next_token()?;

                if let Token::LeftCurlyBrace = self.current_token {
                    self.next_token()?;
                } else {
                    return Err(ParseError::ExpectedToken {
                        expected: TokenKind::LeftCurlyBrace,
                        actual: self.current_token.to_owned(),
                        position: self.current_position,
                    });
                }

                let mut fields = Vec::new();

                loop {
                    if let Token::RightCurlyBrace = self.current_token {
                        let position = (start_position.0, self.current_position.1);

                        self.next_token()?;

                        return Ok(Expression::map(fields, position));
                    }

                    let field_name = self.parse_identifier()?;

                    if let Token::Equal = self.current_token {
                        self.next_token()?;
                    } else {
                        return Err(ParseError::ExpectedToken {
                            expected: TokenKind::Equal,
                            actual: self.current_token.to_owned(),
                            position: self.current_position,
                        });
                    }

                    let field_value = self.parse_expression(0)?;

                    fields.push((field_name, field_value));

                    if let Token::Comma = self.current_token {
                        self.next_token()?;
                    }
                }
            }
            Token::While => {
                self.next_token()?;

                let condition = self.parse_expression(0)?;
                let block = self.parse_block()?;
                let position = (start_position.0, block.position.1);

                Ok(Expression::while_loop(condition, block, position))
            }
            _ => Err(ParseError::ExpectedTokenMultiple {
                expected: vec![
                    TokenKind::Async,
                    TokenKind::Boolean,
                    TokenKind::Float,
                    TokenKind::Identifier,
                    TokenKind::Integer,
                    TokenKind::If,
                    TokenKind::LeftCurlyBrace,
                    TokenKind::LeftParenthesis,
                    TokenKind::LeftSquareBrace,
                    TokenKind::String,
                    TokenKind::While,
                ],
                actual: self.current_token.to_owned(),
                position: self.current_position,
            }),
        }
    }

    fn parse_infix(&mut self, left: Expression) -> Result<Expression, ParseError> {
        log::trace!("Parsing {} as infix operator", self.current_token);

        let operator_precedence = self.current_token.precedence()
            - if self.current_token.is_right_associative() {
                1
            } else {
                0
            };
        let left_start = left.position().0;

        if let Token::Equal = &self.current_token {
            self.next_token()?;

            let value = self.parse_expression(operator_precedence)?;
            let position = (left_start, value.position().1);

            return Ok(Expression::assignment(left, value, position));
        }

        if let Token::PlusEqual | Token::MinusEqual = &self.current_token {
            let math_operator = match self.current_token {
                Token::PlusEqual => MathOperator::Add,
                Token::MinusEqual => MathOperator::Subtract,
                _ => unreachable!(),
            };
            let operator = Node::new(math_operator, self.current_position);

            self.next_token()?;

            let value = self.parse_expression(operator_precedence)?;
            let position = (left_start, value.position().1);

            return Ok(Expression::operator(
                OperatorExpression::CompoundAssignment {
                    assignee: left,
                    operator,
                    modifier: value,
                },
                position,
            ));
        }

        if let Token::DoubleDot = &self.current_token {
            self.next_token()?;

            let end = self.parse_expression(operator_precedence)?;
            let position = (left_start, end.position().1);

            return Ok(Expression::exclusive_range(left, end, position));
        }

        if let Token::Minus | Token::Plus | Token::Star | Token::Slash | Token::Percent =
            &self.current_token
        {
            let math_operator = match &self.current_token {
                Token::Minus => Node::new(MathOperator::Subtract, self.current_position),
                Token::Plus => Node::new(MathOperator::Add, self.current_position),
                Token::Star => Node::new(MathOperator::Multiply, self.current_position),
                Token::Slash => Node::new(MathOperator::Divide, self.current_position),
                Token::Percent => Node::new(MathOperator::Modulo, self.current_position),
                _ => unreachable!(),
            };

            self.next_token()?;

            let right = self.parse_expression(operator_precedence)?;
            let position = (left_start, right.position().1);

            return Ok(Expression::operator(
                OperatorExpression::Math {
                    left,
                    operator: math_operator,
                    right,
                },
                position,
            ));
        }

        if let Token::DoubleEqual
        | Token::BangEqual
        | Token::Less
        | Token::LessEqual
        | Token::Greater
        | Token::GreaterEqual = &self.current_token
        {
            let comparison_operator = match &self.current_token {
                Token::DoubleEqual => Node::new(ComparisonOperator::Equal, self.current_position),
                Token::BangEqual => Node::new(ComparisonOperator::NotEqual, self.current_position),
                Token::Less => Node::new(ComparisonOperator::LessThan, self.current_position),
                Token::LessEqual => {
                    Node::new(ComparisonOperator::LessThanOrEqual, self.current_position)
                }
                Token::Greater => Node::new(ComparisonOperator::GreaterThan, self.current_position),
                Token::GreaterEqual => Node::new(
                    ComparisonOperator::GreaterThanOrEqual,
                    self.current_position,
                ),
                _ => unreachable!(),
            };

            self.next_token()?;

            let right = self.parse_expression(operator_precedence)?;
            let position = (left_start, right.position().1);

            return Ok(Expression::operator(
                OperatorExpression::Comparison {
                    left,
                    operator: comparison_operator,
                    right,
                },
                position,
            ));
        }

        let logic_operator = match &self.current_token {
            Token::DoubleAmpersand => Node::new(LogicOperator::And, self.current_position),
            Token::DoublePipe => Node::new(LogicOperator::Or, self.current_position),
            _ => {
                return Err(ParseError::UnexpectedToken {
                    actual: self.current_token.to_owned(),
                    position: self.current_position,
                })
            }
        };

        self.next_token()?;

        let right = self.parse_expression(operator_precedence)?;
        let position = (left_start, right.position().1);

        Ok(Expression::operator(
            OperatorExpression::Logic {
                left,
                operator: logic_operator,
                right,
            },
            position,
        ))
    }

    fn parse_postfix(&mut self, left: Expression) -> Result<Expression, ParseError> {
        log::trace!("Parsing {} as postfix operator", self.current_token);

        let expression = match &self.current_token {
            Token::Dot => {
                self.next_token()?;

                if let Token::Integer(text) = &self.current_token {
                    let index = text.parse::<usize>().map_err(|error| ParseError::Integer {
                        error,
                        position: self.current_position,
                    })?;
                    let index_node = Node::new(index, self.current_position);
                    let position = (left.position().0, self.current_position.1);

                    self.next_token()?;

                    Expression::tuple_access(left, index_node, position)
                } else {
                    let field = self.parse_identifier()?;
                    let position = (left.position().0, field.position.1);

                    Expression::field_access(left, field, position)
                }
            }
            Token::LeftParenthesis => {
                self.next_token()?;

                let mut arguments = Vec::new();

                while self.current_token != Token::RightParenthesis {
                    let argument = self.parse_expression(0)?;

                    arguments.push(argument);

                    if let Token::Comma = self.current_token {
                        self.next_token()?;
                    } else {
                        break;
                    }
                }

                self.next_token()?;

                let position = (left.position().0, self.current_position.1);

                Expression::call(left, arguments, position)
            }
            Token::LeftSquareBrace => {
                self.next_token()?;

                let index = self.parse_expression(0)?;

                let operator_end = if let Token::RightSquareBrace = self.current_token {
                    let end = self.current_position.1;

                    self.next_token()?;

                    end
                } else {
                    return Err(ParseError::ExpectedToken {
                        expected: TokenKind::RightSquareBrace,
                        actual: self.current_token.to_owned(),
                        position: self.current_position,
                    });
                };

                let position = (left.position().0, operator_end);

                Expression::list_index(left, index, position)
            }
            _ => {
                return Err(ParseError::ExpectedTokenMultiple {
                    expected: vec![
                        TokenKind::Dot,
                        TokenKind::LeftParenthesis,
                        TokenKind::LeftSquareBrace,
                    ],
                    actual: self.current_token.to_owned(),
                    position: self.current_position,
                });
            }
        };

        if self.current_token.is_postfix() {
            self.parse_postfix(expression)
        } else {
            Ok(expression)
        }
    }

    fn parse_if(&mut self) -> Result<IfExpression, ParseError> {
        // Assume that the "if" token has already been consumed

        self.mode = ParserMode::Condition;

        let condition = self.parse_expression(0)?;

        self.mode = ParserMode::Normal;

        let if_block = self.parse_block()?;

        if let Token::Else = self.current_token {
            self.next_token()?;

            let if_keyword_start = self.current_position.0;

            if let Token::If = self.current_token {
                self.next_token()?;

                let if_expression = self.parse_if()?;
                let position = (if_keyword_start, self.current_position.1);

                Ok(IfExpression::IfElse {
                    condition,
                    if_block,
                    r#else: ElseExpression::If(Node::new(Box::new(if_expression), position)),
                })
            } else {
                let else_block = self.parse_block()?;

                Ok(IfExpression::IfElse {
                    condition,
                    if_block,
                    r#else: ElseExpression::Block(else_block),
                })
            }
        } else {
            Ok(IfExpression::If {
                condition,
                if_block,
            })
        }
    }

    fn parse_identifier(&mut self) -> Result<Node<Identifier>, ParseError> {
        if let Token::Identifier(text) = self.current_token {
            let position = self.current_position;

            self.next_token()?;

            Ok(Node::new(Identifier::new(text), position))
        } else {
            Err(ParseError::ExpectedToken {
                expected: TokenKind::Identifier,
                actual: self.current_token.to_owned(),
                position: self.current_position,
            })
        }
    }

    fn parse_block(&mut self) -> Result<Node<BlockExpression>, ParseError> {
        let left_start = self.current_position.0;
        let is_async = if let Token::Async = self.current_token {
            self.next_token()?;

            true
        } else {
            false
        };

        if let Token::LeftCurlyBrace = self.current_token {
            self.next_token()?;
        } else {
            return Err(ParseError::ExpectedToken {
                expected: TokenKind::LeftCurlyBrace,
                actual: self.current_token.to_owned(),
                position: self.current_position,
            });
        }

        let mut statements = VecDeque::new();

        loop {
            if let Token::RightCurlyBrace = self.current_token {
                let position = (left_start, self.current_position.1);

                self.next_token()?;

                let ast = AbstractSyntaxTree {
                    statements,
                    context: self.context.create_child(),
                };

                return if is_async {
                    Ok(Node::new(BlockExpression::Async(ast), position))
                } else {
                    Ok(Node::new(BlockExpression::Sync(ast), position))
                };
            }

            let statement = self.parse_statement()?;

            statements.push_back(statement);
        }
    }

    fn parse_type(&mut self) -> Result<Node<Type>, ParseError> {
        let r#type = match self.current_token {
            Token::Bool => Type::Boolean,
            Token::FloatKeyword => Type::Float,
            Token::Int => Type::Integer,
            _ => {
                return Err(ParseError::ExpectedTokenMultiple {
                    expected: vec![TokenKind::Bool, TokenKind::FloatKeyword, TokenKind::Int],
                    actual: self.current_token.to_owned(),
                    position: self.current_position,
                });
            }
        };
        let position = self.current_position;

        self.next_token()?;

        Ok(Node::new(r#type, position))
    }
}

#[derive(Debug, PartialEq, Clone)]
pub enum ParserMode {
    Condition,
    Normal,
}

#[derive(Debug, PartialEq, Clone)]
pub enum ParseError {
    Boolean {
        error: ParseBoolError,
        position: Span,
    },
    Lex(LexError),
    ExpectedAssignment {
        actual: Statement,
    },
    ExpectedExpression {
        actual: Statement,
    },
    ExpectedIdentifierNode {
        actual: Expression,
    },
    ExpectedIdentifierToken {
        actual: TokenOwned,
        position: Span,
    },
    ExpectedToken {
        expected: TokenKind,
        actual: TokenOwned,
        position: Span,
    },
    ExpectedTokenMultiple {
        expected: Vec<TokenKind>,
        actual: TokenOwned,
        position: Span,
    },
    UnexpectedToken {
        actual: TokenOwned,
        position: Span,
    },
    Float {
        error: ParseFloatError,
        position: Span,
    },
    Integer {
        error: ParseIntError,
        position: Span,
    },
}

impl From<LexError> for ParseError {
    fn from(v: LexError) -> Self {
        Self::Lex(v)
    }
}

impl ParseError {
    pub fn position(&self) -> Span {
        match self {
            ParseError::Boolean { position, .. } => *position,
            ParseError::ExpectedAssignment { actual } => actual.position(),
            ParseError::ExpectedExpression { actual } => actual.position(),
            ParseError::ExpectedIdentifierNode { actual } => actual.position(),
            ParseError::ExpectedIdentifierToken { position, .. } => *position,
            ParseError::ExpectedToken { position, .. } => *position,
            ParseError::ExpectedTokenMultiple { position, .. } => *position,
            ParseError::Float { position, .. } => *position,
            ParseError::Integer { position, .. } => *position,
            ParseError::Lex(error) => error.position(),
            ParseError::UnexpectedToken { position, .. } => *position,
        }
    }
}

impl Display for ParseError {
    fn fmt(&self, f: &mut Formatter) -> fmt::Result {
        match self {
            Self::Boolean { error, .. } => write!(f, "{}", error),
            Self::ExpectedAssignment { .. } => write!(f, "Expected assignment"),
            Self::ExpectedExpression { .. } => write!(f, "Expected expression"),
            Self::ExpectedIdentifierNode { actual } => {
                write!(f, "Expected identifier, found {actual}")
            }
            Self::ExpectedIdentifierToken { actual, .. } => {
                write!(f, "Expected identifier, found {actual}")
            }
            Self::ExpectedToken {
                expected, actual, ..
            } => write!(f, "Expected token {expected}, found {actual}"),
            Self::ExpectedTokenMultiple {
                expected, actual, ..
            } => {
                write!(f, "Expected one of")?;

                for (i, token_kind) in expected.iter().enumerate() {
                    if i == 0 {
                        write!(f, " {token_kind}")?;
                    } else if i == expected.len() - 1 {
                        write!(f, " or {token_kind}")?;
                    } else {
                        write!(f, ", {token_kind}")?;
                    }
                }

                write!(f, ", found {actual}")
            }
            Self::Float { error, .. } => write!(f, "{}", error),
            Self::Integer { error, .. } => write!(f, "{}", error),
            Self::Lex(error) => write!(f, "{}", error),
            Self::UnexpectedToken { actual, .. } => write!(f, "Unexpected token {actual}"),
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::{Identifier, Type};

    use super::*;

    #[test]
    fn dereference() {
        let source = "*a";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::dereference(
                    Expression::identifier(Identifier::new("a"), (1, 2)),
                    (0, 2)
                ),)
            ]))
        );
    }

    #[test]
    fn character_literal() {
        let source = "'a'";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::literal('a', (0, 3)))
            ]))
        );
    }

    #[test]
    fn break_loop() {
        let source = "loop { break; }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::infinite_loop(
                    Node::new(
                        BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                            Statement::ExpressionNullified(Node::new(
                                Expression::r#break(None, (7, 12)),
                                (7, 13)
                            ))
                        ])),
                        (5, 15)
                    ),
                    (0, 15)
                ))
            ]))
        );
    }

    #[test]
    fn built_in_function() {
        let source = "42.to_string()";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::call(
                    Expression::field_access(
                        Expression::literal(42, (0, 2)),
                        Node::new(Identifier::new("to_string"), (3, 12)),
                        (0, 12)
                    ),
                    vec![],
                    (0, 14)
                ))
            ]))
        );
    }

    #[test]
    fn map_expression() {
        let source = "map { x = \"1\", y = 2, z = 3.0 }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::map(
                    vec![
                        (
                            Node::new(Identifier::new("x"), (6, 7)),
                            Expression::literal("1".to_string(), (10, 13)),
                        ),
                        (
                            Node::new(Identifier::new("y"), (15, 16)),
                            Expression::literal(2, (19, 20)),
                        ),
                        (
                            Node::new(Identifier::new("z"), (22, 23)),
                            Expression::literal(3.0, (26, 29)),
                        ),
                    ],
                    (0, 31),
                ))
            ]))
        );
    }

    #[test]
    fn let_mut_while_loop() {
        let source = "let mut x = 0; while x < 10 { x += 1 }; x";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Let(Node::new(
                    LetStatement::LetMut {
                        identifier: Node::new(Identifier::new("x"), (8, 9)),
                        value: Expression::literal(0, (12, 13)),
                    },
                    (0, 14),
                )),
                Statement::ExpressionNullified(Node::new(
                    Expression::while_loop(
                        Expression::comparison(
                            Expression::identifier(Identifier::new("x"), (21, 22)),
                            Node::new(ComparisonOperator::LessThan, (23, 24)),
                            Expression::literal(10, (25, 27)),
                            (21, 27),
                        ),
                        Node::new(
                            BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                                Statement::Expression(Expression::compound_assignment(
                                    Expression::identifier(Identifier::new("x"), (30, 31)),
                                    Node::new(MathOperator::Add, (32, 34)),
                                    Expression::literal(1, (35, 36)),
                                    (30, 36),
                                ),)
                            ])),
                            (28, 38),
                        ),
                        (15, 38),
                    ),
                    (15, 39)
                )),
                Statement::Expression(Expression::identifier(Identifier::new("x"), (40, 41)),),
            ]))
        );
    }

    #[test]
    fn let_statement() {
        let source = "let x = 42;";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([Statement::Let(
                Node::new(
                    LetStatement::Let {
                        identifier: Node::new(Identifier::new("x"), (4, 5)),
                        value: Expression::literal(42, (8, 10)),
                    },
                    (0, 11),
                )
            )]))
        );
    }

    #[test]
    fn let_mut_statement() {
        let source = "let mut x = false;";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([Statement::Let(
                Node::new(
                    LetStatement::LetMut {
                        identifier: Node::new(Identifier::new("x"), (8, 9)),
                        value: Expression::literal(false, (12, 17)),
                    },
                    (0, 18),
                )
            )]))
        );
    }

    #[test]
    fn async_block() {
        let source = "async { x = 42; y = 4.0 }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::block(
                    BlockExpression::Async(AbstractSyntaxTree::with_statements([
                        Statement::ExpressionNullified(Node::new(
                            Expression::operator(
                                OperatorExpression::Assignment {
                                    assignee: Expression::identifier(Identifier::new("x"), (8, 9)),
                                    value: Expression::literal(42, (12, 14)),
                                },
                                (8, 14)
                            ),
                            (8, 15)
                        )),
                        Statement::Expression(Expression::operator(
                            OperatorExpression::Assignment {
                                assignee: Expression::identifier(Identifier::new("y"), (16, 17)),
                                value: Expression::literal(4.0, (20, 23)),
                            },
                            (16, 23)
                        ))
                    ])),
                    (0, 25)
                ))
            ]))
        );
    }

    #[test]
    fn tuple_struct_access() {
        let source = "Foo(42, \"bar\").0";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::tuple_access(
                    Expression::call(
                        Expression::identifier(Identifier::new("Foo"), (0, 3)),
                        vec![
                            Expression::literal(42, (4, 6)),
                            Expression::literal("bar".to_string(), (8, 13)),
                        ],
                        (0, 15)
                    ),
                    Node::new(0, (15, 16)),
                    (0, 16)
                ))
            ]))
        );
    }

    #[test]
    fn fields_struct_instantiation() {
        let source = "Foo { a: 42, b: 4.0 }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::r#struct(
                    StructExpression::Fields {
                        name: Node::new(Identifier::new("Foo"), (0, 3)),
                        fields: vec![
                            (
                                Node::new(Identifier::new("a"), (6, 7)),
                                Expression::literal(42, (9, 11)),
                            ),
                            (
                                Node::new(Identifier::new("b"), (13, 14)),
                                Expression::literal(4.0, (16, 19))
                            )
                        ]
                    },
                    (0, 21)
                ))
            ]))
        );
    }

    #[test]
    fn fields_struct() {
        let source = "struct Foo { a: int, b: float }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::struct_definition(
                    StructDefinition::Fields {
                        name: Node::new(Identifier::new("Foo"), (7, 10)),
                        fields: vec![
                            (
                                Node::new(Identifier::new("a"), (13, 14)),
                                Node::new(Type::Integer, (16, 19))
                            ),
                            (
                                Node::new(Identifier::new("b"), (21, 22)),
                                Node::new(Type::Float, (24, 29))
                            )
                        ]
                    },
                    (0, 31)
                )
            ]))
        );
    }

    #[test]
    fn tuple_struct_instantiation() {
        let source = "Foo(1, 2.0)";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::call(
                    Expression::identifier(Identifier::new("Foo"), (0, 3)),
                    vec![
                        Expression::literal(1, (4, 5)),
                        Expression::literal(2.0, (7, 10))
                    ],
                    (0, 11)
                ))
            ]))
        );
    }

    #[test]
    fn tuple_struct_definition() {
        let source = "struct Foo(int, float);";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::StructDefinition(Node::new(
                    StructDefinition::Tuple {
                        name: Node::new(Identifier::new("Foo"), (7, 10)),
                        items: vec![
                            Node::new(Type::Integer, (11, 14)),
                            Node::new(Type::Float, (16, 21)),
                        ],
                    },
                    (0, 23)
                ))
            ]))
        );
    }

    #[test]
    fn unit_struct() {
        let source = "struct Foo;";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::StructDefinition(Node::new(
                    StructDefinition::Unit {
                        name: Node::new(Identifier::new("Foo"), (7, 10)),
                    },
                    (0, 11)
                ))
            ]))
        );
    }

    #[test]
    fn list_index_nested() {
        let source = "[1, [2], 3][1][0]";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::list_index(
                    Expression::list_index(
                        Expression::list(
                            [
                                Expression::literal(1, (1, 2)),
                                Expression::list([Expression::literal(2, (5, 6))], (4, 7)),
                                Expression::literal(3, (9, 10)),
                            ],
                            (0, 11)
                        ),
                        Expression::literal(1, (12, 13)),
                        (0, 14)
                    ),
                    Expression::literal(0, (15, 16)),
                    (0, 17)
                ),)
            ]))
        );
    }

    #[test]
    fn range() {
        let source = "0..42";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::exclusive_range(
                    Expression::literal(0, (0, 1)),
                    Expression::literal(42, (3, 5)),
                    (0, 5)
                ))
            ]))
        );
    }

    #[test]
    fn negate_variable() {
        let source = "a = 1; -a";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::ExpressionNullified(Node::new(
                    Expression::operator(
                        OperatorExpression::Assignment {
                            assignee: Expression::identifier(Identifier::new("a"), (0, 1)),
                            value: Expression::literal(1, (4, 5)),
                        },
                        (0, 5)
                    ),
                    (0, 6)
                )),
                Statement::Expression(Expression::operator(
                    OperatorExpression::Negation(Expression::identifier(
                        Identifier::new("a"),
                        (8, 9)
                    )),
                    (7, 9)
                ))
            ]))
        );
    }

    #[test]
    fn negate_expression() {
        let source = "-(1 + 1)";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::operator(
                    OperatorExpression::Negation(Expression::grouped(
                        Expression::operator(
                            OperatorExpression::Math {
                                left: Expression::literal(1, (2, 3)),
                                operator: Node::new(MathOperator::Add, (4, 5)),
                                right: Expression::literal(1, (6, 7)),
                            },
                            (2, 7)
                        ),
                        (1, 8)
                    )),
                    (0, 8)
                ))
            ]))
        );
    }

    #[test]
    fn not_expression() {
        let source = "!(1 > 42)";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::operator(
                    OperatorExpression::Not(Expression::grouped(
                        Expression::operator(
                            OperatorExpression::Comparison {
                                left: Expression::literal(1, (2, 3)),
                                operator: Node::new(ComparisonOperator::GreaterThan, (4, 5)),
                                right: Expression::literal(42, (6, 8)),
                            },
                            (2, 8)
                        ),
                        (1, 9)
                    )),
                    (0, 9)
                ))
            ]))
        );
    }

    #[test]
    fn not_variable() {
        let source = "a = false; !a";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::ExpressionNullified(Node::new(
                    Expression::operator(
                        OperatorExpression::Assignment {
                            assignee: Expression::identifier(Identifier::new("a"), (0, 1)),
                            value: Expression::literal(false, (4, 9)),
                        },
                        (0, 9)
                    ),
                    (0, 10)
                )),
                Statement::Expression(Expression::operator(
                    OperatorExpression::Not(Expression::identifier(Identifier::new("a"), (12, 13))),
                    (11, 13)
                )),
            ]))
        );
    }

    #[test]
    fn r#if() {
        let source = "if x { y }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::r#if(
                    IfExpression::If {
                        condition: Expression::identifier(Identifier::new("x"), (3, 4)),
                        if_block: Node::new(
                            BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                                Statement::Expression(Expression::identifier(
                                    Identifier::new("y"),
                                    (7, 8)
                                ))
                            ])),
                            (5, 10)
                        )
                    },
                    (0, 10)
                ))
            ]))
        );
    }

    #[test]
    fn if_else() {
        let source = "if x { y } else { z }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::r#if(
                    IfExpression::IfElse {
                        condition: Expression::identifier(Identifier::new("x"), (3, 4)),
                        if_block: Node::new(
                            BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                                Statement::Expression(Expression::identifier(
                                    Identifier::new("y"),
                                    (7, 8)
                                ))
                            ])),
                            (5, 10)
                        ),
                        r#else: ElseExpression::Block(Node::new(
                            BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                                Statement::Expression(Expression::identifier(
                                    Identifier::new("z"),
                                    (18, 19)
                                ))
                            ])),
                            (16, 21)
                        ))
                    },
                    (0, 21)
                ))
            ]))
        );
    }

    #[test]
    fn if_else_if_else() {
        let source = "if x { y } else if z { a } else { b }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::r#if(
                    IfExpression::IfElse {
                        condition: Expression::identifier(Identifier::new("x"), (3, 4)),
                        if_block: Node::new(
                            BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                                Statement::Expression(Expression::identifier(
                                    Identifier::new("y"),
                                    (7, 8)
                                ))
                            ])),
                            (5, 10)
                        ),
                        r#else: ElseExpression::If(Node::new(
                            Box::new(IfExpression::IfElse {
                                condition: Expression::identifier(Identifier::new("z"), (19, 20)),
                                if_block: Node::new(
                                    BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                                        Statement::Expression(Expression::identifier(
                                            Identifier::new("a"),
                                            (23, 24)
                                        ))
                                    ])),
                                    (21, 26)
                                ),
                                r#else: ElseExpression::Block(Node::new(
                                    BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                                        Statement::Expression(Expression::identifier(
                                            Identifier::new("b"),
                                            (34, 35)
                                        ))
                                    ])),
                                    (32, 37)
                                )),
                            }),
                            (16, 37)
                        )),
                    },
                    (0, 37)
                ))
            ]))
        )
    }

    #[test]
    fn while_loop() {
        let source = "while x < 10 { x += 1 }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::while_loop(
                    Expression::operator(
                        OperatorExpression::Comparison {
                            left: Expression::identifier(Identifier::new("x"), (6, 7)),
                            operator: Node::new(ComparisonOperator::LessThan, (8, 9)),
                            right: Expression::literal(10, (10, 12)),
                        },
                        (6, 12)
                    ),
                    Node::new(
                        BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                            Statement::Expression(Expression::operator(
                                OperatorExpression::CompoundAssignment {
                                    assignee: Expression::identifier(
                                        Identifier::new("x"),
                                        (15, 16)
                                    ),
                                    operator: Node::new(MathOperator::Add, (17, 19)),
                                    modifier: Expression::literal(1, (20, 21)),
                                },
                                (15, 21)
                            ))
                        ])),
                        (13, 23)
                    ),
                    (0, 23)
                ))
            ]))
        )
    }

    #[test]
    fn add_assign() {
        let source = "a += 1";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::operator(
                    OperatorExpression::CompoundAssignment {
                        assignee: Expression::identifier(Identifier::new("a"), (0, 1)),
                        operator: Node::new(MathOperator::Add, (2, 4)),
                        modifier: Expression::literal(1, (5, 6)),
                    },
                    (0, 6)
                ))
            ]))
        )
    }

    #[test]
    fn or() {
        let source = "true || false";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::operator(
                    OperatorExpression::Logic {
                        left: Expression::literal(true, (0, 4)),
                        operator: Node::new(LogicOperator::Or, (5, 7)),
                        right: Expression::literal(false, (8, 13)),
                    },
                    (0, 13)
                ))
            ]))
        )
    }

    #[test]
    fn block_with_one_statement() {
        let source = "{ 40 + 2 }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::block(
                    BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                        Statement::Expression(Expression::operator(
                            OperatorExpression::Math {
                                left: Expression::literal(40, (2, 4)),
                                operator: Node::new(MathOperator::Add, (5, 6)),
                                right: Expression::literal(2, (7, 8)),
                            },
                            (2, 8)
                        ))
                    ])),
                    (0, 10)
                ))
            ]))
        )
    }

    #[test]
    fn block_with_assignment() {
        let source = "{ foo = 42; bar = 42; baz = \"42\" }";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::block(
                    BlockExpression::Sync(AbstractSyntaxTree::with_statements([
                        Statement::ExpressionNullified(Node::new(
                            Expression::assignment(
                                Expression::identifier("foo", (2, 5)),
                                Expression::literal(42, (8, 10)),
                                (2, 10)
                            ),
                            (2, 11)
                        )),
                        Statement::ExpressionNullified(Node::new(
                            Expression::assignment(
                                Expression::identifier("bar", (12, 15)),
                                Expression::literal(42, (18, 20)),
                                (12, 20)
                            ),
                            (12, 21)
                        )),
                        Statement::Expression(Expression::assignment(
                            Expression::identifier("baz", (22, 25)),
                            Expression::literal("42", (28, 32)),
                            (22, 32)
                        ))
                    ])),
                    (0, 34)
                ))
            ]))
        )
    }

    #[test]
    fn equal() {
        let source = "42 == 42";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::comparison(
                    Expression::literal(42, (0, 2)),
                    Node::new(ComparisonOperator::Equal, (3, 5)),
                    Expression::literal(42, (6, 8)),
                    (0, 8)
                ))
            ]))
        );
    }

    #[test]
    fn less_than() {
        let source = "1 < 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::comparison(
                    Expression::literal(1, (0, 1)),
                    Node::new(ComparisonOperator::LessThan, (2, 3)),
                    Expression::literal(2, (4, 5)),
                    (0, 5)
                ))
            ]))
        );
    }

    #[test]
    fn less_than_or_equal() {
        let source = "1 <= 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::comparison(
                    Expression::literal(1, (0, 1)),
                    Node::new(ComparisonOperator::LessThanOrEqual, (2, 4)),
                    Expression::literal(2, (5, 6)),
                    (0, 6)
                ))
            ]))
        );
    }

    #[test]
    fn greater_than_or_equal() {
        let source = "1 >= 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::comparison(
                    Expression::literal(1, (0, 1)),
                    Node::new(ComparisonOperator::GreaterThanOrEqual, (2, 4)),
                    Expression::literal(2, (5, 6)),
                    (0, 6)
                ))
            ]))
        );
    }

    #[test]
    fn greater_than() {
        let source = "1 > 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::comparison(
                    Expression::literal(1, (0, 1)),
                    Node::new(ComparisonOperator::GreaterThan, (2, 3)),
                    Expression::literal(2, (4, 5)),
                    (0, 5)
                ))
            ]))
        );
    }

    #[test]
    fn subtract_negative_integers() {
        let source = "-1 - -2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::math(
                    Expression::literal(-1, (0, 2)),
                    MathOperator::subtract((3, 4)),
                    Expression::literal(-2, (5, 7)),
                    (0, 7)
                ))
            ]))
        );
    }

    #[test]
    fn modulo() {
        let source = "42 % 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::math(
                    Expression::literal(42, (0, 2)),
                    MathOperator::modulo((3, 4)),
                    Expression::literal(2, (5, 6)),
                    (0, 6)
                ))
            ]))
        );
    }

    #[test]
    fn divide() {
        let source = "42 / 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::math(
                    Expression::literal(42, (0, 2)),
                    MathOperator::divide((3, 4)),
                    Expression::literal(2, (5, 6)),
                    (0, 6)
                ))
            ]))
        );
    }

    #[test]
    fn string_concatenation() {
        let source = "\"Hello, \" + \"World!\"";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::math(
                    Expression::literal("Hello, ", (0, 9)),
                    MathOperator::add((10, 11)),
                    Expression::literal("World!", (12, 20)),
                    (0, 20)
                ))
            ]))
        );
    }

    #[test]
    fn string() {
        let source = "\"Hello, World!\"";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::literal("Hello, World!", (0, 15)))
            ]))
        );
    }

    #[test]
    fn boolean() {
        let source = "true";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::literal(true, (0, 4)))
            ]))
        );
    }

    #[test]
    fn list_index() {
        let source = "[1, 2, 3][0]";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::list_index(
                    Expression::list(
                        [
                            Expression::literal(1, (1, 2)),
                            Expression::literal(2, (4, 5)),
                            Expression::literal(3, (7, 8)),
                        ],
                        (0, 9)
                    ),
                    Expression::literal(0, (10, 11)),
                    (0, 12)
                ))
            ]))
        );
    }

    #[test]
    fn property_access() {
        let source = "a.b";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::field_access(
                    Expression::identifier(Identifier::new("a"), (0, 1)),
                    Node::new(Identifier::new("b"), (2, 3)),
                    (0, 3)
                ))
            ]))
        );
    }

    #[test]
    fn complex_list() {
        let source = "[1, 1 + 1, 2 + (4 * 10)]";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::list(
                    [
                        Expression::literal(1, (1, 2)),
                        Expression::math(
                            Expression::literal(1, (4, 5)),
                            MathOperator::add((6, 7)),
                            Expression::literal(1, (8, 9)),
                            (4, 9)
                        ),
                        Expression::math(
                            Expression::literal(2, (11, 12)),
                            Node::new(MathOperator::Add, (13, 14)),
                            Expression::grouped(
                                Expression::math(
                                    Expression::literal(4, (16, 17)),
                                    MathOperator::multiply((18, 19)),
                                    Expression::literal(10, (20, 22)),
                                    (16, 22)
                                ),
                                (15, 23)
                            ),
                            (11, 23)
                        ),
                    ],
                    (0, 24)
                ))
            ]))
        );
    }

    #[test]
    fn list() {
        let source = "[1, 2]";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::list(
                    [
                        Expression::literal(1, (1, 2)),
                        Expression::literal(2, (4, 5))
                    ],
                    (0, 6)
                ))
            ]))
        );
    }

    #[test]
    fn empty_list() {
        let source = "[]";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::list(vec![], (0, 2)))
            ]))
        );
    }

    #[test]
    fn float() {
        let source = "42.0";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::literal(42.0, (0, 4)))
            ]))
        );
    }

    #[test]
    fn add() {
        let source = "1 + 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::math(
                    Expression::literal(1, (0, 1)),
                    MathOperator::add((2, 3)),
                    Expression::literal(2, (4, 5)),
                    (0, 5)
                ))
            ]))
        );
    }

    #[test]
    fn multiply() {
        let source = "1 * 2";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::math(
                    Expression::literal(1, (0, 1)),
                    MathOperator::multiply((2, 3)),
                    Expression::literal(2, (4, 5)),
                    (0, 5)
                ),)
            ]))
        );
    }

    #[test]
    fn add_and_multiply() {
        let source = "1 + 2 * 3";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::math(
                    Expression::literal(1, (0, 1)),
                    MathOperator::add((2, 3)),
                    Expression::math(
                        Expression::literal(2, (4, 5)),
                        MathOperator::multiply((6, 7)),
                        Expression::literal(3, (8, 9)),
                        (4, 9)
                    ),
                    (0, 9)
                )),
            ]))
        );
    }

    #[test]
    fn assignment() {
        let source = "a = 1 + 2 * 3";

        assert_eq!(
            parse(source),
            Ok(AbstractSyntaxTree::with_statements([
                Statement::Expression(Expression::assignment(
                    Expression::identifier("a", (0, 1)),
                    Expression::math(
                        Expression::literal(1, (4, 5)),
                        MathOperator::add((6, 7)),
                        Expression::math(
                            Expression::literal(2, (8, 9)),
                            MathOperator::multiply((10, 11)),
                            Expression::literal(3, (12, 13)),
                            (8, 13)
                        ),
                        (4, 13)
                    ),
                    (0, 13)
                ))
            ]))
        );
    }
}