Add table parsing wip

Author: luki446

src/ast.rs

@@ -1,3 +1,5 @@
+use std::collections::BTreeMap;
+
 use crate::vm::VirtualMachine;
 
 #[derive(Debug, Clone, PartialEq, PartialOrd)]
@@ -27,6 +29,7 @@ pub enum Expression {
     NumberLiteral(f64),
     BooleanLiteral(bool),
     StringLiteral(String),
+    TableLiteral(BTreeMap<Expression, Expression>),
     NilLiteral,
     IdentifierExpression(String),
     BinaryExpression(Box<Expression>, String, Box<Expression>),
@@ -158,6 +161,7 @@ impl Expression {
                     _ => Err(format!("Function '{}' not found", function_name)),
                 }
             }
+            Expression::TableLiteral(btree_map) => todo!(),
         }
     }
 }

src/lex.rs

@@ -29,6 +29,8 @@ pub enum Token {
     RightParen,
     LeftBracket,
     RightBracket,
+    LeftSquareBracket,
+    RightSquareBracket,
     Assigment,
     Dot,
     Comma,
@@ -161,12 +163,19 @@ impl<'a> Lexer<'a> {
                 '{' => {
                     tokens.push(Token::LeftBracket);
                     self.advance();
-
                 }
                 '}' => {
                     tokens.push(Token::RightBracket);
                     self.advance();
                 }
+                '[' => {
+                    tokens.push(Token::LeftSquareBracket);
+                    self.advance();
+                }
+                ']' => {
+                    tokens.push(Token::RightSquareBracket);
+                    self.advance();
+                }
                 '<' => {
                     if Some('=') == self.input.clone().next() {
                         tokens.push(Token::LessThanOrEqual);

src/parser.rs

@@ -1,3 +1,5 @@
+use std::collections::BTreeMap;
+
 use crate::{
     ast::{Expression, Statement},
     lex::{self, Lexer, LiteralType},
@@ -287,6 +289,15 @@ impl<'a> Parser<'a> {
                         Err("Expected ')'".to_string())
                     }
                 }
+                lex::Token::LeftBracket => {
+                    let table_literal = self.parse_table(tokens)?;
+
+                    if let Some(lex::Token::RightBracket) = tokens.next() {
+                        Ok(table_literal)
+                    } else {
+                        Err("Expected '}'".to_string())
+                    }
+                }
                 lex::Token::Literal(LiteralType::Number(number)) => {
                     Ok(Expression::NumberLiteral(number))
                 }
@@ -423,4 +434,16 @@ impl<'a> Parser<'a> {
             loop_condition,
         })
     }
+
+    fn parse_table(
+        &mut self,
+        tokens: &mut std::iter::Peekable<std::vec::IntoIter<lex::Token>>,
+    ) -> Result<Expression, String> {
+        tokens.next();
+
+        // while tokens.peek() != Some(&lex::Token::RightBracket) {
+        //      tokens.next()
+        // }
+        unimplemented!();
+    }
 }