Add FragmentArgumentKind

Author: JoviDeCroock

src/__tests__/parser.test.ts

@@ -244,7 +244,7 @@ describe('parse', () => {
         },
         arguments: [
           {
-            kind: 'Argument',
+            kind: 'FragmentArgument',
             name: {
               kind: 'Name',
               value: 'varA',
@@ -255,7 +255,7 @@ describe('parse', () => {
             },
           },
           {
-            kind: 'Argument',
+            kind: 'FragmentArgument',
             name: {
               kind: 'Name',
               value: 'varB',

src/ast.ts

@@ -38,6 +38,7 @@ export type ASTNode = Or<
   | FieldNode
   | ArgumentNode
   | FragmentSpreadNode
+  | FragmentArgumentNode
   | InlineFragmentNode
   | FragmentDefinitionNode
   | IntValueNode
@@ -185,6 +186,13 @@ export type ConstArgumentNode = Or<
   }
 >;
 
+export type FragmentArgumentNode = {
+  readonly kind: Kind.FRAGMENT_ARGUMENT;
+  readonly name: NameNode;
+  readonly value: ValueNode;
+  readonly loc?: Location;
+};
+
 export type FragmentSpreadNode = Or<
   GraphQL.FragmentSpreadNode,
   {
@@ -194,7 +202,7 @@ export type FragmentSpreadNode = Or<
     readonly loc?: Location;
   }
 > & {
-  readonly arguments?: ReadonlyArray<ArgumentNode>;
+  readonly arguments?: ReadonlyArray<FragmentArgumentNode>;
 };
 
 export type InlineFragmentNode = Or<

src/parser.ts

@@ -246,17 +246,22 @@ function value(constant: boolean): ast.ValueNode {
   };
 }
 
-function arguments_(constant: boolean): ast.ArgumentNode[] | undefined {
+function arguments_(
+  constant: boolean,
+  fragmentArgument?: boolean
+): ast.ArgumentNode[] | ast.FragmentArgumentNode[] | undefined {
   if (input.charCodeAt(idx) === 40 /*'('*/) {
-    const args: ast.ArgumentNode[] = [];
+    const args: ast.ArgumentNode[] | ast.FragmentArgumentNode[] = [];
     idx++;
     ignored();
     do {
       const name = nameNode();
       if (input.charCodeAt(idx++) !== 58 /*':'*/) throw error('Argument');
       ignored();
       args.push({
-        kind: 'Argument' as Kind.ARGUMENT,
+        kind: fragmentArgument
+          ? ('FragmentArgument' as Kind.FRAGMENT_ARGUMENT)
+          : ('Argument' as Kind.ARGUMENT),
         name,
         value: value(constant),
       });
@@ -358,7 +363,8 @@ function selectionSet(): ast.SelectionSetNode {
             selections.push({
               kind: 'FragmentSpread' as Kind.FRAGMENT_SPREAD,
               name: nameNode(),
-              arguments: arguments_(false),
+              // @ts-expect-error
+              arguments: arguments_(false, true) as readonly ast.FragmentArgumentNode[],
               directives: directives(false),
             });
           }
@@ -379,7 +385,8 @@ function selectionSet(): ast.SelectionSetNode {
           selections.push({
             kind: 'FragmentSpread' as Kind.FRAGMENT_SPREAD,
             name: nameNode(),
-            arguments: arguments_(false),
+            // @ts-expect-error
+            arguments: arguments_(false, true) as readonly ast.FragmentArgumentNode[],
             directives: directives(false),
           });
       }