Fix: Dictionary indexing support for d[key] inside function calls like print.

Author: dylan-sutton-chavez

compiler/src/main.rs

@@ -29,8 +29,8 @@ fn main() {
 
     initialize_logger();
 
-    let source = "def add(a, b): return a + b\nresult = add(1, 2)";
-
+    let source = "d = {1: 'one', 2: True, 'x': 3.14}\nprint(d)\nprint(d[2])\nprint(d['x'])";
+    
     let chunk = modules::parser::Parser::new(source, modules::lexer::lexer(source)).parse();
 
     // Instructions.

compiler/src/modules/parser.rs

@@ -316,13 +316,13 @@ impl<'src, I: Iterator<Item = Token>> Parser<'src, I> {
     fn dict_literal(&mut self) {
         let mut pairs = 0u16;
         while !matches!(self.peek(), Some(TokenType::Rbrace) | None) {
-            self.expr(); // clave
-            if matches!(self.peek(), Some(TokenType::Colon)) { self.advance(); }
-            self.expr(); // valor
+            self.expr();                    // clave
+            self.eat(TokenType::Colon);     // ← ahora OBLIGATORIO
+            self.expr();                    // valor
             pairs += 1;
-            if matches!(self.peek(), Some(TokenType::Comma)) { self.advance(); }
+            self.eat_if(TokenType::Comma);
         }
-        self.advance(); // consume '}'
+        self.advance(); // '}'
         self.chunk.emit(OpCode::BuildDict, pairs);
     }
 
@@ -483,6 +483,14 @@ impl<'src, I: Iterator<Item = Token>> Parser<'src, I> {
             Some(TokenType::Lpar)  => self.call(name),
             _                      => self.emit_load_ssa(name),
         }
+
+        // NUEVO: soporte para d[expr], d["x"], etc. (funciona en cualquier sitio)
+        while matches!(self.peek(), Some(TokenType::Lsqb)) {
+            self.advance();           // consume [
+            self.expr();
+            self.eat(TokenType::Rsqb);
+            self.chunk.emit(OpCode::GetItem, 0);
+        }
     }
 
     fn assign(&mut self, name: String) {

compiler/tests/cases/parser_cases.json

@@ -75,5 +75,12 @@
     "names": ["x_1", "a_0", "x_2", "b_0", "x_3", "x_4", "x_5", "y_1"],
     "instructions": [["LoadConst",0], ["StoreName",0], ["PopTop",0], ["LoadName",1], ["JumpIfFalse",9], ["LoadConst",1], ["StoreName",2], ["PopTop",0], ["Jump",18], ["LoadName",3], ["JumpIfFalse",15], ["LoadConst",2], ["StoreName",4], ["PopTop",0], ["Jump",18], ["LoadConst",3], ["StoreName",5], ["PopTop",0], ["Phi",6], ["LoadName",6], ["StoreName",7], ["PopTop",0], ["ReturnValue",0]],
     "annotations": {}
+  },
+  {
+  "src": "d = {1: 'one', 2: True, 'x': 3.14}\nprint(d)\nprint(d[2])\nprint(d['x'])",
+  "constants": ["1", "one", "2", "true", "x", "3.14", "2", "x"],
+  "names": ["d_1"],
+  "instructions": [["LoadConst",0], ["LoadConst",1], ["LoadConst",2], ["LoadConst",3], ["LoadConst",4], ["LoadConst",5], ["BuildDict",3], ["StoreName",0], ["PopTop",0], ["LoadName",0], ["CallPrint",0], ["PopTop",0], ["LoadName",0], ["LoadConst",6], ["GetItem",0], ["CallPrint",0], ["PopTop",0], ["LoadName",0], ["LoadConst",7], ["GetItem",0], ["CallPrint",0], ["PopTop",0], ["ReturnValue",0]],
+  "annotations": {}
   }
 ]