Update parser.py

parser.py

@@ -2,6 +2,13 @@ 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"
+}
+
 class Parser:
     def __init__(self, tokens: List[Token]):
         self.tokens = tokens
@@ -25,10 +32,14 @@ class Parser:
 
     def instruction(self) -> Any:
         token = self.current()
-        if token[0] == "IDENTIFIER":
-            return self.assignment()
-        elif token[0] == "IF":
+        if token[0] == "IF":
             return self.if_statement()
+        elif token[0] == "WHILE":
+            return self.while_statement()
+        elif token[0] in FUNCTIONS:
+            return self.function_call()
+        elif token[0] == "IDENTIFIER":
+            return self.assignment()
         else:
             raise SyntaxError(f"Instrucción no válida: {token}")
 
@@ -41,7 +52,9 @@ class Parser:
 
     def if_statement(self):
         self.match("IF")
+        self.match("OPEN_PAREN")
         condition = self.expression()
+        self.match("CLOSE_PAREN")
         self.match("THEN")
         self.match("OPEN_BRACE")
         body = []
@@ -50,9 +63,33 @@ class Parser:
         self.match("CLOSE_BRACE")
         return {"type": "if", "condition": condition, "body": body}
 
+    def while_statement(self):
+        self.match("WHILE")
+        self.match("OPEN_PAREN")
+        condition = self.expression()
+        self.match("CLOSE_PAREN")
+        self.match("THEN")
+        self.match("OPEN_BRACE")
+        body = []
+        while self.current()[0] != "CLOSE_BRACE":
+            body.append(self.instruction())
+        self.match("CLOSE_BRACE")
+        return {"type": "while", "condition": condition, "body": body}
+
+    def function_call(self):
+        func_name = self.current()[0]
+        self.match(func_name)
+        self.match("OPEN_PAREN")
+        arg = None
+        if self.current()[0] not in ("CLOSE_PAREN",):
+            arg = self.expression()
+        self.match("CLOSE_PAREN")
+        self.match("SEMICOLON")
+        return {"type": "function", "name": func_name, "arg": arg}
+
     def expression(self):
         left = self.term()
-        while self.current()[0] in ("PLUS", "MINUS"):
+        while self.current()[0] in ("PLUS", "MINUS", "EQUAL", "NOT_EQUAL", "GREATER", "LESS", "AND", "OR"):
             op = self.match(self.current()[0])[0]
             right = self.term()
             left = {"type": "binop", "op": op, "left": left, "right": right}
@@ -64,5 +101,14 @@ class Parser:
             return {"type": "var", "value": self.match("IDENTIFIER")[1]}
         elif token_type in ("INT", "FLOAT"):
             return {"type": "num", "value": self.match(token_type)[1]}
+        elif token_type in ("TRUE", "FALSE"):
+            return {"type": "bool", "value": self.match(token_type)[0]}
+        elif token_type == "STRING":
+            return {"type": "string", "value": self.match("STRING")[1]}
+        elif token_type == "OPEN_PAREN":
+            self.match("OPEN_PAREN")
+            expr = self.expression()
+            self.match("CLOSE_PAREN")
+            return expr
         else:
             raise SyntaxError(f"Expresión inválida: {self.current()}")