Implement DeclaredVariables for new nodes

1 files changed, 13 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java b/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java
index 5627d462..faa8f4e2 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java
@@ -22,6 +22,9 @@ class DeclaredVariables {
         PVector<UntypedVariableDeclarationNode> variables
     ) {
         return switch (statement) {
+            case UntypedAssignNode assign ->
+                variables;
+
             case UntypedBlockReturnNode blockReturn ->
                 variables;
 
@@ -33,6 +36,9 @@ class DeclaredVariables {
 
             case UntypedValNode val ->
                 forVal(val, variables);
+
+            case UntypedVarNode var ->
+                forVar(var, variables);
         };
     }
 
@@ -67,6 +73,13 @@ class DeclaredVariables {
         return forTarget(statement.target(), variables);
     }
 
+    static PVector<UntypedVariableDeclarationNode> forVar(
+        UntypedVarNode statement,
+        PVector<UntypedVariableDeclarationNode> variables
+    ) {
+        return forTarget(statement.target(), variables);
+    }
+
     private static PVector<UntypedVariableDeclarationNode> forTarget(
         UntypedTargetNode target,
         PVector<UntypedVariableDeclarationNode> variables