thp/src/syntax/binding.rs

use super::ast::var_binding::Binding;
use super::utils::{parse_token_type, try_operator};
use super::{expression, ParseResult};
use crate::error_handling::SyntaxError;
use crate::lexic::token::{Token, TokenType};
use crate::utils::Result3;

pub fn try_parse<'a>(tokens: &'a Vec<Token>, pos: usize) -> ParseResult<Binding> {
    let mut current_pos = pos;

    // TODO: Detect if the binding starts with a datatype

    /*
     * let keyword
     */
    let (is_mutable, binding_token, next_pos) = {
        let let_token = parse_token_type(tokens, current_pos, TokenType::LET);
        match let_token {
            ParseResult::Ok(let_token, next_let) => {
                let mut_token = parse_token_type(tokens, next_let, TokenType::MUT);
                match mut_token {
                    ParseResult::Ok(_mut_token, next_mut) => (true, let_token, next_mut),
                    _ => (false, let_token, next_let),
                }
            }
            _ => return ParseResult::Unmatched,
        }
    };
    current_pos = next_pos;

    /*
     * identifier
     */
    let (identifier, next_pos) = match parse_token_type(tokens, current_pos, TokenType::Identifier)
    {
        ParseResult::Ok(t, n) => (t, n),
        ParseResult::Mismatch(token) => {
            // The parser found a token, but it's not an identifier
            return ParseResult::Err(SyntaxError {
                error_start: token.position,
                error_end: token.get_end_position(),
                reason: "??".into(),
            });
        }
        ParseResult::Err(error) => {
            return ParseResult::Err(error);
        }
        _ => {
            // The parser didn't find an Identifier after VAL/VAR
            return ParseResult::Err(SyntaxError {
                reason: format!(
                    "There should be an identifier after a `{}` token",
                    if is_mutable { "val" } else { "var" }
                ),
                error_start: binding_token.position,
                error_end: binding_token.get_end_position(),
            });
        }
    };
    current_pos = next_pos;

    /*
     * Equal (=) operator
     */
    let equal_operator = match try_operator(tokens, current_pos, String::from("=")) {
        Result3::Ok(t) => t,
        Result3::Err(t) => {
            // The parser found a token, but it's not the `=` operator
            return ParseResult::Err(SyntaxError {
                reason: format!("There should be an equal sign `=` after the identifier"),
                error_start: t.position,
                error_end: t.get_end_position(),
            });
        }
        Result3::None => {
            // The parser didn't find the `=` operator after the identifier
            return ParseResult::Err(SyntaxError {
                reason: format!("There should be an equal sign `=` after the identifier",),
                error_start: identifier.position,
                error_end: identifier.get_end_position(),
            });
        }
    };
    current_pos += 1;

    let (expression, next_pos) = match expression::try_parse(tokens, current_pos) {
        ParseResult::Ok(exp, next) => (exp, next),
        _ => {
            return ParseResult::Err(SyntaxError {
                reason: String::from("Expected an expression after the equal `=` operator"),
                error_start: equal_operator.position,
                error_end: equal_operator.get_end_position(),
            });
        }
    };
    current_pos = next_pos;

    let binding = Binding {
        datatype: None,
        identifier: &identifier,
        expression,
        is_mutable,
    };

    ParseResult::Ok(binding, current_pos)
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{lexic::get_tokens, syntax::utils::try_token_type};

    #[test]
    fn should_parse_val_binding() {
        let tokens = get_tokens(&String::from("let identifier = 20")).unwrap();
        let ParseResult::Ok(binding, _) = try_parse(&tokens, 0) else {
            panic!()
        };

        assert_eq!("identifier", format!("{}", binding.identifier.value));
    }

    #[test]
    fn should_parse_val() {
        let tokens = get_tokens(&String::from("let")).unwrap();
        let token = *try_token_type(&tokens, 0, TokenType::LET).unwrap();

        assert_eq!(TokenType::LET, token.token_type);
        assert_eq!("let", token.value);
    }

    #[test]
    fn should_parse_identifier() {
        let tokens = get_tokens(&String::from("identifier")).unwrap();
        let token = *try_token_type(&tokens, 0, TokenType::Identifier).unwrap();

        assert_eq!("identifier", token.value);
    }

    #[test]
    fn should_parse_operator() {
        let tokens = get_tokens(&String::from("=")).unwrap();
        let token = *try_operator(&tokens, 0, String::from("=")).unwrap();

        assert_eq!("=", token.value);
    }

    /*
    #[test]
    fn should_parse_binding_with_datatype() {
        let tokens = get_tokens(&String::from("Num val identifier = 20")).unwrap();
        let ParseResult::Ok(Binding::Val(binding), _) = try_parse(&tokens, 0) else {
            panic!()
        };

        assert_eq!(Some(String::from("Num")), binding.datatype);
        assert_eq!("identifier", format!("{}", binding.identifier));

        let tokens = get_tokens(&String::from("Bool var identifier = 20")).unwrap();
        let ParseResult::Ok(Binding::Var(binding), _) = try_parse(&tokens, 0) else {
            panic!()
        };

        assert_eq!(Some(String::from("Bool")), binding.datatype);
        assert_eq!("identifier", format!("{}", binding.identifier));
    }
     */

    #[test]
    fn should_return_correct_error() {
        let tokens = get_tokens(&String::from("let")).unwrap();
        assert_eq!(TokenType::LET, tokens[0].token_type);
        assert_eq!(0, tokens[0].position);
        let binding = try_parse(&tokens, 0);

        match binding {
            ParseResult::Err(error) => {
                assert_eq!(0, error.error_start);
                assert_eq!(3, error.error_end);
            }
            _ => panic!("Error expected"),
        }
    }

    #[test]
    fn should_return_error_when_identifier_is_wrong() {
        let tokens = get_tokens(&String::from("let 322")).unwrap();
        assert_eq!(TokenType::LET, tokens[0].token_type);
        assert_eq!(0, tokens[0].position);
        let binding = try_parse(&tokens, 0);

        match binding {
            ParseResult::Err(error) => {
                assert_eq!(4, error.error_start);
                assert_eq!(7, error.error_end);
            }
            _ => panic!("Error expected"),
        }

        let tokens = get_tokens(&String::from("let \"hello\"")).unwrap();
        let binding = try_parse(&tokens, 0);

        match binding {
            ParseResult::Err(error) => {
                assert_eq!(4, error.error_start);
                assert_eq!(11, error.error_end);
            }
            _ => panic!("Error expected"),
        }
    }

    #[test]
    fn should_return_error_when_equal_op_is_wrong() {
        let tokens = get_tokens(&String::from("let id \"error\"")).unwrap();
        let binding = try_parse(&tokens, 0);

        match binding {
            ParseResult::Err(error) => {
                assert_eq!(7, error.error_start);
                assert_eq!(14, error.error_end);
            }
            _ => panic!("Error expected"),
        }
    }
}