@osdk/generator 2.4.0-beta.16 → 2.4.0-beta.18

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @osdk/generator
 
+## 2.4.0-beta.18
+
+### Patch Changes
+
+- 44bbbe0: Add interface query support.
+- Updated dependencies [722c89a]
+- Updated dependencies [44bbbe0]
+  - @osdk/api@2.4.0-beta.18
+  - @osdk/generator-converters@2.4.0-beta.18
+
+## 2.4.0-beta.17
+
+### Patch Changes
+
+- @osdk/api@2.4.0-beta.17
+- @osdk/generator-converters@2.4.0-beta.17
+
 ## 2.4.0-beta.16
 
 ### Patch Changes

package/build/browser/shared/getInterfaceTypeApiNamesFromQuery.js

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2025 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { getInterfaceTypesFromQueryDataType } from "./getInterfaceTypesFromQueryDataType.js";
+export function getInterfaceTypeApiNamesFromQuery(query) {
+  const types = new Set();
+  for (const {
+    dataType
+  } of Object.values(query.parameters)) {
+    getInterfaceTypesFromQueryDataType(dataType, types);
+  }
+  getInterfaceTypesFromQueryDataType(query.output, types);
+  return Array.from(types);
+}
+//# sourceMappingURL=getInterfaceTypeApiNamesFromQuery.js.map

package/build/browser/shared/getInterfaceTypeApiNamesFromQuery.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getInterfaceTypeApiNamesFromQuery.js","names":["getInterfaceTypesFromQueryDataType","getInterfaceTypeApiNamesFromQuery","query","types","Set","dataType","Object","values","parameters","output","Array","from"],"sources":["getInterfaceTypeApiNamesFromQuery.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { QueryTypeV2 } from \"@osdk/foundry.ontologies\";\nimport type { EnhancedQuery } from \"../GenerateContext/EnhancedQuery.js\";\nimport { getInterfaceTypesFromQueryDataType } from \"./getInterfaceTypesFromQueryDataType.js\";\n\nexport function getInterfaceTypeApiNamesFromQuery(\n  query: QueryTypeV2 | EnhancedQuery,\n): string[] {\n  const types = new Set<string>();\n\n  for (const { dataType } of Object.values(query.parameters)) {\n    getInterfaceTypesFromQueryDataType(dataType, types);\n  }\n  getInterfaceTypesFromQueryDataType(query.output, types);\n\n  return Array.from(types);\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAIA,SAASA,kCAAkC,QAAQ,yCAAyC;AAE5F,OAAO,SAASC,iCAAiCA,CAC/CC,KAAkC,EACxB;EACV,MAAMC,KAAK,GAAG,IAAIC,GAAG,CAAS,CAAC;EAE/B,KAAK,MAAM;IAAEC;EAAS,CAAC,IAAIC,MAAM,CAACC,MAAM,CAACL,KAAK,CAACM,UAAU,CAAC,EAAE;IAC1DR,kCAAkC,CAACK,QAAQ,EAAEF,KAAK,CAAC;EACrD;EACAH,kCAAkC,CAACE,KAAK,CAACO,MAAM,EAAEN,KAAK,CAAC;EAEvD,OAAOO,KAAK,CAACC,IAAI,CAACR,KAAK,CAAC;AAC1B","ignoreList":[]}

package/build/browser/shared/getInterfaceTypesFromQueryDataType.js

@@ -0,0 +1,63 @@
+/*
+ * Copyright 2025 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+export function getInterfaceTypesFromQueryDataType(dataType, types) {
+  switch (dataType.type) {
+    case "array":
+    case "set":
+      getInterfaceTypesFromQueryDataType(dataType.subType, types);
+      return;
+    case "interfaceObject":
+    case "interfaceObjectSet":
+      types.add(dataType.interfaceTypeApiName);
+      return;
+    case "struct":
+      for (const prop of Object.values(dataType.fields)) {
+        getInterfaceTypesFromQueryDataType(prop.fieldType, types);
+      }
+      return;
+    case "union":
+      for (const type of dataType.unionTypes) {
+        getInterfaceTypesFromQueryDataType(type, types);
+      }
+      return;
+    case "entrySet":
+      getInterfaceTypesFromQueryDataType(dataType.keyType, types);
+      getInterfaceTypesFromQueryDataType(dataType.valueType, types);
+      return;
+    case "attachment":
+    case "boolean":
+    case "date":
+    case "double":
+    case "float":
+    case "integer":
+    case "long":
+    case "null":
+    case "string":
+    case "threeDimensionalAggregation":
+    case "timestamp":
+    case "twoDimensionalAggregation":
+    case "entrySet":
+    case "unsupported":
+    case "object":
+    case "objectSet":
+      /* complete no-op */
+      return;
+    default:
+      throw new Error(`Unknown query data type ${dataType.type}`);
+  }
+}
+//# sourceMappingURL=getInterfaceTypesFromQueryDataType.js.map

package/build/browser/shared/getInterfaceTypesFromQueryDataType.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getInterfaceTypesFromQueryDataType.js","names":["getInterfaceTypesFromQueryDataType","dataType","types","type","subType","add","interfaceTypeApiName","prop","Object","values","fields","fieldType","unionTypes","keyType","valueType","Error"],"sources":["getInterfaceTypesFromQueryDataType.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { QueryDataType } from \"@osdk/foundry.ontologies\";\n\nexport function getInterfaceTypesFromQueryDataType(\n  dataType: QueryDataType,\n  types: Set<string>,\n): void {\n  switch (dataType.type) {\n    case \"array\":\n    case \"set\":\n      getInterfaceTypesFromQueryDataType(dataType.subType, types);\n      return;\n    case \"interfaceObject\":\n    case \"interfaceObjectSet\":\n      types.add(dataType.interfaceTypeApiName!);\n      return;\n\n    case \"struct\":\n      for (const prop of Object.values(dataType.fields)) {\n        getInterfaceTypesFromQueryDataType(prop.fieldType, types);\n      }\n      return;\n\n    case \"union\":\n      for (const type of dataType.unionTypes) {\n        getInterfaceTypesFromQueryDataType(type, types);\n      }\n      return;\n\n    case \"entrySet\":\n      getInterfaceTypesFromQueryDataType(dataType.keyType, types);\n      getInterfaceTypesFromQueryDataType(dataType.valueType, types);\n      return;\n\n    case \"attachment\":\n    case \"boolean\":\n    case \"date\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"null\":\n    case \"string\":\n    case \"threeDimensionalAggregation\":\n    case \"timestamp\":\n    case \"twoDimensionalAggregation\":\n    case \"entrySet\":\n    case \"unsupported\":\n    case \"object\":\n    case \"objectSet\":\n      /* complete no-op */\n      return;\n\n    default:\n      const _: never = dataType;\n      throw new Error(\n        `Unknown query data type ${(dataType as any).type}`,\n      );\n  }\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAIA,OAAO,SAASA,kCAAkCA,CAChDC,QAAuB,EACvBC,KAAkB,EACZ;EACN,QAAQD,QAAQ,CAACE,IAAI;IACnB,KAAK,OAAO;IACZ,KAAK,KAAK;MACRH,kCAAkC,CAACC,QAAQ,CAACG,OAAO,EAAEF,KAAK,CAAC;MAC3D;IACF,KAAK,iBAAiB;IACtB,KAAK,oBAAoB;MACvBA,KAAK,CAACG,GAAG,CAACJ,QAAQ,CAACK,oBAAqB,CAAC;MACzC;IAEF,KAAK,QAAQ;MACX,KAAK,MAAMC,IAAI,IAAIC,MAAM,CAACC,MAAM,CAACR,QAAQ,CAACS,MAAM,CAAC,EAAE;QACjDV,kCAAkC,CAACO,IAAI,CAACI,SAAS,EAAET,KAAK,CAAC;MAC3D;MACA;IAEF,KAAK,OAAO;MACV,KAAK,MAAMC,IAAI,IAAIF,QAAQ,CAACW,UAAU,EAAE;QACtCZ,kCAAkC,CAACG,IAAI,EAAED,KAAK,CAAC;MACjD;MACA;IAEF,KAAK,UAAU;MACbF,kCAAkC,CAACC,QAAQ,CAACY,OAAO,EAAEX,KAAK,CAAC;MAC3DF,kCAAkC,CAACC,QAAQ,CAACa,SAAS,EAAEZ,KAAK,CAAC;MAC7D;IAEF,KAAK,YAAY;IACjB,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,6BAA6B;IAClC,KAAK,WAAW;IAChB,KAAK,2BAA2B;IAChC,KAAK,UAAU;IACf,KAAK,aAAa;IAClB,KAAK,QAAQ;IACb,KAAK,WAAW;MACd;MACA;IAEF;MAEE,MAAM,IAAIa,KAAK,CACb,2BAA4Bd,QAAQ,CAASE,IAAI,EACnD,CAAC;EACL;AACF","ignoreList":[]}

package/build/browser/v2.0/generatePerQueryDataFiles.js

@@ -16,6 +16,7 @@
 
 import { wireQueryDataTypeToQueryDataTypeDefinition, wireQueryParameterV2ToQueryParameterDefinition as paramToDef, wireQueryTypeV2ToSdkQueryDefinitionNoParams } from "@osdk/generator-converters";
 import path from "node:path";
+import { getInterfaceTypeApiNamesFromQuery } from "../shared/getInterfaceTypeApiNamesFromQuery.js";
 import { getObjectImports } from "../shared/getObjectImports.js";
 import { getObjectTypeApiNamesFromQuery } from "../shared/getObjectTypeApiNamesFromQuery.js";
 import { deleteUndefineds } from "../util/deleteUndefineds.js";
