Add DocumentNodeLike type

0no-co · Jan 10, 2024 · 479632a · 479632a
1 parent a985f12
commit 479632a
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/src/api.ts b/src/api.ts
@@ -2,7 +2,7 @@ import { parse as _parse } from '@0no-co/graphql.web';
 import type { stringLiteral } from './utils';
 import type { parseDocument } from './parser';
 
-function parse<const Input extends stringLiteral<Input>>(input: Input): parseDocument<Input> {
+function parse<const In extends stringLiteral<In>>(input: In): parseDocument<In> {
   return _parse(input) as any;
 }
 

diff --git a/src/fragments.ts b/src/fragments.ts
@@ -1,4 +1,5 @@
 import type { Kind } from '@0no-co/graphql.web';
+import type { DocumentNodeLike } from './parser';
 
 type _getFragmentMapRec<Definitions> = Definitions extends readonly [
   infer Definition,
@@ -10,5 +11,6 @@ type _getFragmentMapRec<Definitions> = Definitions extends readonly [
       _getFragmentMapRec<Rest>
   : {};
 
-export type getFragmentMap<Document extends { kind: Kind.DOCUMENT; definitions: any[] }> =
-  _getFragmentMapRec<Document['definitions']>;
+export type getFragmentMap<Document extends DocumentNodeLike> = _getFragmentMapRec<
+  Document['definitions']
+>;
diff --git a/src/parser.ts b/src/parser.ts
@@ -506,4 +506,9 @@ type parseType<In> = takeType<skipIgnored<In>> extends [infer Node, string] ? No
 
 type parseOperation<In> = TakeOperation<skipIgnored<In>> extends [infer Node, string] ? Node : void;
 
+export type DocumentNodeLike = {
+  kind: Kind.DOCUMENT;
+  definitions: any[];
+};
+
 export type { parseConstValue, parseOperation, parseDocument, parseValue, parseType };
diff --git a/src/selection.ts b/src/selection.ts
@@ -9,6 +9,7 @@ import type {
 import type { tada } from './namespace';
 import type { obj, objValues } from './utils';
 import type { getFragmentMap } from './fragments';
+import type { DocumentNodeLike } from './parser';
 
 import type {
   IntrospectionField,
@@ -217,7 +218,7 @@ type getDefinitionSelectionRec<
   : {};
 
 type getDocumentType<
-  Document extends { kind: Kind.DOCUMENT; definitions: any[] },
+  Document extends DocumentNodeLike,
   Introspection extends IntrospectionLikeType,
   Fragments extends { [name: string]: any } = {},
 > = getDefinitionSelectionRec<
@@ -227,7 +228,7 @@ type getDocumentType<
 >;
 
 type getFragmentType<
-  Document extends { kind: Kind.DOCUMENT; definitions: any[] },
+  Document extends DocumentNodeLike,
   Introspection extends IntrospectionLikeType,
   Fragments extends { [name: string]: any } = {},
 > = Document['definitions'][0] extends {