Add support for list value nodes (ChilliCream#745)

nigel-sampson · May 15, 2019 · 0d4e164 · 0d4e164
1 parent bb9d0c6
commit 0d4e164
Showing 1 changed file with 15 additions and 15 deletions.
diff --git a/src/Core/Core/Validation/AllVariablesUsedVisitor.cs b/src/Core/Core/Validation/AllVariablesUsedVisitor.cs
@@ -81,30 +81,30 @@ private void VisitArguments(IEnumerable<ArgumentNode> arguments)
         {
             foreach (ArgumentNode argumentNode in arguments)
             {
-                if (argumentNode.Value is VariableNode v)
-                {
-                    _usedVariables.Add(v.Value);
-                }
-
-                if (argumentNode.Value is ObjectValueNode o)
-                {
-                    VisitObjectValue(o);
-                }
+                VisitValue(argumentNode.Value);
             }
         }
 
-        private void VisitObjectValue(ObjectValueNode objectNode)
+        private void VisitValue(IValueNode value)
         {
-            foreach (ObjectFieldNode field in objectNode.Fields)
+            if (value is VariableNode v)
             {
-                if (field.Value is VariableNode v)
+                _usedVariables.Add(v.Value);
+            }
+
+            if (value is ObjectValueNode o)
+            {
+                foreach(ObjectFieldNode field in o.Fields)
                 {
-                    _usedVariables.Add(v.Value);
+                    VisitValue(field.Value);
                 }
+            }
 
-                if (field.Value is ObjectValueNode o)
+            if (value is ListValueNode l)
+            {
+                foreach (IValueNode item in l.Items)
                 {
-                    VisitObjectValue(o);
+                    VisitValue(item);
                 }
             }
         }