From f5628441e7a525041d5b3a532490dc21dda13545 Mon Sep 17 00:00:00 2001
From: ahmedsamyh <ahmedsamyh10@gmail.com>
Date: Wed, 20 Nov 2024 13:49:03 +0500
Subject: [PATCH] WIP: parsing Statements.

---
 main.momo |  2 +-
 main.py   | 59 +++++++++++++++++++++++++++++++++----------------------
 2 files changed, 36 insertions(+), 25 deletions(-)

diff --git a/main.momo b/main.momo
index 78ff711..32a41ca 100644
--- a/main.momo
+++ b/main.momo
@@ -1 +1 @@
-position = initial + rate * 60;
+msg: string = "Hello, World";
diff --git a/main.py b/main.py
index 10355c8..1dda1c5 100644
--- a/main.py
+++ b/main.py
@@ -337,8 +337,8 @@ NOTE: ¢ is a zero length string, meaning nothing will be substituted
 Grammar:
     Statement       => Ident :? Ident? =? Expression* ;
     Expression      => Name Binop Name
-    Name            => Value | Ident
-    Value           => Int | Float
+    Name            => LitValue | Ident
+    LitValue        => Int | Float | String
     BinaryOperator  => ArithmeticOp | ComparisionOp | LogicalOp
     ComparisionOp   => > | >= | < | <= | == | !=
     ArithmeticOp    => + | - | * | / | %
@@ -350,15 +350,15 @@ class AstNode:
     def __init__(self, typ: AstNodeType):
         self.typ = typ
 
-class StatementAstNode(AstNode):
+class AstNodeStatement(AstNode):
     def __init__(self):
         __super__(AstNodeType.STMT)
 
-class ExpressionAstNode(AstNode):
+class AstNodeExpression(AstNode):
     def __init__(self):
         __super__(AstNodeType.EXPR)
 
-class IntAstNode(AstNode):
+class AstNodeInt(AstNode):
     def __init__(self, value: int):
         __super__(AstNodeType.IDENT)
         self.value = value
@@ -367,28 +367,37 @@ class Parser:
     def __init__(self, tokens):
         self.tokens = tokens
 
-    def parse(self, tokens: [Token]) -> AstNode:
-        stmt = self.parseStatement(tokens)
+    def syntax_error(self, msg: str, token: Token):
+        fatal(f"{str(token.loc)}: Syntax Error: {msg}")
 
-    def parseStatement(self, tokens: [Token]) -> StatementAstNode:
-        if len(tokens) <= 0: return None
-        t: Token = tokens.pop(0)
-        expr = self.parseExpr(tokens)
-
-    def parseExpr(self, tokens: [Token]) -> ExpressionAstNode:
-        lhs = self.parseIdent(tokens)
-        op  = self.parseBinaryOp(tokens)
-        rhs = self.parseIdent(tokens)
-
-    # def parseIdent(self, tokens: [Token]) -> IdentAstNode:
-    #     if len(tokens) <= 0: return None
-    #     t: Token = tokens.pop(0)
+    def parse(self) -> AstNode:
+        stmt = self.parseStatement()
+        return stmt
 
-    #     if t.typ == TokenType.NUMBER:
-    #         return IdentAstNode(int(t.lexeme))
-    #     elif t.typ == TokenType.IDENT:
-    #         return IdentAstNode(t.lexeme)
+    def parseStatement(self) -> AstNodeStatement:
+        tokens = self.tokens
+        if len(tokens) <= 0: return None
+        var_name = tokens.pop(0)
+        var_typ  = None
+        # Check if colon is there
+        if tokens[0].typ == TokenType.COLON:
+            colon = tokens.pop(0)
+            var_typ = None if len(tokens) <= 0 else tokens.pop(0)
+            eof_msg = "Reached End of File"
+            typ_mismatch_msg = f"Got {token_type_as_str_map[var_typ.typ]}"
+            if var_typ == None or var_typ.typ != TokenType.IDENT:
+                self.syntax_error("Expected type of variable after colon, but %s" % eof_msg if var_typ == None else typ_mismatch_msg, colon)
+
+        dlog(var_name)
+        if var_typ != None:
+            dlog(var_typ)
+
+        print("TODO: Implement parseStatement()")
+        exit(1)
+        # expr = self.parseExpr()
 
+    def parseExpr(self) -> AstNodeExpression:
+        pass
 
 def main():
     program: str = sys.argv.pop(0)
@@ -407,6 +416,8 @@ def main():
     # TODO: Parse
     parser = Parser(tokens)
 
+    parser.parse()
+
     for t in tokens:
         pprint.pp(str(t))
 
-- 
2.39.5