From e4772ab0d4466f3844091ad603938b24c442e75b Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 28 Jul 2023 21:45:24 +0100
Subject: Implement DeclaredVariables for new nodes

---
 .../shedlang/compiler/typechecker/DeclaredVariables.java    | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'src/main/java')

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
-- 
cgit v1.2.3