@sanity/codegen 5.9.1 → 5.9.2

package/dist/typescript/schemaTypeGenerator.js

@@ -16,6 +16,7 @@ export class SchemaTypeGenerator {
             tsType
         };
     });
+    arrayOfUsed = false;
     identifiers = new Map();
     tsTypes = new Map();
     constructor(schema){
@@ -50,6 +51,9 @@ export class SchemaTypeGenerator {
     hasType(typeName) {
         return this.tsTypes.has(typeName);
     }
+    isArrayOfUsed() {
+        return this.arrayOfUsed;
+    }
     *[Symbol.iterator]() {
         for (const { name } of this.schema){
             yield {
@@ -65,6 +69,7 @@ export class SchemaTypeGenerator {
    * Helper function used to generate TS types for arrays of inline types, or arrays of inline types
    * wrapped in the ArrayOf wrapper that adds _key prop
    */ generateArrayOfTsType(typeNode) {
+        this.arrayOfUsed = true;
         const typeNodes = this.generateTsType(typeNode.of);
         return t.tsTypeReference(ARRAY_OF, t.tsTypeParameterInstantiation([
             typeNodes

package/dist/typescript/schemaTypeGenerator.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/typescript/schemaTypeGenerator.ts"],"sourcesContent":["import * as t from '@babel/types'\nimport {\n  type ArrayTypeNode,\n  type DocumentSchemaType,\n  type InlineTypeNode,\n  type ObjectAttribute,\n  type ObjectTypeNode,\n  type SchemaType,\n  type TypeDeclarationSchemaType,\n  typeEvaluate,\n  type TypeNode,\n  type UnionTypeNode,\n} from 'groq-js'\n\nimport {safeParseQuery} from '../safeParseQuery.js'\nimport {ARRAY_OF, INTERNAL_REFERENCE_SYMBOL} from './constants.js'\nimport {\n  getFilterArrayUnionType,\n  getUniqueIdentifierForName,\n  isIdentifierName,\n  weakMapMemo,\n} from './helpers.js'\nimport {type ExtractedQuery, type TypeEvaluationStats} from './types.js'\n\nexport class SchemaTypeGenerator {\n  public readonly schema: SchemaType\n  evaluateQuery = weakMapMemo(\n    // eslint-disable-next-line unicorn/consistent-function-scoping\n    ({query}: Pick<ExtractedQuery, 'query'>): {stats: TypeEvaluationStats; tsType: t.TSType} => {\n      const ast = safeParseQuery(query)\n      const typeNode = typeEvaluate(ast, this.schema)\n      const tsType = this.generateTsType(typeNode)\n      const stats = walkAndCountQueryTypeNodeStats(typeNode)\n      return {stats, tsType}\n    },\n  )\n  private identifiers = new Map<string, t.Identifier>()\n\n  private tsTypes = new Map<string, t.TSType>()\n\n  constructor(schema: SchemaType) {\n    this.schema = schema\n\n    const uniqueTypeNames = new Set<string>()\n    for (const type of schema) {\n      if (uniqueTypeNames.has(type.name)) {\n        throw new Error(\n          `Duplicate type name \"${type.name}\" in schema. Type names must be unique within the same schema.`,\n        )\n      }\n      uniqueTypeNames.add(type.name)\n    }\n\n    for (const type of schema) {\n      const currentIdentifierNames = new Set([...this.identifiers.values()].map((id) => id.name))\n      const uniqueIdentifier = getUniqueIdentifierForName(type.name, currentIdentifierNames)\n      this.identifiers.set(type.name, uniqueIdentifier)\n    }\n\n    for (const type of schema) {\n      this.tsTypes.set(type.name, this.generateTsType(type))\n    }\n  }\n\n  getType(typeName: string): {id: t.Identifier; tsType: t.TSType} | undefined {\n    const tsType = this.tsTypes.get(typeName)\n    const id = this.identifiers.get(typeName)\n    if (tsType && id) return {id, tsType}\n    return undefined\n  }\n\n  hasType(typeName: string): boolean {\n    return this.tsTypes.has(typeName)\n  }\n\n  *[Symbol.iterator]() {\n    for (const {name} of this.schema) {\n      yield {name, ...this.getType(name)!}\n    }\n  }\n\n  typeNames(): string[] {\n    return this.schema.map((schemaType) => schemaType.name)\n  }\n\n  /**\n   * Helper function used to generate TS types for arrays of inline types, or arrays of inline types\n   * wrapped in the ArrayOf wrapper that adds _key prop\n   */\n  private generateArrayOfTsType(typeNode: ArrayTypeNode): t.TSTypeReference {\n    const typeNodes = this.generateTsType(typeNode.of)\n    return t.tsTypeReference(ARRAY_OF, t.tsTypeParameterInstantiation([typeNodes]))\n  }\n\n  // Helper function used to generate TS types for array type nodes.\n  private generateArrayTsType(typeNode: ArrayTypeNode): t.TSTypeReference | t.TSUnionType {\n    // if it's an array of a single inline type, wrap it in ArrayOf\n    if (typeNode.of.type === 'inline') {\n      return this.generateArrayOfTsType(typeNode)\n    }\n\n    // if it's not an inline object and not a union, wrap in Array\n    if (typeNode.of.type !== 'union') {\n      const typeNodes = this.generateTsType(typeNode.of)\n      return t.tsTypeReference(t.identifier('Array'), t.tsTypeParameterInstantiation([typeNodes]))\n    }\n\n    // if it's not a union type or all of the union type members are non-inlines, wrap type in Array\n    if (typeNode.of.of.every((unionTypeNode) => unionTypeNode.type !== 'inline')) {\n      const typeNodes = this.generateTsType(typeNode.of)\n      return t.tsTypeReference(t.identifier('Array'), t.tsTypeParameterInstantiation([typeNodes]))\n    }\n\n    // all the union types nodes are inline\n    if (typeNode.of.of.every((unionMember) => unionMember.type === 'inline')) {\n      return this.generateArrayOfTsType(typeNode)\n    }\n\n    // some of the union types are inlines, while some are not - split and recurse\n    const arrayOfNonInline = getFilterArrayUnionType(typeNode, (member) => member.type !== 'inline')\n    const arrayOfInline = getFilterArrayUnionType(typeNode, (member) => member.type === 'inline')\n\n    return t.tsUnionType([\n      this.generateArrayTsType(arrayOfNonInline),\n      this.generateArrayTsType(arrayOfInline),\n    ])\n  }\n\n  // Helper function used to generate TS types for document type nodes.\n  private generateDocumentTsType(document: DocumentSchemaType): t.TSType {\n    const props = Object.entries(document.attributes).map(([key, node]) =>\n      this.generateTsObjectProperty(key, node),\n    )\n\n    return t.tsTypeLiteral(props)\n  }\n\n  private generateInlineTsType(typeNode: InlineTypeNode): t.TSType {\n    const id = this.identifiers.get(typeNode.name)\n    if (!id) {\n      // Not found in schema, return unknown type\n      return t.addComment(\n        t.tsUnknownKeyword(),\n        'trailing',\n        ` Unable to locate the referenced type \"${typeNode.name}\" in schema`,\n        true,\n      )\n    }\n\n    return t.tsTypeReference(id)\n  }\n\n  // Helper function used to generate TS types for object type nodes.\n  private generateObjectTsType(typeNode: ObjectTypeNode): t.TSType {\n    const props: t.TSPropertySignature[] = []\n    for (const [key, attribute] of Object.entries(typeNode.attributes)) {\n      props.push(this.generateTsObjectProperty(key, attribute))\n    }\n    const rest = typeNode.rest\n\n    if (rest) {\n      switch (rest.type) {\n        case 'inline': {\n          const resolved = this.generateInlineTsType(rest)\n          // if object rest is unknown, we can't generate a type literal for it\n          if (t.isTSUnknownKeyword(resolved)) return resolved\n          return t.tsIntersectionType([t.tsTypeLiteral(props), resolved])\n        }\n        case 'object': {\n          for (const [key, attribute] of Object.entries(rest.attributes)) {\n            props.push(this.generateTsObjectProperty(key, attribute))\n          }\n          break\n        }\n        case 'unknown': {\n          return t.tsUnknownKeyword()\n        }\n        default: {\n          // @ts-expect-error This should never happen\n          throw new Error(`Type \"${rest.type}\" not found in schema`)\n        }\n      }\n    }\n\n    if (typeNode.dereferencesTo) {\n      const derefType = Object.assign(\n        t.tsPropertySignature(\n          INTERNAL_REFERENCE_SYMBOL,\n          t.tsTypeAnnotation(t.tsLiteralType(t.stringLiteral(typeNode.dereferencesTo))),\n        ),\n        {computed: true, optional: true},\n      )\n      props.push(derefType)\n    }\n\n    return t.tsTypeLiteral(props)\n  }\n\n  // Helper function used to generate TS types for object properties.\n  private generateTsObjectProperty(key: string, attribute: ObjectAttribute): t.TSPropertySignature {\n    const type = this.generateTsType(attribute.value)\n    const keyNode = isIdentifierName(key) ? t.identifier(key) : t.stringLiteral(key)\n    const propertySignature = t.tsPropertySignature(keyNode, t.tsTypeAnnotation(type))\n    propertySignature.optional = attribute.optional\n\n    return propertySignature\n  }\n\n  private generateTsType(\n    typeNode: DocumentSchemaType | TypeDeclarationSchemaType | TypeNode,\n  ): t.TSType {\n    switch (typeNode.type) {\n      case 'array': {\n        return this.generateArrayTsType(typeNode)\n      }\n      case 'boolean': {\n        if (typeNode.value !== undefined) {\n          return t.tsLiteralType(t.booleanLiteral(typeNode.value))\n        }\n        return t.tsBooleanKeyword()\n      }\n      case 'document': {\n        return this.generateDocumentTsType(typeNode)\n      }\n      case 'inline': {\n        return this.generateInlineTsType(typeNode)\n      }\n      case 'null': {\n        return t.tsNullKeyword()\n      }\n      case 'number': {\n        if (typeNode.value !== undefined) {\n          return t.tsLiteralType(t.numericLiteral(typeNode.value))\n        }\n        return t.tsNumberKeyword()\n      }\n      case 'object': {\n        return this.generateObjectTsType(typeNode)\n      }\n      case 'string': {\n        if (typeNode.value !== undefined) {\n          return t.tsLiteralType(t.stringLiteral(typeNode.value))\n        }\n        return t.tsStringKeyword()\n      }\n      case 'type': {\n        return this.generateTsType(typeNode.value)\n      }\n      case 'union': {\n        return this.generateUnionTsType(typeNode)\n      }\n      case 'unknown': {\n        return t.tsUnknownKeyword()\n      }\n\n      default: {\n        throw new Error(\n          `Encountered unsupported node type \"${\n            // @ts-expect-error This should never happen\n            typeNode.type\n          }\" while generating schema types`,\n        )\n      }\n    }\n  }\n\n  // Helper function used to generate TS types for union type nodes.\n  private generateUnionTsType(typeNode: UnionTypeNode): t.TSType {\n    if (typeNode.of.length === 0) return t.tsNeverKeyword()\n    if (typeNode.of.length === 1) return this.generateTsType(typeNode.of[0]!)\n    return t.tsUnionType(typeNode.of.map((node) => this.generateTsType(node)))\n  }\n}\n\nexport function walkAndCountQueryTypeNodeStats(typeNode: TypeNode): TypeEvaluationStats {\n  switch (typeNode.type) {\n    case 'array': {\n      const acc = walkAndCountQueryTypeNodeStats(typeNode.of)\n      acc.allTypes += 1 // count the array type itself\n      return acc\n    }\n    case 'object': {\n      // if the rest is unknown, we count it as one unknown type\n      if (typeNode.rest && typeNode.rest.type === 'unknown') {\n        return {allTypes: 2, emptyUnions: 0, unknownTypes: 1} // count the object type itself as well\n      }\n\n      const restStats = typeNode.rest\n        ? walkAndCountQueryTypeNodeStats(typeNode.rest)\n        : {allTypes: 0, emptyUnions: 0, unknownTypes: 0}\n\n      // count the object type itself\n      restStats.allTypes += 1\n\n      const attrs = Object.values(typeNode.attributes)\n      let acc = restStats\n      for (const attribute of attrs) {\n        const {allTypes, emptyUnions, unknownTypes} = walkAndCountQueryTypeNodeStats(\n          attribute.value,\n        )\n        acc = {\n          allTypes: acc.allTypes + allTypes,\n          emptyUnions: acc.emptyUnions + emptyUnions,\n          unknownTypes: acc.unknownTypes + unknownTypes,\n        }\n      }\n      return acc\n    }\n    case 'union': {\n      if (typeNode.of.length === 0) {\n        return {allTypes: 1, emptyUnions: 1, unknownTypes: 0}\n      }\n\n      let acc = {allTypes: 1, emptyUnions: 0, unknownTypes: 0} // count the union type itself\n      for (const type of typeNode.of) {\n        const {allTypes, emptyUnions, unknownTypes} = walkAndCountQueryTypeNodeStats(type)\n        acc = {\n          allTypes: acc.allTypes + allTypes,\n          emptyUnions: acc.emptyUnions + emptyUnions,\n          unknownTypes: acc.unknownTypes + unknownTypes,\n        }\n      }\n      return acc\n    }\n    case 'unknown': {\n      return {allTypes: 1, emptyUnions: 0, unknownTypes: 1}\n    }\n    default: {\n      return {allTypes: 1, emptyUnions: 0, unknownTypes: 0}\n    }\n  }\n}\n"],"names":["t","typeEvaluate","safeParseQuery","ARRAY_OF","INTERNAL_REFERENCE_SYMBOL","getFilterArrayUnionType","getUniqueIdentifierForName","isIdentifierName","weakMapMemo","SchemaTypeGenerator","schema","evaluateQuery","query","ast","typeNode","tsType","generateTsType","stats","walkAndCountQueryTypeNodeStats","identifiers","Map","tsTypes","uniqueTypeNames","Set","type","has","name","Error","add","currentIdentifierNames","values","map","id","uniqueIdentifier","set","getType","typeName","get","undefined","hasType","Symbol","iterator","typeNames","schemaType","generateArrayOfTsType","typeNodes","of","tsTypeReference","tsTypeParameterInstantiation","generateArrayTsType","identifier","every","unionTypeNode","unionMember","arrayOfNonInline","member","arrayOfInline","tsUnionType","generateDocumentTsType","document","props","Object","entries","attributes","key","node","generateTsObjectProperty","tsTypeLiteral","generateInlineTsType","addComment","tsUnknownKeyword","generateObjectTsType","attribute","push","rest","resolved","isTSUnknownKeyword","tsIntersectionType","dereferencesTo","derefType","assign","tsPropertySignature","tsTypeAnnotation","tsLiteralType","stringLiteral","computed","optional","value","keyNode","propertySignature","booleanLiteral","tsBooleanKeyword","tsNullKeyword","numericLiteral","tsNumberKeyword","tsStringKeyword","generateUnionTsType","length","tsNeverKeyword","acc","allTypes","emptyUnions","unknownTypes","restStats","attrs"],"mappings":"AAAA,YAAYA,OAAO,eAAc;AACjC,SAQEC,YAAY,QAGP,UAAS;AAEhB,SAAQC,cAAc,QAAO,uBAAsB;AACnD,SAAQC,QAAQ,EAAEC,yBAAyB,QAAO,iBAAgB;AAClE,SACEC,uBAAuB,EACvBC,0BAA0B,EAC1BC,gBAAgB,EAChBC,WAAW,QACN,eAAc;AAGrB,OAAO,MAAMC;IACKC,OAAkB;IAClCC,gBAAgBH,YACd,+DAA+D;IAC/D,CAAC,EAACI,KAAK,EAAgC;QACrC,MAAMC,MAAMX,eAAeU;QAC3B,MAAME,WAAWb,aAAaY,KAAK,IAAI,CAACH,MAAM;QAC9C,MAAMK,SAAS,IAAI,CAACC,cAAc,CAACF;QACnC,MAAMG,QAAQC,+BAA+BJ;QAC7C,OAAO;YAACG;YAAOF;QAAM;IACvB,GACD;IACOI,cAAc,IAAIC,MAA2B;IAE7CC,UAAU,IAAID,MAAuB;IAE7C,YAAYV,MAAkB,CAAE;QAC9B,IAAI,CAACA,MAAM,GAAGA;QAEd,MAAMY,kBAAkB,IAAIC;QAC5B,KAAK,MAAMC,QAAQd,OAAQ;YACzB,IAAIY,gBAAgBG,GAAG,CAACD,KAAKE,IAAI,GAAG;gBAClC,MAAM,IAAIC,MACR,CAAC,qBAAqB,EAAEH,KAAKE,IAAI,CAAC,8DAA8D,CAAC;YAErG;YACAJ,gBAAgBM,GAAG,CAACJ,KAAKE,IAAI;QAC/B;QAEA,KAAK,MAAMF,QAAQd,OAAQ;YACzB,MAAMmB,yBAAyB,IAAIN,IAAI;mBAAI,IAAI,CAACJ,WAAW,CAACW,MAAM;aAAG,CAACC,GAAG,CAAC,CAACC,KAAOA,GAAGN,IAAI;YACzF,MAAMO,mBAAmB3B,2BAA2BkB,KAAKE,IAAI,EAAEG;YAC/D,IAAI,CAACV,WAAW,CAACe,GAAG,CAACV,KAAKE,IAAI,EAAEO;QAClC;QAEA,KAAK,MAAMT,QAAQd,OAAQ;YACzB,IAAI,CAACW,OAAO,CAACa,GAAG,CAACV,KAAKE,IAAI,EAAE,IAAI,CAACV,cAAc,CAACQ;QAClD;IACF;IAEAW,QAAQC,QAAgB,EAAoD;QAC1E,MAAMrB,SAAS,IAAI,CAACM,OAAO,CAACgB,GAAG,CAACD;QAChC,MAAMJ,KAAK,IAAI,CAACb,WAAW,CAACkB,GAAG,CAACD;QAChC,IAAIrB,UAAUiB,IAAI,OAAO;YAACA;YAAIjB;QAAM;QACpC,OAAOuB;IACT;IAEAC,QAAQH,QAAgB,EAAW;QACjC,OAAO,IAAI,CAACf,OAAO,CAACI,GAAG,CAACW;IAC1B;IAEA,CAAC,CAACI,OAAOC,QAAQ,CAAC,GAAG;QACnB,KAAK,MAAM,EAACf,IAAI,EAAC,IAAI,IAAI,CAAChB,MAAM,CAAE;YAChC,MAAM;gBAACgB;gBAAM,GAAG,IAAI,CAACS,OAAO,CAACT,KAAK;YAAC;QACrC;IACF;IAEAgB,YAAsB;QACpB,OAAO,IAAI,CAAChC,MAAM,CAACqB,GAAG,CAAC,CAACY,aAAeA,WAAWjB,IAAI;IACxD;IAEA;;;GAGC,GACD,AAAQkB,sBAAsB9B,QAAuB,EAAqB;QACxE,MAAM+B,YAAY,IAAI,CAAC7B,cAAc,CAACF,SAASgC,EAAE;QACjD,OAAO9C,EAAE+C,eAAe,CAAC5C,UAAUH,EAAEgD,4BAA4B,CAAC;YAACH;SAAU;IAC/E;IAEA,kEAAkE;IAC1DI,oBAAoBnC,QAAuB,EAAqC;QACtF,+DAA+D;QAC/D,IAAIA,SAASgC,EAAE,CAACtB,IAAI,KAAK,UAAU;YACjC,OAAO,IAAI,CAACoB,qBAAqB,CAAC9B;QACpC;QAEA,8DAA8D;QAC9D,IAAIA,SAASgC,EAAE,CAACtB,IAAI,KAAK,SAAS;YAChC,MAAMqB,YAAY,IAAI,CAAC7B,cAAc,CAACF,SAASgC,EAAE;YACjD,OAAO9C,EAAE+C,eAAe,CAAC/C,EAAEkD,UAAU,CAAC,UAAUlD,EAAEgD,4BAA4B,CAAC;gBAACH;aAAU;QAC5F;QAEA,gGAAgG;QAChG,IAAI/B,SAASgC,EAAE,CAACA,EAAE,CAACK,KAAK,CAAC,CAACC,gBAAkBA,cAAc5B,IAAI,KAAK,WAAW;YAC5E,MAAMqB,YAAY,IAAI,CAAC7B,cAAc,CAACF,SAASgC,EAAE;YACjD,OAAO9C,EAAE+C,eAAe,CAAC/C,EAAEkD,UAAU,CAAC,UAAUlD,EAAEgD,4BAA4B,CAAC;gBAACH;aAAU;QAC5F;QAEA,uCAAuC;QACvC,IAAI/B,SAASgC,EAAE,CAACA,EAAE,CAACK,KAAK,CAAC,CAACE,cAAgBA,YAAY7B,IAAI,KAAK,WAAW;YACxE,OAAO,IAAI,CAACoB,qBAAqB,CAAC9B;QACpC;QAEA,8EAA8E;QAC9E,MAAMwC,mBAAmBjD,wBAAwBS,UAAU,CAACyC,SAAWA,OAAO/B,IAAI,KAAK;QACvF,MAAMgC,gBAAgBnD,wBAAwBS,UAAU,CAACyC,SAAWA,OAAO/B,IAAI,KAAK;QAEpF,OAAOxB,EAAEyD,WAAW,CAAC;YACnB,IAAI,CAACR,mBAAmB,CAACK;YACzB,IAAI,CAACL,mBAAmB,CAACO;SAC1B;IACH;IAEA,qEAAqE;IAC7DE,uBAAuBC,QAA4B,EAAY;QACrE,MAAMC,QAAQC,OAAOC,OAAO,CAACH,SAASI,UAAU,EAAEhC,GAAG,CAAC,CAAC,CAACiC,KAAKC,KAAK,GAChE,IAAI,CAACC,wBAAwB,CAACF,KAAKC;QAGrC,OAAOjE,EAAEmE,aAAa,CAACP;IACzB;IAEQQ,qBAAqBtD,QAAwB,EAAY;QAC/D,MAAMkB,KAAK,IAAI,CAACb,WAAW,CAACkB,GAAG,CAACvB,SAASY,IAAI;QAC7C,IAAI,CAACM,IAAI;YACP,2CAA2C;YAC3C,OAAOhC,EAAEqE,UAAU,CACjBrE,EAAEsE,gBAAgB,IAClB,YACA,CAAC,uCAAuC,EAAExD,SAASY,IAAI,CAAC,WAAW,CAAC,EACpE;QAEJ;QAEA,OAAO1B,EAAE+C,eAAe,CAACf;IAC3B;IAEA,mEAAmE;IAC3DuC,qBAAqBzD,QAAwB,EAAY;QAC/D,MAAM8C,QAAiC,EAAE;QACzC,KAAK,MAAM,CAACI,KAAKQ,UAAU,IAAIX,OAAOC,OAAO,CAAChD,SAASiD,UAAU,EAAG;YAClEH,MAAMa,IAAI,CAAC,IAAI,CAACP,wBAAwB,CAACF,KAAKQ;QAChD;QACA,MAAME,OAAO5D,SAAS4D,IAAI;QAE1B,IAAIA,MAAM;YACR,OAAQA,KAAKlD,IAAI;gBACf,KAAK;oBAAU;wBACb,MAAMmD,WAAW,IAAI,CAACP,oBAAoB,CAACM;wBAC3C,qEAAqE;wBACrE,IAAI1E,EAAE4E,kBAAkB,CAACD,WAAW,OAAOA;wBAC3C,OAAO3E,EAAE6E,kBAAkB,CAAC;4BAAC7E,EAAEmE,aAAa,CAACP;4BAAQe;yBAAS;oBAChE;gBACA,KAAK;oBAAU;wBACb,KAAK,MAAM,CAACX,KAAKQ,UAAU,IAAIX,OAAOC,OAAO,CAACY,KAAKX,UAAU,EAAG;4BAC9DH,MAAMa,IAAI,CAAC,IAAI,CAACP,wBAAwB,CAACF,KAAKQ;wBAChD;wBACA;oBACF;gBACA,KAAK;oBAAW;wBACd,OAAOxE,EAAEsE,gBAAgB;oBAC3B;gBACA;oBAAS;wBACP,4CAA4C;wBAC5C,MAAM,IAAI3C,MAAM,CAAC,MAAM,EAAE+C,KAAKlD,IAAI,CAAC,qBAAqB,CAAC;oBAC3D;YACF;QACF;QAEA,IAAIV,SAASgE,cAAc,EAAE;YAC3B,MAAMC,YAAYlB,OAAOmB,MAAM,CAC7BhF,EAAEiF,mBAAmB,CACnB7E,2BACAJ,EAAEkF,gBAAgB,CAAClF,EAAEmF,aAAa,CAACnF,EAAEoF,aAAa,CAACtE,SAASgE,cAAc,MAE5E;gBAACO,UAAU;gBAAMC,UAAU;YAAI;YAEjC1B,MAAMa,IAAI,CAACM;QACb;QAEA,OAAO/E,EAAEmE,aAAa,CAACP;IACzB;IAEA,mEAAmE;IAC3DM,yBAAyBF,GAAW,EAAEQ,SAA0B,EAAyB;QAC/F,MAAMhD,OAAO,IAAI,CAACR,cAAc,CAACwD,UAAUe,KAAK;QAChD,MAAMC,UAAUjF,iBAAiByD,OAAOhE,EAAEkD,UAAU,CAACc,OAAOhE,EAAEoF,aAAa,CAACpB;QAC5E,MAAMyB,oBAAoBzF,EAAEiF,mBAAmB,CAACO,SAASxF,EAAEkF,gBAAgB,CAAC1D;QAC5EiE,kBAAkBH,QAAQ,GAAGd,UAAUc,QAAQ;QAE/C,OAAOG;IACT;IAEQzE,eACNF,QAAmE,EACzD;QACV,OAAQA,SAASU,IAAI;YACnB,KAAK;gBAAS;oBACZ,OAAO,IAAI,CAACyB,mBAAmB,CAACnC;gBAClC;YACA,KAAK;gBAAW;oBACd,IAAIA,SAASyE,KAAK,KAAKjD,WAAW;wBAChC,OAAOtC,EAAEmF,aAAa,CAACnF,EAAE0F,cAAc,CAAC5E,SAASyE,KAAK;oBACxD;oBACA,OAAOvF,EAAE2F,gBAAgB;gBAC3B;YACA,KAAK;gBAAY;oBACf,OAAO,IAAI,CAACjC,sBAAsB,CAAC5C;gBACrC;YACA,KAAK;gBAAU;oBACb,OAAO,IAAI,CAACsD,oBAAoB,CAACtD;gBACnC;YACA,KAAK;gBAAQ;oBACX,OAAOd,EAAE4F,aAAa;gBACxB;YACA,KAAK;gBAAU;oBACb,IAAI9E,SAASyE,KAAK,KAAKjD,WAAW;wBAChC,OAAOtC,EAAEmF,aAAa,CAACnF,EAAE6F,cAAc,CAAC/E,SAASyE,KAAK;oBACxD;oBACA,OAAOvF,EAAE8F,eAAe;gBAC1B;YACA,KAAK;gBAAU;oBACb,OAAO,IAAI,CAACvB,oBAAoB,CAACzD;gBACnC;YACA,KAAK;gBAAU;oBACb,IAAIA,SAASyE,KAAK,KAAKjD,WAAW;wBAChC,OAAOtC,EAAEmF,aAAa,CAACnF,EAAEoF,aAAa,CAACtE,SAASyE,KAAK;oBACvD;oBACA,OAAOvF,EAAE+F,eAAe;gBAC1B;YACA,KAAK;gBAAQ;oBACX,OAAO,IAAI,CAAC/E,cAAc,CAACF,SAASyE,KAAK;gBAC3C;YACA,KAAK;gBAAS;oBACZ,OAAO,IAAI,CAACS,mBAAmB,CAAClF;gBAClC;YACA,KAAK;gBAAW;oBACd,OAAOd,EAAEsE,gBAAgB;gBAC3B;YAEA;gBAAS;oBACP,MAAM,IAAI3C,MACR,CAAC,mCAAmC,EAClC,4CAA4C;oBAC5Cb,SAASU,IAAI,CACd,+BAA+B,CAAC;gBAErC;QACF;IACF;IAEA,kEAAkE;IAC1DwE,oBAAoBlF,QAAuB,EAAY;QAC7D,IAAIA,SAASgC,EAAE,CAACmD,MAAM,KAAK,GAAG,OAAOjG,EAAEkG,cAAc;QACrD,IAAIpF,SAASgC,EAAE,CAACmD,MAAM,KAAK,GAAG,OAAO,IAAI,CAACjF,cAAc,CAACF,SAASgC,EAAE,CAAC,EAAE;QACvE,OAAO9C,EAAEyD,WAAW,CAAC3C,SAASgC,EAAE,CAACf,GAAG,CAAC,CAACkC,OAAS,IAAI,CAACjD,cAAc,CAACiD;IACrE;AACF;AAEA,OAAO,SAAS/C,+BAA+BJ,QAAkB;IAC/D,OAAQA,SAASU,IAAI;QACnB,KAAK;YAAS;gBACZ,MAAM2E,MAAMjF,+BAA+BJ,SAASgC,EAAE;gBACtDqD,IAAIC,QAAQ,IAAI,GAAE,8BAA8B;gBAChD,OAAOD;YACT;QACA,KAAK;YAAU;gBACb,0DAA0D;gBAC1D,IAAIrF,SAAS4D,IAAI,IAAI5D,SAAS4D,IAAI,CAAClD,IAAI,KAAK,WAAW;oBACrD,OAAO;wBAAC4E,UAAU;wBAAGC,aAAa;wBAAGC,cAAc;oBAAC,EAAE,uCAAuC;;gBAC/F;gBAEA,MAAMC,YAAYzF,SAAS4D,IAAI,GAC3BxD,+BAA+BJ,SAAS4D,IAAI,IAC5C;oBAAC0B,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC;gBAEjD,+BAA+B;gBAC/BC,UAAUH,QAAQ,IAAI;gBAEtB,MAAMI,QAAQ3C,OAAO/B,MAAM,CAAChB,SAASiD,UAAU;gBAC/C,IAAIoC,MAAMI;gBACV,KAAK,MAAM/B,aAAagC,MAAO;oBAC7B,MAAM,EAACJ,QAAQ,EAAEC,WAAW,EAAEC,YAAY,EAAC,GAAGpF,+BAC5CsD,UAAUe,KAAK;oBAEjBY,MAAM;wBACJC,UAAUD,IAAIC,QAAQ,GAAGA;wBACzBC,aAAaF,IAAIE,WAAW,GAAGA;wBAC/BC,cAAcH,IAAIG,YAAY,GAAGA;oBACnC;gBACF;gBACA,OAAOH;YACT;QACA,KAAK;YAAS;gBACZ,IAAIrF,SAASgC,EAAE,CAACmD,MAAM,KAAK,GAAG;oBAC5B,OAAO;wBAACG,UAAU;wBAAGC,aAAa;wBAAGC,cAAc;oBAAC;gBACtD;gBAEA,IAAIH,MAAM;oBAACC,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC,EAAE,8BAA8B;;gBACvF,KAAK,MAAM9E,QAAQV,SAASgC,EAAE,CAAE;oBAC9B,MAAM,EAACsD,QAAQ,EAAEC,WAAW,EAAEC,YAAY,EAAC,GAAGpF,+BAA+BM;oBAC7E2E,MAAM;wBACJC,UAAUD,IAAIC,QAAQ,GAAGA;wBACzBC,aAAaF,IAAIE,WAAW,GAAGA;wBAC/BC,cAAcH,IAAIG,YAAY,GAAGA;oBACnC;gBACF;gBACA,OAAOH;YACT;QACA,KAAK;YAAW;gBACd,OAAO;oBAACC,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC;YACtD;QACA;YAAS;gBACP,OAAO;oBAACF,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC;YACtD;IACF;AACF"}
+{"version":3,"sources":["../../src/typescript/schemaTypeGenerator.ts"],"sourcesContent":["import * as t from '@babel/types'\nimport {\n  type ArrayTypeNode,\n  type DocumentSchemaType,\n  type InlineTypeNode,\n  type ObjectAttribute,\n  type ObjectTypeNode,\n  type SchemaType,\n  type TypeDeclarationSchemaType,\n  typeEvaluate,\n  type TypeNode,\n  type UnionTypeNode,\n} from 'groq-js'\n\nimport {safeParseQuery} from '../safeParseQuery.js'\nimport {ARRAY_OF, INTERNAL_REFERENCE_SYMBOL} from './constants.js'\nimport {\n  getFilterArrayUnionType,\n  getUniqueIdentifierForName,\n  isIdentifierName,\n  weakMapMemo,\n} from './helpers.js'\nimport {type ExtractedQuery, type TypeEvaluationStats} from './types.js'\n\nexport class SchemaTypeGenerator {\n  public readonly schema: SchemaType\n  evaluateQuery = weakMapMemo(\n    // eslint-disable-next-line unicorn/consistent-function-scoping\n    ({query}: Pick<ExtractedQuery, 'query'>): {stats: TypeEvaluationStats; tsType: t.TSType} => {\n      const ast = safeParseQuery(query)\n      const typeNode = typeEvaluate(ast, this.schema)\n      const tsType = this.generateTsType(typeNode)\n      const stats = walkAndCountQueryTypeNodeStats(typeNode)\n      return {stats, tsType}\n    },\n  )\n  private arrayOfUsed = false\n\n  private identifiers = new Map<string, t.Identifier>()\n\n  private tsTypes = new Map<string, t.TSType>()\n\n  constructor(schema: SchemaType) {\n    this.schema = schema\n\n    const uniqueTypeNames = new Set<string>()\n    for (const type of schema) {\n      if (uniqueTypeNames.has(type.name)) {\n        throw new Error(\n          `Duplicate type name \"${type.name}\" in schema. Type names must be unique within the same schema.`,\n        )\n      }\n      uniqueTypeNames.add(type.name)\n    }\n\n    for (const type of schema) {\n      const currentIdentifierNames = new Set([...this.identifiers.values()].map((id) => id.name))\n      const uniqueIdentifier = getUniqueIdentifierForName(type.name, currentIdentifierNames)\n      this.identifiers.set(type.name, uniqueIdentifier)\n    }\n\n    for (const type of schema) {\n      this.tsTypes.set(type.name, this.generateTsType(type))\n    }\n  }\n\n  getType(typeName: string): {id: t.Identifier; tsType: t.TSType} | undefined {\n    const tsType = this.tsTypes.get(typeName)\n    const id = this.identifiers.get(typeName)\n    if (tsType && id) return {id, tsType}\n    return undefined\n  }\n\n  hasType(typeName: string): boolean {\n    return this.tsTypes.has(typeName)\n  }\n\n  isArrayOfUsed(): boolean {\n    return this.arrayOfUsed\n  }\n\n  *[Symbol.iterator]() {\n    for (const {name} of this.schema) {\n      yield {name, ...this.getType(name)!}\n    }\n  }\n\n  typeNames(): string[] {\n    return this.schema.map((schemaType) => schemaType.name)\n  }\n\n  /**\n   * Helper function used to generate TS types for arrays of inline types, or arrays of inline types\n   * wrapped in the ArrayOf wrapper that adds _key prop\n   */\n  private generateArrayOfTsType(typeNode: ArrayTypeNode): t.TSTypeReference {\n    this.arrayOfUsed = true\n    const typeNodes = this.generateTsType(typeNode.of)\n    return t.tsTypeReference(ARRAY_OF, t.tsTypeParameterInstantiation([typeNodes]))\n  }\n\n  // Helper function used to generate TS types for array type nodes.\n  private generateArrayTsType(typeNode: ArrayTypeNode): t.TSTypeReference | t.TSUnionType {\n    // if it's an array of a single inline type, wrap it in ArrayOf\n    if (typeNode.of.type === 'inline') {\n      return this.generateArrayOfTsType(typeNode)\n    }\n\n    // if it's not an inline object and not a union, wrap in Array\n    if (typeNode.of.type !== 'union') {\n      const typeNodes = this.generateTsType(typeNode.of)\n      return t.tsTypeReference(t.identifier('Array'), t.tsTypeParameterInstantiation([typeNodes]))\n    }\n\n    // if it's not a union type or all of the union type members are non-inlines, wrap type in Array\n    if (typeNode.of.of.every((unionTypeNode) => unionTypeNode.type !== 'inline')) {\n      const typeNodes = this.generateTsType(typeNode.of)\n      return t.tsTypeReference(t.identifier('Array'), t.tsTypeParameterInstantiation([typeNodes]))\n    }\n\n    // all the union types nodes are inline\n    if (typeNode.of.of.every((unionMember) => unionMember.type === 'inline')) {\n      return this.generateArrayOfTsType(typeNode)\n    }\n\n    // some of the union types are inlines, while some are not - split and recurse\n    const arrayOfNonInline = getFilterArrayUnionType(typeNode, (member) => member.type !== 'inline')\n    const arrayOfInline = getFilterArrayUnionType(typeNode, (member) => member.type === 'inline')\n\n    return t.tsUnionType([\n      this.generateArrayTsType(arrayOfNonInline),\n      this.generateArrayTsType(arrayOfInline),\n    ])\n  }\n\n  // Helper function used to generate TS types for document type nodes.\n  private generateDocumentTsType(document: DocumentSchemaType): t.TSType {\n    const props = Object.entries(document.attributes).map(([key, node]) =>\n      this.generateTsObjectProperty(key, node),\n    )\n\n    return t.tsTypeLiteral(props)\n  }\n\n  private generateInlineTsType(typeNode: InlineTypeNode): t.TSType {\n    const id = this.identifiers.get(typeNode.name)\n    if (!id) {\n      // Not found in schema, return unknown type\n      return t.addComment(\n        t.tsUnknownKeyword(),\n        'trailing',\n        ` Unable to locate the referenced type \"${typeNode.name}\" in schema`,\n        true,\n      )\n    }\n\n    return t.tsTypeReference(id)\n  }\n\n  // Helper function used to generate TS types for object type nodes.\n  private generateObjectTsType(typeNode: ObjectTypeNode): t.TSType {\n    const props: t.TSPropertySignature[] = []\n    for (const [key, attribute] of Object.entries(typeNode.attributes)) {\n      props.push(this.generateTsObjectProperty(key, attribute))\n    }\n    const rest = typeNode.rest\n\n    if (rest) {\n      switch (rest.type) {\n        case 'inline': {\n          const resolved = this.generateInlineTsType(rest)\n          // if object rest is unknown, we can't generate a type literal for it\n          if (t.isTSUnknownKeyword(resolved)) return resolved\n          return t.tsIntersectionType([t.tsTypeLiteral(props), resolved])\n        }\n        case 'object': {\n          for (const [key, attribute] of Object.entries(rest.attributes)) {\n            props.push(this.generateTsObjectProperty(key, attribute))\n          }\n          break\n        }\n        case 'unknown': {\n          return t.tsUnknownKeyword()\n        }\n        default: {\n          // @ts-expect-error This should never happen\n          throw new Error(`Type \"${rest.type}\" not found in schema`)\n        }\n      }\n    }\n\n    if (typeNode.dereferencesTo) {\n      const derefType = Object.assign(\n        t.tsPropertySignature(\n          INTERNAL_REFERENCE_SYMBOL,\n          t.tsTypeAnnotation(t.tsLiteralType(t.stringLiteral(typeNode.dereferencesTo))),\n        ),\n        {computed: true, optional: true},\n      )\n      props.push(derefType)\n    }\n\n    return t.tsTypeLiteral(props)\n  }\n\n  // Helper function used to generate TS types for object properties.\n  private generateTsObjectProperty(key: string, attribute: ObjectAttribute): t.TSPropertySignature {\n    const type = this.generateTsType(attribute.value)\n    const keyNode = isIdentifierName(key) ? t.identifier(key) : t.stringLiteral(key)\n    const propertySignature = t.tsPropertySignature(keyNode, t.tsTypeAnnotation(type))\n    propertySignature.optional = attribute.optional\n\n    return propertySignature\n  }\n\n  private generateTsType(\n    typeNode: DocumentSchemaType | TypeDeclarationSchemaType | TypeNode,\n  ): t.TSType {\n    switch (typeNode.type) {\n      case 'array': {\n        return this.generateArrayTsType(typeNode)\n      }\n      case 'boolean': {\n        if (typeNode.value !== undefined) {\n          return t.tsLiteralType(t.booleanLiteral(typeNode.value))\n        }\n        return t.tsBooleanKeyword()\n      }\n      case 'document': {\n        return this.generateDocumentTsType(typeNode)\n      }\n      case 'inline': {\n        return this.generateInlineTsType(typeNode)\n      }\n      case 'null': {\n        return t.tsNullKeyword()\n      }\n      case 'number': {\n        if (typeNode.value !== undefined) {\n          return t.tsLiteralType(t.numericLiteral(typeNode.value))\n        }\n        return t.tsNumberKeyword()\n      }\n      case 'object': {\n        return this.generateObjectTsType(typeNode)\n      }\n      case 'string': {\n        if (typeNode.value !== undefined) {\n          return t.tsLiteralType(t.stringLiteral(typeNode.value))\n        }\n        return t.tsStringKeyword()\n      }\n      case 'type': {\n        return this.generateTsType(typeNode.value)\n      }\n      case 'union': {\n        return this.generateUnionTsType(typeNode)\n      }\n      case 'unknown': {\n        return t.tsUnknownKeyword()\n      }\n\n      default: {\n        throw new Error(\n          `Encountered unsupported node type \"${\n            // @ts-expect-error This should never happen\n            typeNode.type\n          }\" while generating schema types`,\n        )\n      }\n    }\n  }\n\n  // Helper function used to generate TS types for union type nodes.\n  private generateUnionTsType(typeNode: UnionTypeNode): t.TSType {\n    if (typeNode.of.length === 0) return t.tsNeverKeyword()\n    if (typeNode.of.length === 1) return this.generateTsType(typeNode.of[0]!)\n    return t.tsUnionType(typeNode.of.map((node) => this.generateTsType(node)))\n  }\n}\n\nexport function walkAndCountQueryTypeNodeStats(typeNode: TypeNode): TypeEvaluationStats {\n  switch (typeNode.type) {\n    case 'array': {\n      const acc = walkAndCountQueryTypeNodeStats(typeNode.of)\n      acc.allTypes += 1 // count the array type itself\n      return acc\n    }\n    case 'object': {\n      // if the rest is unknown, we count it as one unknown type\n      if (typeNode.rest && typeNode.rest.type === 'unknown') {\n        return {allTypes: 2, emptyUnions: 0, unknownTypes: 1} // count the object type itself as well\n      }\n\n      const restStats = typeNode.rest\n        ? walkAndCountQueryTypeNodeStats(typeNode.rest)\n        : {allTypes: 0, emptyUnions: 0, unknownTypes: 0}\n\n      // count the object type itself\n      restStats.allTypes += 1\n\n      const attrs = Object.values(typeNode.attributes)\n      let acc = restStats\n      for (const attribute of attrs) {\n        const {allTypes, emptyUnions, unknownTypes} = walkAndCountQueryTypeNodeStats(\n          attribute.value,\n        )\n        acc = {\n          allTypes: acc.allTypes + allTypes,\n          emptyUnions: acc.emptyUnions + emptyUnions,\n          unknownTypes: acc.unknownTypes + unknownTypes,\n        }\n      }\n      return acc\n    }\n    case 'union': {\n      if (typeNode.of.length === 0) {\n        return {allTypes: 1, emptyUnions: 1, unknownTypes: 0}\n      }\n\n      let acc = {allTypes: 1, emptyUnions: 0, unknownTypes: 0} // count the union type itself\n      for (const type of typeNode.of) {\n        const {allTypes, emptyUnions, unknownTypes} = walkAndCountQueryTypeNodeStats(type)\n        acc = {\n          allTypes: acc.allTypes + allTypes,\n          emptyUnions: acc.emptyUnions + emptyUnions,\n          unknownTypes: acc.unknownTypes + unknownTypes,\n        }\n      }\n      return acc\n    }\n    case 'unknown': {\n      return {allTypes: 1, emptyUnions: 0, unknownTypes: 1}\n    }\n    default: {\n      return {allTypes: 1, emptyUnions: 0, unknownTypes: 0}\n    }\n  }\n}\n"],"names":["t","typeEvaluate","safeParseQuery","ARRAY_OF","INTERNAL_REFERENCE_SYMBOL","getFilterArrayUnionType","getUniqueIdentifierForName","isIdentifierName","weakMapMemo","SchemaTypeGenerator","schema","evaluateQuery","query","ast","typeNode","tsType","generateTsType","stats","walkAndCountQueryTypeNodeStats","arrayOfUsed","identifiers","Map","tsTypes","uniqueTypeNames","Set","type","has","name","Error","add","currentIdentifierNames","values","map","id","uniqueIdentifier","set","getType","typeName","get","undefined","hasType","isArrayOfUsed","Symbol","iterator","typeNames","schemaType","generateArrayOfTsType","typeNodes","of","tsTypeReference","tsTypeParameterInstantiation","generateArrayTsType","identifier","every","unionTypeNode","unionMember","arrayOfNonInline","member","arrayOfInline","tsUnionType","generateDocumentTsType","document","props","Object","entries","attributes","key","node","generateTsObjectProperty","tsTypeLiteral","generateInlineTsType","addComment","tsUnknownKeyword","generateObjectTsType","attribute","push","rest","resolved","isTSUnknownKeyword","tsIntersectionType","dereferencesTo","derefType","assign","tsPropertySignature","tsTypeAnnotation","tsLiteralType","stringLiteral","computed","optional","value","keyNode","propertySignature","booleanLiteral","tsBooleanKeyword","tsNullKeyword","numericLiteral","tsNumberKeyword","tsStringKeyword","generateUnionTsType","length","tsNeverKeyword","acc","allTypes","emptyUnions","unknownTypes","restStats","attrs"],"mappings":"AAAA,YAAYA,OAAO,eAAc;AACjC,SAQEC,YAAY,QAGP,UAAS;AAEhB,SAAQC,cAAc,QAAO,uBAAsB;AACnD,SAAQC,QAAQ,EAAEC,yBAAyB,QAAO,iBAAgB;AAClE,SACEC,uBAAuB,EACvBC,0BAA0B,EAC1BC,gBAAgB,EAChBC,WAAW,QACN,eAAc;AAGrB,OAAO,MAAMC;IACKC,OAAkB;IAClCC,gBAAgBH,YACd,+DAA+D;IAC/D,CAAC,EAACI,KAAK,EAAgC;QACrC,MAAMC,MAAMX,eAAeU;QAC3B,MAAME,WAAWb,aAAaY,KAAK,IAAI,CAACH,MAAM;QAC9C,MAAMK,SAAS,IAAI,CAACC,cAAc,CAACF;QACnC,MAAMG,QAAQC,+BAA+BJ;QAC7C,OAAO;YAACG;YAAOF;QAAM;IACvB,GACD;IACOI,cAAc,MAAK;IAEnBC,cAAc,IAAIC,MAA2B;IAE7CC,UAAU,IAAID,MAAuB;IAE7C,YAAYX,MAAkB,CAAE;QAC9B,IAAI,CAACA,MAAM,GAAGA;QAEd,MAAMa,kBAAkB,IAAIC;QAC5B,KAAK,MAAMC,QAAQf,OAAQ;YACzB,IAAIa,gBAAgBG,GAAG,CAACD,KAAKE,IAAI,GAAG;gBAClC,MAAM,IAAIC,MACR,CAAC,qBAAqB,EAAEH,KAAKE,IAAI,CAAC,8DAA8D,CAAC;YAErG;YACAJ,gBAAgBM,GAAG,CAACJ,KAAKE,IAAI;QAC/B;QAEA,KAAK,MAAMF,QAAQf,OAAQ;YACzB,MAAMoB,yBAAyB,IAAIN,IAAI;mBAAI,IAAI,CAACJ,WAAW,CAACW,MAAM;aAAG,CAACC,GAAG,CAAC,CAACC,KAAOA,GAAGN,IAAI;YACzF,MAAMO,mBAAmB5B,2BAA2BmB,KAAKE,IAAI,EAAEG;YAC/D,IAAI,CAACV,WAAW,CAACe,GAAG,CAACV,KAAKE,IAAI,EAAEO;QAClC;QAEA,KAAK,MAAMT,QAAQf,OAAQ;YACzB,IAAI,CAACY,OAAO,CAACa,GAAG,CAACV,KAAKE,IAAI,EAAE,IAAI,CAACX,cAAc,CAACS;QAClD;IACF;IAEAW,QAAQC,QAAgB,EAAoD;QAC1E,MAAMtB,SAAS,IAAI,CAACO,OAAO,CAACgB,GAAG,CAACD;QAChC,MAAMJ,KAAK,IAAI,CAACb,WAAW,CAACkB,GAAG,CAACD;QAChC,IAAItB,UAAUkB,IAAI,OAAO;YAACA;YAAIlB;QAAM;QACpC,OAAOwB;IACT;IAEAC,QAAQH,QAAgB,EAAW;QACjC,OAAO,IAAI,CAACf,OAAO,CAACI,GAAG,CAACW;IAC1B;IAEAI,gBAAyB;QACvB,OAAO,IAAI,CAACtB,WAAW;IACzB;IAEA,CAAC,CAACuB,OAAOC,QAAQ,CAAC,GAAG;QACnB,KAAK,MAAM,EAAChB,IAAI,EAAC,IAAI,IAAI,CAACjB,MAAM,CAAE;YAChC,MAAM;gBAACiB;gBAAM,GAAG,IAAI,CAACS,OAAO,CAACT,KAAK;YAAC;QACrC;IACF;IAEAiB,YAAsB;QACpB,OAAO,IAAI,CAAClC,MAAM,CAACsB,GAAG,CAAC,CAACa,aAAeA,WAAWlB,IAAI;IACxD;IAEA;;;GAGC,GACD,AAAQmB,sBAAsBhC,QAAuB,EAAqB;QACxE,IAAI,CAACK,WAAW,GAAG;QACnB,MAAM4B,YAAY,IAAI,CAAC/B,cAAc,CAACF,SAASkC,EAAE;QACjD,OAAOhD,EAAEiD,eAAe,CAAC9C,UAAUH,EAAEkD,4BAA4B,CAAC;YAACH;SAAU;IAC/E;IAEA,kEAAkE;IAC1DI,oBAAoBrC,QAAuB,EAAqC;QACtF,+DAA+D;QAC/D,IAAIA,SAASkC,EAAE,CAACvB,IAAI,KAAK,UAAU;YACjC,OAAO,IAAI,CAACqB,qBAAqB,CAAChC;QACpC;QAEA,8DAA8D;QAC9D,IAAIA,SAASkC,EAAE,CAACvB,IAAI,KAAK,SAAS;YAChC,MAAMsB,YAAY,IAAI,CAAC/B,cAAc,CAACF,SAASkC,EAAE;YACjD,OAAOhD,EAAEiD,eAAe,CAACjD,EAAEoD,UAAU,CAAC,UAAUpD,EAAEkD,4BAA4B,CAAC;gBAACH;aAAU;QAC5F;QAEA,gGAAgG;QAChG,IAAIjC,SAASkC,EAAE,CAACA,EAAE,CAACK,KAAK,CAAC,CAACC,gBAAkBA,cAAc7B,IAAI,KAAK,WAAW;YAC5E,MAAMsB,YAAY,IAAI,CAAC/B,cAAc,CAACF,SAASkC,EAAE;YACjD,OAAOhD,EAAEiD,eAAe,CAACjD,EAAEoD,UAAU,CAAC,UAAUpD,EAAEkD,4BAA4B,CAAC;gBAACH;aAAU;QAC5F;QAEA,uCAAuC;QACvC,IAAIjC,SAASkC,EAAE,CAACA,EAAE,CAACK,KAAK,CAAC,CAACE,cAAgBA,YAAY9B,IAAI,KAAK,WAAW;YACxE,OAAO,IAAI,CAACqB,qBAAqB,CAAChC;QACpC;QAEA,8EAA8E;QAC9E,MAAM0C,mBAAmBnD,wBAAwBS,UAAU,CAAC2C,SAAWA,OAAOhC,IAAI,KAAK;QACvF,MAAMiC,gBAAgBrD,wBAAwBS,UAAU,CAAC2C,SAAWA,OAAOhC,IAAI,KAAK;QAEpF,OAAOzB,EAAE2D,WAAW,CAAC;YACnB,IAAI,CAACR,mBAAmB,CAACK;YACzB,IAAI,CAACL,mBAAmB,CAACO;SAC1B;IACH;IAEA,qEAAqE;IAC7DE,uBAAuBC,QAA4B,EAAY;QACrE,MAAMC,QAAQC,OAAOC,OAAO,CAACH,SAASI,UAAU,EAAEjC,GAAG,CAAC,CAAC,CAACkC,KAAKC,KAAK,GAChE,IAAI,CAACC,wBAAwB,CAACF,KAAKC;QAGrC,OAAOnE,EAAEqE,aAAa,CAACP;IACzB;IAEQQ,qBAAqBxD,QAAwB,EAAY;QAC/D,MAAMmB,KAAK,IAAI,CAACb,WAAW,CAACkB,GAAG,CAACxB,SAASa,IAAI;QAC7C,IAAI,CAACM,IAAI;YACP,2CAA2C;YAC3C,OAAOjC,EAAEuE,UAAU,CACjBvE,EAAEwE,gBAAgB,IAClB,YACA,CAAC,uCAAuC,EAAE1D,SAASa,IAAI,CAAC,WAAW,CAAC,EACpE;QAEJ;QAEA,OAAO3B,EAAEiD,eAAe,CAAChB;IAC3B;IAEA,mEAAmE;IAC3DwC,qBAAqB3D,QAAwB,EAAY;QAC/D,MAAMgD,QAAiC,EAAE;QACzC,KAAK,MAAM,CAACI,KAAKQ,UAAU,IAAIX,OAAOC,OAAO,CAAClD,SAASmD,UAAU,EAAG;YAClEH,MAAMa,IAAI,CAAC,IAAI,CAACP,wBAAwB,CAACF,KAAKQ;QAChD;QACA,MAAME,OAAO9D,SAAS8D,IAAI;QAE1B,IAAIA,MAAM;YACR,OAAQA,KAAKnD,IAAI;gBACf,KAAK;oBAAU;wBACb,MAAMoD,WAAW,IAAI,CAACP,oBAAoB,CAACM;wBAC3C,qEAAqE;wBACrE,IAAI5E,EAAE8E,kBAAkB,CAACD,WAAW,OAAOA;wBAC3C,OAAO7E,EAAE+E,kBAAkB,CAAC;4BAAC/E,EAAEqE,aAAa,CAACP;4BAAQe;yBAAS;oBAChE;gBACA,KAAK;oBAAU;wBACb,KAAK,MAAM,CAACX,KAAKQ,UAAU,IAAIX,OAAOC,OAAO,CAACY,KAAKX,UAAU,EAAG;4BAC9DH,MAAMa,IAAI,CAAC,IAAI,CAACP,wBAAwB,CAACF,KAAKQ;wBAChD;wBACA;oBACF;gBACA,KAAK;oBAAW;wBACd,OAAO1E,EAAEwE,gBAAgB;oBAC3B;gBACA;oBAAS;wBACP,4CAA4C;wBAC5C,MAAM,IAAI5C,MAAM,CAAC,MAAM,EAAEgD,KAAKnD,IAAI,CAAC,qBAAqB,CAAC;oBAC3D;YACF;QACF;QAEA,IAAIX,SAASkE,cAAc,EAAE;YAC3B,MAAMC,YAAYlB,OAAOmB,MAAM,CAC7BlF,EAAEmF,mBAAmB,CACnB/E,2BACAJ,EAAEoF,gBAAgB,CAACpF,EAAEqF,aAAa,CAACrF,EAAEsF,aAAa,CAACxE,SAASkE,cAAc,MAE5E;gBAACO,UAAU;gBAAMC,UAAU;YAAI;YAEjC1B,MAAMa,IAAI,CAACM;QACb;QAEA,OAAOjF,EAAEqE,aAAa,CAACP;IACzB;IAEA,mEAAmE;IAC3DM,yBAAyBF,GAAW,EAAEQ,SAA0B,EAAyB;QAC/F,MAAMjD,OAAO,IAAI,CAACT,cAAc,CAAC0D,UAAUe,KAAK;QAChD,MAAMC,UAAUnF,iBAAiB2D,OAAOlE,EAAEoD,UAAU,CAACc,OAAOlE,EAAEsF,aAAa,CAACpB;QAC5E,MAAMyB,oBAAoB3F,EAAEmF,mBAAmB,CAACO,SAAS1F,EAAEoF,gBAAgB,CAAC3D;QAC5EkE,kBAAkBH,QAAQ,GAAGd,UAAUc,QAAQ;QAE/C,OAAOG;IACT;IAEQ3E,eACNF,QAAmE,EACzD;QACV,OAAQA,SAASW,IAAI;YACnB,KAAK;gBAAS;oBACZ,OAAO,IAAI,CAAC0B,mBAAmB,CAACrC;gBAClC;YACA,KAAK;gBAAW;oBACd,IAAIA,SAAS2E,KAAK,KAAKlD,WAAW;wBAChC,OAAOvC,EAAEqF,aAAa,CAACrF,EAAE4F,cAAc,CAAC9E,SAAS2E,KAAK;oBACxD;oBACA,OAAOzF,EAAE6F,gBAAgB;gBAC3B;YACA,KAAK;gBAAY;oBACf,OAAO,IAAI,CAACjC,sBAAsB,CAAC9C;gBACrC;YACA,KAAK;gBAAU;oBACb,OAAO,IAAI,CAACwD,oBAAoB,CAACxD;gBACnC;YACA,KAAK;gBAAQ;oBACX,OAAOd,EAAE8F,aAAa;gBACxB;YACA,KAAK;gBAAU;oBACb,IAAIhF,SAAS2E,KAAK,KAAKlD,WAAW;wBAChC,OAAOvC,EAAEqF,aAAa,CAACrF,EAAE+F,cAAc,CAACjF,SAAS2E,KAAK;oBACxD;oBACA,OAAOzF,EAAEgG,eAAe;gBAC1B;YACA,KAAK;gBAAU;oBACb,OAAO,IAAI,CAACvB,oBAAoB,CAAC3D;gBACnC;YACA,KAAK;gBAAU;oBACb,IAAIA,SAAS2E,KAAK,KAAKlD,WAAW;wBAChC,OAAOvC,EAAEqF,aAAa,CAACrF,EAAEsF,aAAa,CAACxE,SAAS2E,KAAK;oBACvD;oBACA,OAAOzF,EAAEiG,eAAe;gBAC1B;YACA,KAAK;gBAAQ;oBACX,OAAO,IAAI,CAACjF,cAAc,CAACF,SAAS2E,KAAK;gBAC3C;YACA,KAAK;gBAAS;oBACZ,OAAO,IAAI,CAACS,mBAAmB,CAACpF;gBAClC;YACA,KAAK;gBAAW;oBACd,OAAOd,EAAEwE,gBAAgB;gBAC3B;YAEA;gBAAS;oBACP,MAAM,IAAI5C,MACR,CAAC,mCAAmC,EAClC,4CAA4C;oBAC5Cd,SAASW,IAAI,CACd,+BAA+B,CAAC;gBAErC;QACF;IACF;IAEA,kEAAkE;IAC1DyE,oBAAoBpF,QAAuB,EAAY;QAC7D,IAAIA,SAASkC,EAAE,CAACmD,MAAM,KAAK,GAAG,OAAOnG,EAAEoG,cAAc;QACrD,IAAItF,SAASkC,EAAE,CAACmD,MAAM,KAAK,GAAG,OAAO,IAAI,CAACnF,cAAc,CAACF,SAASkC,EAAE,CAAC,EAAE;QACvE,OAAOhD,EAAE2D,WAAW,CAAC7C,SAASkC,EAAE,CAAChB,GAAG,CAAC,CAACmC,OAAS,IAAI,CAACnD,cAAc,CAACmD;IACrE;AACF;AAEA,OAAO,SAASjD,+BAA+BJ,QAAkB;IAC/D,OAAQA,SAASW,IAAI;QACnB,KAAK;YAAS;gBACZ,MAAM4E,MAAMnF,+BAA+BJ,SAASkC,EAAE;gBACtDqD,IAAIC,QAAQ,IAAI,GAAE,8BAA8B;gBAChD,OAAOD;YACT;QACA,KAAK;YAAU;gBACb,0DAA0D;gBAC1D,IAAIvF,SAAS8D,IAAI,IAAI9D,SAAS8D,IAAI,CAACnD,IAAI,KAAK,WAAW;oBACrD,OAAO;wBAAC6E,UAAU;wBAAGC,aAAa;wBAAGC,cAAc;oBAAC,EAAE,uCAAuC;;gBAC/F;gBAEA,MAAMC,YAAY3F,SAAS8D,IAAI,GAC3B1D,+BAA+BJ,SAAS8D,IAAI,IAC5C;oBAAC0B,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC;gBAEjD,+BAA+B;gBAC/BC,UAAUH,QAAQ,IAAI;gBAEtB,MAAMI,QAAQ3C,OAAOhC,MAAM,CAACjB,SAASmD,UAAU;gBAC/C,IAAIoC,MAAMI;gBACV,KAAK,MAAM/B,aAAagC,MAAO;oBAC7B,MAAM,EAACJ,QAAQ,EAAEC,WAAW,EAAEC,YAAY,EAAC,GAAGtF,+BAC5CwD,UAAUe,KAAK;oBAEjBY,MAAM;wBACJC,UAAUD,IAAIC,QAAQ,GAAGA;wBACzBC,aAAaF,IAAIE,WAAW,GAAGA;wBAC/BC,cAAcH,IAAIG,YAAY,GAAGA;oBACnC;gBACF;gBACA,OAAOH;YACT;QACA,KAAK;YAAS;gBACZ,IAAIvF,SAASkC,EAAE,CAACmD,MAAM,KAAK,GAAG;oBAC5B,OAAO;wBAACG,UAAU;wBAAGC,aAAa;wBAAGC,cAAc;oBAAC;gBACtD;gBAEA,IAAIH,MAAM;oBAACC,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC,EAAE,8BAA8B;;gBACvF,KAAK,MAAM/E,QAAQX,SAASkC,EAAE,CAAE;oBAC9B,MAAM,EAACsD,QAAQ,EAAEC,WAAW,EAAEC,YAAY,EAAC,GAAGtF,+BAA+BO;oBAC7E4E,MAAM;wBACJC,UAAUD,IAAIC,QAAQ,GAAGA;wBACzBC,aAAaF,IAAIE,WAAW,GAAGA;wBAC/BC,cAAcH,IAAIG,YAAY,GAAGA;oBACnC;gBACF;gBACA,OAAOH;YACT;QACA,KAAK;YAAW;gBACd,OAAO;oBAACC,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC;YACtD;QACA;YAAS;gBACP,OAAO;oBAACF,UAAU;oBAAGC,aAAa;oBAAGC,cAAc;gBAAC;YACtD;IACF;AACF"}

package/dist/typescript/typeGenerator.js

@@ -190,7 +190,7 @@ import { QueryEvaluationError } from './types.js';
     async generateTypes(options) {
         const { reporter: report } = options;
         const internalReferenceSymbol = this.getInternalReferenceSymbolDeclaration();
-        const arrayOfDeclaration = this.getArrayOfDeclaration();
+        const schemaTypeGenerator = this.getSchemaTypeGenerator(options);
         const schemaTypeDeclarations = this.getSchemaTypeDeclarations(options);
         const allSanitySchemaTypesDeclaration = this.getAllSanitySchemaTypesDeclaration(options);
         report?.event.generatedSchemaTypes({
@@ -208,13 +208,17 @@ import { QueryEvaluationError } from './types.js';
         code += allSanitySchemaTypesDeclaration.code;
         program.body.push(internalReferenceSymbol.ast);
         code += internalReferenceSymbol.code;
-        program.body.push(arrayOfDeclaration.ast);
-        code += arrayOfDeclaration.code;
         const evaluatedModules = await TypeGenerator.getEvaluatedModules({
             ...options,
             schemaTypeDeclarations,
-            schemaTypeGenerator: this.getSchemaTypeGenerator(options)
+            schemaTypeGenerator
         });
+        // Only generate ArrayOf if it's actually used
+        if (schemaTypeGenerator.isArrayOfUsed()) {
+            const arrayOfDeclaration = this.getArrayOfDeclaration();
+            program.body.push(arrayOfDeclaration.ast);
+            code += arrayOfDeclaration.code;
+        }
         for (const { queries } of evaluatedModules){
             for (const query of queries){
                 program.body.push(query.ast);

package/dist/typescript/typeGenerator.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/typescript/typeGenerator.ts"],"sourcesContent":["/* eslint-disable unicorn/consistent-function-scoping */\nimport process from 'node:process'\n\nimport * as t from '@babel/types'\nimport {type WorkerChannel, type WorkerChannelReporter} from '@sanity/worker-channels'\nimport {type SchemaType} from 'groq-js'\nimport {createSelector} from 'reselect'\n\nimport {resultSuffix} from '../casing.js'\nimport {\n  ALL_SANITY_SCHEMA_TYPES,\n  ARRAY_OF,\n  INTERNAL_REFERENCE_SYMBOL,\n  SANITY_QUERIES,\n} from './constants.js'\nimport {\n  computeOnce,\n  generateCode,\n  getUniqueIdentifierForName,\n  normalizePrintablePath,\n} from './helpers.js'\nimport {SchemaTypeGenerator} from './schemaTypeGenerator.js'\nimport {\n  type EvaluatedModule,\n  type EvaluatedQuery,\n  type ExtractedModule,\n  QueryEvaluationError,\n  type QueryExtractionError,\n} from './types.js'\n\nexport type TypegenWorkerChannel = WorkerChannel.Definition<{\n  evaluatedModules: WorkerChannel.Stream<EvaluatedModule>\n  generatedQueryTypes: WorkerChannel.Event<{\n    queryMapDeclaration: {ast: t.Program; code: string}\n  }>\n  generatedSchemaTypes: WorkerChannel.Event<{\n    allSanitySchemaTypesDeclaration: {\n      ast: t.ExportNamedDeclaration\n      code: string\n      id: t.Identifier\n    }\n    internalReferenceSymbol: {\n      ast: t.ExportNamedDeclaration\n      code: string\n      id: t.Identifier\n    }\n    schemaTypeDeclarations: {\n      ast: t.ExportNamedDeclaration\n      code: string\n      id: t.Identifier\n      name: string\n      tsType: t.TSType\n    }[]\n  }>\n}>\n\nexport interface GenerateTypesOptions {\n  schema: SchemaType\n\n  overloadClientMethods?: boolean\n  queries?: AsyncIterable<ExtractedModule>\n  reporter?: WorkerChannelReporter<TypegenWorkerChannel>\n  root?: string\n  schemaPath?: string\n}\n\ntype GetEvaluatedModulesOptions = GenerateTypesOptions & {\n  schemaTypeDeclarations: ReturnType<TypeGenerator['getSchemaTypeDeclarations']>\n  schemaTypeGenerator: SchemaTypeGenerator\n}\ntype GetQueryMapDeclarationOptions = GenerateTypesOptions & {\n  evaluatedModules: EvaluatedModule[]\n}\n\n/**\n * A class used to generate TypeScript types from a given schema\n * @beta\n */\nexport class TypeGenerator {\n  private getSchemaTypeGenerator = createSelector(\n    [(options: GenerateTypesOptions) => options.schema],\n\n    (schema) => new SchemaTypeGenerator(schema),\n  )\n\n  private getSchemaTypeDeclarations = createSelector(\n    [\n      (options: GenerateTypesOptions) => options.root,\n      (options: GenerateTypesOptions) => options.schemaPath,\n      this.getSchemaTypeGenerator,\n    ],\n\n    (root = process.cwd(), schemaPath, schema) =>\n      [...schema].map(({id, name, tsType}, index) => {\n        const typeAlias = t.tsTypeAliasDeclaration(id, null, tsType)\n        let ast = t.exportNamedDeclaration(typeAlias)\n\n        if (index === 0 && schemaPath) {\n          ast = t.addComments(ast, 'leading', [\n            {type: 'CommentLine', value: ` Source: ${normalizePrintablePath(root, schemaPath)}`},\n          ])\n        }\n        const code = generateCode(ast)\n        return {ast, code, id, name, tsType}\n      }),\n  )\n\n  private getAllSanitySchemaTypesDeclaration = createSelector(\n    [this.getSchemaTypeDeclarations],\n    (schemaTypes) => {\n      const ast = t.exportNamedDeclaration(\n        t.tsTypeAliasDeclaration(\n          ALL_SANITY_SCHEMA_TYPES,\n          null,\n          schemaTypes.length > 0\n            ? t.tsUnionType(schemaTypes.map(({id}) => t.tsTypeReference(id)))\n            : t.tsNeverKeyword(),\n        ),\n      )\n      const code = generateCode(ast)\n\n      return {ast, code, id: ALL_SANITY_SCHEMA_TYPES}\n    },\n  )\n\n  private getArrayOfDeclaration = computeOnce(() => {\n    // Creates: type ArrayOf<T> = Array<T & { _key: string }>;\n    const typeParam = t.tsTypeParameter(null, null, 'T')\n    const intersectionType = t.tsIntersectionType([\n      t.tsTypeReference(t.identifier('T')),\n      t.tsTypeLiteral([\n        t.tsPropertySignature(t.identifier('_key'), t.tsTypeAnnotation(t.tsStringKeyword())),\n      ]),\n    ])\n    const arrayType = t.tsTypeReference(\n      t.identifier('Array'),\n      t.tsTypeParameterInstantiation([intersectionType]),\n    )\n\n    const ast = t.tsTypeAliasDeclaration(\n      ARRAY_OF,\n      t.tsTypeParameterDeclaration([typeParam]),\n      arrayType,\n    )\n    const code = generateCode(ast)\n\n    return {ast, code, id: ARRAY_OF}\n  })\n\n  private getInternalReferenceSymbolDeclaration = computeOnce(() => {\n    const typeOperator = t.tsTypeOperator(t.tsSymbolKeyword(), 'unique')\n\n    const id = INTERNAL_REFERENCE_SYMBOL\n    id.typeAnnotation = t.tsTypeAnnotation(typeOperator)\n\n    const declaration = t.variableDeclaration('const', [t.variableDeclarator(id)])\n    declaration.declare = true\n    const ast = t.exportNamedDeclaration(declaration)\n    const code = generateCode(ast)\n\n    return {ast, code, id}\n  })\n\n  private static async getEvaluatedModules({\n    queries: extractedModules,\n    reporter: report,\n    root = process.cwd(),\n    schemaTypeDeclarations,\n    schemaTypeGenerator,\n  }: GetEvaluatedModulesOptions) {\n    if (!extractedModules) {\n      report?.stream.evaluatedModules.end()\n      return []\n    }\n\n    const currentIdentifiers = new Set<string>(schemaTypeDeclarations.map(({id}) => id.name))\n    const evaluatedModuleResults: EvaluatedModule[] = []\n\n    for await (const {filename, ...extractedModule} of extractedModules) {\n      const queries: EvaluatedQuery[] = []\n      const errors: (QueryEvaluationError | QueryExtractionError)[] = [...extractedModule.errors]\n\n      for (const extractedQuery of extractedModule.queries) {\n        const {variable} = extractedQuery\n        try {\n          const {stats, tsType} = schemaTypeGenerator.evaluateQuery(extractedQuery)\n          const id = getUniqueIdentifierForName(resultSuffix(variable.id.name), currentIdentifiers)\n          const typeAlias = t.tsTypeAliasDeclaration(id, null, tsType)\n          const trimmedQuery = extractedQuery.query.replaceAll(/(\\r\\n|\\n|\\r)/gm, '').trim()\n          const ast = t.addComments(t.exportNamedDeclaration(typeAlias), 'leading', [\n            {type: 'CommentLine', value: ` Source: ${normalizePrintablePath(root, filename)}`},\n            {type: 'CommentLine', value: ` Variable: ${variable.id.name}`},\n            {type: 'CommentLine', value: ` Query: ${trimmedQuery}`},\n          ])\n\n          const evaluatedQueryResult: EvaluatedQuery = {\n            ast,\n            code: generateCode(ast),\n            id,\n            stats,\n            tsType,\n            ...extractedQuery,\n          }\n\n          currentIdentifiers.add(id.name)\n          queries.push(evaluatedQueryResult)\n        } catch (cause) {\n          errors.push(new QueryEvaluationError({cause, filename, variable}))\n        }\n      }\n\n      const evaluatedModule: EvaluatedModule = {\n        errors,\n        filename,\n        queries,\n      }\n      report?.stream.evaluatedModules.emit(evaluatedModule)\n      evaluatedModuleResults.push(evaluatedModule)\n    }\n    report?.stream.evaluatedModules.end()\n\n    return evaluatedModuleResults\n  }\n\n  private static async getQueryMapDeclaration({\n    evaluatedModules,\n    overloadClientMethods = true,\n  }: GetQueryMapDeclarationOptions) {\n    if (!overloadClientMethods) return {ast: t.program([]), code: ''}\n\n    const queries = evaluatedModules.flatMap((module) => module.queries)\n    if (queries.length === 0) return {ast: t.program([]), code: ''}\n\n    const typesByQuerystring: {[query: string]: string[]} = {}\n    for (const {id, query} of queries) {\n      typesByQuerystring[query] ??= []\n      typesByQuerystring[query].push(id.name)\n    }\n\n    const queryReturnInterface = t.tsInterfaceDeclaration(\n      SANITY_QUERIES,\n      null,\n      [],\n      t.tsInterfaceBody(\n        Object.entries(typesByQuerystring).map(([query, types]) => {\n          return t.tsPropertySignature(\n            t.stringLiteral(query),\n            t.tsTypeAnnotation(\n              types.length > 0\n                ? t.tsUnionType(types.map((type) => t.tsTypeReference(t.identifier(type))))\n                : t.tsNeverKeyword(),\n            ),\n          )\n        }),\n      ),\n    )\n\n    const declareModule = t.declareModule(\n      t.stringLiteral('@sanity/client'),\n      t.blockStatement([queryReturnInterface]),\n    )\n\n    const clientImport = t.addComments(\n      t.importDeclaration([], t.stringLiteral('@sanity/client')),\n      'leading',\n      [{type: 'CommentLine', value: ' Query TypeMap'}],\n    )\n\n    const ast = t.program([clientImport, declareModule])\n    const code = generateCode(ast)\n    return {ast, code}\n  }\n\n  async generateTypes(options: GenerateTypesOptions) {\n    const {reporter: report} = options\n    const internalReferenceSymbol = this.getInternalReferenceSymbolDeclaration()\n    const arrayOfDeclaration = this.getArrayOfDeclaration()\n    const schemaTypeDeclarations = this.getSchemaTypeDeclarations(options)\n    const allSanitySchemaTypesDeclaration = this.getAllSanitySchemaTypesDeclaration(options)\n\n    report?.event.generatedSchemaTypes({\n      allSanitySchemaTypesDeclaration,\n      internalReferenceSymbol,\n      schemaTypeDeclarations,\n    })\n\n    const program = t.program([])\n    let code = ''\n\n    for (const declaration of schemaTypeDeclarations) {\n      program.body.push(declaration.ast)\n      code += declaration.code\n    }\n\n    program.body.push(allSanitySchemaTypesDeclaration.ast)\n    code += allSanitySchemaTypesDeclaration.code\n\n    program.body.push(internalReferenceSymbol.ast)\n    code += internalReferenceSymbol.code\n\n    program.body.push(arrayOfDeclaration.ast)\n    code += arrayOfDeclaration.code\n\n    const evaluatedModules = await TypeGenerator.getEvaluatedModules({\n      ...options,\n      schemaTypeDeclarations,\n      schemaTypeGenerator: this.getSchemaTypeGenerator(options),\n    })\n    for (const {queries} of evaluatedModules) {\n      for (const query of queries) {\n        program.body.push(query.ast)\n        code += query.code\n      }\n    }\n\n    const queryMapDeclaration = await TypeGenerator.getQueryMapDeclaration({\n      ...options,\n      evaluatedModules,\n    })\n    program.body.push(...queryMapDeclaration.ast.body)\n    code += queryMapDeclaration.code\n\n    report?.event.generatedQueryTypes({queryMapDeclaration})\n\n    return {ast: program, code}\n  }\n}\n"],"names":["process","t","createSelector","resultSuffix","ALL_SANITY_SCHEMA_TYPES","ARRAY_OF","INTERNAL_REFERENCE_SYMBOL","SANITY_QUERIES","computeOnce","generateCode","getUniqueIdentifierForName","normalizePrintablePath","SchemaTypeGenerator","QueryEvaluationError","TypeGenerator","getSchemaTypeGenerator","options","schema","getSchemaTypeDeclarations","root","schemaPath","cwd","map","id","name","tsType","index","typeAlias","tsTypeAliasDeclaration","ast","exportNamedDeclaration","addComments","type","value","code","getAllSanitySchemaTypesDeclaration","schemaTypes","length","tsUnionType","tsTypeReference","tsNeverKeyword","getArrayOfDeclaration","typeParam","tsTypeParameter","intersectionType","tsIntersectionType","identifier","tsTypeLiteral","tsPropertySignature","tsTypeAnnotation","tsStringKeyword","arrayType","tsTypeParameterInstantiation","tsTypeParameterDeclaration","getInternalReferenceSymbolDeclaration","typeOperator","tsTypeOperator","tsSymbolKeyword","typeAnnotation","declaration","variableDeclaration","variableDeclarator","declare","getEvaluatedModules","queries","extractedModules","reporter","report","schemaTypeDeclarations","schemaTypeGenerator","stream","evaluatedModules","end","currentIdentifiers","Set","evaluatedModuleResults","filename","extractedModule","errors","extractedQuery","variable","stats","evaluateQuery","trimmedQuery","query","replaceAll","trim","evaluatedQueryResult","add","push","cause","evaluatedModule","emit","getQueryMapDeclaration","overloadClientMethods","program","flatMap","module","typesByQuerystring","queryReturnInterface","tsInterfaceDeclaration","tsInterfaceBody","Object","entries","types","stringLiteral","declareModule","blockStatement","clientImport","importDeclaration","generateTypes","internalReferenceSymbol","arrayOfDeclaration","allSanitySchemaTypesDeclaration","event","generatedSchemaTypes","body","queryMapDeclaration","generatedQueryTypes"],"mappings":"AAAA,sDAAsD,GACtD,OAAOA,aAAa,eAAc;AAElC,YAAYC,OAAO,eAAc;AAGjC,SAAQC,cAAc,QAAO,WAAU;AAEvC,SAAQC,YAAY,QAAO,eAAc;AACzC,SACEC,uBAAuB,EACvBC,QAAQ,EACRC,yBAAyB,EACzBC,cAAc,QACT,iBAAgB;AACvB,SACEC,WAAW,EACXC,YAAY,EACZC,0BAA0B,EAC1BC,sBAAsB,QACjB,eAAc;AACrB,SAAQC,mBAAmB,QAAO,2BAA0B;AAC5D,SAIEC,oBAAoB,QAEf,aAAY;AA8CnB;;;CAGC,GACD,OAAO,MAAMC;IACHC,yBAAyBb,eAC/B;QAAC,CAACc,UAAkCA,QAAQC,MAAM;KAAC,EAEnD,CAACA,SAAW,IAAIL,oBAAoBK,SACrC;IAEOC,4BAA4BhB,eAClC;QACE,CAACc,UAAkCA,QAAQG,IAAI;QAC/C,CAACH,UAAkCA,QAAQI,UAAU;QACrD,IAAI,CAACL,sBAAsB;KAC5B,EAED,CAACI,OAAOnB,QAAQqB,GAAG,EAAE,EAAED,YAAYH,SACjC;eAAIA;SAAO,CAACK,GAAG,CAAC,CAAC,EAACC,EAAE,EAAEC,IAAI,EAAEC,MAAM,EAAC,EAAEC;YACnC,MAAMC,YAAY1B,EAAE2B,sBAAsB,CAACL,IAAI,MAAME;YACrD,IAAII,MAAM5B,EAAE6B,sBAAsB,CAACH;YAEnC,IAAID,UAAU,KAAKN,YAAY;gBAC7BS,MAAM5B,EAAE8B,WAAW,CAACF,KAAK,WAAW;oBAClC;wBAACG,MAAM;wBAAeC,OAAO,CAAC,SAAS,EAAEtB,uBAAuBQ,MAAMC,aAAa;oBAAA;iBACpF;YACH;YACA,MAAMc,OAAOzB,aAAaoB;YAC1B,OAAO;gBAACA;gBAAKK;gBAAMX;gBAAIC;gBAAMC;YAAM;QACrC,IACH;IAEOU,qCAAqCjC,eAC3C;QAAC,IAAI,CAACgB,yBAAyB;KAAC,EAChC,CAACkB;QACC,MAAMP,MAAM5B,EAAE6B,sBAAsB,CAClC7B,EAAE2B,sBAAsB,CACtBxB,yBACA,MACAgC,YAAYC,MAAM,GAAG,IACjBpC,EAAEqC,WAAW,CAACF,YAAYd,GAAG,CAAC,CAAC,EAACC,EAAE,EAAC,GAAKtB,EAAEsC,eAAe,CAAChB,QAC1DtB,EAAEuC,cAAc;QAGxB,MAAMN,OAAOzB,aAAaoB;QAE1B,OAAO;YAACA;YAAKK;YAAMX,IAAInB;QAAuB;IAChD,GACD;IAEOqC,wBAAwBjC,YAAY;QAC1C,0DAA0D;QAC1D,MAAMkC,YAAYzC,EAAE0C,eAAe,CAAC,MAAM,MAAM;QAChD,MAAMC,mBAAmB3C,EAAE4C,kBAAkB,CAAC;YAC5C5C,EAAEsC,eAAe,CAACtC,EAAE6C,UAAU,CAAC;YAC/B7C,EAAE8C,aAAa,CAAC;gBACd9C,EAAE+C,mBAAmB,CAAC/C,EAAE6C,UAAU,CAAC,SAAS7C,EAAEgD,gBAAgB,CAAChD,EAAEiD,eAAe;aACjF;SACF;QACD,MAAMC,YAAYlD,EAAEsC,eAAe,CACjCtC,EAAE6C,UAAU,CAAC,UACb7C,EAAEmD,4BAA4B,CAAC;YAACR;SAAiB;QAGnD,MAAMf,MAAM5B,EAAE2B,sBAAsB,CAClCvB,UACAJ,EAAEoD,0BAA0B,CAAC;YAACX;SAAU,GACxCS;QAEF,MAAMjB,OAAOzB,aAAaoB;QAE1B,OAAO;YAACA;YAAKK;YAAMX,IAAIlB;QAAQ;IACjC,GAAE;IAEMiD,wCAAwC9C,YAAY;QAC1D,MAAM+C,eAAetD,EAAEuD,cAAc,CAACvD,EAAEwD,eAAe,IAAI;QAE3D,MAAMlC,KAAKjB;QACXiB,GAAGmC,cAAc,GAAGzD,EAAEgD,gBAAgB,CAACM;QAEvC,MAAMI,cAAc1D,EAAE2D,mBAAmB,CAAC,SAAS;YAAC3D,EAAE4D,kBAAkB,CAACtC;SAAI;QAC7EoC,YAAYG,OAAO,GAAG;QACtB,MAAMjC,MAAM5B,EAAE6B,sBAAsB,CAAC6B;QACrC,MAAMzB,OAAOzB,aAAaoB;QAE1B,OAAO;YAACA;YAAKK;YAAMX;QAAE;IACvB,GAAE;IAEF,aAAqBwC,oBAAoB,EACvCC,SAASC,gBAAgB,EACzBC,UAAUC,MAAM,EAChBhD,OAAOnB,QAAQqB,GAAG,EAAE,EACpB+C,sBAAsB,EACtBC,mBAAmB,EACQ,EAAE;QAC7B,IAAI,CAACJ,kBAAkB;YACrBE,QAAQG,OAAOC,iBAAiBC;YAChC,OAAO,EAAE;QACX;QAEA,MAAMC,qBAAqB,IAAIC,IAAYN,uBAAuB9C,GAAG,CAAC,CAAC,EAACC,EAAE,EAAC,GAAKA,GAAGC,IAAI;QACvF,MAAMmD,yBAA4C,EAAE;QAEpD,WAAW,MAAM,EAACC,QAAQ,EAAE,GAAGC,iBAAgB,IAAIZ,iBAAkB;YACnE,MAAMD,UAA4B,EAAE;YACpC,MAAMc,SAA0D;mBAAID,gBAAgBC,MAAM;aAAC;YAE3F,KAAK,MAAMC,kBAAkBF,gBAAgBb,OAAO,CAAE;gBACpD,MAAM,EAACgB,QAAQ,EAAC,GAAGD;gBACnB,IAAI;oBACF,MAAM,EAACE,KAAK,EAAExD,MAAM,EAAC,GAAG4C,oBAAoBa,aAAa,CAACH;oBAC1D,MAAMxD,KAAKb,2BAA2BP,aAAa6E,SAASzD,EAAE,CAACC,IAAI,GAAGiD;oBACtE,MAAM9C,YAAY1B,EAAE2B,sBAAsB,CAACL,IAAI,MAAME;oBACrD,MAAM0D,eAAeJ,eAAeK,KAAK,CAACC,UAAU,CAAC,kBAAkB,IAAIC,IAAI;oBAC/E,MAAMzD,MAAM5B,EAAE8B,WAAW,CAAC9B,EAAE6B,sBAAsB,CAACH,YAAY,WAAW;wBACxE;4BAACK,MAAM;4BAAeC,OAAO,CAAC,SAAS,EAAEtB,uBAAuBQ,MAAMyD,WAAW;wBAAA;wBACjF;4BAAC5C,MAAM;4BAAeC,OAAO,CAAC,WAAW,EAAE+C,SAASzD,EAAE,CAACC,IAAI,EAAE;wBAAA;wBAC7D;4BAACQ,MAAM;4BAAeC,OAAO,CAAC,QAAQ,EAAEkD,cAAc;wBAAA;qBACvD;oBAED,MAAMI,uBAAuC;wBAC3C1D;wBACAK,MAAMzB,aAAaoB;wBACnBN;wBACA0D;wBACAxD;wBACA,GAAGsD,cAAc;oBACnB;oBAEAN,mBAAmBe,GAAG,CAACjE,GAAGC,IAAI;oBAC9BwC,QAAQyB,IAAI,CAACF;gBACf,EAAE,OAAOG,OAAO;oBACdZ,OAAOW,IAAI,CAAC,IAAI5E,qBAAqB;wBAAC6E;wBAAOd;wBAAUI;oBAAQ;gBACjE;YACF;YAEA,MAAMW,kBAAmC;gBACvCb;gBACAF;gBACAZ;YACF;YACAG,QAAQG,OAAOC,iBAAiBqB,KAAKD;YACrChB,uBAAuBc,IAAI,CAACE;QAC9B;QACAxB,QAAQG,OAAOC,iBAAiBC;QAEhC,OAAOG;IACT;IAEA,aAAqBkB,uBAAuB,EAC1CtB,gBAAgB,EAChBuB,wBAAwB,IAAI,EACE,EAAE;QAChC,IAAI,CAACA,uBAAuB,OAAO;YAACjE,KAAK5B,EAAE8F,OAAO,CAAC,EAAE;YAAG7D,MAAM;QAAE;QAEhE,MAAM8B,UAAUO,iBAAiByB,OAAO,CAAC,CAACC,SAAWA,OAAOjC,OAAO;QACnE,IAAIA,QAAQ3B,MAAM,KAAK,GAAG,OAAO;YAACR,KAAK5B,EAAE8F,OAAO,CAAC,EAAE;YAAG7D,MAAM;QAAE;QAE9D,MAAMgE,qBAAkD,CAAC;QACzD,KAAK,MAAM,EAAC3E,EAAE,EAAE6D,KAAK,EAAC,IAAIpB,QAAS;YACjCkC,kBAAkB,CAACd,MAAM,KAAK,EAAE;YAChCc,kBAAkB,CAACd,MAAM,CAACK,IAAI,CAAClE,GAAGC,IAAI;QACxC;QAEA,MAAM2E,uBAAuBlG,EAAEmG,sBAAsB,CACnD7F,gBACA,MACA,EAAE,EACFN,EAAEoG,eAAe,CACfC,OAAOC,OAAO,CAACL,oBAAoB5E,GAAG,CAAC,CAAC,CAAC8D,OAAOoB,MAAM;YACpD,OAAOvG,EAAE+C,mBAAmB,CAC1B/C,EAAEwG,aAAa,CAACrB,QAChBnF,EAAEgD,gBAAgB,CAChBuD,MAAMnE,MAAM,GAAG,IACXpC,EAAEqC,WAAW,CAACkE,MAAMlF,GAAG,CAAC,CAACU,OAAS/B,EAAEsC,eAAe,CAACtC,EAAE6C,UAAU,CAACd,WACjE/B,EAAEuC,cAAc;QAG1B;QAIJ,MAAMkE,gBAAgBzG,EAAEyG,aAAa,CACnCzG,EAAEwG,aAAa,CAAC,mBAChBxG,EAAE0G,cAAc,CAAC;YAACR;SAAqB;QAGzC,MAAMS,eAAe3G,EAAE8B,WAAW,CAChC9B,EAAE4G,iBAAiB,CAAC,EAAE,EAAE5G,EAAEwG,aAAa,CAAC,oBACxC,WACA;YAAC;gBAACzE,MAAM;gBAAeC,OAAO;YAAgB;SAAE;QAGlD,MAAMJ,MAAM5B,EAAE8F,OAAO,CAAC;YAACa;YAAcF;SAAc;QACnD,MAAMxE,OAAOzB,aAAaoB;QAC1B,OAAO;YAACA;YAAKK;QAAI;IACnB;IAEA,MAAM4E,cAAc9F,OAA6B,EAAE;QACjD,MAAM,EAACkD,UAAUC,MAAM,EAAC,GAAGnD;QAC3B,MAAM+F,0BAA0B,IAAI,CAACzD,qCAAqC;QAC1E,MAAM0D,qBAAqB,IAAI,CAACvE,qBAAqB;QACrD,MAAM2B,yBAAyB,IAAI,CAAClD,yBAAyB,CAACF;QAC9D,MAAMiG,kCAAkC,IAAI,CAAC9E,kCAAkC,CAACnB;QAEhFmD,QAAQ+C,MAAMC,qBAAqB;YACjCF;YACAF;YACA3C;QACF;QAEA,MAAM2B,UAAU9F,EAAE8F,OAAO,CAAC,EAAE;QAC5B,IAAI7D,OAAO;QAEX,KAAK,MAAMyB,eAAeS,uBAAwB;YAChD2B,QAAQqB,IAAI,CAAC3B,IAAI,CAAC9B,YAAY9B,GAAG;YACjCK,QAAQyB,YAAYzB,IAAI;QAC1B;QAEA6D,QAAQqB,IAAI,CAAC3B,IAAI,CAACwB,gCAAgCpF,GAAG;QACrDK,QAAQ+E,gCAAgC/E,IAAI;QAE5C6D,QAAQqB,IAAI,CAAC3B,IAAI,CAACsB,wBAAwBlF,GAAG;QAC7CK,QAAQ6E,wBAAwB7E,IAAI;QAEpC6D,QAAQqB,IAAI,CAAC3B,IAAI,CAACuB,mBAAmBnF,GAAG;QACxCK,QAAQ8E,mBAAmB9E,IAAI;QAE/B,MAAMqC,mBAAmB,MAAMzD,cAAciD,mBAAmB,CAAC;YAC/D,GAAG/C,OAAO;YACVoD;YACAC,qBAAqB,IAAI,CAACtD,sBAAsB,CAACC;QACnD;QACA,KAAK,MAAM,EAACgD,OAAO,EAAC,IAAIO,iBAAkB;YACxC,KAAK,MAAMa,SAASpB,QAAS;gBAC3B+B,QAAQqB,IAAI,CAAC3B,IAAI,CAACL,MAAMvD,GAAG;gBAC3BK,QAAQkD,MAAMlD,IAAI;YACpB;QACF;QAEA,MAAMmF,sBAAsB,MAAMvG,cAAc+E,sBAAsB,CAAC;YACrE,GAAG7E,OAAO;YACVuD;QACF;QACAwB,QAAQqB,IAAI,CAAC3B,IAAI,IAAI4B,oBAAoBxF,GAAG,CAACuF,IAAI;QACjDlF,QAAQmF,oBAAoBnF,IAAI;QAEhCiC,QAAQ+C,MAAMI,oBAAoB;YAACD;QAAmB;QAEtD,OAAO;YAACxF,KAAKkE;YAAS7D;QAAI;IAC5B;AACF"}
+{"version":3,"sources":["../../src/typescript/typeGenerator.ts"],"sourcesContent":["/* eslint-disable unicorn/consistent-function-scoping */\nimport process from 'node:process'\n\nimport * as t from '@babel/types'\nimport {type WorkerChannel, type WorkerChannelReporter} from '@sanity/worker-channels'\nimport {type SchemaType} from 'groq-js'\nimport {createSelector} from 'reselect'\n\nimport {resultSuffix} from '../casing.js'\nimport {\n  ALL_SANITY_SCHEMA_TYPES,\n  ARRAY_OF,\n  INTERNAL_REFERENCE_SYMBOL,\n  SANITY_QUERIES,\n} from './constants.js'\nimport {\n  computeOnce,\n  generateCode,\n  getUniqueIdentifierForName,\n  normalizePrintablePath,\n} from './helpers.js'\nimport {SchemaTypeGenerator} from './schemaTypeGenerator.js'\nimport {\n  type EvaluatedModule,\n  type EvaluatedQuery,\n  type ExtractedModule,\n  QueryEvaluationError,\n  type QueryExtractionError,\n} from './types.js'\n\nexport type TypegenWorkerChannel = WorkerChannel.Definition<{\n  evaluatedModules: WorkerChannel.Stream<EvaluatedModule>\n  generatedQueryTypes: WorkerChannel.Event<{\n    queryMapDeclaration: {ast: t.Program; code: string}\n  }>\n  generatedSchemaTypes: WorkerChannel.Event<{\n    allSanitySchemaTypesDeclaration: {\n      ast: t.ExportNamedDeclaration\n      code: string\n      id: t.Identifier\n    }\n    internalReferenceSymbol: {\n      ast: t.ExportNamedDeclaration\n      code: string\n      id: t.Identifier\n    }\n    schemaTypeDeclarations: {\n      ast: t.ExportNamedDeclaration\n      code: string\n      id: t.Identifier\n      name: string\n      tsType: t.TSType\n    }[]\n  }>\n}>\n\nexport interface GenerateTypesOptions {\n  schema: SchemaType\n\n  overloadClientMethods?: boolean\n  queries?: AsyncIterable<ExtractedModule>\n  reporter?: WorkerChannelReporter<TypegenWorkerChannel>\n  root?: string\n  schemaPath?: string\n}\n\ntype GetEvaluatedModulesOptions = GenerateTypesOptions & {\n  schemaTypeDeclarations: ReturnType<TypeGenerator['getSchemaTypeDeclarations']>\n  schemaTypeGenerator: SchemaTypeGenerator\n}\ntype GetQueryMapDeclarationOptions = GenerateTypesOptions & {\n  evaluatedModules: EvaluatedModule[]\n}\n\n/**\n * A class used to generate TypeScript types from a given schema\n * @beta\n */\nexport class TypeGenerator {\n  private getSchemaTypeGenerator = createSelector(\n    [(options: GenerateTypesOptions) => options.schema],\n\n    (schema) => new SchemaTypeGenerator(schema),\n  )\n\n  private getSchemaTypeDeclarations = createSelector(\n    [\n      (options: GenerateTypesOptions) => options.root,\n      (options: GenerateTypesOptions) => options.schemaPath,\n      this.getSchemaTypeGenerator,\n    ],\n\n    (root = process.cwd(), schemaPath, schema) =>\n      [...schema].map(({id, name, tsType}, index) => {\n        const typeAlias = t.tsTypeAliasDeclaration(id, null, tsType)\n        let ast = t.exportNamedDeclaration(typeAlias)\n\n        if (index === 0 && schemaPath) {\n          ast = t.addComments(ast, 'leading', [\n            {type: 'CommentLine', value: ` Source: ${normalizePrintablePath(root, schemaPath)}`},\n          ])\n        }\n        const code = generateCode(ast)\n        return {ast, code, id, name, tsType}\n      }),\n  )\n\n  private getAllSanitySchemaTypesDeclaration = createSelector(\n    [this.getSchemaTypeDeclarations],\n    (schemaTypes) => {\n      const ast = t.exportNamedDeclaration(\n        t.tsTypeAliasDeclaration(\n          ALL_SANITY_SCHEMA_TYPES,\n          null,\n          schemaTypes.length > 0\n            ? t.tsUnionType(schemaTypes.map(({id}) => t.tsTypeReference(id)))\n            : t.tsNeverKeyword(),\n        ),\n      )\n      const code = generateCode(ast)\n\n      return {ast, code, id: ALL_SANITY_SCHEMA_TYPES}\n    },\n  )\n\n  private getArrayOfDeclaration = computeOnce(() => {\n    // Creates: type ArrayOf<T> = Array<T & { _key: string }>;\n    const typeParam = t.tsTypeParameter(null, null, 'T')\n    const intersectionType = t.tsIntersectionType([\n      t.tsTypeReference(t.identifier('T')),\n      t.tsTypeLiteral([\n        t.tsPropertySignature(t.identifier('_key'), t.tsTypeAnnotation(t.tsStringKeyword())),\n      ]),\n    ])\n    const arrayType = t.tsTypeReference(\n      t.identifier('Array'),\n      t.tsTypeParameterInstantiation([intersectionType]),\n    )\n\n    const ast = t.tsTypeAliasDeclaration(\n      ARRAY_OF,\n      t.tsTypeParameterDeclaration([typeParam]),\n      arrayType,\n    )\n    const code = generateCode(ast)\n\n    return {ast, code, id: ARRAY_OF}\n  })\n\n  private getInternalReferenceSymbolDeclaration = computeOnce(() => {\n    const typeOperator = t.tsTypeOperator(t.tsSymbolKeyword(), 'unique')\n\n    const id = INTERNAL_REFERENCE_SYMBOL\n    id.typeAnnotation = t.tsTypeAnnotation(typeOperator)\n\n    const declaration = t.variableDeclaration('const', [t.variableDeclarator(id)])\n    declaration.declare = true\n    const ast = t.exportNamedDeclaration(declaration)\n    const code = generateCode(ast)\n\n    return {ast, code, id}\n  })\n\n  private static async getEvaluatedModules({\n    queries: extractedModules,\n    reporter: report,\n    root = process.cwd(),\n    schemaTypeDeclarations,\n    schemaTypeGenerator,\n  }: GetEvaluatedModulesOptions) {\n    if (!extractedModules) {\n      report?.stream.evaluatedModules.end()\n      return []\n    }\n\n    const currentIdentifiers = new Set<string>(schemaTypeDeclarations.map(({id}) => id.name))\n    const evaluatedModuleResults: EvaluatedModule[] = []\n\n    for await (const {filename, ...extractedModule} of extractedModules) {\n      const queries: EvaluatedQuery[] = []\n      const errors: (QueryEvaluationError | QueryExtractionError)[] = [...extractedModule.errors]\n\n      for (const extractedQuery of extractedModule.queries) {\n        const {variable} = extractedQuery\n        try {\n          const {stats, tsType} = schemaTypeGenerator.evaluateQuery(extractedQuery)\n          const id = getUniqueIdentifierForName(resultSuffix(variable.id.name), currentIdentifiers)\n          const typeAlias = t.tsTypeAliasDeclaration(id, null, tsType)\n          const trimmedQuery = extractedQuery.query.replaceAll(/(\\r\\n|\\n|\\r)/gm, '').trim()\n          const ast = t.addComments(t.exportNamedDeclaration(typeAlias), 'leading', [\n            {type: 'CommentLine', value: ` Source: ${normalizePrintablePath(root, filename)}`},\n            {type: 'CommentLine', value: ` Variable: ${variable.id.name}`},\n            {type: 'CommentLine', value: ` Query: ${trimmedQuery}`},\n          ])\n\n          const evaluatedQueryResult: EvaluatedQuery = {\n            ast,\n            code: generateCode(ast),\n            id,\n            stats,\n            tsType,\n            ...extractedQuery,\n          }\n\n          currentIdentifiers.add(id.name)\n          queries.push(evaluatedQueryResult)\n        } catch (cause) {\n          errors.push(new QueryEvaluationError({cause, filename, variable}))\n        }\n      }\n\n      const evaluatedModule: EvaluatedModule = {\n        errors,\n        filename,\n        queries,\n      }\n      report?.stream.evaluatedModules.emit(evaluatedModule)\n      evaluatedModuleResults.push(evaluatedModule)\n    }\n    report?.stream.evaluatedModules.end()\n\n    return evaluatedModuleResults\n  }\n\n  private static async getQueryMapDeclaration({\n    evaluatedModules,\n    overloadClientMethods = true,\n  }: GetQueryMapDeclarationOptions) {\n    if (!overloadClientMethods) return {ast: t.program([]), code: ''}\n\n    const queries = evaluatedModules.flatMap((module) => module.queries)\n    if (queries.length === 0) return {ast: t.program([]), code: ''}\n\n    const typesByQuerystring: {[query: string]: string[]} = {}\n    for (const {id, query} of queries) {\n      typesByQuerystring[query] ??= []\n      typesByQuerystring[query].push(id.name)\n    }\n\n    const queryReturnInterface = t.tsInterfaceDeclaration(\n      SANITY_QUERIES,\n      null,\n      [],\n      t.tsInterfaceBody(\n        Object.entries(typesByQuerystring).map(([query, types]) => {\n          return t.tsPropertySignature(\n            t.stringLiteral(query),\n            t.tsTypeAnnotation(\n              types.length > 0\n                ? t.tsUnionType(types.map((type) => t.tsTypeReference(t.identifier(type))))\n                : t.tsNeverKeyword(),\n            ),\n          )\n        }),\n      ),\n    )\n\n    const declareModule = t.declareModule(\n      t.stringLiteral('@sanity/client'),\n      t.blockStatement([queryReturnInterface]),\n    )\n\n    const clientImport = t.addComments(\n      t.importDeclaration([], t.stringLiteral('@sanity/client')),\n      'leading',\n      [{type: 'CommentLine', value: ' Query TypeMap'}],\n    )\n\n    const ast = t.program([clientImport, declareModule])\n    const code = generateCode(ast)\n    return {ast, code}\n  }\n\n  async generateTypes(options: GenerateTypesOptions) {\n    const {reporter: report} = options\n    const internalReferenceSymbol = this.getInternalReferenceSymbolDeclaration()\n    const schemaTypeGenerator = this.getSchemaTypeGenerator(options)\n    const schemaTypeDeclarations = this.getSchemaTypeDeclarations(options)\n    const allSanitySchemaTypesDeclaration = this.getAllSanitySchemaTypesDeclaration(options)\n\n    report?.event.generatedSchemaTypes({\n      allSanitySchemaTypesDeclaration,\n      internalReferenceSymbol,\n      schemaTypeDeclarations,\n    })\n\n    const program = t.program([])\n    let code = ''\n\n    for (const declaration of schemaTypeDeclarations) {\n      program.body.push(declaration.ast)\n      code += declaration.code\n    }\n\n    program.body.push(allSanitySchemaTypesDeclaration.ast)\n    code += allSanitySchemaTypesDeclaration.code\n\n    program.body.push(internalReferenceSymbol.ast)\n    code += internalReferenceSymbol.code\n\n    const evaluatedModules = await TypeGenerator.getEvaluatedModules({\n      ...options,\n      schemaTypeDeclarations,\n      schemaTypeGenerator,\n    })\n\n    // Only generate ArrayOf if it's actually used\n    if (schemaTypeGenerator.isArrayOfUsed()) {\n      const arrayOfDeclaration = this.getArrayOfDeclaration()\n      program.body.push(arrayOfDeclaration.ast)\n      code += arrayOfDeclaration.code\n    }\n\n    for (const {queries} of evaluatedModules) {\n      for (const query of queries) {\n        program.body.push(query.ast)\n        code += query.code\n      }\n    }\n\n    const queryMapDeclaration = await TypeGenerator.getQueryMapDeclaration({\n      ...options,\n      evaluatedModules,\n    })\n    program.body.push(...queryMapDeclaration.ast.body)\n    code += queryMapDeclaration.code\n\n    report?.event.generatedQueryTypes({queryMapDeclaration})\n\n    return {ast: program, code}\n  }\n}\n"],"names":["process","t","createSelector","resultSuffix","ALL_SANITY_SCHEMA_TYPES","ARRAY_OF","INTERNAL_REFERENCE_SYMBOL","SANITY_QUERIES","computeOnce","generateCode","getUniqueIdentifierForName","normalizePrintablePath","SchemaTypeGenerator","QueryEvaluationError","TypeGenerator","getSchemaTypeGenerator","options","schema","getSchemaTypeDeclarations","root","schemaPath","cwd","map","id","name","tsType","index","typeAlias","tsTypeAliasDeclaration","ast","exportNamedDeclaration","addComments","type","value","code","getAllSanitySchemaTypesDeclaration","schemaTypes","length","tsUnionType","tsTypeReference","tsNeverKeyword","getArrayOfDeclaration","typeParam","tsTypeParameter","intersectionType","tsIntersectionType","identifier","tsTypeLiteral","tsPropertySignature","tsTypeAnnotation","tsStringKeyword","arrayType","tsTypeParameterInstantiation","tsTypeParameterDeclaration","getInternalReferenceSymbolDeclaration","typeOperator","tsTypeOperator","tsSymbolKeyword","typeAnnotation","declaration","variableDeclaration","variableDeclarator","declare","getEvaluatedModules","queries","extractedModules","reporter","report","schemaTypeDeclarations","schemaTypeGenerator","stream","evaluatedModules","end","currentIdentifiers","Set","evaluatedModuleResults","filename","extractedModule","errors","extractedQuery","variable","stats","evaluateQuery","trimmedQuery","query","replaceAll","trim","evaluatedQueryResult","add","push","cause","evaluatedModule","emit","getQueryMapDeclaration","overloadClientMethods","program","flatMap","module","typesByQuerystring","queryReturnInterface","tsInterfaceDeclaration","tsInterfaceBody","Object","entries","types","stringLiteral","declareModule","blockStatement","clientImport","importDeclaration","generateTypes","internalReferenceSymbol","allSanitySchemaTypesDeclaration","event","generatedSchemaTypes","body","isArrayOfUsed","arrayOfDeclaration","queryMapDeclaration","generatedQueryTypes"],"mappings":"AAAA,sDAAsD,GACtD,OAAOA,aAAa,eAAc;AAElC,YAAYC,OAAO,eAAc;AAGjC,SAAQC,cAAc,QAAO,WAAU;AAEvC,SAAQC,YAAY,QAAO,eAAc;AACzC,SACEC,uBAAuB,EACvBC,QAAQ,EACRC,yBAAyB,EACzBC,cAAc,QACT,iBAAgB;AACvB,SACEC,WAAW,EACXC,YAAY,EACZC,0BAA0B,EAC1BC,sBAAsB,QACjB,eAAc;AACrB,SAAQC,mBAAmB,QAAO,2BAA0B;AAC5D,SAIEC,oBAAoB,QAEf,aAAY;AA8CnB;;;CAGC,GACD,OAAO,MAAMC;IACHC,yBAAyBb,eAC/B;QAAC,CAACc,UAAkCA,QAAQC,MAAM;KAAC,EAEnD,CAACA,SAAW,IAAIL,oBAAoBK,SACrC;IAEOC,4BAA4BhB,eAClC;QACE,CAACc,UAAkCA,QAAQG,IAAI;QAC/C,CAACH,UAAkCA,QAAQI,UAAU;QACrD,IAAI,CAACL,sBAAsB;KAC5B,EAED,CAACI,OAAOnB,QAAQqB,GAAG,EAAE,EAAED,YAAYH,SACjC;eAAIA;SAAO,CAACK,GAAG,CAAC,CAAC,EAACC,EAAE,EAAEC,IAAI,EAAEC,MAAM,EAAC,EAAEC;YACnC,MAAMC,YAAY1B,EAAE2B,sBAAsB,CAACL,IAAI,MAAME;YACrD,IAAII,MAAM5B,EAAE6B,sBAAsB,CAACH;YAEnC,IAAID,UAAU,KAAKN,YAAY;gBAC7BS,MAAM5B,EAAE8B,WAAW,CAACF,KAAK,WAAW;oBAClC;wBAACG,MAAM;wBAAeC,OAAO,CAAC,SAAS,EAAEtB,uBAAuBQ,MAAMC,aAAa;oBAAA;iBACpF;YACH;YACA,MAAMc,OAAOzB,aAAaoB;YAC1B,OAAO;gBAACA;gBAAKK;gBAAMX;gBAAIC;gBAAMC;YAAM;QACrC,IACH;IAEOU,qCAAqCjC,eAC3C;QAAC,IAAI,CAACgB,yBAAyB;KAAC,EAChC,CAACkB;QACC,MAAMP,MAAM5B,EAAE6B,sBAAsB,CAClC7B,EAAE2B,sBAAsB,CACtBxB,yBACA,MACAgC,YAAYC,MAAM,GAAG,IACjBpC,EAAEqC,WAAW,CAACF,YAAYd,GAAG,CAAC,CAAC,EAACC,EAAE,EAAC,GAAKtB,EAAEsC,eAAe,CAAChB,QAC1DtB,EAAEuC,cAAc;QAGxB,MAAMN,OAAOzB,aAAaoB;QAE1B,OAAO;YAACA;YAAKK;YAAMX,IAAInB;QAAuB;IAChD,GACD;IAEOqC,wBAAwBjC,YAAY;QAC1C,0DAA0D;QAC1D,MAAMkC,YAAYzC,EAAE0C,eAAe,CAAC,MAAM,MAAM;QAChD,MAAMC,mBAAmB3C,EAAE4C,kBAAkB,CAAC;YAC5C5C,EAAEsC,eAAe,CAACtC,EAAE6C,UAAU,CAAC;YAC/B7C,EAAE8C,aAAa,CAAC;gBACd9C,EAAE+C,mBAAmB,CAAC/C,EAAE6C,UAAU,CAAC,SAAS7C,EAAEgD,gBAAgB,CAAChD,EAAEiD,eAAe;aACjF;SACF;QACD,MAAMC,YAAYlD,EAAEsC,eAAe,CACjCtC,EAAE6C,UAAU,CAAC,UACb7C,EAAEmD,4BAA4B,CAAC;YAACR;SAAiB;QAGnD,MAAMf,MAAM5B,EAAE2B,sBAAsB,CAClCvB,UACAJ,EAAEoD,0BAA0B,CAAC;YAACX;SAAU,GACxCS;QAEF,MAAMjB,OAAOzB,aAAaoB;QAE1B,OAAO;YAACA;YAAKK;YAAMX,IAAIlB;QAAQ;IACjC,GAAE;IAEMiD,wCAAwC9C,YAAY;QAC1D,MAAM+C,eAAetD,EAAEuD,cAAc,CAACvD,EAAEwD,eAAe,IAAI;QAE3D,MAAMlC,KAAKjB;QACXiB,GAAGmC,cAAc,GAAGzD,EAAEgD,gBAAgB,CAACM;QAEvC,MAAMI,cAAc1D,EAAE2D,mBAAmB,CAAC,SAAS;YAAC3D,EAAE4D,kBAAkB,CAACtC;SAAI;QAC7EoC,YAAYG,OAAO,GAAG;QACtB,MAAMjC,MAAM5B,EAAE6B,sBAAsB,CAAC6B;QACrC,MAAMzB,OAAOzB,aAAaoB;QAE1B,OAAO;YAACA;YAAKK;YAAMX;QAAE;IACvB,GAAE;IAEF,aAAqBwC,oBAAoB,EACvCC,SAASC,gBAAgB,EACzBC,UAAUC,MAAM,EAChBhD,OAAOnB,QAAQqB,GAAG,EAAE,EACpB+C,sBAAsB,EACtBC,mBAAmB,EACQ,EAAE;QAC7B,IAAI,CAACJ,kBAAkB;YACrBE,QAAQG,OAAOC,iBAAiBC;YAChC,OAAO,EAAE;QACX;QAEA,MAAMC,qBAAqB,IAAIC,IAAYN,uBAAuB9C,GAAG,CAAC,CAAC,EAACC,EAAE,EAAC,GAAKA,GAAGC,IAAI;QACvF,MAAMmD,yBAA4C,EAAE;QAEpD,WAAW,MAAM,EAACC,QAAQ,EAAE,GAAGC,iBAAgB,IAAIZ,iBAAkB;YACnE,MAAMD,UAA4B,EAAE;YACpC,MAAMc,SAA0D;mBAAID,gBAAgBC,MAAM;aAAC;YAE3F,KAAK,MAAMC,kBAAkBF,gBAAgBb,OAAO,CAAE;gBACpD,MAAM,EAACgB,QAAQ,EAAC,GAAGD;gBACnB,IAAI;oBACF,MAAM,EAACE,KAAK,EAAExD,MAAM,EAAC,GAAG4C,oBAAoBa,aAAa,CAACH;oBAC1D,MAAMxD,KAAKb,2BAA2BP,aAAa6E,SAASzD,EAAE,CAACC,IAAI,GAAGiD;oBACtE,MAAM9C,YAAY1B,EAAE2B,sBAAsB,CAACL,IAAI,MAAME;oBACrD,MAAM0D,eAAeJ,eAAeK,KAAK,CAACC,UAAU,CAAC,kBAAkB,IAAIC,IAAI;oBAC/E,MAAMzD,MAAM5B,EAAE8B,WAAW,CAAC9B,EAAE6B,sBAAsB,CAACH,YAAY,WAAW;wBACxE;4BAACK,MAAM;4BAAeC,OAAO,CAAC,SAAS,EAAEtB,uBAAuBQ,MAAMyD,WAAW;wBAAA;wBACjF;4BAAC5C,MAAM;4BAAeC,OAAO,CAAC,WAAW,EAAE+C,SAASzD,EAAE,CAACC,IAAI,EAAE;wBAAA;wBAC7D;4BAACQ,MAAM;4BAAeC,OAAO,CAAC,QAAQ,EAAEkD,cAAc;wBAAA;qBACvD;oBAED,MAAMI,uBAAuC;wBAC3C1D;wBACAK,MAAMzB,aAAaoB;wBACnBN;wBACA0D;wBACAxD;wBACA,GAAGsD,cAAc;oBACnB;oBAEAN,mBAAmBe,GAAG,CAACjE,GAAGC,IAAI;oBAC9BwC,QAAQyB,IAAI,CAACF;gBACf,EAAE,OAAOG,OAAO;oBACdZ,OAAOW,IAAI,CAAC,IAAI5E,qBAAqB;wBAAC6E;wBAAOd;wBAAUI;oBAAQ;gBACjE;YACF;YAEA,MAAMW,kBAAmC;gBACvCb;gBACAF;gBACAZ;YACF;YACAG,QAAQG,OAAOC,iBAAiBqB,KAAKD;YACrChB,uBAAuBc,IAAI,CAACE;QAC9B;QACAxB,QAAQG,OAAOC,iBAAiBC;QAEhC,OAAOG;IACT;IAEA,aAAqBkB,uBAAuB,EAC1CtB,gBAAgB,EAChBuB,wBAAwB,IAAI,EACE,EAAE;QAChC,IAAI,CAACA,uBAAuB,OAAO;YAACjE,KAAK5B,EAAE8F,OAAO,CAAC,EAAE;YAAG7D,MAAM;QAAE;QAEhE,MAAM8B,UAAUO,iBAAiByB,OAAO,CAAC,CAACC,SAAWA,OAAOjC,OAAO;QACnE,IAAIA,QAAQ3B,MAAM,KAAK,GAAG,OAAO;YAACR,KAAK5B,EAAE8F,OAAO,CAAC,EAAE;YAAG7D,MAAM;QAAE;QAE9D,MAAMgE,qBAAkD,CAAC;QACzD,KAAK,MAAM,EAAC3E,EAAE,EAAE6D,KAAK,EAAC,IAAIpB,QAAS;YACjCkC,kBAAkB,CAACd,MAAM,KAAK,EAAE;YAChCc,kBAAkB,CAACd,MAAM,CAACK,IAAI,CAAClE,GAAGC,IAAI;QACxC;QAEA,MAAM2E,uBAAuBlG,EAAEmG,sBAAsB,CACnD7F,gBACA,MACA,EAAE,EACFN,EAAEoG,eAAe,CACfC,OAAOC,OAAO,CAACL,oBAAoB5E,GAAG,CAAC,CAAC,CAAC8D,OAAOoB,MAAM;YACpD,OAAOvG,EAAE+C,mBAAmB,CAC1B/C,EAAEwG,aAAa,CAACrB,QAChBnF,EAAEgD,gBAAgB,CAChBuD,MAAMnE,MAAM,GAAG,IACXpC,EAAEqC,WAAW,CAACkE,MAAMlF,GAAG,CAAC,CAACU,OAAS/B,EAAEsC,eAAe,CAACtC,EAAE6C,UAAU,CAACd,WACjE/B,EAAEuC,cAAc;QAG1B;QAIJ,MAAMkE,gBAAgBzG,EAAEyG,aAAa,CACnCzG,EAAEwG,aAAa,CAAC,mBAChBxG,EAAE0G,cAAc,CAAC;YAACR;SAAqB;QAGzC,MAAMS,eAAe3G,EAAE8B,WAAW,CAChC9B,EAAE4G,iBAAiB,CAAC,EAAE,EAAE5G,EAAEwG,aAAa,CAAC,oBACxC,WACA;YAAC;gBAACzE,MAAM;gBAAeC,OAAO;YAAgB;SAAE;QAGlD,MAAMJ,MAAM5B,EAAE8F,OAAO,CAAC;YAACa;YAAcF;SAAc;QACnD,MAAMxE,OAAOzB,aAAaoB;QAC1B,OAAO;YAACA;YAAKK;QAAI;IACnB;IAEA,MAAM4E,cAAc9F,OAA6B,EAAE;QACjD,MAAM,EAACkD,UAAUC,MAAM,EAAC,GAAGnD;QAC3B,MAAM+F,0BAA0B,IAAI,CAACzD,qCAAqC;QAC1E,MAAMe,sBAAsB,IAAI,CAACtD,sBAAsB,CAACC;QACxD,MAAMoD,yBAAyB,IAAI,CAAClD,yBAAyB,CAACF;QAC9D,MAAMgG,kCAAkC,IAAI,CAAC7E,kCAAkC,CAACnB;QAEhFmD,QAAQ8C,MAAMC,qBAAqB;YACjCF;YACAD;YACA3C;QACF;QAEA,MAAM2B,UAAU9F,EAAE8F,OAAO,CAAC,EAAE;QAC5B,IAAI7D,OAAO;QAEX,KAAK,MAAMyB,eAAeS,uBAAwB;YAChD2B,QAAQoB,IAAI,CAAC1B,IAAI,CAAC9B,YAAY9B,GAAG;YACjCK,QAAQyB,YAAYzB,IAAI;QAC1B;QAEA6D,QAAQoB,IAAI,CAAC1B,IAAI,CAACuB,gCAAgCnF,GAAG;QACrDK,QAAQ8E,gCAAgC9E,IAAI;QAE5C6D,QAAQoB,IAAI,CAAC1B,IAAI,CAACsB,wBAAwBlF,GAAG;QAC7CK,QAAQ6E,wBAAwB7E,IAAI;QAEpC,MAAMqC,mBAAmB,MAAMzD,cAAciD,mBAAmB,CAAC;YAC/D,GAAG/C,OAAO;YACVoD;YACAC;QACF;QAEA,8CAA8C;QAC9C,IAAIA,oBAAoB+C,aAAa,IAAI;YACvC,MAAMC,qBAAqB,IAAI,CAAC5E,qBAAqB;YACrDsD,QAAQoB,IAAI,CAAC1B,IAAI,CAAC4B,mBAAmBxF,GAAG;YACxCK,QAAQmF,mBAAmBnF,IAAI;QACjC;QAEA,KAAK,MAAM,EAAC8B,OAAO,EAAC,IAAIO,iBAAkB;YACxC,KAAK,MAAMa,SAASpB,QAAS;gBAC3B+B,QAAQoB,IAAI,CAAC1B,IAAI,CAACL,MAAMvD,GAAG;gBAC3BK,QAAQkD,MAAMlD,IAAI;YACpB;QACF;QAEA,MAAMoF,sBAAsB,MAAMxG,cAAc+E,sBAAsB,CAAC;YACrE,GAAG7E,OAAO;YACVuD;QACF;QACAwB,QAAQoB,IAAI,CAAC1B,IAAI,IAAI6B,oBAAoBzF,GAAG,CAACsF,IAAI;QACjDjF,QAAQoF,oBAAoBpF,IAAI;QAEhCiC,QAAQ8C,MAAMM,oBAAoB;YAACD;QAAmB;QAEtD,OAAO;YAACzF,KAAKkE;YAAS7D;QAAI;IAC5B;AACF"}

package/oclif.manifest.json

@@ -41,5 +41,5 @@
       ]
     }
   },
-  "version": "5.9.1"
+  "version": "5.9.2"
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sanity/codegen",
-  "version": "5.9.1",
+  "version": "5.9.2",
   "description": "Codegen toolkit for Sanity.io",
   "keywords": [
     "sanity",