hackc: LetStatement accepts SubscriptTermHEADmain

1 files changed, 14 insertions, 8 deletions

diff --git a/projects/hackc/statements.py b/projects/hackc/statements.py
index 458ba01..dc76228 100644
--- a/projects/hackc/statements.py
+++ b/projects/hackc/statements.py
@@ -1,4 +1,4 @@
-from .expressions import Expression, SubroutineCall
+from .expressions import Expression, SubroutineCall, VarTerm, SubscriptTerm
 from .utils import *
 
 
@@ -55,19 +55,25 @@ class LetStatement:
         """Construct let statement.
 
         Format:
-        'let' <name> '=' <expression> ';'
+        'let' (<var term> | <subscript term>) '=' <expression> ';'
         """
         if len(tokens) < 5 or tokens[0] != "let":
             return (None, 0)
 
-        name = tokens[1]
-        if name.type != "identifier":
-            raise UnexpectedToken("variable name", name)
+        if tokens[1].type != "identifier":
+            raise UnexpectedToken("variable name", tokens[1])
 
-        if tokens[2] != "=":
-            raise UnexpectedToken("=", tokens[2])
+        t = 1
+        name, dt = SubscriptTerm.from_tokens(tokens[1:])
+        if name is None:
+            name = VarTerm(tokens[1])
+            dt = 1
+        t += dt
+
+        if tokens[t] != "=":
+            raise UnexpectedToken("=", tokens[t])
+        t += 1
 
-        t = 3
         expr, dt = Expression.from_tokens(tokens[t:])
         if expr is None:
             raise UnexpectedToken(f"Expected expression", tokens[3])