hackc: idk, statements and expressions?

1 files changed, 11 insertions, 1 deletions

diff --git a/projects/hackc/expressions.py b/projects/hackc/expressions.py
index de58528..3bc83d2 100644
--- a/projects/hackc/expressions.py
+++ b/projects/hackc/expressions.py
@@ -1,4 +1,5 @@
 from .tokens import Token
+from .utils import *
 
 OPS = "+-*/&|<>="
 UNARY_OPS = "-~"
@@ -15,7 +16,7 @@ class Term:
             return (None, 0)
         if tokens[0].type in ["integer", "string"] or tokens[0] in CONSTANTS:
             return (ConstantTerm(tokens[0]), 1)
-        if tokens[0] in UNARY_OPS:
+        if tokens[0].token in UNARY_OPS:
             term, dt = Term.from_tokens(tokens[1:])
             return (UnaryTerm(tokens[0], term), dt + 1)
 
@@ -24,6 +25,9 @@ class ConstantTerm:
     def __init__(self, term: Token):
         self.term = term
 
+    def __str__(self):
+        return self.term.token
+
 
 class VarTerm:
     def __init__(self, var: Token, subscript=None):
@@ -74,3 +78,9 @@ class Expression:
         t += dt
 
         return (Expression(lhs, op, rhs), t)
+
+    def __str__(self):
+        if self.op is not None:
+            return f"({self.lhs}) {self.op} ({self.rhs})"
+        else:
+            return str(self.lhs)