Backport GraphQL-Tag fix

0no-co · Oct 31, 2024 · 704c5d9 · 704c5d9
1 parent 0e92930
commit 704c5d9
Show file tree

Hide file tree

Showing 2 changed files with 104 additions and 4 deletions.
diff --git a/src/__tests__/__snapshots__/parser.test.ts.snap b/src/__tests__/__snapshots__/parser.test.ts.snap
@@ -644,7 +644,8 @@ exports[`parse > parses the kitchen sink document like graphql.js does 1`] = `
                       "value": {
                         "block": true,
                         "kind": "StringValue",
-                        "value": "block string uses """",
+                        "value": "block string uses """
+",
                       },
                     },
                   ],
@@ -768,5 +769,86 @@ exports[`parse > parses the kitchen sink document like graphql.js does 1`] = `
     },
   ],
   "kind": "Document",
+  "loc": {
+    "end": 1257,
+    "source": {
+      "body": "# Copyright (c) 2015-present, Facebook, Inc.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
+query queryName($foo: ComplexType, $site: Site = MOBILE) @onQuery {
+  whoever123is: node(id: [123, 456]) {
+    id
+    ... on User @onInlineFragment {
+      field2 {
+        id
+        alias: field1(first: 10, after: $foo) @include(if: $foo) {
+          id
+          ...frag @onFragmentSpread
+        }
+      }
+    }
+    ... @skip(unless: $foo) {
+      id
+    }
+    ... {
+      id
+    }
+  }
+}
+
+mutation likeStory @onMutation {
+  like(story: 123) @onField {
+    story {
+      id @onField
+    }
+  }
+}
+
+subscription StoryLikeSubscription($input: StoryLikeSubscribeInput)
+@onSubscription {
+  storyLikeSubscribe(input: $input) {
+    story {
+      likers {
+        count
+      }
+      likeSentence {
+        text
+      }
+    }
+  }
+}
+
+fragment frag on Friend @onFragmentDefinition {
+  foo(
+    size: $site
+    bar: 12
+    obj: {
+      key: "value"
+      block: """
+      block string uses \\"""
+      """
+    }
+  )
+}
+
+query teeny {
+  unnamed(truthy: true, falsey: false, nullish: null)
+  query
+}
+
+query tiny {
+  __typename
+}
+",
+      "locationOffset": {
+        "column": 1,
+        "line": 1,
+      },
+      "name": "graphql.web",
+    },
+    "start": 0,
+  },
 }
 `;
diff --git a/src/parser.ts b/src/parser.ts
@@ -483,7 +483,7 @@ function operationDefinition(
   }
 }
 
-function document(): ast.DocumentNode {
+function document(input: string, noLoc: boolean): ast.DocumentNode {
   let match: string | undefined;
   let definition: ast.OperationDefinitionNode | undefined;
   ignored();
@@ -498,6 +498,24 @@ function document(): ast.DocumentNode {
       throw error('Document');
     }
   } while (idx < input.length);
+
+  if (!noLoc) {
+    return {
+      kind: 'Document' as Kind.DOCUMENT,
+      definitions,
+      loc: {
+        start: 0,
+        end: input.length,
+        // @ts-ignore
+        source: {
+          body: input,
+          name: 'graphql.web',
+          locationOffset: { line: 1, column: 1 },
+        },
+      },
+    };
+  }
+
   return {
     kind: 'Document' as Kind.DOCUMENT,
     definitions,
@@ -510,11 +528,11 @@ type ParseOptions = {
 
 export function parse(
   string: string | Source,
-  _options?: ParseOptions | undefined
+  options?: ParseOptions | undefined
 ): ast.DocumentNode {
   input = typeof string.body === 'string' ? string.body : string;
   idx = 0;
-  return document();
+  return document(input, options && options.noLocation);
 }
 
 export function parseValue(