Fix buildVariableDefinitions

.github/workflows/ci.yml

@@ -35,7 +35,10 @@ jobs:
           run_install: true
 
       - name: 'Build'
-        run: yarn build
+        run: pnpm build
+
+      - name: 'Type Test'
+        run: pnpm type-test
 
       - name: 'Test'
-        run: yarn test
+        run: pnpm test

src/AST.ts

@@ -251,6 +251,7 @@ export const document = <T extends ReadonlyArray<Definition>>(
   definitions,
 });
 
+// @todo use the proper `astFromValue`
 export const toValueNode = (value: any, enums: any[] = []): ValueNode => {
   if (typeof value === "string") {
     if (enums.some((e) => Object.values(e).includes(value)))

src/Selection.ts

@@ -79,7 +79,6 @@ export class Selection<
     const op = this.type.toLowerCase() as OperationTypeNode;
     const selectionSet = this.toSelectionSet();
     const variableDefinitions = buildVariableDefinitions(
-      op,
       this.schema,
       selectionSet
     );

src/Variables.ts

@@ -1,6 +1,6 @@
 import type { GraphQLSchema } from "graphql";
 import { Kind, visitWithTypeInfo, TypeInfo, visit } from "graphql";
-import { O, U } from "ts-toolbelt";
+import type { O, U } from "ts-toolbelt";
 
 import {
   SelectionSet,
@@ -13,41 +13,44 @@ import {
   variable,
   variableDefinition,
   NamedType,
+  operation,
 } from "./AST";
 
 export const $ = <Name extends string>(name: Name): Variable<Name> =>
   variable(name);
 
 export const buildVariableDefinitions = <T extends SelectionSet<any>>(
-  op: "query" | "mutation" | "subscription",
   schema: GraphQLSchema,
   selectionSet: T
 ): Array<VariableDefinition<any, any>> => {
   const variableDefinitions: VariableDefinition<any, any>[] = [];
   const typeInfo = new TypeInfo(schema);
+
+  // @note need to wrap selectionset in operation for TypeInfo to track correctly
+  const operationDefinition = operation(
+    "query",
+    "",
+    selectionSet,
+    variableDefinitions
+  );
+
   const visitor = visitWithTypeInfo(typeInfo, {
-    [Kind.ARGUMENT]: (argNode) => {
-      if (isVariable(argNode.value) && typeInfo.getArgument()?.astNode?.type) {
+    [Kind.ARGUMENT]: (node) => {
+      const type = typeInfo.getArgument()?.astNode?.type!;
+
+      if (node.value.kind === "Variable") {
         // define the `VariableDefinition`
-        variableDefinitions.push(
-          variableDefinition(
-            argNode.value,
-            typeInfo.getArgument()?.astNode?.type!
-          )
-        );
+        variableDefinitions.push(variableDefinition(node.value, type));
       }
     },
   });
 
   // @todo return from here
-  visit(selectionSet, visitor);
+  visit(operationDefinition, visitor);
 
   return variableDefinitions;
 };
 
-export const isVariable = (value: any): value is Variable<string> =>
-  typeof value === "object" && value.kind === Kind.VARIABLE;
-
 // @note Traverse the AST extracting `Argument` nodes w/ values of `Variable`.
 // Extract the type the Variable value needs to be against/the Schema.
 export type Variables<

src/__tests__/Variables.test.ts

@@ -29,7 +29,6 @@ describe("Variables", () => {
 
     // @note we need a way to get the input type at runtime
     const variableDefinitions = buildVariableDefinitions(
-      "query",
       schema,
       selectionSet(selection)
     );
@@ -48,7 +47,7 @@ describe("Variables", () => {
 
       type User {
         id: ID!
-        friends(limit: Int = 10): [User!]
+        friends(limit: Int!): [User!]
       }
     `,
       { noLocation: true }
@@ -69,7 +68,6 @@ describe("Variables", () => {
     ]);
 
     const variableDefinitions = buildVariableDefinitions(
-      "query",
       schema,
       selectionSet(selection)
     );