Update parser.py

parser.py

@@ -2,13 +2,14 @@ from typing import List, Tuple, Any
 
 Token = Tuple[str, str]
 
-# Lista de funciones robóticas válidas
 FUNCTIONS = {
     "MOVE_FORWARD", "MOVE_BACKWARD", "ROTATE", "STOP", "WAIT", "TURN_LEFT",
     "TURN_RIGHT", "TOGGLE_LIGHT", "CALIBRATE", "PRINT", "SET", "READ_SENSOR",
     "ACTIVATE_SENSOR", "DEACTIVATE_SENSOR", "START", "REBOOT", "SHUTDOWN"
 }
 
+TYPES = {"int", "float", "boolean"}
+
 class Parser:
     def __init__(self, tokens: List[Token]):
         self.tokens = tokens
@@ -32,7 +33,9 @@ class Parser:
 
     def instruction(self) -> Any:
         token = self.current()
-        if token[0] == "IF":
+        if token[0] in TYPES:
+            return self.declaration()
+        elif token[0] == "IF":
             return self.if_statement()
         elif token[0] == "WHILE":
             return self.while_statement()
@@ -43,6 +46,12 @@ class Parser:
         else:
             raise SyntaxError(f"Instrucción no válida: {token}")
 
+    def declaration(self):
+        tipo = self.match(self.current()[0])[0]
+        nombre = self.match("IDENTIFIER")[1]
+        self.match("SEMICOLON")
+        return {"type": "declaration", "var": nombre, "datatype": tipo}
+
     def assignment(self):
         ident = self.match("IDENTIFIER")[1]
         self.match("ASSIGN")