@@ -46,8 +47,9 @@ export async function generatePerQueryDataFilesV2({
 async function generateV2QueryFile(fs, outDir, relOutDir, query, importExt, ontology, forInternalUse, fixedVersionQueryTypes) {
   const relFilePath = path.join(relOutDir, `${query.shortApiName}.ts`);
   const objectTypes = getObjectTypeApiNamesFromQuery(query);
-  const objectTypeObjects = new Set(objectTypes.map(o => ontology.requireObjectType(o)));
-  const importObjects = getObjectImports(objectTypeObjects, "", relFilePath, true);
+  const interfaceTypes = getInterfaceTypeApiNamesFromQuery(query);
+  const interfaceAndObjectTypes = new Set([...objectTypes.map(o => ontology.requireObjectType(o)), ...interfaceTypes.map(o => ontology.requireInterfaceType(o))]);
+  const importObjects = getObjectImports(interfaceAndObjectTypes, "", relFilePath, true);
   const baseProps = deleteUndefineds(wireQueryTypeV2ToSdkQueryDefinitionNoParams(query.raw));
   const outputBase = deleteUndefineds(wireQueryDataTypeToQueryDataTypeDefinition(query.output));
   const isUsingFixedVersion = fixedVersionQueryTypes.includes(query.fullApiName);
@@ -194,9 +196,15 @@ export function getQueryParamType(enhancedOntology, input, type, isMapKey = fals
       }
       inner = `Query${type}.ObjectType<${enhancedOntology.requireObjectType(input.object).getImportedDefinitionIdentifier(true)}>`;
       break;
+    case "interface":
+      inner = `Query${type}.InterfaceType<${enhancedOntology.requireInterfaceType(input.interface).getImportedDefinitionIdentifier(true)}>`;
+      break;
     case "objectSet":
       inner = `Query${type}.ObjectSetType<${enhancedOntology.requireObjectType(input.objectSet).getImportedDefinitionIdentifier(true)}>`;
       break;
+    case "interfaceObjectSet":
+      inner = `Query${type}.ObjectSetType<${enhancedOntology.requireInterfaceType(input.objectSet).getImportedDefinitionIdentifier(true)}>`;
+      break;
     case "set":
       inner = `${type === "Param" ? "Readonly" : ""}Set<${getQueryParamType(enhancedOntology, input.set, type)}>`;
       break;

package/build/browser/v2.0/generatePerQueryDataFiles.js.map

@@ -1 +1 @@
-{"version":3,"file":"generatePerQueryDataFiles.js","names":["wireQueryDataTypeToQueryDataTypeDefinition","wireQueryParameterV2ToQueryParameterDefinition","paramToDef","wireQueryTypeV2ToSdkQueryDefinitionNoParams","path","getObjectImports","getObjectTypeApiNamesFromQuery","deleteUndefineds","stringify","formatTs","getDescriptionIfPresent","generatePerQueryDataFilesV2","fs","outDir","rootOutDir","ontology","fixedVersionQueryTypes","importExt","forInternalUse","relOutDir","join","mkdir","recursive","Promise","all","Object","values","queryTypes","map","query","generateV2QueryFile","writeFile","shortApiName","getImportPathRelTo","keys","length","relFilePath","objectTypes","objectTypeObjects","Set","o","requireObjectType","importObjects","baseProps","raw","outputBase","output","isUsingFixedVersion","includes","fullApiName","description","parameters","paramsIdentifier","*","parameter","formatter","apiName","q","queryParamJsDoc","nullable","getQueryParamType","type","dataType","parameterDefsForType","getLineFor__OsdkTargetType","undefined","displayName","rid","definitionIdentifier","valueFormatter","qdt","objectTypeApiName","getImportedDefinitionIdentifier","param","ret","enhancedOntology","input","isMapKey","inner","JSON","struct","p","twoDimensionalAggregation","keyType","keySubtype","valueType","threeDimensionalAggregation","object","objectSet","set","union","u","multiplicity"],"sources":["generatePerQueryDataFiles.ts"],"sourcesContent":["/*\n * Copyright 2023 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { QueryParameterDefinition } from \"@osdk/api\";\nimport type { QueryDataType } from \"@osdk/foundry.ontologies\";\nimport {\n  wireQueryDataTypeToQueryDataTypeDefinition,\n  wireQueryParameterV2ToQueryParameterDefinition as paramToDef,\n  wireQueryTypeV2ToSdkQueryDefinitionNoParams,\n} from \"@osdk/generator-converters\";\nimport path from \"node:path\";\nimport type { EnhancedOntologyDefinition } from \"../GenerateContext/EnhancedOntologyDefinition.js\";\nimport type { EnhancedQuery } from \"../GenerateContext/EnhancedQuery.js\";\nimport type { GenerateContext } from \"../GenerateContext/GenerateContext.js\";\nimport type { MinimalFs } from \"../MinimalFs.js\";\nimport { getObjectImports } from \"../shared/getObjectImports.js\";\nimport { getObjectTypeApiNamesFromQuery } from \"../shared/getObjectTypeApiNamesFromQuery.js\";\nimport { deleteUndefineds } from \"../util/deleteUndefineds.js\";\nimport { stringify } from \"../util/stringify.js\";\nimport { formatTs } from \"../util/test/formatTs.js\";\nimport { getDescriptionIfPresent } from \"./getDescriptionIfPresent.js\";\n\nexport async function generatePerQueryDataFilesV2(\n  {\n    fs,\n    outDir: rootOutDir,\n    ontology,\n    fixedVersionQueryTypes,\n    importExt = \"\",\n    forInternalUse = false,\n  }: Pick<\n    GenerateContext,\n    | \"fs\"\n    | \"outDir\"\n    | \"importExt\"\n    | \"ontology\"\n    | \"forInternalUse\"\n    | \"fixedVersionQueryTypes\"\n  >,\n  v2: boolean,\n): Promise<void> {\n  const relOutDir = path.join(\".\", \"ontology\", \"queries\");\n  const outDir = path.join(rootOutDir, \"ontology\", \"queries\");\n  await fs.mkdir(outDir, { recursive: true });\n  await Promise.all(\n    Object.values(ontology.queryTypes).map(async query => {\n      await generateV2QueryFile(\n        fs,\n        outDir,\n        relOutDir,\n        query,\n        importExt,\n        ontology,\n        forInternalUse,\n        fixedVersionQueryTypes,\n      );\n    }),\n  );\n\n  const indexFilePath = `${outDir}.ts`;\n  await fs.writeFile(\n    indexFilePath,\n    await formatTs(`\n    ${\n      Object.values(ontology.queryTypes).map(query =>\n        `export {${query.shortApiName}} from \"${\n          query.getImportPathRelTo(relOutDir)\n        }\";`\n      )\n        .join(\"\\n\")\n    }\n      ${Object.keys(ontology.queryTypes).length === 0 ? \"export {};\" : \"\"}\n    `),\n  );\n}\n\nasync function generateV2QueryFile(\n  fs: MinimalFs,\n  outDir: string,\n  relOutDir: string,\n  query: EnhancedQuery,\n  importExt: string,\n  ontology: EnhancedOntologyDefinition,\n  forInternalUse: boolean,\n  fixedVersionQueryTypes: string[],\n) {\n  const relFilePath = path.join(relOutDir, `${query.shortApiName}.ts`);\n  const objectTypes = getObjectTypeApiNamesFromQuery(query);\n  const objectTypeObjects = new Set(\n    objectTypes.map(o => ontology.requireObjectType(o)),\n  );\n  const importObjects = getObjectImports(\n    objectTypeObjects,\n    \"\",\n    relFilePath,\n    true,\n  );\n\n  const baseProps = deleteUndefineds(\n    wireQueryTypeV2ToSdkQueryDefinitionNoParams(query.raw),\n  );\n\n  const outputBase = deleteUndefineds(\n    wireQueryDataTypeToQueryDataTypeDefinition(query.output),\n  );\n\n  const isUsingFixedVersion = fixedVersionQueryTypes.includes(\n    query.fullApiName,\n  );\n\n  await fs.writeFile(\n    path.join(outDir, `${query.shortApiName}.ts`),\n    await formatTs(`\n        import type { ObjectSpecifier, QueryDefinition, QueryParam, QueryResult, VersionBound} from \"${\n      forInternalUse ? \"@osdk/api\" : \"@osdk/client\"\n    }\";\n        import type { $ExpectedClientVersion } from \"../../OntologyMetadata${importExt}\";\n        import { $osdkMetadata} from \"../../OntologyMetadata${importExt}\";\n        ${importObjects}\n\n        export namespace ${query.shortApiName} {\n          export interface Signature {\n            ${getDescriptionIfPresent(query.description)}\n            (${\n      Object.keys(query.parameters).length > 0\n        ? `query: ${query.paramsIdentifier}`\n        : \"\"\n    }): Promise<${query.shortApiName}.ReturnType>\n          }\n\n        ${\n      Object.keys(query.parameters).length > 0\n        ? `\n            export interface Parameters {\n            ${\n          stringify(query.parameters, {\n            \"*\": (parameter, formatter, apiName) => {\n              const q = paramToDef(parameter);\n              return [\n                `\n                ${\n                  queryParamJsDoc(paramToDef(parameter), { apiName })\n                }readonly \"${apiName}\"${q.nullable ? \"?\" : \"\"}`,\n                getQueryParamType(ontology, q, \"Param\"),\n              ];\n            },\n          })\n        }\n            }`\n        : \"\"\n    }\n\n            ${\n      query.output.type === \"struct\"\n        ? `\n            export interface ReturnType \n            ${\n          getQueryParamType(\n            ontology,\n            paramToDef({ dataType: query.output }),\n            \"Result\",\n          )\n        }\n        `\n        : `\n        export type ReturnType = ${\n          getQueryParamType(\n            ontology,\n            paramToDef({ dataType: query.output }),\n            \"Result\",\n          )\n        }\n          `\n    }\n      }\n    \n            \n        export interface ${query.shortApiName} extends QueryDefinition<\n          ${query.shortApiName}.Signature\n        >, VersionBound<$ExpectedClientVersion>{\n         __DefinitionMetadata?: {\n             ${stringify(baseProps)}\n             isFixedVersion: ${isUsingFixedVersion};\n            parameters: {\n            ${parameterDefsForType(ontology, query)}\n            };\n            output: {\n            ${stringify(outputBase)},\n            ${getLineFor__OsdkTargetType(ontology, query.output)}\n            };\n            signature: ${query.shortApiName}.Signature;\n        }, \n        ${\n      stringify(baseProps, {\n        \"description\": () => undefined,\n        \"displayName\": () => undefined,\n        \"rid\": () => undefined,\n      })\n    }, \n          osdkMetadata: typeof $osdkMetadata;\n              }\n\n\n        export const ${query.shortApiName}: ${query.definitionIdentifier} = {\n            ${\n      stringify(baseProps, {\n        \"description\": () => undefined,\n        \"displayName\": () => undefined,\n        \"rid\": () => undefined,\n      })\n    },\n    isFixedVersion: ${isUsingFixedVersion},\n    osdkMetadata: $osdkMetadata\n        };\n        `),\n  );\n}\n\nfunction parameterDefsForType(\n  ontology: EnhancedOntologyDefinition,\n  query: EnhancedQuery,\n) {\n  return stringify(query.parameters, {\n    \"*\": (parameter, valueFormatter, apiName) => [\n      `${queryParamJsDoc(paramToDef(parameter), { apiName })} ${apiName}`,\n      ` {\n          ${stringify(deleteUndefineds(paramToDef(parameter)))},\n          ${getLineFor__OsdkTargetType(ontology, parameter.dataType)}\n        }`,\n    ],\n  });\n}\n\nfunction getLineFor__OsdkTargetType(\n  ontology: EnhancedOntologyDefinition,\n  qdt: QueryDataType,\n) {\n  if (qdt.type === \"object\" || qdt.type === \"objectSet\") {\n    return `__OsdkTargetType?: ${\n      ontology.requireObjectType(\n        qdt.objectTypeApiName!,\n      ).getImportedDefinitionIdentifier(true)\n    }`;\n  }\n  return \"\";\n}\n\nexport function queryParamJsDoc(\n  param: QueryParameterDefinition<any>,\n  { apiName }: { apiName: string },\n): string {\n  let ret = `/**\\n`;\n\n  if (param.description) {\n    if (param.description) {\n      ret += ` *   description: ${param.description}\\n`;\n    }\n  } else {\n    ret += ` * (no ontology metadata)\\n`;\n  }\n\n  ret += ` */\\n`;\n  return ret;\n}\n\nexport function getQueryParamType(\n  enhancedOntology: EnhancedOntologyDefinition,\n  input: QueryParameterDefinition,\n  type: \"Param\" | \"Result\",\n  isMapKey = false,\n): string {\n  let inner = `unknown /* ${input.type} */`;\n\n  switch (input.type) {\n    case \"date\":\n      inner = `Query${type}.PrimitiveType<${JSON.stringify(\"datetime\")}>`;\n      break;\n\n    case \"attachment\":\n    case \"boolean\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"timestamp\":\n      inner = `Query${type}.PrimitiveType<${JSON.stringify(input.type)}>`;\n      break;\n    case \"struct\":\n      inner = `{\n            ${\n        stringify(input.struct, {\n          \"*\": (p, formatter, apiName) => {\n            return [\n              `\n                ${type === \"Param\" ? \"readonly \" : \"\"}\"${apiName}\"${\n                p.nullable ? \"?\" : \"\"\n              }`,\n              getQueryParamType(enhancedOntology, p, type),\n            ];\n          },\n        })\n      }\n            }`;\n      break;\n    case \"twoDimensionalAggregation\":\n      inner = `Query${type}.TwoDimensionalAggregationType<${\n        input.twoDimensionalAggregation.keyType === \"range\"\n          ? `Query${type}.RangeKey<\"${input.twoDimensionalAggregation.keySubtype}\">`\n          : `\"${input.twoDimensionalAggregation.keyType}\"`\n      }, \"${input.twoDimensionalAggregation.valueType}\">`;\n      break;\n\n    case \"threeDimensionalAggregation\":\n      inner = `Query${type}.ThreeDimensionalAggregationType<${\n        input.threeDimensionalAggregation.keyType === \"range\"\n          ? `Query${type}.RangeKey<\"${input.threeDimensionalAggregation.keySubtype}\">`\n          : `\"${input.threeDimensionalAggregation.keyType}\"`\n      },${\n        input.threeDimensionalAggregation.valueType.keyType === \"range\"\n          ? `Query${type}.RangeKey<\"${input.threeDimensionalAggregation.valueType.keySubtype}\">`\n          : `\"${input.threeDimensionalAggregation.valueType.keyType}\"`\n      }, \n        \"${input.threeDimensionalAggregation.valueType.valueType}\">`;\n      break;\n    case \"object\":\n      if (isMapKey) {\n        inner = `ObjectSpecifier<${\n          enhancedOntology.requireObjectType(input.object)\n            .getImportedDefinitionIdentifier(true)\n        }>`;\n        break;\n      }\n      inner = `Query${type}.ObjectType<${\n        enhancedOntology.requireObjectType(input.object)\n          .getImportedDefinitionIdentifier(true)\n      }>`;\n      break;\n\n    case \"objectSet\":\n      inner = `Query${type}.ObjectSetType<${\n        enhancedOntology.requireObjectType(input.objectSet)\n          .getImportedDefinitionIdentifier(true)\n      }>`;\n      break;\n\n    case \"set\":\n      inner = `${type === \"Param\" ? \"Readonly\" : \"\"}Set<${\n        getQueryParamType(enhancedOntology, input.set, type)\n      }>`;\n      break;\n\n    case \"union\":\n      inner = input.union.map((u) =>\n        getQueryParamType(enhancedOntology, u, type)\n      ).join(\" | \");\n      break;\n\n    case \"map\":\n      inner = `Partial<Record<${\n        getQueryParamType(enhancedOntology, input.keyType, type, true)\n      }, ${getQueryParamType(enhancedOntology, input.valueType, type)}>>`;\n  }\n\n  if (input.multiplicity && type === \"Param\") {\n    return `ReadonlyArray<${inner}>`;\n  } else if (input.multiplicity) {\n    return `Array<${inner}>`;\n  }\n  return inner;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAIA,SACEA,0CAA0C,EAC1CC,8CAA8C,IAAIC,UAAU,EAC5DC,2CAA2C,QACtC,4BAA4B;AACnC,OAAOC,IAAI,MAAM,WAAW;AAK5B,SAASC,gBAAgB,QAAQ,+BAA+B;AAChE,SAASC,8BAA8B,QAAQ,6CAA6C;AAC5F,SAASC,gBAAgB,QAAQ,6BAA6B;AAC9D,SAASC,SAAS,QAAQ,sBAAsB;AAChD,SAASC,QAAQ,QAAQ,0BAA0B;AACnD,SAASC,uBAAuB,QAAQ,8BAA8B;AAEtE,OAAO,eAAeC,2BAA2BA,CAC/C;EACEC,EAAE;EACFC,MAAM,EAAEC,UAAU;EAClBC,QAAQ;EACRC,sBAAsB;EACtBC,SAAS,GAAG,EAAE;EACdC,cAAc,GAAG;AASnB,CAAC,EAEc;EACf,MAAMC,SAAS,GAAGf,IAAI,CAACgB,IAAI,CAAC,GAAG,EAAE,UAAU,EAAE,SAAS,CAAC;EACvD,MAAMP,MAAM,GAAGT,IAAI,CAACgB,IAAI,CAACN,UAAU,EAAE,UAAU,EAAE,SAAS,CAAC;EAC3D,MAAMF,EAAE,CAACS,KAAK,CAACR,MAAM,EAAE;IAAES,SAAS,EAAE;EAAK,CAAC,CAAC;EAC3C,MAAMC,OAAO,CAACC,GAAG,CACfC,MAAM,CAACC,MAAM,CAACX,QAAQ,CAACY,UAAU,CAAC,CAACC,GAAG,CAAC,MAAMC,KAAK,IAAI;IACpD,MAAMC,mBAAmB,CACvBlB,EAAE,EACFC,MAAM,EACNM,SAAS,EACTU,KAAK,EACLZ,SAAS,EACTF,QAAQ,EACRG,cAAc,EACdF,sBACF,CAAC;EACH,CAAC,CACH,CAAC;EAGD,MAAMJ,EAAE,CAACmB,SAAS,CADI,GAAGlB,MAAM,KAAK,EAGlC,MAAMJ,QAAQ,CAAC;AACnB,MACMgB,MAAM,CAACC,MAAM,CAACX,QAAQ,CAACY,UAAU,CAAC,CAACC,GAAG,CAACC,KAAK,IAC1C,WAAWA,KAAK,CAACG,YAAY,WAC3BH,KAAK,CAACI,kBAAkB,CAACd,SAAS,CAAC,IAEvC,CAAC,CACEC,IAAI,CAAC,IAAI,CAAC;AACnB,QACQK,MAAM,CAACS,IAAI,CAACnB,QAAQ,CAACY,UAAU,CAAC,CAACQ,MAAM,KAAK,CAAC,GAAG,YAAY,GAAG,EAAE;AACzE,KAAK,CACH,CAAC;AACH;AAEA,eAAeL,mBAAmBA,CAChClB,EAAa,EACbC,MAAc,EACdM,SAAiB,EACjBU,KAAoB,EACpBZ,SAAiB,EACjBF,QAAoC,EACpCG,cAAuB,EACvBF,sBAAgC,EAChC;EACA,MAAMoB,WAAW,GAAGhC,IAAI,CAACgB,IAAI,CAACD,SAAS,EAAE,GAAGU,KAAK,CAACG,YAAY,KAAK,CAAC;EACpE,MAAMK,WAAW,GAAG/B,8BAA8B,CAACuB,KAAK,CAAC;EACzD,MAAMS,iBAAiB,GAAG,IAAIC,GAAG,CAC/BF,WAAW,CAACT,GAAG,CAACY,CAAC,IAAIzB,QAAQ,CAAC0B,iBAAiB,CAACD,CAAC,CAAC,CACpD,CAAC;EACD,MAAME,aAAa,GAAGrC,gBAAgB,CACpCiC,iBAAiB,EACjB,EAAE,EACFF,WAAW,EACX,IACF,CAAC;EAED,MAAMO,SAAS,GAAGpC,gBAAgB,CAChCJ,2CAA2C,CAAC0B,KAAK,CAACe,GAAG,CACvD,CAAC;EAED,MAAMC,UAAU,GAAGtC,gBAAgB,CACjCP,0CAA0C,CAAC6B,KAAK,CAACiB,MAAM,CACzD,CAAC;EAED,MAAMC,mBAAmB,GAAG/B,sBAAsB,CAACgC,QAAQ,CACzDnB,KAAK,CAACoB,WACR,CAAC;EAED,MAAMrC,EAAE,CAACmB,SAAS,CAChB3B,IAAI,CAACgB,IAAI,CAACP,MAAM,EAAE,GAAGgB,KAAK,CAACG,YAAY,KAAK,CAAC,EAC7C,MAAMvB,QAAQ,CAAC;AACnB,uGACMS,cAAc,GAAG,WAAW,GAAG,cAAc;AACnD,6EAC6ED,SAAS;AACtF,8DAA8DA,SAAS;AACvE,UAAUyB,aAAa;AACvB;AACA,2BAA2Bb,KAAK,CAACG,YAAY;AAC7C;AACA,cAActB,uBAAuB,CAACmB,KAAK,CAACqB,WAAW,CAAC;AACxD,eACMzB,MAAM,CAACS,IAAI,CAACL,KAAK,CAACsB,UAAU,CAAC,CAAChB,MAAM,GAAG,CAAC,GACpC,UAAUN,KAAK,CAACuB,gBAAgB,EAAE,GAClC,EAAE,cACMvB,KAAK,CAACG,YAAY;AACpC;AACA;AACA,UACMP,MAAM,CAACS,IAAI,CAACL,KAAK,CAACsB,UAAU,CAAC,CAAChB,MAAM,GAAG,CAAC,GACpC;AACV;AACA,cACU3B,SAAS,CAACqB,KAAK,CAACsB,UAAU,EAAE;IAC1B,GAAG,EAAEE,CAACC,SAAS,EAAEC,SAAS,EAAEC,OAAO,KAAK;MACtC,MAAMC,CAAC,GAAGvD,UAAU,CAACoD,SAAS,CAAC;MAC/B,OAAO,CACL;AAChB,kBACkBI,eAAe,CAACxD,UAAU,CAACoD,SAAS,CAAC,EAAE;QAAEE;MAAQ,CAAC,CAAC,aACxCA,OAAO,IAAIC,CAAC,CAACE,QAAQ,GAAG,GAAG,GAAG,EAAE,EAAE,EAC/CC,iBAAiB,CAAC7C,QAAQ,EAAE0C,CAAC,EAAE,OAAO,CAAC,CACxC;IACH;EACF,CAAC,CAAC;AACZ,cACc,GACJ,EAAE;AACZ;AACA,cAEM5B,KAAK,CAACiB,MAAM,CAACe,IAAI,KAAK,QAAQ,GAC1B;AACV;AACA,cACUD,iBAAiB,CACf7C,QAAQ,EACRb,UAAU,CAAC;IAAE4D,QAAQ,EAAEjC,KAAK,CAACiB;EAAO,CAAC,CAAC,EACtC,QACF,CAAC;AACX,SACS,GACC;AACV,mCACUc,iBAAiB,CACf7C,QAAQ,EACRb,UAAU,CAAC;IAAE4D,QAAQ,EAAEjC,KAAK,CAACiB;EAAO,CAAC,CAAC,EACtC,QACF,CAAC;AACX,WACW;AACX;AACA;AACA;AACA,2BAC2BjB,KAAK,CAACG,YAAY;AAC7C,YAAYH,KAAK,CAACG,YAAY;AAC9B;AACA;AACA,eAAexB,SAAS,CAACmC,SAAS,CAAC;AACnC,+BAA+BI,mBAAmB;AAClD;AACA,cAAcgB,oBAAoB,CAAChD,QAAQ,EAAEc,KAAK,CAAC;AACnD;AACA;AACA,cAAcrB,SAAS,CAACqC,UAAU,CAAC;AACnC,cAAcmB,0BAA0B,CAACjD,QAAQ,EAAEc,KAAK,CAACiB,MAAM,CAAC;AAChE;AACA,yBAAyBjB,KAAK,CAACG,YAAY;AAC3C;AACA,UACMxB,SAAS,CAACmC,SAAS,EAAE;IACnB,aAAa,EAAEO,CAAA,KAAMe,SAAS;IAC9B,aAAa,EAAEC,CAAA,KAAMD,SAAS;IAC9B,KAAK,EAAEE,CAAA,KAAMF;EACf,CAAC,CAAC;AACR;AACA;AACA;AACA;AACA,uBACuBpC,KAAK,CAACG,YAAY,KAAKH,KAAK,CAACuC,oBAAoB;AACxE,cACM5D,SAAS,CAACmC,SAAS,EAAE;IACnB,aAAa,EAAEO,CAAA,KAAMe,SAAS;IAC9B,aAAa,EAAEC,CAAA,KAAMD,SAAS;IAC9B,KAAK,EAAEE,CAAA,KAAMF;EACf,CAAC,CAAC;AACR,sBACsBlB,mBAAmB;AACzC;AACA;AACA,SAAS,CACP,CAAC;AACH;AAEA,SAASgB,oBAAoBA,CAC3BhD,QAAoC,EACpCc,KAAoB,EACpB;EACA,OAAOrB,SAAS,CAACqB,KAAK,CAACsB,UAAU,EAAE;IACjC,GAAG,EAAEE,CAACC,SAAS,EAAEe,cAAc,EAAEb,OAAO,KAAK,CAC3C,GAAGE,eAAe,CAACxD,UAAU,CAACoD,SAAS,CAAC,EAAE;MAAEE;IAAQ,CAAC,CAAC,IAAIA,OAAO,EAAE,EACnE;AACN,YAAYhD,SAAS,CAACD,gBAAgB,CAACL,UAAU,CAACoD,SAAS,CAAC,CAAC,CAAC;AAC9D,YAAYU,0BAA0B,CAACjD,QAAQ,EAAEuC,SAAS,CAACQ,QAAQ,CAAC;AACpE,UAAU;EAER,CAAC,CAAC;AACJ;AAEA,SAASE,0BAA0BA,CACjCjD,QAAoC,EACpCuD,GAAkB,EAClB;EACA,IAAIA,GAAG,CAACT,IAAI,KAAK,QAAQ,IAAIS,GAAG,CAACT,IAAI,KAAK,WAAW,EAAE;IACrD,OAAO,sBACL9C,QAAQ,CAAC0B,iBAAiB,CACxB6B,GAAG,CAACC,iBACN,CAAC,CAACC,+BAA+B,CAAC,IAAI,CAAC,EACvC;EACJ;EACA,OAAO,EAAE;AACX;AAEA,OAAO,SAASd,eAAeA,CAC7Be,KAAoC,EACpC;EAAEjB;AAA6B,CAAC,EACxB;EACR,IAAIkB,GAAG,GAAG,OAAO;EAEjB,IAAID,KAAK,CAACvB,WAAW,EAAE;IACrB,IAAIuB,KAAK,CAACvB,WAAW,EAAE;MACrBwB,GAAG,IAAI,qBAAqBD,KAAK,CAACvB,WAAW,IAAI;IACnD;EACF,CAAC,MAAM;IACLwB,GAAG,IAAI,6BAA6B;EACtC;EAEAA,GAAG,IAAI,OAAO;EACd,OAAOA,GAAG;AACZ;AAEA,OAAO,SAASd,iBAAiBA,CAC/Be,gBAA4C,EAC5CC,KAA+B,EAC/Bf,IAAwB,EACxBgB,QAAQ,GAAG,KAAK,EACR;EACR,IAAIC,KAAK,GAAG,cAAcF,KAAK,CAACf,IAAI,KAAK;EAEzC,QAAQe,KAAK,CAACf,IAAI;IAChB,KAAK,MAAM;MACTiB,KAAK,GAAG,QAAQjB,IAAI,kBAAkBkB,IAAI,CAACvE,SAAS,CAAC,UAAU,CAAC,GAAG;MACnE;IAEF,KAAK,YAAY;IACjB,KAAK,SAAS;IACd,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,WAAW;MACdsE,KAAK,GAAG,QAAQjB,IAAI,kBAAkBkB,IAAI,CAACvE,SAAS,CAACoE,KAAK,CAACf,IAAI,CAAC,GAAG;MACnE;IACF,KAAK,QAAQ;MACXiB,KAAK,GAAG;AACd,cACQtE,SAAS,CAACoE,KAAK,CAACI,MAAM,EAAE;QACtB,GAAG,EAAE3B,CAAC4B,CAAC,EAAE1B,SAAS,EAAEC,OAAO,KAAK;UAC9B,OAAO,CACL;AACd,kBAAkBK,IAAI,KAAK,OAAO,GAAG,WAAW,GAAG,EAAE,IAAIL,OAAO,IAChDyB,CAAC,CAACtB,QAAQ,GAAG,GAAG,GAAG,EAAE,EACrB,EACFC,iBAAiB,CAACe,gBAAgB,EAAEM,CAAC,EAAEpB,IAAI,CAAC,CAC7C;QACH;MACF,CAAC,CAAC;AACV,cACc;MACR;IACF,KAAK,2BAA2B;MAC9BiB,KAAK,GAAG,QAAQjB,IAAI,kCAClBe,KAAK,CAACM,yBAAyB,CAACC,OAAO,KAAK,OAAO,GAC/C,QAAQtB,IAAI,cAAce,KAAK,CAACM,yBAAyB,CAACE,UAAU,IAAI,GACxE,IAAIR,KAAK,CAACM,yBAAyB,CAACC,OAAO,GAAG,MAC9CP,KAAK,CAACM,yBAAyB,CAACG,SAAS,IAAI;MACnD;IAEF,KAAK,6BAA6B;MAChCP,KAAK,GAAG,QAAQjB,IAAI,oCAClBe,KAAK,CAACU,2BAA2B,CAACH,OAAO,KAAK,OAAO,GACjD,QAAQtB,IAAI,cAAce,KAAK,CAACU,2BAA2B,CAACF,UAAU,IAAI,GAC1E,IAAIR,KAAK,CAACU,2BAA2B,CAACH,OAAO,GAAG,IAEpDP,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACF,OAAO,KAAK,OAAO,GAC3D,QAAQtB,IAAI,cAAce,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACD,UAAU,IAAI,GACpF,IAAIR,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACF,OAAO,GAAG;AACtE,WACWP,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACA,SAAS,IAAI;MAC9D;IACF,KAAK,QAAQ;MACX,IAAIR,QAAQ,EAAE;QACZC,KAAK,GAAG,mBACNH,gBAAgB,CAAClC,iBAAiB,CAACmC,KAAK,CAACW,MAAM,CAAC,CAC7Cf,+BAA+B,CAAC,IAAI,CAAC,GACvC;QACH;MACF;MACAM,KAAK,GAAG,QAAQjB,IAAI,eAClBc,gBAAgB,CAAClC,iBAAiB,CAACmC,KAAK,CAACW,MAAM,CAAC,CAC7Cf,+BAA+B,CAAC,IAAI,CAAC,GACvC;MACH;IAEF,KAAK,WAAW;MACdM,KAAK,GAAG,QAAQjB,IAAI,kBAClBc,gBAAgB,CAAClC,iBAAiB,CAACmC,KAAK,CAACY,SAAS,CAAC,CAChDhB,+BAA+B,CAAC,IAAI,CAAC,GACvC;MACH;IAEF,KAAK,KAAK;MACRM,KAAK,GAAG,GAAGjB,IAAI,KAAK,OAAO,GAAG,UAAU,GAAG,EAAE,OAC3CD,iBAAiB,CAACe,gBAAgB,EAAEC,KAAK,CAACa,GAAG,EAAE5B,IAAI,CAAC,GACnD;MACH;IAEF,KAAK,OAAO;MACViB,KAAK,GAAGF,KAAK,CAACc,KAAK,CAAC9D,GAAG,CAAE+D,CAAC,IACxB/B,iBAAiB,CAACe,gBAAgB,EAAEgB,CAAC,EAAE9B,IAAI,CAC7C,CAAC,CAACzC,IAAI,CAAC,KAAK,CAAC;MACb;IAEF,KAAK,KAAK;MACR0D,KAAK,GAAG,kBACNlB,iBAAiB,CAACe,gBAAgB,EAAEC,KAAK,CAACO,OAAO,EAAEtB,IAAI,EAAE,IAAI,CAAC,KAC3DD,iBAAiB,CAACe,gBAAgB,EAAEC,KAAK,CAACS,SAAS,EAAExB,IAAI,CAAC,IAAI;EACvE;EAEA,IAAIe,KAAK,CAACgB,YAAY,IAAI/B,IAAI,KAAK,OAAO,EAAE;IAC1C,OAAO,iBAAiBiB,KAAK,GAAG;EAClC,CAAC,MAAM,IAAIF,KAAK,CAACgB,YAAY,EAAE;IAC7B,OAAO,SAASd,KAAK,GAAG;EAC1B;EACA,OAAOA,KAAK;AACd","ignoreList":[]}
+{"version":3,"file":"generatePerQueryDataFiles.js","names":["wireQueryDataTypeToQueryDataTypeDefinition","wireQueryParameterV2ToQueryParameterDefinition","paramToDef","wireQueryTypeV2ToSdkQueryDefinitionNoParams","path","getInterfaceTypeApiNamesFromQuery","getObjectImports","getObjectTypeApiNamesFromQuery","deleteUndefineds","stringify","formatTs","getDescriptionIfPresent","generatePerQueryDataFilesV2","fs","outDir","rootOutDir","ontology","fixedVersionQueryTypes","importExt","forInternalUse","relOutDir","join","mkdir","recursive","Promise","all","Object","values","queryTypes","map","query","generateV2QueryFile","writeFile","shortApiName","getImportPathRelTo","keys","length","relFilePath","objectTypes","interfaceTypes","interfaceAndObjectTypes","Set","o","requireObjectType","requireInterfaceType","importObjects","baseProps","raw","outputBase","output","isUsingFixedVersion","includes","fullApiName","description","parameters","paramsIdentifier","*","parameter","formatter","apiName","q","queryParamJsDoc","nullable","getQueryParamType","type","dataType","parameterDefsForType","getLineFor__OsdkTargetType","undefined","displayName","rid","definitionIdentifier","valueFormatter","qdt","objectTypeApiName","getImportedDefinitionIdentifier","param","ret","enhancedOntology","input","isMapKey","inner","JSON","struct","p","twoDimensionalAggregation","keyType","keySubtype","valueType","threeDimensionalAggregation","object","interface","objectSet","set","union","u","multiplicity"],"sources":["generatePerQueryDataFiles.ts"],"sourcesContent":["/*\n * Copyright 2023 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { QueryParameterDefinition } from \"@osdk/api\";\nimport type { QueryDataType } from \"@osdk/foundry.ontologies\";\nimport {\n  wireQueryDataTypeToQueryDataTypeDefinition,\n  wireQueryParameterV2ToQueryParameterDefinition as paramToDef,\n  wireQueryTypeV2ToSdkQueryDefinitionNoParams,\n} from \"@osdk/generator-converters\";\nimport path from \"node:path\";\nimport type { EnhancedOntologyDefinition } from \"../GenerateContext/EnhancedOntologyDefinition.js\";\nimport type { EnhancedQuery } from \"../GenerateContext/EnhancedQuery.js\";\nimport type { GenerateContext } from \"../GenerateContext/GenerateContext.js\";\nimport type { MinimalFs } from \"../MinimalFs.js\";\nimport { getInterfaceTypeApiNamesFromQuery } from \"../shared/getInterfaceTypeApiNamesFromQuery.js\";\nimport { getObjectImports } from \"../shared/getObjectImports.js\";\nimport { getObjectTypeApiNamesFromQuery } from \"../shared/getObjectTypeApiNamesFromQuery.js\";\nimport { deleteUndefineds } from \"../util/deleteUndefineds.js\";\nimport { stringify } from \"../util/stringify.js\";\nimport { formatTs } from \"../util/test/formatTs.js\";\nimport { getDescriptionIfPresent } from \"./getDescriptionIfPresent.js\";\n\nexport async function generatePerQueryDataFilesV2(\n  {\n    fs,\n    outDir: rootOutDir,\n    ontology,\n    fixedVersionQueryTypes,\n    importExt = \"\",\n    forInternalUse = false,\n  }: Pick<\n    GenerateContext,\n    | \"fs\"\n    | \"outDir\"\n    | \"importExt\"\n    | \"ontology\"\n    | \"forInternalUse\"\n    | \"fixedVersionQueryTypes\"\n  >,\n  v2: boolean,\n): Promise<void> {\n  const relOutDir = path.join(\".\", \"ontology\", \"queries\");\n  const outDir = path.join(rootOutDir, \"ontology\", \"queries\");\n  await fs.mkdir(outDir, { recursive: true });\n  await Promise.all(\n    Object.values(ontology.queryTypes).map(async query => {\n      await generateV2QueryFile(\n        fs,\n        outDir,\n        relOutDir,\n        query,\n        importExt,\n        ontology,\n        forInternalUse,\n        fixedVersionQueryTypes,\n      );\n    }),\n  );\n\n  const indexFilePath = `${outDir}.ts`;\n  await fs.writeFile(\n    indexFilePath,\n    await formatTs(`\n    ${\n      Object.values(ontology.queryTypes).map(query =>\n        `export {${query.shortApiName}} from \"${\n          query.getImportPathRelTo(relOutDir)\n        }\";`\n      )\n        .join(\"\\n\")\n    }\n      ${Object.keys(ontology.queryTypes).length === 0 ? \"export {};\" : \"\"}\n    `),\n  );\n}\n\nasync function generateV2QueryFile(\n  fs: MinimalFs,\n  outDir: string,\n  relOutDir: string,\n  query: EnhancedQuery,\n  importExt: string,\n  ontology: EnhancedOntologyDefinition,\n  forInternalUse: boolean,\n  fixedVersionQueryTypes: string[],\n) {\n  const relFilePath = path.join(relOutDir, `${query.shortApiName}.ts`);\n  const objectTypes = getObjectTypeApiNamesFromQuery(query);\n  const interfaceTypes = getInterfaceTypeApiNamesFromQuery(query);\n  const interfaceAndObjectTypes = new Set(\n    [\n      ...objectTypes.map(o => ontology.requireObjectType(o)),\n      ...interfaceTypes.map(o => ontology.requireInterfaceType(o)),\n    ],\n  );\n\n  const importObjects = getObjectImports(\n    interfaceAndObjectTypes,\n    \"\",\n    relFilePath,\n    true,\n  );\n\n  const baseProps = deleteUndefineds(\n    wireQueryTypeV2ToSdkQueryDefinitionNoParams(query.raw),\n  );\n\n  const outputBase = deleteUndefineds(\n    wireQueryDataTypeToQueryDataTypeDefinition(query.output),\n  );\n\n  const isUsingFixedVersion = fixedVersionQueryTypes.includes(\n    query.fullApiName,\n  );\n\n  await fs.writeFile(\n    path.join(outDir, `${query.shortApiName}.ts`),\n    await formatTs(`\n        import type { ObjectSpecifier, QueryDefinition, QueryParam, QueryResult, VersionBound} from \"${\n      forInternalUse ? \"@osdk/api\" : \"@osdk/client\"\n    }\";\n        import type { $ExpectedClientVersion } from \"../../OntologyMetadata${importExt}\";\n        import { $osdkMetadata} from \"../../OntologyMetadata${importExt}\";\n        ${importObjects}\n\n        export namespace ${query.shortApiName} {\n          export interface Signature {\n            ${getDescriptionIfPresent(query.description)}\n            (${\n      Object.keys(query.parameters).length > 0\n        ? `query: ${query.paramsIdentifier}`\n        : \"\"\n    }): Promise<${query.shortApiName}.ReturnType>\n          }\n\n        ${\n      Object.keys(query.parameters).length > 0\n        ? `\n            export interface Parameters {\n            ${\n          stringify(query.parameters, {\n            \"*\": (parameter, formatter, apiName) => {\n              const q = paramToDef(parameter);\n              return [\n                `\n                ${\n                  queryParamJsDoc(paramToDef(parameter), { apiName })\n                }readonly \"${apiName}\"${q.nullable ? \"?\" : \"\"}`,\n                getQueryParamType(ontology, q, \"Param\"),\n              ];\n            },\n          })\n        }\n            }`\n        : \"\"\n    }\n\n            ${\n      query.output.type === \"struct\"\n        ? `\n            export interface ReturnType \n            ${\n          getQueryParamType(\n            ontology,\n            paramToDef({ dataType: query.output }),\n            \"Result\",\n          )\n        }\n        `\n        : `\n        export type ReturnType = ${\n          getQueryParamType(\n            ontology,\n            paramToDef({ dataType: query.output }),\n            \"Result\",\n          )\n        }\n          `\n    }\n      }\n    \n            \n        export interface ${query.shortApiName} extends QueryDefinition<\n          ${query.shortApiName}.Signature\n        >, VersionBound<$ExpectedClientVersion>{\n         __DefinitionMetadata?: {\n             ${stringify(baseProps)}\n             isFixedVersion: ${isUsingFixedVersion};\n            parameters: {\n            ${parameterDefsForType(ontology, query)}\n            };\n            output: {\n            ${stringify(outputBase)},\n            ${getLineFor__OsdkTargetType(ontology, query.output)}\n            };\n            signature: ${query.shortApiName}.Signature;\n        }, \n        ${\n      stringify(baseProps, {\n        \"description\": () => undefined,\n        \"displayName\": () => undefined,\n        \"rid\": () => undefined,\n      })\n    }, \n          osdkMetadata: typeof $osdkMetadata;\n              }\n\n\n        export const ${query.shortApiName}: ${query.definitionIdentifier} = {\n            ${\n      stringify(baseProps, {\n        \"description\": () => undefined,\n        \"displayName\": () => undefined,\n        \"rid\": () => undefined,\n      })\n    },\n    isFixedVersion: ${isUsingFixedVersion},\n    osdkMetadata: $osdkMetadata\n        };\n        `),\n  );\n}\n\nfunction parameterDefsForType(\n  ontology: EnhancedOntologyDefinition,\n  query: EnhancedQuery,\n) {\n  return stringify(query.parameters, {\n    \"*\": (parameter, valueFormatter, apiName) => [\n      `${queryParamJsDoc(paramToDef(parameter), { apiName })} ${apiName}`,\n      ` {\n          ${stringify(deleteUndefineds(paramToDef(parameter)))},\n          ${getLineFor__OsdkTargetType(ontology, parameter.dataType)}\n        }`,\n    ],\n  });\n}\n\nfunction getLineFor__OsdkTargetType(\n  ontology: EnhancedOntologyDefinition,\n  qdt: QueryDataType,\n) {\n  if (qdt.type === \"object\" || qdt.type === \"objectSet\") {\n    return `__OsdkTargetType?: ${\n      ontology.requireObjectType(\n        qdt.objectTypeApiName!,\n      ).getImportedDefinitionIdentifier(true)\n    }`;\n  }\n  return \"\";\n}\n\nexport function queryParamJsDoc(\n  param: QueryParameterDefinition<any>,\n  { apiName }: { apiName: string },\n): string {\n  let ret = `/**\\n`;\n\n  if (param.description) {\n    if (param.description) {\n      ret += ` *   description: ${param.description}\\n`;\n    }\n  } else {\n    ret += ` * (no ontology metadata)\\n`;\n  }\n\n  ret += ` */\\n`;\n  return ret;\n}\n\nexport function getQueryParamType(\n  enhancedOntology: EnhancedOntologyDefinition,\n  input: QueryParameterDefinition,\n  type: \"Param\" | \"Result\",\n  isMapKey = false,\n): string {\n  let inner = `unknown /* ${input.type} */`;\n\n  switch (input.type) {\n    case \"date\":\n      inner = `Query${type}.PrimitiveType<${JSON.stringify(\"datetime\")}>`;\n      break;\n\n    case \"attachment\":\n    case \"boolean\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"timestamp\":\n      inner = `Query${type}.PrimitiveType<${JSON.stringify(input.type)}>`;\n      break;\n    case \"struct\":\n      inner = `{\n            ${\n        stringify(input.struct, {\n          \"*\": (p, formatter, apiName) => {\n            return [\n              `\n                ${type === \"Param\" ? \"readonly \" : \"\"}\"${apiName}\"${\n                p.nullable ? \"?\" : \"\"\n              }`,\n              getQueryParamType(enhancedOntology, p, type),\n            ];\n          },\n        })\n      }\n            }`;\n      break;\n    case \"twoDimensionalAggregation\":\n      inner = `Query${type}.TwoDimensionalAggregationType<${\n        input.twoDimensionalAggregation.keyType === \"range\"\n          ? `Query${type}.RangeKey<\"${input.twoDimensionalAggregation.keySubtype}\">`\n          : `\"${input.twoDimensionalAggregation.keyType}\"`\n      }, \"${input.twoDimensionalAggregation.valueType}\">`;\n      break;\n\n    case \"threeDimensionalAggregation\":\n      inner = `Query${type}.ThreeDimensionalAggregationType<${\n        input.threeDimensionalAggregation.keyType === \"range\"\n          ? `Query${type}.RangeKey<\"${input.threeDimensionalAggregation.keySubtype}\">`\n          : `\"${input.threeDimensionalAggregation.keyType}\"`\n      },${\n        input.threeDimensionalAggregation.valueType.keyType === \"range\"\n          ? `Query${type}.RangeKey<\"${input.threeDimensionalAggregation.valueType.keySubtype}\">`\n          : `\"${input.threeDimensionalAggregation.valueType.keyType}\"`\n      }, \n        \"${input.threeDimensionalAggregation.valueType.valueType}\">`;\n      break;\n    case \"object\":\n      if (isMapKey) {\n        inner = `ObjectSpecifier<${\n          enhancedOntology.requireObjectType(input.object)\n            .getImportedDefinitionIdentifier(true)\n        }>`;\n        break;\n      }\n      inner = `Query${type}.ObjectType<${\n        enhancedOntology.requireObjectType(input.object)\n          .getImportedDefinitionIdentifier(true)\n      }>`;\n      break;\n    case \"interface\":\n      inner = `Query${type}.InterfaceType<${\n        enhancedOntology.requireInterfaceType(input.interface)\n          .getImportedDefinitionIdentifier(true)\n      }>`;\n      break;\n\n    case \"objectSet\":\n      inner = `Query${type}.ObjectSetType<${\n        enhancedOntology.requireObjectType(input.objectSet)\n          .getImportedDefinitionIdentifier(true)\n      }>`;\n      break;\n\n    case \"interfaceObjectSet\":\n      inner = `Query${type}.ObjectSetType<${\n        enhancedOntology.requireInterfaceType(input.objectSet)\n          .getImportedDefinitionIdentifier(true)\n      }>`;\n      break;\n\n    case \"set\":\n      inner = `${type === \"Param\" ? \"Readonly\" : \"\"}Set<${\n        getQueryParamType(enhancedOntology, input.set, type)\n      }>`;\n      break;\n\n    case \"union\":\n      inner = input.union.map((u) =>\n        getQueryParamType(enhancedOntology, u, type)\n      ).join(\" | \");\n      break;\n\n    case \"map\":\n      inner = `Partial<Record<${\n        getQueryParamType(enhancedOntology, input.keyType, type, true)\n      }, ${getQueryParamType(enhancedOntology, input.valueType, type)}>>`;\n  }\n\n  if (input.multiplicity && type === \"Param\") {\n    return `ReadonlyArray<${inner}>`;\n  } else if (input.multiplicity) {\n    return `Array<${inner}>`;\n  }\n  return inner;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAIA,SACEA,0CAA0C,EAC1CC,8CAA8C,IAAIC,UAAU,EAC5DC,2CAA2C,QACtC,4BAA4B;AACnC,OAAOC,IAAI,MAAM,WAAW;AAK5B,SAASC,iCAAiC,QAAQ,gDAAgD;AAClG,SAASC,gBAAgB,QAAQ,+BAA+B;AAChE,SAASC,8BAA8B,QAAQ,6CAA6C;AAC5F,SAASC,gBAAgB,QAAQ,6BAA6B;AAC9D,SAASC,SAAS,QAAQ,sBAAsB;AAChD,SAASC,QAAQ,QAAQ,0BAA0B;AACnD,SAASC,uBAAuB,QAAQ,8BAA8B;AAEtE,OAAO,eAAeC,2BAA2BA,CAC/C;EACEC,EAAE;EACFC,MAAM,EAAEC,UAAU;EAClBC,QAAQ;EACRC,sBAAsB;EACtBC,SAAS,GAAG,EAAE;EACdC,cAAc,GAAG;AASnB,CAAC,EAEc;EACf,MAAMC,SAAS,GAAGhB,IAAI,CAACiB,IAAI,CAAC,GAAG,EAAE,UAAU,EAAE,SAAS,CAAC;EACvD,MAAMP,MAAM,GAAGV,IAAI,CAACiB,IAAI,CAACN,UAAU,EAAE,UAAU,EAAE,SAAS,CAAC;EAC3D,MAAMF,EAAE,CAACS,KAAK,CAACR,MAAM,EAAE;IAAES,SAAS,EAAE;EAAK,CAAC,CAAC;EAC3C,MAAMC,OAAO,CAACC,GAAG,CACfC,MAAM,CAACC,MAAM,CAACX,QAAQ,CAACY,UAAU,CAAC,CAACC,GAAG,CAAC,MAAMC,KAAK,IAAI;IACpD,MAAMC,mBAAmB,CACvBlB,EAAE,EACFC,MAAM,EACNM,SAAS,EACTU,KAAK,EACLZ,SAAS,EACTF,QAAQ,EACRG,cAAc,EACdF,sBACF,CAAC;EACH,CAAC,CACH,CAAC;EAGD,MAAMJ,EAAE,CAACmB,SAAS,CADI,GAAGlB,MAAM,KAAK,EAGlC,MAAMJ,QAAQ,CAAC;AACnB,MACMgB,MAAM,CAACC,MAAM,CAACX,QAAQ,CAACY,UAAU,CAAC,CAACC,GAAG,CAACC,KAAK,IAC1C,WAAWA,KAAK,CAACG,YAAY,WAC3BH,KAAK,CAACI,kBAAkB,CAACd,SAAS,CAAC,IAEvC,CAAC,CACEC,IAAI,CAAC,IAAI,CAAC;AACnB,QACQK,MAAM,CAACS,IAAI,CAACnB,QAAQ,CAACY,UAAU,CAAC,CAACQ,MAAM,KAAK,CAAC,GAAG,YAAY,GAAG,EAAE;AACzE,KAAK,CACH,CAAC;AACH;AAEA,eAAeL,mBAAmBA,CAChClB,EAAa,EACbC,MAAc,EACdM,SAAiB,EACjBU,KAAoB,EACpBZ,SAAiB,EACjBF,QAAoC,EACpCG,cAAuB,EACvBF,sBAAgC,EAChC;EACA,MAAMoB,WAAW,GAAGjC,IAAI,CAACiB,IAAI,CAACD,SAAS,EAAE,GAAGU,KAAK,CAACG,YAAY,KAAK,CAAC;EACpE,MAAMK,WAAW,GAAG/B,8BAA8B,CAACuB,KAAK,CAAC;EACzD,MAAMS,cAAc,GAAGlC,iCAAiC,CAACyB,KAAK,CAAC;EAC/D,MAAMU,uBAAuB,GAAG,IAAIC,GAAG,CACrC,CACE,GAAGH,WAAW,CAACT,GAAG,CAACa,CAAC,IAAI1B,QAAQ,CAAC2B,iBAAiB,CAACD,CAAC,CAAC,CAAC,EACtD,GAAGH,cAAc,CAACV,GAAG,CAACa,CAAC,IAAI1B,QAAQ,CAAC4B,oBAAoB,CAACF,CAAC,CAAC,CAAC,CAEhE,CAAC;EAED,MAAMG,aAAa,GAAGvC,gBAAgB,CACpCkC,uBAAuB,EACvB,EAAE,EACFH,WAAW,EACX,IACF,CAAC;EAED,MAAMS,SAAS,GAAGtC,gBAAgB,CAChCL,2CAA2C,CAAC2B,KAAK,CAACiB,GAAG,CACvD,CAAC;EAED,MAAMC,UAAU,GAAGxC,gBAAgB,CACjCR,0CAA0C,CAAC8B,KAAK,CAACmB,MAAM,CACzD,CAAC;EAED,MAAMC,mBAAmB,GAAGjC,sBAAsB,CAACkC,QAAQ,CACzDrB,KAAK,CAACsB,WACR,CAAC;EAED,MAAMvC,EAAE,CAACmB,SAAS,CAChB5B,IAAI,CAACiB,IAAI,CAACP,MAAM,EAAE,GAAGgB,KAAK,CAACG,YAAY,KAAK,CAAC,EAC7C,MAAMvB,QAAQ,CAAC;AACnB,uGACMS,cAAc,GAAG,WAAW,GAAG,cAAc;AACnD,6EAC6ED,SAAS;AACtF,8DAA8DA,SAAS;AACvE,UAAU2B,aAAa;AACvB;AACA,2BAA2Bf,KAAK,CAACG,YAAY;AAC7C;AACA,cAActB,uBAAuB,CAACmB,KAAK,CAACuB,WAAW,CAAC;AACxD,eACM3B,MAAM,CAACS,IAAI,CAACL,KAAK,CAACwB,UAAU,CAAC,CAAClB,MAAM,GAAG,CAAC,GACpC,UAAUN,KAAK,CAACyB,gBAAgB,EAAE,GAClC,EAAE,cACMzB,KAAK,CAACG,YAAY;AACpC;AACA;AACA,UACMP,MAAM,CAACS,IAAI,CAACL,KAAK,CAACwB,UAAU,CAAC,CAAClB,MAAM,GAAG,CAAC,GACpC;AACV;AACA,cACU3B,SAAS,CAACqB,KAAK,CAACwB,UAAU,EAAE;IAC1B,GAAG,EAAEE,CAACC,SAAS,EAAEC,SAAS,EAAEC,OAAO,KAAK;MACtC,MAAMC,CAAC,GAAG1D,UAAU,CAACuD,SAAS,CAAC;MAC/B,OAAO,CACL;AAChB,kBACkBI,eAAe,CAAC3D,UAAU,CAACuD,SAAS,CAAC,EAAE;QAAEE;MAAQ,CAAC,CAAC,aACxCA,OAAO,IAAIC,CAAC,CAACE,QAAQ,GAAG,GAAG,GAAG,EAAE,EAAE,EAC/CC,iBAAiB,CAAC/C,QAAQ,EAAE4C,CAAC,EAAE,OAAO,CAAC,CACxC;IACH;EACF,CAAC,CAAC;AACZ,cACc,GACJ,EAAE;AACZ;AACA,cAEM9B,KAAK,CAACmB,MAAM,CAACe,IAAI,KAAK,QAAQ,GAC1B;AACV;AACA,cACUD,iBAAiB,CACf/C,QAAQ,EACRd,UAAU,CAAC;IAAE+D,QAAQ,EAAEnC,KAAK,CAACmB;EAAO,CAAC,CAAC,EACtC,QACF,CAAC;AACX,SACS,GACC;AACV,mCACUc,iBAAiB,CACf/C,QAAQ,EACRd,UAAU,CAAC;IAAE+D,QAAQ,EAAEnC,KAAK,CAACmB;EAAO,CAAC,CAAC,EACtC,QACF,CAAC;AACX,WACW;AACX;AACA;AACA;AACA,2BAC2BnB,KAAK,CAACG,YAAY;AAC7C,YAAYH,KAAK,CAACG,YAAY;AAC9B;AACA;AACA,eAAexB,SAAS,CAACqC,SAAS,CAAC;AACnC,+BAA+BI,mBAAmB;AAClD;AACA,cAAcgB,oBAAoB,CAAClD,QAAQ,EAAEc,KAAK,CAAC;AACnD;AACA;AACA,cAAcrB,SAAS,CAACuC,UAAU,CAAC;AACnC,cAAcmB,0BAA0B,CAACnD,QAAQ,EAAEc,KAAK,CAACmB,MAAM,CAAC;AAChE;AACA,yBAAyBnB,KAAK,CAACG,YAAY;AAC3C;AACA,UACMxB,SAAS,CAACqC,SAAS,EAAE;IACnB,aAAa,EAAEO,CAAA,KAAMe,SAAS;IAC9B,aAAa,EAAEC,CAAA,KAAMD,SAAS;IAC9B,KAAK,EAAEE,CAAA,KAAMF;EACf,CAAC,CAAC;AACR;AACA;AACA;AACA;AACA,uBACuBtC,KAAK,CAACG,YAAY,KAAKH,KAAK,CAACyC,oBAAoB;AACxE,cACM9D,SAAS,CAACqC,SAAS,EAAE;IACnB,aAAa,EAAEO,CAAA,KAAMe,SAAS;IAC9B,aAAa,EAAEC,CAAA,KAAMD,SAAS;IAC9B,KAAK,EAAEE,CAAA,KAAMF;EACf,CAAC,CAAC;AACR,sBACsBlB,mBAAmB;AACzC;AACA;AACA,SAAS,CACP,CAAC;AACH;AAEA,SAASgB,oBAAoBA,CAC3BlD,QAAoC,EACpCc,KAAoB,EACpB;EACA,OAAOrB,SAAS,CAACqB,KAAK,CAACwB,UAAU,EAAE;IACjC,GAAG,EAAEE,CAACC,SAAS,EAAEe,cAAc,EAAEb,OAAO,KAAK,CAC3C,GAAGE,eAAe,CAAC3D,UAAU,CAACuD,SAAS,CAAC,EAAE;MAAEE;IAAQ,CAAC,CAAC,IAAIA,OAAO,EAAE,EACnE;AACN,YAAYlD,SAAS,CAACD,gBAAgB,CAACN,UAAU,CAACuD,SAAS,CAAC,CAAC,CAAC;AAC9D,YAAYU,0BAA0B,CAACnD,QAAQ,EAAEyC,SAAS,CAACQ,QAAQ,CAAC;AACpE,UAAU;EAER,CAAC,CAAC;AACJ;AAEA,SAASE,0BAA0BA,CACjCnD,QAAoC,EACpCyD,GAAkB,EAClB;EACA,IAAIA,GAAG,CAACT,IAAI,KAAK,QAAQ,IAAIS,GAAG,CAACT,IAAI,KAAK,WAAW,EAAE;IACrD,OAAO,sBACLhD,QAAQ,CAAC2B,iBAAiB,CACxB8B,GAAG,CAACC,iBACN,CAAC,CAACC,+BAA+B,CAAC,IAAI,CAAC,EACvC;EACJ;EACA,OAAO,EAAE;AACX;AAEA,OAAO,SAASd,eAAeA,CAC7Be,KAAoC,EACpC;EAAEjB;AAA6B,CAAC,EACxB;EACR,IAAIkB,GAAG,GAAG,OAAO;EAEjB,IAAID,KAAK,CAACvB,WAAW,EAAE;IACrB,IAAIuB,KAAK,CAACvB,WAAW,EAAE;MACrBwB,GAAG,IAAI,qBAAqBD,KAAK,CAACvB,WAAW,IAAI;IACnD;EACF,CAAC,MAAM;IACLwB,GAAG,IAAI,6BAA6B;EACtC;EAEAA,GAAG,IAAI,OAAO;EACd,OAAOA,GAAG;AACZ;AAEA,OAAO,SAASd,iBAAiBA,CAC/Be,gBAA4C,EAC5CC,KAA+B,EAC/Bf,IAAwB,EACxBgB,QAAQ,GAAG,KAAK,EACR;EACR,IAAIC,KAAK,GAAG,cAAcF,KAAK,CAACf,IAAI,KAAK;EAEzC,QAAQe,KAAK,CAACf,IAAI;IAChB,KAAK,MAAM;MACTiB,KAAK,GAAG,QAAQjB,IAAI,kBAAkBkB,IAAI,CAACzE,SAAS,CAAC,UAAU,CAAC,GAAG;MACnE;IAEF,KAAK,YAAY;IACjB,KAAK,SAAS;IACd,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,WAAW;MACdwE,KAAK,GAAG,QAAQjB,IAAI,kBAAkBkB,IAAI,CAACzE,SAAS,CAACsE,KAAK,CAACf,IAAI,CAAC,GAAG;MACnE;IACF,KAAK,QAAQ;MACXiB,KAAK,GAAG;AACd,cACQxE,SAAS,CAACsE,KAAK,CAACI,MAAM,EAAE;QACtB,GAAG,EAAE3B,CAAC4B,CAAC,EAAE1B,SAAS,EAAEC,OAAO,KAAK;UAC9B,OAAO,CACL;AACd,kBAAkBK,IAAI,KAAK,OAAO,GAAG,WAAW,GAAG,EAAE,IAAIL,OAAO,IAChDyB,CAAC,CAACtB,QAAQ,GAAG,GAAG,GAAG,EAAE,EACrB,EACFC,iBAAiB,CAACe,gBAAgB,EAAEM,CAAC,EAAEpB,IAAI,CAAC,CAC7C;QACH;MACF,CAAC,CAAC;AACV,cACc;MACR;IACF,KAAK,2BAA2B;MAC9BiB,KAAK,GAAG,QAAQjB,IAAI,kCAClBe,KAAK,CAACM,yBAAyB,CAACC,OAAO,KAAK,OAAO,GAC/C,QAAQtB,IAAI,cAAce,KAAK,CAACM,yBAAyB,CAACE,UAAU,IAAI,GACxE,IAAIR,KAAK,CAACM,yBAAyB,CAACC,OAAO,GAAG,MAC9CP,KAAK,CAACM,yBAAyB,CAACG,SAAS,IAAI;MACnD;IAEF,KAAK,6BAA6B;MAChCP,KAAK,GAAG,QAAQjB,IAAI,oCAClBe,KAAK,CAACU,2BAA2B,CAACH,OAAO,KAAK,OAAO,GACjD,QAAQtB,IAAI,cAAce,KAAK,CAACU,2BAA2B,CAACF,UAAU,IAAI,GAC1E,IAAIR,KAAK,CAACU,2BAA2B,CAACH,OAAO,GAAG,IAEpDP,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACF,OAAO,KAAK,OAAO,GAC3D,QAAQtB,IAAI,cAAce,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACD,UAAU,IAAI,GACpF,IAAIR,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACF,OAAO,GAAG;AACtE,WACWP,KAAK,CAACU,2BAA2B,CAACD,SAAS,CAACA,SAAS,IAAI;MAC9D;IACF,KAAK,QAAQ;MACX,IAAIR,QAAQ,EAAE;QACZC,KAAK,GAAG,mBACNH,gBAAgB,CAACnC,iBAAiB,CAACoC,KAAK,CAACW,MAAM,CAAC,CAC7Cf,+BAA+B,CAAC,IAAI,CAAC,GACvC;QACH;MACF;MACAM,KAAK,GAAG,QAAQjB,IAAI,eAClBc,gBAAgB,CAACnC,iBAAiB,CAACoC,KAAK,CAACW,MAAM,CAAC,CAC7Cf,+BAA+B,CAAC,IAAI,CAAC,GACvC;MACH;IACF,KAAK,WAAW;MACdM,KAAK,GAAG,QAAQjB,IAAI,kBAClBc,gBAAgB,CAAClC,oBAAoB,CAACmC,KAAK,CAACY,SAAS,CAAC,CACnDhB,+BAA+B,CAAC,IAAI,CAAC,GACvC;MACH;IAEF,KAAK,WAAW;MACdM,KAAK,GAAG,QAAQjB,IAAI,kBAClBc,gBAAgB,CAACnC,iBAAiB,CAACoC,KAAK,CAACa,SAAS,CAAC,CAChDjB,+BAA+B,CAAC,IAAI,CAAC,GACvC;MACH;IAEF,KAAK,oBAAoB;MACvBM,KAAK,GAAG,QAAQjB,IAAI,kBAClBc,gBAAgB,CAAClC,oBAAoB,CAACmC,KAAK,CAACa,SAAS,CAAC,CACnDjB,+BAA+B,CAAC,IAAI,CAAC,GACvC;MACH;IAEF,KAAK,KAAK;MACRM,KAAK,GAAG,GAAGjB,IAAI,KAAK,OAAO,GAAG,UAAU,GAAG,EAAE,OAC3CD,iBAAiB,CAACe,gBAAgB,EAAEC,KAAK,CAACc,GAAG,EAAE7B,IAAI,CAAC,GACnD;MACH;IAEF,KAAK,OAAO;MACViB,KAAK,GAAGF,KAAK,CAACe,KAAK,CAACjE,GAAG,CAAEkE,CAAC,IACxBhC,iBAAiB,CAACe,gBAAgB,EAAEiB,CAAC,EAAE/B,IAAI,CAC7C,CAAC,CAAC3C,IAAI,CAAC,KAAK,CAAC;MACb;IAEF,KAAK,KAAK;MACR4D,KAAK,GAAG,kBACNlB,iBAAiB,CAACe,gBAAgB,EAAEC,KAAK,CAACO,OAAO,EAAEtB,IAAI,EAAE,IAAI,CAAC,KAC3DD,iBAAiB,CAACe,gBAAgB,EAAEC,KAAK,CAACS,SAAS,EAAExB,IAAI,CAAC,IAAI;EACvE;EAEA,IAAIe,KAAK,CAACiB,YAAY,IAAIhC,IAAI,KAAK,OAAO,EAAE;IAC1C,OAAO,iBAAiBiB,KAAK,GAAG;EAClC,CAAC,MAAM,IAAIF,KAAK,CAACiB,YAAY,EAAE;IAC7B,OAAO,SAASf,KAAK,GAAG;EAC1B;EACA,OAAOA,KAAK;AACd","ignoreList":[]}

package/build/cjs/index.cjs

@@ -1012,6 +1012,65 @@ async function generatePerObjectDataFiles(ctx) {
     `));
 }
 
+// src/shared/getInterfaceTypesFromQueryDataType.ts
+function getInterfaceTypesFromQueryDataType(dataType, types) {
+  switch (dataType.type) {
+    case "array":
+    case "set":
+      getInterfaceTypesFromQueryDataType(dataType.subType, types);
+      return;
+    case "interfaceObject":
+    case "interfaceObjectSet":
+      types.add(dataType.interfaceTypeApiName);
+      return;
+    case "struct":
+      for (const prop of Object.values(dataType.fields)) {
+        getInterfaceTypesFromQueryDataType(prop.fieldType, types);
+      }
+      return;
+    case "union":
+      for (const type of dataType.unionTypes) {
+        getInterfaceTypesFromQueryDataType(type, types);
+      }
+      return;
+    case "entrySet":
+      getInterfaceTypesFromQueryDataType(dataType.keyType, types);
+      getInterfaceTypesFromQueryDataType(dataType.valueType, types);
+      return;
+    case "attachment":
+    case "boolean":
+    case "date":
+    case "double":
+    case "float":
+    case "integer":
+    case "long":
+    case "null":
+    case "string":
+    case "threeDimensionalAggregation":
+    case "timestamp":
+    case "twoDimensionalAggregation":
+    case "entrySet":
+    case "unsupported":
+    case "object":
+    case "objectSet":
+      return;
+    default:
+      throw new Error(`Unknown query data type ${dataType.type}`);
+  }
+}
+
+// src/shared/getInterfaceTypeApiNamesFromQuery.ts
+function getInterfaceTypeApiNamesFromQuery(query) {
+  const types = /* @__PURE__ */ new Set();
+  for (const {
+    dataType
+  } of Object.values(query.parameters)) {
+    getInterfaceTypesFromQueryDataType(dataType, types);
+  }
+  getInterfaceTypesFromQueryDataType(query.output, types);
+  return Array.from(types);
+}
+
 // src/shared/getObjectTypesFromQueryDataType.ts
 function getObjectTypesFromQueryDataType(dataType, types) {
   switch (dataType.type) {
@@ -1099,8 +1158,9 @@ async function generatePerQueryDataFilesV2({
 async function generateV2QueryFile(fs2, outDir, relOutDir, query, importExt, ontology, forInternalUse, fixedVersionQueryTypes) {
   const relFilePath = path8__namespace.default.join(relOutDir, `${query.shortApiName}.ts`);
   const objectTypes = getObjectTypeApiNamesFromQuery(query);
-  const objectTypeObjects = new Set(objectTypes.map((o) => ontology.requireObjectType(o)));
-  const importObjects = getObjectImports(objectTypeObjects, "", relFilePath, true);
+  const interfaceTypes = getInterfaceTypeApiNamesFromQuery(query);
+  const interfaceAndObjectTypes = /* @__PURE__ */ new Set([...objectTypes.map((o) => ontology.requireObjectType(o)), ...interfaceTypes.map((o) => ontology.requireInterfaceType(o))]);
+  const importObjects = getObjectImports(interfaceAndObjectTypes, "", relFilePath, true);
   const baseProps = deleteUndefineds(generatorConverters.wireQueryTypeV2ToSdkQueryDefinitionNoParams(query.raw));
   const outputBase = deleteUndefineds(generatorConverters.wireQueryDataTypeToQueryDataTypeDefinition(query.output));
   const isUsingFixedVersion = fixedVersionQueryTypes.includes(query.fullApiName);
@@ -1251,9 +1311,15 @@ function getQueryParamType(enhancedOntology, input, type, isMapKey = false) {
       }
       inner = `Query${type}.ObjectType<${enhancedOntology.requireObjectType(input.object).getImportedDefinitionIdentifier(true)}>`;
       break;
+    case "interface":
+      inner = `Query${type}.InterfaceType<${enhancedOntology.requireInterfaceType(input.interface).getImportedDefinitionIdentifier(true)}>`;
+      break;
     case "objectSet":
       inner = `Query${type}.ObjectSetType<${enhancedOntology.requireObjectType(input.objectSet).getImportedDefinitionIdentifier(true)}>`;
       break;
+    case "interfaceObjectSet":
+      inner = `Query${type}.ObjectSetType<${enhancedOntology.requireInterfaceType(input.objectSet).getImportedDefinitionIdentifier(true)}>`;
+      break;
     case "set":
       inner = `${type === "Param" ? "Readonly" : ""}Set<${getQueryParamType(enhancedOntology, input.set, type)}>`;
       break;