added equation support &more bug fixes

src/nodes/__init__.py

@@ -7,6 +7,7 @@
 from .function_definition import FunctionDefinition
 from .function_call import FunctionCall
 from .assignment import VariableAssignment
+from .equation import Equation
 
 __all__ = [
     'Node',
@@ -23,4 +24,5 @@
     'FunctionDefinition',
     'FunctionCall',
     'VariableAssignment',
+    'Equation'
 ]

src/nodes/equation.py

@@ -0,0 +1,15 @@
+from .node import Node
+
+class Equation(Node):
+    def __init__(self, left, right):
+        self.left = left
+        self.right = right
+
+    def __repr__(self):
+        return str(
+            { "Equation": {
+                "Left": self.left,
+                "Right": self.right
+            }
+            }
+        )

src/parser.py

@@ -4,7 +4,6 @@
 TODO:
 parenthesis means multiplication ( well sometimes )
 statement separation
-variable assignments
 equation
 inequality
 
@@ -27,7 +26,11 @@ def advance(self):
             self.current_token = None
 
     def parse(self):
-        node = self.parse_expression() 
+        node = self.parse_expression()
+        # fix errors like let x = 4x = 6 -> {'Equation': {'Left': {'Variable': 'x', 'Value': {'BinaryOperation': '*', 'Left': {'Number': '4'}, 'Right': {'Symbol': 'x'}}}, 'Right': None}}
+        if self.current_token != None and self.current_token.value == "=":
+            right = self.parse_expression()
+            return Equation(node, right)
         return node
 
     def parse_expression(self):