@palantir/pack.document-schema.type-gen 0.1.1 → 0.2.1

package/build/esm/utils/ir/__tests__/generateModelsFromIr.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateModelsFromIr.test.js","names":["path","describe","expect","it","generateModelsFromIr","formatWithPrettier","snapshotDir","join","__dirname","result","name","description","version","primaryModelKeys","models","Person","type","record","key","fields","value","string","minLength","maxLength","meta","addedIn","integer","minValue","maxValue","formatted","lines","split","importLines","filter","line","startsWith","length","toBeGreaterThan","forEach","toMatch","toMatchFileSnapshot","Event","docRef","documentTypeRids","userRef","datetime","toContain","ObjectNode","double","TextBox","Node","union","discriminant","variants","object"],"sources":["generateModelsFromIr.test.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 path from \"path\";\nimport { describe, expect, it } from \"vitest\";\nimport type {\n  IFieldDef,\n  IModelDef,\n  IRealTimeDocumentSchema,\n  IRecordDef,\n  IUnionDef,\n} from \"../../../lib/pack-docschema-api/pack-docschema-ir/index.js\";\nimport { generateModelsFromIr } from \"../generateModelsFromIr.js\";\nimport { formatWithPrettier } from \"./formatWithPrettier.js\";\n\ndescribe(\"generateModelsFromIr\", () => {\n  const snapshotDir = path.join(__dirname, \"__snapshots__\", \"generateModelsFromIr\");\n\n  it(\"should generate Model constants for simple record types\", async () => {\n    const personField1: IFieldDef = {\n      key: \"name\",\n      name: \"Name\",\n      description: \"Person name\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"string\",\n          string: { minLength: 2, maxLength: 50 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personField2: IFieldDef = {\n      key: \"age\",\n      name: \"Age\",\n      description: \"Person age\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"integer\",\n          integer: { minValue: 0, maxValue: 150 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personRecord: IRecordDef = {\n      key: \"Person\",\n      name: \"Person\",\n      description: \"A person record\",\n      fields: [personField1, personField2],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Person\"],\n      models: {\n        Person: {\n          type: \"record\",\n          record: personRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateModelsFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify imports are properly formatted (no weird indentation)\n    const lines = formatted.split(\"\\n\");\n    const importLines = lines.filter(line => line.startsWith(\"import\"));\n    expect(importLines.length).toBeGreaterThan(0);\n    // All imports should start at column 0 (no leading whitespace)\n    importLines.forEach(line => {\n      expect(line).toMatch(/^import/);\n    });\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"simple-record.ts\"));\n  });\n\n  it(\"should generate Model constants with external ref fields\", async () => {\n    const eventField1: IFieldDef = {\n      key: \"documentRef\",\n      name: \"Document Reference\",\n      description: \"Reference to a document\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"docRef\",\n          docRef: {\n            documentTypeRids: [],\n          },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventField2: IFieldDef = {\n      key: \"userRef\",\n      name: \"User Reference\",\n      description: \"Reference to a user\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"userRef\",\n          userRef: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventField3: IFieldDef = {\n      key: \"timestamp\",\n      name: \"Timestamp\",\n      description: \"Event timestamp\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"datetime\",\n          datetime: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventRecord: IRecordDef = {\n      key: \"Event\",\n      name: \"Event\",\n      description: \"An event record with external refs\",\n      fields: [eventField1, eventField2, eventField3],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with external refs\",\n      version: 1,\n      primaryModelKeys: [\"Event\"],\n      models: {\n        Event: {\n          type: \"record\",\n          record: eventRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateModelsFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify external ref fields are included in metadata\n    expect(formatted).toContain(\"externalRefFieldTypes\");\n    expect(formatted).toContain(\"documentRef: 'docRef'\");\n    expect(formatted).toContain(\"userRef: 'userRef'\");\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"external-refs.ts\"));\n  });\n\n  it(\"should generate Model constants for union types\", async () => {\n    // Define the ObjectNode record\n    const objectNodeRecord: IRecordDef = {\n      key: \"ObjectNode\",\n      name: \"ObjectNode\",\n      description: \"A node in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"y\",\n          name: \"Y\",\n          description: \"Y coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the TextBox record\n    const textBoxRecord: IRecordDef = {\n      key: \"TextBox\",\n      name: \"TextBox\",\n      description: \"A text box in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"text\",\n          name: \"Text\",\n          description: \"Text content\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the Node union\n    const nodeUnion: IUnionDef = {\n      key: \"Node\",\n      discriminant: \"type\",\n      name: \"Node\",\n      description: \"A node in the graph\",\n      variants: {\n        object: \"ObjectNode\",\n        \"text-box\": \"TextBox\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with unions\",\n      version: 1,\n      primaryModelKeys: [\"ObjectNode\", \"TextBox\", \"Node\"],\n      models: {\n        ObjectNode: {\n          type: \"record\",\n          record: objectNodeRecord,\n        } as IModelDef,\n        TextBox: {\n          type: \"record\",\n          record: textBoxRecord,\n        } as IModelDef,\n        Node: {\n          type: \"union\",\n          union: nodeUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateModelsFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify all models are generated (primary models + union variants)\n    expect(formatted).toContain(\"export const ObjectNodeModel\");\n    expect(formatted).toContain(\"export const TextBoxModel\");\n    expect(formatted).toContain(\"export const NodeModel\");\n    expect(formatted).toContain(\"export const NodeObjectModel\");\n    expect(formatted).toContain(\"export const NodeTextBoxModel\");\n\n    // Verify DocumentSchema contains all models\n    expect(formatted).toContain(\"export const DocumentModel\");\n    expect(formatted).toContain(\"ObjectNode: ObjectNodeModel\");\n    expect(formatted).toContain(\"TextBox: TextBoxModel\");\n    expect(formatted).toContain(\"Node: NodeModel\");\n    expect(formatted).toContain(\"NodeObject: NodeObjectModel\");\n    expect(formatted).toContain(\"NodeTextBox: NodeTextBoxModel\");\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"union-types.ts\"));\n  });\n});\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,OAAOA,IAAI,MAAM,MAAM;AACvB,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,QAAQ;AAQ7C,SAASC,oBAAoB,QAAQ,4BAA4B;AACjE,SAASC,kBAAkB,QAAQ,yBAAyB;AAE5DJ,QAAQ,CAAC,sBAAsB,EAAE,MAAM;EACrC,MAAMK,WAAW,GAAGN,IAAI,CAACO,IAAI,CAACC,SAAS,EAAE,eAAe,EAAE,sBAAsB,CAAC;EAEjFL,EAAE,CAAC,yDAAyD,EAAE,YAAY;IAkDxE,MAAMM,MAAM,GAAG,MAAML,oBAAoB,CAbD;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,QAAQ,CAAC;MAC5BC,MAAM,EAAE;QACNC,MAAM,EAAE;UACNC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBqB;YAC/BC,GAAG,EAAE,QAAQ;YACbR,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,iBAAiB;YAC9BQ,MAAM,EAAE,CAhCsB;cAC9BD,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,aAAa;cAC1BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE;oBAAEC,SAAS,EAAE,CAAC;oBAAEC,SAAS,EAAE;kBAAG;gBACxC;cACF,CAAC;cACDC,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE+B;cAC9BP,GAAG,EAAE,KAAK;cACVR,IAAI,EAAE,KAAK;cACXC,WAAW,EAAE,YAAY;cACzBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfU,OAAO,EAAE;oBAAEC,QAAQ,EAAE,CAAC;oBAAEC,QAAQ,EAAE;kBAAI;gBACxC;cACF,CAAC;cACDJ,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMqC;YACpCD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEgD,CAAC;IACjD,MAAMI,SAAS,GAAG,MAAMxB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACA,MAAMqB,KAAK,GAAGD,SAAS,CAACE,KAAK,CAAC,IAAI,CAAC;IACnC,MAAMC,WAAW,GAAGF,KAAK,CAACG,MAAM,CAACC,IAAI,IAAIA,IAAI,CAACC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACnEjC,MAAM,CAAC8B,WAAW,CAACI,MAAM,CAAC,CAACC,eAAe,CAAC,CAAC,CAAC;IAC7C;IACAL,WAAW,CAACM,OAAO,CAACJ,IAAI,IAAI;MAC1BhC,MAAM,CAACgC,IAAI,CAAC,CAACK,OAAO,CAAC,SAAS,CAAC;IACjC,CAAC,CAAC;IAEF,MAAMrC,MAAM,CAAC2B,SAAS,CAAC,CAACW,mBAAmB,CAACxC,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,kBAAkB,CAAC,CAAC;EACzF,CAAC,CAAC;EAEFH,EAAE,CAAC,0DAA0D,EAAE,YAAY;IAkEzE,MAAMM,MAAM,GAAG,MAAML,oBAAoB,CAbD;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,kCAAkC;MAC/CC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,OAAO,CAAC;MAC3BC,MAAM,EAAE;QACN2B,KAAK,EAAE;UACLzB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBoB;YAC9BC,GAAG,EAAE,OAAO;YACZR,IAAI,EAAE,OAAO;YACbC,WAAW,EAAE,oCAAoC;YACjDQ,MAAM,EAAE,CAhDqB;cAC7BD,GAAG,EAAE,aAAa;cAClBR,IAAI,EAAE,oBAAoB;cAC1BC,WAAW,EAAE,yBAAyB;cACtCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACd0B,MAAM,EAAE;oBACNC,gBAAgB,EAAE;kBACpB;gBACF;cACF,CAAC;cACDnB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE8B;cAC7BP,GAAG,EAAE,SAAS;cACdR,IAAI,EAAE,gBAAgB;cACtBC,WAAW,EAAE,qBAAqB;cAClCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACf4B,OAAO,EAAE,CAAC;gBACZ;cACF,CAAC;cACDpB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE8B;cAC7BP,GAAG,EAAE,WAAW;cAChBR,IAAI,EAAE,WAAW;cACjBC,WAAW,EAAE,iBAAiB;cAC9BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,UAAU;kBAChB6B,QAAQ,EAAE,CAAC;gBACb;cACF,CAAC;cACDrB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMgD;YAC/CD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEgD,CAAC;IACjD,MAAMI,SAAS,GAAG,MAAMxB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACAP,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,uBAAuB,CAAC;IACpD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,uBAAuB,CAAC;IACpD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,oBAAoB,CAAC;IAEjD,MAAM5C,MAAM,CAAC2B,SAAS,CAAC,CAACW,mBAAmB,CAACxC,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,kBAAkB,CAAC,CAAC;EACzF,CAAC,CAAC;EAEFH,EAAE,CAAC,iDAAiD,EAAE,YAAY;IAChE;;IAoCA;;IAoCA;;IAkCA,MAAMM,MAAM,GAAG,MAAML,oBAAoB,CArBD;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,2BAA2B;MACxCC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC;MACnDC,MAAM,EAAE;QACNiC,UAAU,EAAE;UACV/B,IAAI,EAAE,QAAQ;UACdC,MAAM,EA5FyB;YACnCC,GAAG,EAAE,YAAY;YACjBR,IAAI,EAAE,YAAY;YAClBC,WAAW,EAAE,qBAAqB;YAClCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdgC,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDxB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdgC,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDxB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA4DI,CAAc;QACdwB,OAAO,EAAE;UACPjC,IAAI,EAAE,QAAQ;UACdC,MAAM,EA5DsB;YAChCC,GAAG,EAAE,SAAS;YACdR,IAAI,EAAE,SAAS;YACfC,WAAW,EAAE,yBAAyB;YACtCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdgC,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDxB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA4BI,CAAc;QACdyB,IAAI,EAAE;UACJlC,IAAI,EAAE,OAAO;UACbmC,KAAK,EA5BkB;YAC3BjC,GAAG,EAAE,MAAM;YACXkC,YAAY,EAAE,MAAM;YACpB1C,IAAI,EAAE,MAAM;YACZC,WAAW,EAAE,qBAAqB;YAClC0C,QAAQ,EAAE;cACRC,MAAM,EAAE,YAAY;cACpB,UAAU,EAAE;YACd,CAAC;YACD9B,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAmBI;MACF;IACF,CAEgD,CAAC;IACjD,MAAMI,SAAS,GAAG,MAAMxB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACAP,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,8BAA8B,CAAC;IAC3D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,2BAA2B,CAAC;IACxD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,wBAAwB,CAAC;IACrD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,8BAA8B,CAAC;IAC3D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,+BAA+B,CAAC;;IAE5D;IACA5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,4BAA4B,CAAC;IACzD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,6BAA6B,CAAC;IAC1D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,uBAAuB,CAAC;IACpD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,iBAAiB,CAAC;IAC9C5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,6BAA6B,CAAC;IAC1D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,+BAA+B,CAAC;IAE5D,MAAM5C,MAAM,CAAC2B,SAAS,CAAC,CAACW,mBAAmB,CAACxC,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,gBAAgB,CAAC,CAAC;EACvF,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}
+{"version":3,"file":"generateModelsFromIr.test.js","names":["path","describe","expect","it","generateModelsFromIr","formatWithPrettier","snapshotDir","join","__dirname","result","name","description","version","primaryModelKeys","models","Person","type","record","key","fields","value","string","minLength","maxLength","meta","addedIn","integer","minValue","maxValue","formatted","lines","split","importLines","filter","line","startsWith","length","toBeGreaterThan","forEach","toMatch","toMatchFileSnapshot","Event","docRef","documentTypeRids","userRef","datetime","toContain","ObjectNode","double","TextBox","Node","union","discriminant","variants","object","Cat","Dog","Animal","cat","dog"],"sources":["generateModelsFromIr.test.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 path from \"path\";\nimport { describe, expect, it } from \"vitest\";\nimport type {\n  IFieldDef,\n  IModelDef,\n  IRealTimeDocumentSchema,\n  IRecordDef,\n  IUnionDef,\n} from \"../../../lib/pack-docschema-api/pack-docschema-ir/index.js\";\nimport { generateModelsFromIr } from \"../generateModelsFromIr.js\";\nimport { formatWithPrettier } from \"./formatWithPrettier.js\";\n\ndescribe(\"generateModelsFromIr\", () => {\n  const snapshotDir = path.join(__dirname, \"__snapshots__\", \"generateModelsFromIr\");\n\n  it(\"should generate Model constants for simple record types\", async () => {\n    const personField1: IFieldDef = {\n      key: \"name\",\n      name: \"Name\",\n      description: \"Person name\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"string\",\n          string: { minLength: 2, maxLength: 50 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personField2: IFieldDef = {\n      key: \"age\",\n      name: \"Age\",\n      description: \"Person age\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"integer\",\n          integer: { minValue: 0, maxValue: 150 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personRecord: IRecordDef = {\n      key: \"Person\",\n      name: \"Person\",\n      description: \"A person record\",\n      fields: [personField1, personField2],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Person\"],\n      models: {\n        Person: {\n          type: \"record\",\n          record: personRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateModelsFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify imports are properly formatted (no weird indentation)\n    const lines = formatted.split(\"\\n\");\n    const importLines = lines.filter(line => line.startsWith(\"import\"));\n    expect(importLines.length).toBeGreaterThan(0);\n    // All imports should start at column 0 (no leading whitespace)\n    importLines.forEach(line => {\n      expect(line).toMatch(/^import/);\n    });\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"simple-record.ts\"));\n  });\n\n  it(\"should generate Model constants with external ref fields\", async () => {\n    const eventField1: IFieldDef = {\n      key: \"documentRef\",\n      name: \"Document Reference\",\n      description: \"Reference to a document\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"docRef\",\n          docRef: {\n            documentTypeRids: [],\n          },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventField2: IFieldDef = {\n      key: \"userRef\",\n      name: \"User Reference\",\n      description: \"Reference to a user\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"userRef\",\n          userRef: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventField3: IFieldDef = {\n      key: \"timestamp\",\n      name: \"Timestamp\",\n      description: \"Event timestamp\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"datetime\",\n          datetime: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventRecord: IRecordDef = {\n      key: \"Event\",\n      name: \"Event\",\n      description: \"An event record with external refs\",\n      fields: [eventField1, eventField2, eventField3],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with external refs\",\n      version: 1,\n      primaryModelKeys: [\"Event\"],\n      models: {\n        Event: {\n          type: \"record\",\n          record: eventRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateModelsFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify external ref fields are included in metadata\n    expect(formatted).toContain(\"externalRefFieldTypes\");\n    expect(formatted).toContain(\"documentRef: 'docRef'\");\n    expect(formatted).toContain(\"userRef: 'userRef'\");\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"external-refs.ts\"));\n  });\n\n  it(\"should generate Model constants for union types\", async () => {\n    // Define the ObjectNode record\n    const objectNodeRecord: IRecordDef = {\n      key: \"ObjectNode\",\n      name: \"ObjectNode\",\n      description: \"A node in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"y\",\n          name: \"Y\",\n          description: \"Y coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the TextBox record\n    const textBoxRecord: IRecordDef = {\n      key: \"TextBox\",\n      name: \"TextBox\",\n      description: \"A text box in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"text\",\n          name: \"Text\",\n          description: \"Text content\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the Node union\n    const nodeUnion: IUnionDef = {\n      key: \"Node\",\n      discriminant: \"type\",\n      name: \"Node\",\n      description: \"A node in the graph\",\n      variants: {\n        object: \"ObjectNode\",\n        \"text-box\": \"TextBox\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with unions\",\n      version: 1,\n      primaryModelKeys: [\"ObjectNode\", \"TextBox\", \"Node\"],\n      models: {\n        ObjectNode: {\n          type: \"record\",\n          record: objectNodeRecord,\n        } as IModelDef,\n        TextBox: {\n          type: \"record\",\n          record: textBoxRecord,\n        } as IModelDef,\n        Node: {\n          type: \"union\",\n          union: nodeUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateModelsFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify all models are generated (primary models + union variants)\n    expect(formatted).toContain(\"export const ObjectNodeModel\");\n    expect(formatted).toContain(\"export const TextBoxModel\");\n    expect(formatted).toContain(\"export const NodeModel\");\n    expect(formatted).toContain(\"export const NodeObjectModel\");\n    expect(formatted).toContain(\"export const NodeTextBoxModel\");\n\n    // Verify DocumentSchema contains all models\n    expect(formatted).toContain(\"export const DocumentModel\");\n    expect(formatted).toContain(\"ObjectNode: ObjectNodeModel\");\n    expect(formatted).toContain(\"TextBox: TextBoxModel\");\n    expect(formatted).toContain(\"Node: NodeModel\");\n    expect(formatted).toContain(\"NodeObject: NodeObjectModel\");\n    expect(formatted).toContain(\"NodeTextBox: NodeTextBoxModel\");\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"union-types.ts\"));\n  });\n\n  it(\"should generate Model constants with custom discriminant\", async () => {\n    const catRecord: IRecordDef = {\n      key: \"Cat\",\n      name: \"Cat\",\n      description: \"A cat\",\n      fields: [\n        {\n          key: \"meow\",\n          name: \"Meow\",\n          description: \"Cat sound\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    const dogRecord: IRecordDef = {\n      key: \"Dog\",\n      name: \"Dog\",\n      description: \"A dog\",\n      fields: [\n        {\n          key: \"bark\",\n          name: \"Bark\",\n          description: \"Dog sound\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    const animalUnion: IUnionDef = {\n      key: \"Animal\",\n      discriminant: \"kind\",\n      name: \"Animal\",\n      description: \"An animal union with custom discriminant\",\n      variants: {\n        cat: \"Cat\",\n        dog: \"Dog\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with custom discriminant\",\n      version: 1,\n      primaryModelKeys: [\"Cat\", \"Dog\", \"Animal\"],\n      models: {\n        Cat: {\n          type: \"record\",\n          record: catRecord,\n        } as IModelDef,\n        Dog: {\n          type: \"record\",\n          record: dogRecord,\n        } as IModelDef,\n        Animal: {\n          type: \"union\",\n          union: animalUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateModelsFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    expect(formatted).toContain(\"discriminant: 'kind'\");\n    expect(formatted).toContain(\"export const AnimalModel\");\n    expect(formatted).toContain(\"export const AnimalCatModel\");\n    expect(formatted).toContain(\"export const AnimalDogModel\");\n\n    await expect(formatted).toMatchFileSnapshot(\n      path.join(snapshotDir, \"custom-discriminant.ts\"),\n    );\n  });\n});\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,OAAOA,IAAI,MAAM,MAAM;AACvB,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,QAAQ;AAQ7C,SAASC,oBAAoB,QAAQ,4BAA4B;AACjE,SAASC,kBAAkB,QAAQ,yBAAyB;AAE5DJ,QAAQ,CAAC,sBAAsB,EAAE,MAAM;EACrC,MAAMK,WAAW,GAAGN,IAAI,CAACO,IAAI,CAACC,SAAS,EAAE,eAAe,EAAE,sBAAsB,CAAC;EAEjFL,EAAE,CAAC,yDAAyD,EAAE,YAAY;IAkDxE,MAAMM,MAAM,GAAG,MAAML,oBAAoB,CAbD;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,QAAQ,CAAC;MAC5BC,MAAM,EAAE;QACNC,MAAM,EAAE;UACNC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBqB;YAC/BC,GAAG,EAAE,QAAQ;YACbR,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,iBAAiB;YAC9BQ,MAAM,EAAE,CAhCsB;cAC9BD,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,aAAa;cAC1BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE;oBAAEC,SAAS,EAAE,CAAC;oBAAEC,SAAS,EAAE;kBAAG;gBACxC;cACF,CAAC;cACDC,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE+B;cAC9BP,GAAG,EAAE,KAAK;cACVR,IAAI,EAAE,KAAK;cACXC,WAAW,EAAE,YAAY;cACzBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfU,OAAO,EAAE;oBAAEC,QAAQ,EAAE,CAAC;oBAAEC,QAAQ,EAAE;kBAAI;gBACxC;cACF,CAAC;cACDJ,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMqC;YACpCD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEgD,CAAC;IACjD,MAAMI,SAAS,GAAG,MAAMxB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACA,MAAMqB,KAAK,GAAGD,SAAS,CAACE,KAAK,CAAC,IAAI,CAAC;IACnC,MAAMC,WAAW,GAAGF,KAAK,CAACG,MAAM,CAACC,IAAI,IAAIA,IAAI,CAACC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACnEjC,MAAM,CAAC8B,WAAW,CAACI,MAAM,CAAC,CAACC,eAAe,CAAC,CAAC,CAAC;IAC7C;IACAL,WAAW,CAACM,OAAO,CAACJ,IAAI,IAAI;MAC1BhC,MAAM,CAACgC,IAAI,CAAC,CAACK,OAAO,CAAC,SAAS,CAAC;IACjC,CAAC,CAAC;IAEF,MAAMrC,MAAM,CAAC2B,SAAS,CAAC,CAACW,mBAAmB,CAACxC,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,kBAAkB,CAAC,CAAC;EACzF,CAAC,CAAC;EAEFH,EAAE,CAAC,0DAA0D,EAAE,YAAY;IAkEzE,MAAMM,MAAM,GAAG,MAAML,oBAAoB,CAbD;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,kCAAkC;MAC/CC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,OAAO,CAAC;MAC3BC,MAAM,EAAE;QACN2B,KAAK,EAAE;UACLzB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBoB;YAC9BC,GAAG,EAAE,OAAO;YACZR,IAAI,EAAE,OAAO;YACbC,WAAW,EAAE,oCAAoC;YACjDQ,MAAM,EAAE,CAhDqB;cAC7BD,GAAG,EAAE,aAAa;cAClBR,IAAI,EAAE,oBAAoB;cAC1BC,WAAW,EAAE,yBAAyB;cACtCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACd0B,MAAM,EAAE;oBACNC,gBAAgB,EAAE;kBACpB;gBACF;cACF,CAAC;cACDnB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE8B;cAC7BP,GAAG,EAAE,SAAS;cACdR,IAAI,EAAE,gBAAgB;cACtBC,WAAW,EAAE,qBAAqB;cAClCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACf4B,OAAO,EAAE,CAAC;gBACZ;cACF,CAAC;cACDpB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE8B;cAC7BP,GAAG,EAAE,WAAW;cAChBR,IAAI,EAAE,WAAW;cACjBC,WAAW,EAAE,iBAAiB;cAC9BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,UAAU;kBAChB6B,QAAQ,EAAE,CAAC;gBACb;cACF,CAAC;cACDrB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMgD;YAC/CD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEgD,CAAC;IACjD,MAAMI,SAAS,GAAG,MAAMxB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACAP,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,uBAAuB,CAAC;IACpD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,uBAAuB,CAAC;IACpD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,oBAAoB,CAAC;IAEjD,MAAM5C,MAAM,CAAC2B,SAAS,CAAC,CAACW,mBAAmB,CAACxC,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,kBAAkB,CAAC,CAAC;EACzF,CAAC,CAAC;EAEFH,EAAE,CAAC,iDAAiD,EAAE,YAAY;IAChE;;IAoCA;;IAoCA;;IAkCA,MAAMM,MAAM,GAAG,MAAML,oBAAoB,CArBD;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,2BAA2B;MACxCC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC;MACnDC,MAAM,EAAE;QACNiC,UAAU,EAAE;UACV/B,IAAI,EAAE,QAAQ;UACdC,MAAM,EA5FyB;YACnCC,GAAG,EAAE,YAAY;YACjBR,IAAI,EAAE,YAAY;YAClBC,WAAW,EAAE,qBAAqB;YAClCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdgC,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDxB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdgC,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDxB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA4DI,CAAc;QACdwB,OAAO,EAAE;UACPjC,IAAI,EAAE,QAAQ;UACdC,MAAM,EA5DsB;YAChCC,GAAG,EAAE,SAAS;YACdR,IAAI,EAAE,SAAS;YACfC,WAAW,EAAE,yBAAyB;YACtCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdgC,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDxB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA4BI,CAAc;QACdyB,IAAI,EAAE;UACJlC,IAAI,EAAE,OAAO;UACbmC,KAAK,EA5BkB;YAC3BjC,GAAG,EAAE,MAAM;YACXkC,YAAY,EAAE,MAAM;YACpB1C,IAAI,EAAE,MAAM;YACZC,WAAW,EAAE,qBAAqB;YAClC0C,QAAQ,EAAE;cACRC,MAAM,EAAE,YAAY;cACpB,UAAU,EAAE;YACd,CAAC;YACD9B,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAmBI;MACF;IACF,CAEgD,CAAC;IACjD,MAAMI,SAAS,GAAG,MAAMxB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACAP,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,8BAA8B,CAAC;IAC3D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,2BAA2B,CAAC;IACxD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,wBAAwB,CAAC;IACrD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,8BAA8B,CAAC;IAC3D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,+BAA+B,CAAC;;IAE5D;IACA5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,4BAA4B,CAAC;IACzD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,6BAA6B,CAAC;IAC1D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,uBAAuB,CAAC;IACpD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,iBAAiB,CAAC;IAC9C5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,6BAA6B,CAAC;IAC1D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,+BAA+B,CAAC;IAE5D,MAAM5C,MAAM,CAAC2B,SAAS,CAAC,CAACW,mBAAmB,CAACxC,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,gBAAgB,CAAC,CAAC;EACvF,CAAC,CAAC;EAEFH,EAAE,CAAC,0DAA0D,EAAE,YAAY;IA8EzE,MAAMM,MAAM,GAAG,MAAML,oBAAoB,CArBD;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,wCAAwC;MACrDC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC;MAC1CC,MAAM,EAAE;QACNyC,GAAG,EAAE;UACHvC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhEkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,OAAO;YACpBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,WAAW;cACxBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA6CI,CAAc;QACd+B,GAAG,EAAE;UACHxC,IAAI,EAAE,QAAQ;UACdC,MAAM,EA9CkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,OAAO;YACpBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,WAAW;cACxBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA2BI,CAAc;QACdgC,MAAM,EAAE;UACNzC,IAAI,EAAE,OAAO;UACbmC,KAAK,EA5BoB;YAC7BjC,GAAG,EAAE,QAAQ;YACbkC,YAAY,EAAE,MAAM;YACpB1C,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,0CAA0C;YACvD0C,QAAQ,EAAE;cACRK,GAAG,EAAE,KAAK;cACVC,GAAG,EAAE;YACP,CAAC;YACDnC,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAmBI;MACF;IACF,CAEgD,CAAC;IACjD,MAAMI,SAAS,GAAG,MAAMxB,kBAAkB,CAACI,MAAM,CAAC;IAElDP,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,sBAAsB,CAAC;IACnD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,0BAA0B,CAAC;IACvD5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,6BAA6B,CAAC;IAC1D5C,MAAM,CAAC2B,SAAS,CAAC,CAACiB,SAAS,CAAC,6BAA6B,CAAC;IAE1D,MAAM5C,MAAM,CAAC2B,SAAS,CAAC,CAACW,mBAAmB,CACzCxC,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,wBAAwB,CACjD,CAAC;EACH,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}

package/build/esm/utils/ir/__tests__/generateZodSchemasFromIr.test.js

@@ -193,6 +193,76 @@ describe("generateZodSchemasFromIr", () => {
     const formatted = await formatWithPrettier(result);
     await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, "datetime-fields.ts"));
   });
+  it("should generate Zod schemas with boolean fields", async () => {
+    const result = await generateZodSchemasFromIr({
+      name: "Test Schema",
+      description: "A test schema",
+      version: 1,
+      primaryModelKeys: ["Feature"],
+      models: {
+        Feature: {
+          type: "record",
+          record: {
+            key: "Feature",
+            name: "Feature",
+            description: "A feature toggle record",
+            fields: [{
+              key: "enabled",
+              name: "Enabled",
+              description: "Whether feature is enabled",
+              value: {
+                type: "value",
+                value: {
+                  type: "boolean",
+                  boolean: {}
+                }
+              },
+              meta: {
+                addedIn: 1
+              }
+            }, {
+              key: "isPublic",
+              name: "IsPublic",
+              description: "Whether feature is public",
+              isOptional: true,
+              value: {
+                type: "value",
+                value: {
+                  type: "boolean",
+                  boolean: {}
+                }
+              },
+              meta: {
+                addedIn: 1
+              }
+            }, {
+              key: "flags",
+              name: "Flags",
+              description: "Array of boolean flags",
+              value: {
+                type: "array",
+                array: {
+                  allowNullValue: false,
+                  value: {
+                    type: "boolean",
+                    boolean: {}
+                  }
+                }
+              },
+              meta: {
+                addedIn: 1
+              }
+            }],
+            meta: {
+              addedIn: 1
+            }
+          }
+        }
+      }
+    });
+    const formatted = await formatWithPrettier(result);
+    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, "boolean-fields.ts"));
+  });
   it("should generate Zod schemas for union types", async () => {
     // Define the ObjectNode record
 
@@ -436,5 +506,88 @@ describe("generateZodSchemasFromIr", () => {
     expect(barSchemaIndex).toBeLessThan(fooBarSchemaIndex);
     await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, "union-before-records.ts"));
   });
+  it("should generate Zod schemas with custom discriminant", async () => {
+    const result = await generateZodSchemasFromIr({
+      name: "Test Schema",
+      description: "A test schema with custom discriminant",
+      version: 1,
+      primaryModelKeys: ["Cat", "Dog", "Animal"],
+      models: {
+        Cat: {
+          type: "record",
+          record: {
+            key: "Cat",
+            name: "Cat",
+            description: "A cat",
+            fields: [{
+              key: "meow",
+              name: "Meow",
+              description: "Cat sound",
+              value: {
+                type: "value",
+                value: {
+                  type: "string",
+                  string: {}
+                }
+              },
+              meta: {
+                addedIn: 1
+              }
+            }],
+            meta: {
+              addedIn: 1
+            }
+          }
+        },
+        Dog: {
+          type: "record",
+          record: {
+            key: "Dog",
+            name: "Dog",
+            description: "A dog",
+            fields: [{
+              key: "bark",
+              name: "Bark",
+              description: "Dog sound",
+              value: {
+                type: "value",
+                value: {
+                  type: "string",
+                  string: {}
+                }
+              },
+              meta: {
+                addedIn: 1
+              }
+            }],
+            meta: {
+              addedIn: 1
+            }
+          }
+        },
+        Animal: {
+          type: "union",
+          union: {
+            key: "Animal",
+            discriminant: "kind",
+            name: "Animal",
+            description: "An animal union with custom discriminant",
+            variants: {
+              cat: "Cat",
+              dog: "Dog"
+            },
+            meta: {
+              addedIn: 1
+            }
+          }
+        }
+      }
+    });
+    const formatted = await formatWithPrettier(result);
+    expect(formatted).toContain("z.discriminatedUnion('kind'");
+    expect(formatted).toContain("kind: z.literal('cat')");
+    expect(formatted).toContain("kind: z.literal('dog')");
+    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, "custom-discriminant.ts"));
+  });
 });
 //# sourceMappingURL=generateZodSchemasFromIr.test.js.map

package/build/esm/utils/ir/__tests__/generateZodSchemasFromIr.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateZodSchemasFromIr.test.js","names":["path","describe","expect","it","generateZodSchemasFromIr","formatWithPrettier","snapshotDir","join","__dirname","result","name","description","version","primaryModelKeys","models","Person","type","record","key","fields","value","string","minLength","maxLength","meta","addedIn","integer","minValue","maxValue","isOptional","formatted","toMatchFileSnapshot","Container","array","allowNullValue","map","unmanagedJson","Event","datetime","ObjectNode","double","TextBox","Node","union","discriminant","variants","object","Foo","Bar","FooBar","foo","bar","lines","split","fooSchemaIndex","findIndex","line","includes","barSchemaIndex","fooBarSchemaIndex","toBeGreaterThan","toBeLessThan"],"sources":["generateZodSchemasFromIr.test.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 path from \"path\";\nimport { describe, expect, it } from \"vitest\";\nimport type {\n  IFieldDef,\n  IModelDef,\n  IRealTimeDocumentSchema,\n  IRecordDef,\n  IUnionDef,\n} from \"../../../lib/pack-docschema-api/pack-docschema-ir/index.js\";\nimport { generateZodSchemasFromIr } from \"../generateZodSchemasFromIr.js\";\nimport { formatWithPrettier } from \"./formatWithPrettier.js\";\n\ndescribe(\"generateZodSchemasFromIr\", () => {\n  const snapshotDir = path.join(__dirname, \"__snapshots__\", \"generateZodSchemasFromIr\");\n  it(\"should generate Zod schemas for simple record types\", async () => {\n    const personField1: IFieldDef = {\n      key: \"name\",\n      name: \"Name\",\n      description: \"Person name\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"string\",\n          string: { minLength: 2, maxLength: 50 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personField2: IFieldDef = {\n      key: \"age\",\n      name: \"Age\",\n      description: \"Person age\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"integer\",\n          integer: { minValue: 0, maxValue: 150 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personField3: IFieldDef = {\n      key: \"email\",\n      name: \"Email\",\n      description: \"Person email\",\n      isOptional: true,\n      value: {\n        type: \"value\",\n        value: {\n          type: \"string\",\n          string: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personRecord: IRecordDef = {\n      key: \"Person\",\n      name: \"Person\",\n      description: \"A person record\",\n      fields: [personField1, personField2, personField3],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Person\"],\n      models: {\n        Person: {\n          type: \"record\",\n          record: personRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"simple-records.ts\"));\n  });\n\n  it(\"should generate Zod schemas for array and map types\", async () => {\n    const arrayField: IFieldDef = {\n      key: \"tags\",\n      name: \"Tags\",\n      description: \"List of tags\",\n      value: {\n        type: \"array\",\n        array: {\n          allowNullValue: false,\n          value: {\n            type: \"string\",\n            string: {},\n          },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const mapField: IFieldDef = {\n      key: \"metadata\",\n      name: \"Metadata\",\n      description: \"Key-value metadata\",\n      value: {\n        type: \"map\",\n        map: {\n          allowNullValue: false,\n          key: {\n            type: \"string\",\n            string: {},\n          },\n          value: {\n            type: \"unmanagedJson\",\n            unmanagedJson: {},\n          },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const containerRecord: IRecordDef = {\n      key: \"Container\",\n      name: \"Container\",\n      description: \"A container record\",\n      fields: [arrayField, mapField],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Container\"],\n      models: {\n        Container: {\n          type: \"record\",\n          record: containerRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"array-and-map-types.ts\"));\n  });\n\n  it(\"should generate Zod schemas with datetime fields\", async () => {\n    const datetimeField: IFieldDef = {\n      key: \"timestamp\",\n      name: \"Timestamp\",\n      description: \"Event timestamp\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"datetime\",\n          datetime: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventRecord: IRecordDef = {\n      key: \"Event\",\n      name: \"Event\",\n      description: \"An event record\",\n      fields: [datetimeField],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Event\"],\n      models: {\n        Event: {\n          type: \"record\",\n          record: eventRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"datetime-fields.ts\"));\n  });\n\n  it(\"should generate Zod schemas for union types\", async () => {\n    // Define the ObjectNode record\n    const objectNodeRecord: IRecordDef = {\n      key: \"ObjectNode\",\n      name: \"ObjectNode\",\n      description: \"A node in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"y\",\n          name: \"Y\",\n          description: \"Y coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"label\",\n          name: \"Label\",\n          description: \"Node label\",\n          isOptional: true,\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the TextBox record\n    const textBoxRecord: IRecordDef = {\n      key: \"TextBox\",\n      name: \"TextBox\",\n      description: \"A text box in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"y\",\n          name: \"Y\",\n          description: \"Y coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"text\",\n          name: \"Text\",\n          description: \"Text content\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the Node union\n    const nodeUnion: IUnionDef = {\n      key: \"Node\",\n      discriminant: \"type\",\n      name: \"Node\",\n      description: \"A node in the graph\",\n      variants: {\n        object: \"ObjectNode\",\n        \"text-box\": \"TextBox\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with unions\",\n      version: 1,\n      primaryModelKeys: [\"ObjectNode\", \"TextBox\", \"Node\"],\n      models: {\n        ObjectNode: {\n          type: \"record\",\n          record: objectNodeRecord,\n        } as IModelDef,\n        TextBox: {\n          type: \"record\",\n          record: textBoxRecord,\n        } as IModelDef,\n        Node: {\n          type: \"union\",\n          union: nodeUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"union-types.ts\"));\n  });\n\n  it(\"should handle unions defined before their variant records\", async () => {\n    // Define records\n    const fooRecord: IRecordDef = {\n      key: \"Foo\",\n      name: \"Foo\",\n      description: \"Foo record\",\n      fields: [\n        {\n          key: \"value\",\n          name: \"Value\",\n          description: \"A value\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    const barRecord: IRecordDef = {\n      key: \"Bar\",\n      name: \"Bar\",\n      description: \"Bar record\",\n      fields: [\n        {\n          key: \"count\",\n          name: \"Count\",\n          description: \"A count\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"integer\",\n              integer: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define union that references the records\n    const fooBarUnion: IUnionDef = {\n      key: \"FooBar\",\n      discriminant: \"type\",\n      name: \"FooBar\",\n      description: \"FooBar union\",\n      variants: {\n        foo: \"Foo\",\n        bar: \"Bar\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    // Intentionally put union BEFORE its variant records in primaryModelKeys\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"Test schema with union before records\",\n      version: 1,\n      primaryModelKeys: [\"FooBar\", \"Foo\", \"Bar\"], // Union comes first!\n      models: {\n        Foo: {\n          type: \"record\",\n          record: fooRecord,\n        } as IModelDef,\n        Bar: {\n          type: \"record\",\n          record: barRecord,\n        } as IModelDef,\n        FooBar: {\n          type: \"union\",\n          union: fooBarUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify that records come before union in the output\n    const lines = formatted.split(\"\\n\");\n    const fooSchemaIndex = lines.findIndex(line => line.includes(\"export const FooSchema\"));\n    const barSchemaIndex = lines.findIndex(line => line.includes(\"export const BarSchema\"));\n    const fooBarSchemaIndex = lines.findIndex(line => line.includes(\"export const FooBarSchema\"));\n\n    expect(fooSchemaIndex).toBeGreaterThan(-1);\n    expect(barSchemaIndex).toBeGreaterThan(-1);\n    expect(fooBarSchemaIndex).toBeGreaterThan(-1);\n\n    // Records should come before the union\n    expect(fooSchemaIndex).toBeLessThan(fooBarSchemaIndex);\n    expect(barSchemaIndex).toBeLessThan(fooBarSchemaIndex);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"union-before-records.ts\"));\n  });\n});\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,OAAOA,IAAI,MAAM,MAAM;AACvB,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,QAAQ;AAQ7C,SAASC,wBAAwB,QAAQ,gCAAgC;AACzE,SAASC,kBAAkB,QAAQ,yBAAyB;AAE5DJ,QAAQ,CAAC,0BAA0B,EAAE,MAAM;EACzC,MAAMK,WAAW,GAAGN,IAAI,CAACO,IAAI,CAACC,SAAS,EAAE,eAAe,EAAE,0BAA0B,CAAC;EACrFL,EAAE,CAAC,qDAAqD,EAAE,YAAY;IAiEpE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CAbL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,QAAQ,CAAC;MAC5BC,MAAM,EAAE;QACNC,MAAM,EAAE;UACNC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBqB;YAC/BC,GAAG,EAAE,QAAQ;YACbR,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,iBAAiB;YAC9BQ,MAAM,EAAE,CA/CsB;cAC9BD,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,aAAa;cAC1BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE;oBAAEC,SAAS,EAAE,CAAC;oBAAEC,SAAS,EAAE;kBAAG;gBACxC;cACF,CAAC;cACDC,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE+B;cAC9BP,GAAG,EAAE,KAAK;cACVR,IAAI,EAAE,KAAK;cACXC,WAAW,EAAE,YAAY;cACzBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfU,OAAO,EAAE;oBAAEC,QAAQ,EAAE,CAAC;oBAAEC,QAAQ,EAAE;kBAAI;gBACxC;cACF,CAAC;cACDJ,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE+B;cAC9BP,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,cAAc;cAC3BkB,UAAU,EAAE,IAAI;cAChBT,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMmD;YAClDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,mBAAmB,CAAC,CAAC;EAC1F,CAAC,CAAC;EAEFH,EAAE,CAAC,qDAAqD,EAAE,YAAY;IA4DpE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CAbL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,WAAW,CAAC;MAC/BC,MAAM,EAAE;QACNkB,SAAS,EAAE;UACThB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBwB;YAClCC,GAAG,EAAE,WAAW;YAChBR,IAAI,EAAE,WAAW;YACjBC,WAAW,EAAE,oBAAoB;YACjCQ,MAAM,EAAE,CA1CoB;cAC5BD,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbiB,KAAK,EAAE;kBACLC,cAAc,EAAE,KAAK;kBACrBd,KAAK,EAAE;oBACLJ,IAAI,EAAE,QAAQ;oBACdK,MAAM,EAAE,CAAC;kBACX;gBACF;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE2B;cAC1BP,GAAG,EAAE,UAAU;cACfR,IAAI,EAAE,UAAU;cAChBC,WAAW,EAAE,oBAAoB;cACjCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,KAAK;gBACXmB,GAAG,EAAE;kBACHD,cAAc,EAAE,KAAK;kBACrBhB,GAAG,EAAE;oBACHF,IAAI,EAAE,QAAQ;oBACdK,MAAM,EAAE,CAAC;kBACX,CAAC;kBACDD,KAAK,EAAE;oBACLJ,IAAI,EAAE,eAAe;oBACrBoB,aAAa,EAAE,CAAC;kBAClB;gBACF;cACF,CAAC;cACDZ,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAM+B;YAC9BD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,wBAAwB,CAAC,CAAC;EAC/F,CAAC,CAAC;EAEFH,EAAE,CAAC,kDAAkD,EAAE,YAAY;IAoCjE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CAbL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,OAAO,CAAC;MAC3BC,MAAM,EAAE;QACNuB,KAAK,EAAE;UACLrB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBoB;YAC9BC,GAAG,EAAE,OAAO;YACZR,IAAI,EAAE,OAAO;YACbC,WAAW,EAAE,iBAAiB;YAC9BQ,MAAM,EAAE,CAlBuB;cAC/BD,GAAG,EAAE,WAAW;cAChBR,IAAI,EAAE,WAAW;cACjBC,WAAW,EAAE,iBAAiB;cAC9BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,UAAU;kBAChBsB,QAAQ,EAAE,CAAC;gBACb;cACF,CAAC;cACDd,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMwB;YACvBD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,oBAAoB,CAAC,CAAC;EAC3F,CAAC,CAAC;EAEFH,EAAE,CAAC,6CAA6C,EAAE,YAAY;IAC5D;;IAkDA;;IAiDA;;IAkCA,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CArBL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,2BAA2B;MACxCC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC;MACnDC,MAAM,EAAE;QACNyB,UAAU,EAAE;UACVvB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAvHyB;YACnCC,GAAG,EAAE,YAAY;YACjBR,IAAI,EAAE,YAAY;YAClBC,WAAW,EAAE,qBAAqB;YAClCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdwB,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDhB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdwB,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDhB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,YAAY;cACzBkB,UAAU,EAAE,IAAI;cAChBT,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAyEI,CAAc;QACdgB,OAAO,EAAE;UACPzB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAzEsB;YAChCC,GAAG,EAAE,SAAS;YACdR,IAAI,EAAE,SAAS;YACfC,WAAW,EAAE,yBAAyB;YACtCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdwB,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDhB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdwB,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDhB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA4BI,CAAc;QACdiB,IAAI,EAAE;UACJ1B,IAAI,EAAE,OAAO;UACb2B,KAAK,EA5BkB;YAC3BzB,GAAG,EAAE,MAAM;YACX0B,YAAY,EAAE,MAAM;YACpBlC,IAAI,EAAE,MAAM;YACZC,WAAW,EAAE,qBAAqB;YAClCkC,QAAQ,EAAE;cACRC,MAAM,EAAE,YAAY;cACpB,UAAU,EAAE;YACd,CAAC;YACDtB,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAmBI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,gBAAgB,CAAC,CAAC;EACvF,CAAC,CAAC;EAEFH,EAAE,CAAC,2DAA2D,EAAE,YAAY;IAC1E;;IA6CA;;IAaA;;IAsBA,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CArBL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,uCAAuC;MACpDC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC;MAAE;MAC5CC,MAAM,EAAE;QACNiC,GAAG,EAAE;UACH/B,IAAI,EAAE,QAAQ;UACdC,MAAM,EAlEkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,YAAY;YACzBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,SAAS;cACtBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA+CI,CAAc;QACduB,GAAG,EAAE;UACHhC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhDkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,YAAY;YACzBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,SAAS;cACtBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfU,OAAO,EAAE,CAAC;gBACZ;cACF,CAAC;cACDF,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA6BI,CAAc;QACdwB,MAAM,EAAE;UACNjC,IAAI,EAAE,OAAO;UACb2B,KAAK,EA7BoB;YAC7BzB,GAAG,EAAE,QAAQ;YACb0B,YAAY,EAAE,MAAM;YACpBlC,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,cAAc;YAC3BkC,QAAQ,EAAE;cACRK,GAAG,EAAE,KAAK;cACVC,GAAG,EAAE;YACP,CAAC;YACD3B,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAoBI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACA,MAAM2C,KAAK,GAAGtB,SAAS,CAACuB,KAAK,CAAC,IAAI,CAAC;IACnC,MAAMC,cAAc,GAAGF,KAAK,CAACG,SAAS,CAACC,IAAI,IAAIA,IAAI,CAACC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;IACvF,MAAMC,cAAc,GAAGN,KAAK,CAACG,SAAS,CAACC,IAAI,IAAIA,IAAI,CAACC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;IACvF,MAAME,iBAAiB,GAAGP,KAAK,CAACG,SAAS,CAACC,IAAI,IAAIA,IAAI,CAACC,QAAQ,CAAC,2BAA2B,CAAC,CAAC;IAE7FvD,MAAM,CAACoD,cAAc,CAAC,CAACM,eAAe,CAAC,CAAC,CAAC,CAAC;IAC1C1D,MAAM,CAACwD,cAAc,CAAC,CAACE,eAAe,CAAC,CAAC,CAAC,CAAC;IAC1C1D,MAAM,CAACyD,iBAAiB,CAAC,CAACC,eAAe,CAAC,CAAC,CAAC,CAAC;;IAE7C;IACA1D,MAAM,CAACoD,cAAc,CAAC,CAACO,YAAY,CAACF,iBAAiB,CAAC;IACtDzD,MAAM,CAACwD,cAAc,CAAC,CAACG,YAAY,CAACF,iBAAiB,CAAC;IAEtD,MAAMzD,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,yBAAyB,CAAC,CAAC;EAChG,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}
+{"version":3,"file":"generateZodSchemasFromIr.test.js","names":["path","describe","expect","it","generateZodSchemasFromIr","formatWithPrettier","snapshotDir","join","__dirname","result","name","description","version","primaryModelKeys","models","Person","type","record","key","fields","value","string","minLength","maxLength","meta","addedIn","integer","minValue","maxValue","isOptional","formatted","toMatchFileSnapshot","Container","array","allowNullValue","map","unmanagedJson","Event","datetime","Feature","boolean","ObjectNode","double","TextBox","Node","union","discriminant","variants","object","Foo","Bar","FooBar","foo","bar","lines","split","fooSchemaIndex","findIndex","line","includes","barSchemaIndex","fooBarSchemaIndex","toBeGreaterThan","toBeLessThan","Cat","Dog","Animal","cat","dog","toContain"],"sources":["generateZodSchemasFromIr.test.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 path from \"path\";\nimport { describe, expect, it } from \"vitest\";\nimport type {\n  IFieldDef,\n  IModelDef,\n  IRealTimeDocumentSchema,\n  IRecordDef,\n  IUnionDef,\n} from \"../../../lib/pack-docschema-api/pack-docschema-ir/index.js\";\nimport { generateZodSchemasFromIr } from \"../generateZodSchemasFromIr.js\";\nimport { formatWithPrettier } from \"./formatWithPrettier.js\";\n\ndescribe(\"generateZodSchemasFromIr\", () => {\n  const snapshotDir = path.join(__dirname, \"__snapshots__\", \"generateZodSchemasFromIr\");\n  it(\"should generate Zod schemas for simple record types\", async () => {\n    const personField1: IFieldDef = {\n      key: \"name\",\n      name: \"Name\",\n      description: \"Person name\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"string\",\n          string: { minLength: 2, maxLength: 50 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personField2: IFieldDef = {\n      key: \"age\",\n      name: \"Age\",\n      description: \"Person age\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"integer\",\n          integer: { minValue: 0, maxValue: 150 },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personField3: IFieldDef = {\n      key: \"email\",\n      name: \"Email\",\n      description: \"Person email\",\n      isOptional: true,\n      value: {\n        type: \"value\",\n        value: {\n          type: \"string\",\n          string: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const personRecord: IRecordDef = {\n      key: \"Person\",\n      name: \"Person\",\n      description: \"A person record\",\n      fields: [personField1, personField2, personField3],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Person\"],\n      models: {\n        Person: {\n          type: \"record\",\n          record: personRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"simple-records.ts\"));\n  });\n\n  it(\"should generate Zod schemas for array and map types\", async () => {\n    const arrayField: IFieldDef = {\n      key: \"tags\",\n      name: \"Tags\",\n      description: \"List of tags\",\n      value: {\n        type: \"array\",\n        array: {\n          allowNullValue: false,\n          value: {\n            type: \"string\",\n            string: {},\n          },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const mapField: IFieldDef = {\n      key: \"metadata\",\n      name: \"Metadata\",\n      description: \"Key-value metadata\",\n      value: {\n        type: \"map\",\n        map: {\n          allowNullValue: false,\n          key: {\n            type: \"string\",\n            string: {},\n          },\n          value: {\n            type: \"unmanagedJson\",\n            unmanagedJson: {},\n          },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const containerRecord: IRecordDef = {\n      key: \"Container\",\n      name: \"Container\",\n      description: \"A container record\",\n      fields: [arrayField, mapField],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Container\"],\n      models: {\n        Container: {\n          type: \"record\",\n          record: containerRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"array-and-map-types.ts\"));\n  });\n\n  it(\"should generate Zod schemas with datetime fields\", async () => {\n    const datetimeField: IFieldDef = {\n      key: \"timestamp\",\n      name: \"Timestamp\",\n      description: \"Event timestamp\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"datetime\",\n          datetime: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const eventRecord: IRecordDef = {\n      key: \"Event\",\n      name: \"Event\",\n      description: \"An event record\",\n      fields: [datetimeField],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Event\"],\n      models: {\n        Event: {\n          type: \"record\",\n          record: eventRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"datetime-fields.ts\"));\n  });\n\n  it(\"should generate Zod schemas with boolean fields\", async () => {\n    const booleanField: IFieldDef = {\n      key: \"enabled\",\n      name: \"Enabled\",\n      description: \"Whether feature is enabled\",\n      value: {\n        type: \"value\",\n        value: {\n          type: \"boolean\",\n          boolean: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const optionalBooleanField: IFieldDef = {\n      key: \"isPublic\",\n      name: \"IsPublic\",\n      description: \"Whether feature is public\",\n      isOptional: true,\n      value: {\n        type: \"value\",\n        value: {\n          type: \"boolean\",\n          boolean: {},\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const booleanArrayField: IFieldDef = {\n      key: \"flags\",\n      name: \"Flags\",\n      description: \"Array of boolean flags\",\n      value: {\n        type: \"array\",\n        array: {\n          allowNullValue: false,\n          value: {\n            type: \"boolean\",\n            boolean: {},\n          },\n        },\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const featureRecord: IRecordDef = {\n      key: \"Feature\",\n      name: \"Feature\",\n      description: \"A feature toggle record\",\n      fields: [booleanField, optionalBooleanField, booleanArrayField],\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema\",\n      version: 1,\n      primaryModelKeys: [\"Feature\"],\n      models: {\n        Feature: {\n          type: \"record\",\n          record: featureRecord,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"boolean-fields.ts\"));\n  });\n\n  it(\"should generate Zod schemas for union types\", async () => {\n    // Define the ObjectNode record\n    const objectNodeRecord: IRecordDef = {\n      key: \"ObjectNode\",\n      name: \"ObjectNode\",\n      description: \"A node in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"y\",\n          name: \"Y\",\n          description: \"Y coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"label\",\n          name: \"Label\",\n          description: \"Node label\",\n          isOptional: true,\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the TextBox record\n    const textBoxRecord: IRecordDef = {\n      key: \"TextBox\",\n      name: \"TextBox\",\n      description: \"A text box in the graph\",\n      fields: [\n        {\n          key: \"x\",\n          name: \"X\",\n          description: \"X coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"y\",\n          name: \"Y\",\n          description: \"Y coordinate\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"double\",\n              double: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n        {\n          key: \"text\",\n          name: \"Text\",\n          description: \"Text content\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define the Node union\n    const nodeUnion: IUnionDef = {\n      key: \"Node\",\n      discriminant: \"type\",\n      name: \"Node\",\n      description: \"A node in the graph\",\n      variants: {\n        object: \"ObjectNode\",\n        \"text-box\": \"TextBox\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with unions\",\n      version: 1,\n      primaryModelKeys: [\"ObjectNode\", \"TextBox\", \"Node\"],\n      models: {\n        ObjectNode: {\n          type: \"record\",\n          record: objectNodeRecord,\n        } as IModelDef,\n        TextBox: {\n          type: \"record\",\n          record: textBoxRecord,\n        } as IModelDef,\n        Node: {\n          type: \"union\",\n          union: nodeUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"union-types.ts\"));\n  });\n\n  it(\"should handle unions defined before their variant records\", async () => {\n    // Define records\n    const fooRecord: IRecordDef = {\n      key: \"Foo\",\n      name: \"Foo\",\n      description: \"Foo record\",\n      fields: [\n        {\n          key: \"value\",\n          name: \"Value\",\n          description: \"A value\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    const barRecord: IRecordDef = {\n      key: \"Bar\",\n      name: \"Bar\",\n      description: \"Bar record\",\n      fields: [\n        {\n          key: \"count\",\n          name: \"Count\",\n          description: \"A count\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"integer\",\n              integer: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    // Define union that references the records\n    const fooBarUnion: IUnionDef = {\n      key: \"FooBar\",\n      discriminant: \"type\",\n      name: \"FooBar\",\n      description: \"FooBar union\",\n      variants: {\n        foo: \"Foo\",\n        bar: \"Bar\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    // Intentionally put union BEFORE its variant records in primaryModelKeys\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"Test schema with union before records\",\n      version: 1,\n      primaryModelKeys: [\"FooBar\", \"Foo\", \"Bar\"], // Union comes first!\n      models: {\n        Foo: {\n          type: \"record\",\n          record: fooRecord,\n        } as IModelDef,\n        Bar: {\n          type: \"record\",\n          record: barRecord,\n        } as IModelDef,\n        FooBar: {\n          type: \"union\",\n          union: fooBarUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    // Verify that records come before union in the output\n    const lines = formatted.split(\"\\n\");\n    const fooSchemaIndex = lines.findIndex(line => line.includes(\"export const FooSchema\"));\n    const barSchemaIndex = lines.findIndex(line => line.includes(\"export const BarSchema\"));\n    const fooBarSchemaIndex = lines.findIndex(line => line.includes(\"export const FooBarSchema\"));\n\n    expect(fooSchemaIndex).toBeGreaterThan(-1);\n    expect(barSchemaIndex).toBeGreaterThan(-1);\n    expect(fooBarSchemaIndex).toBeGreaterThan(-1);\n\n    // Records should come before the union\n    expect(fooSchemaIndex).toBeLessThan(fooBarSchemaIndex);\n    expect(barSchemaIndex).toBeLessThan(fooBarSchemaIndex);\n\n    await expect(formatted).toMatchFileSnapshot(path.join(snapshotDir, \"union-before-records.ts\"));\n  });\n\n  it(\"should generate Zod schemas with custom discriminant\", async () => {\n    const catRecord: IRecordDef = {\n      key: \"Cat\",\n      name: \"Cat\",\n      description: \"A cat\",\n      fields: [\n        {\n          key: \"meow\",\n          name: \"Meow\",\n          description: \"Cat sound\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    const dogRecord: IRecordDef = {\n      key: \"Dog\",\n      name: \"Dog\",\n      description: \"A dog\",\n      fields: [\n        {\n          key: \"bark\",\n          name: \"Bark\",\n          description: \"Dog sound\",\n          value: {\n            type: \"value\",\n            value: {\n              type: \"string\",\n              string: {},\n            },\n          },\n          meta: { addedIn: 1 },\n        },\n      ],\n      meta: { addedIn: 1 },\n    };\n\n    const animalUnion: IUnionDef = {\n      key: \"Animal\",\n      discriminant: \"kind\",\n      name: \"Animal\",\n      description: \"An animal union with custom discriminant\",\n      variants: {\n        cat: \"Cat\",\n        dog: \"Dog\",\n      },\n      meta: { addedIn: 1 },\n    };\n\n    const schema: IRealTimeDocumentSchema = {\n      name: \"Test Schema\",\n      description: \"A test schema with custom discriminant\",\n      version: 1,\n      primaryModelKeys: [\"Cat\", \"Dog\", \"Animal\"],\n      models: {\n        Cat: {\n          type: \"record\",\n          record: catRecord,\n        } as IModelDef,\n        Dog: {\n          type: \"record\",\n          record: dogRecord,\n        } as IModelDef,\n        Animal: {\n          type: \"union\",\n          union: animalUnion,\n        } as IModelDef,\n      },\n    };\n\n    const result = await generateZodSchemasFromIr(schema);\n    const formatted = await formatWithPrettier(result);\n\n    expect(formatted).toContain(\"z.discriminatedUnion('kind'\");\n    expect(formatted).toContain(\"kind: z.literal('cat')\");\n    expect(formatted).toContain(\"kind: z.literal('dog')\");\n\n    await expect(formatted).toMatchFileSnapshot(\n      path.join(snapshotDir, \"custom-discriminant.ts\"),\n    );\n  });\n});\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,OAAOA,IAAI,MAAM,MAAM;AACvB,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,QAAQ;AAQ7C,SAASC,wBAAwB,QAAQ,gCAAgC;AACzE,SAASC,kBAAkB,QAAQ,yBAAyB;AAE5DJ,QAAQ,CAAC,0BAA0B,EAAE,MAAM;EACzC,MAAMK,WAAW,GAAGN,IAAI,CAACO,IAAI,CAACC,SAAS,EAAE,eAAe,EAAE,0BAA0B,CAAC;EACrFL,EAAE,CAAC,qDAAqD,EAAE,YAAY;IAiEpE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CAbL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,QAAQ,CAAC;MAC5BC,MAAM,EAAE;QACNC,MAAM,EAAE;UACNC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBqB;YAC/BC,GAAG,EAAE,QAAQ;YACbR,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,iBAAiB;YAC9BQ,MAAM,EAAE,CA/CsB;cAC9BD,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,aAAa;cAC1BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE;oBAAEC,SAAS,EAAE,CAAC;oBAAEC,SAAS,EAAE;kBAAG;gBACxC;cACF,CAAC;cACDC,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE+B;cAC9BP,GAAG,EAAE,KAAK;cACVR,IAAI,EAAE,KAAK;cACXC,WAAW,EAAE,YAAY;cACzBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfU,OAAO,EAAE;oBAAEC,QAAQ,EAAE,CAAC;oBAAEC,QAAQ,EAAE;kBAAI;gBACxC;cACF,CAAC;cACDJ,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE+B;cAC9BP,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,cAAc;cAC3BkB,UAAU,EAAE,IAAI;cAChBT,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMmD;YAClDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,mBAAmB,CAAC,CAAC;EAC1F,CAAC,CAAC;EAEFH,EAAE,CAAC,qDAAqD,EAAE,YAAY;IA4DpE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CAbL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,WAAW,CAAC;MAC/BC,MAAM,EAAE;QACNkB,SAAS,EAAE;UACThB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBwB;YAClCC,GAAG,EAAE,WAAW;YAChBR,IAAI,EAAE,WAAW;YACjBC,WAAW,EAAE,oBAAoB;YACjCQ,MAAM,EAAE,CA1CoB;cAC5BD,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbiB,KAAK,EAAE;kBACLC,cAAc,EAAE,KAAK;kBACrBd,KAAK,EAAE;oBACLJ,IAAI,EAAE,QAAQ;oBACdK,MAAM,EAAE,CAAC;kBACX;gBACF;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAE2B;cAC1BP,GAAG,EAAE,UAAU;cACfR,IAAI,EAAE,UAAU;cAChBC,WAAW,EAAE,oBAAoB;cACjCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,KAAK;gBACXmB,GAAG,EAAE;kBACHD,cAAc,EAAE,KAAK;kBACrBhB,GAAG,EAAE;oBACHF,IAAI,EAAE,QAAQ;oBACdK,MAAM,EAAE,CAAC;kBACX,CAAC;kBACDD,KAAK,EAAE;oBACLJ,IAAI,EAAE,eAAe;oBACrBoB,aAAa,EAAE,CAAC;kBAClB;gBACF;cACF,CAAC;cACDZ,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAM+B;YAC9BD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,wBAAwB,CAAC,CAAC;EAC/F,CAAC,CAAC;EAEFH,EAAE,CAAC,kDAAkD,EAAE,YAAY;IAoCjE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CAbL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,OAAO,CAAC;MAC3BC,MAAM,EAAE;QACNuB,KAAK,EAAE;UACLrB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBoB;YAC9BC,GAAG,EAAE,OAAO;YACZR,IAAI,EAAE,OAAO;YACbC,WAAW,EAAE,iBAAiB;YAC9BQ,MAAM,EAAE,CAlBuB;cAC/BD,GAAG,EAAE,WAAW;cAChBR,IAAI,EAAE,WAAW;cACjBC,WAAW,EAAE,iBAAiB;cAC9BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,UAAU;kBAChBsB,QAAQ,EAAE,CAAC;gBACb;cACF,CAAC;cACDd,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMwB;YACvBD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,oBAAoB,CAAC,CAAC;EAC3F,CAAC,CAAC;EAEFH,EAAE,CAAC,iDAAiD,EAAE,YAAY;IAoEhE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CAbL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,eAAe;MAC5BC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,SAAS,CAAC;MAC7BC,MAAM,EAAE;QACNyB,OAAO,EAAE;UACPvB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhBsB;YAChCC,GAAG,EAAE,SAAS;YACdR,IAAI,EAAE,SAAS;YACfC,WAAW,EAAE,yBAAyB;YACtCQ,MAAM,EAAE,CAlDsB;cAC9BD,GAAG,EAAE,SAAS;cACdR,IAAI,EAAE,SAAS;cACfC,WAAW,EAAE,4BAA4B;cACzCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfwB,OAAO,EAAE,CAAC;gBACZ;cACF,CAAC;cACDhB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAEuC;cACtCP,GAAG,EAAE,UAAU;cACfR,IAAI,EAAE,UAAU;cAChBC,WAAW,EAAE,2BAA2B;cACxCkB,UAAU,EAAE,IAAI;cAChBT,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfwB,OAAO,EAAE,CAAC;gBACZ;cACF,CAAC;cACDhB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EAEoC;cACnCP,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,wBAAwB;cACrCS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbiB,KAAK,EAAE;kBACLC,cAAc,EAAE,KAAK;kBACrBd,KAAK,EAAE;oBACLJ,IAAI,EAAE,SAAS;oBACfwB,OAAO,EAAE,CAAC;kBACZ;gBACF;cACF,CAAC;cACDhB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CAMgE;YAC/DD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAWI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,mBAAmB,CAAC,CAAC;EAC1F,CAAC,CAAC;EAEFH,EAAE,CAAC,6CAA6C,EAAE,YAAY;IAC5D;;IAkDA;;IAiDA;;IAkCA,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CArBL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,2BAA2B;MACxCC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC;MACnDC,MAAM,EAAE;QACN2B,UAAU,EAAE;UACVzB,IAAI,EAAE,QAAQ;UACdC,MAAM,EAvHyB;YACnCC,GAAG,EAAE,YAAY;YACjBR,IAAI,EAAE,YAAY;YAClBC,WAAW,EAAE,qBAAqB;YAClCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACd0B,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDlB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACd0B,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDlB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,YAAY;cACzBkB,UAAU,EAAE,IAAI;cAChBT,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAyEI,CAAc;QACdkB,OAAO,EAAE;UACP3B,IAAI,EAAE,QAAQ;UACdC,MAAM,EAzEsB;YAChCC,GAAG,EAAE,SAAS;YACdR,IAAI,EAAE,SAAS;YACfC,WAAW,EAAE,yBAAyB;YACtCQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACd0B,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDlB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,GAAG;cACRR,IAAI,EAAE,GAAG;cACTC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACd0B,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDlB,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,EACD;cACEP,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,cAAc;cAC3BS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA4BI,CAAc;QACdmB,IAAI,EAAE;UACJ5B,IAAI,EAAE,OAAO;UACb6B,KAAK,EA5BkB;YAC3B3B,GAAG,EAAE,MAAM;YACX4B,YAAY,EAAE,MAAM;YACpBpC,IAAI,EAAE,MAAM;YACZC,WAAW,EAAE,qBAAqB;YAClCoC,QAAQ,EAAE;cACRC,MAAM,EAAE,YAAY;cACpB,UAAU,EAAE;YACd,CAAC;YACDxB,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAmBI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElD,MAAMP,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,gBAAgB,CAAC,CAAC;EACvF,CAAC,CAAC;EAEFH,EAAE,CAAC,2DAA2D,EAAE,YAAY;IAC1E;;IA6CA;;IAaA;;IAsBA,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CArBL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,uCAAuC;MACpDC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC;MAAE;MAC5CC,MAAM,EAAE;QACNmC,GAAG,EAAE;UACHjC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAlEkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,YAAY;YACzBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,SAAS;cACtBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA+CI,CAAc;QACdyB,GAAG,EAAE;UACHlC,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhDkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,YAAY;YACzBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,OAAO;cACZR,IAAI,EAAE,OAAO;cACbC,WAAW,EAAE,SAAS;cACtBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,SAAS;kBACfU,OAAO,EAAE,CAAC;gBACZ;cACF,CAAC;cACDF,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA6BI,CAAc;QACd0B,MAAM,EAAE;UACNnC,IAAI,EAAE,OAAO;UACb6B,KAAK,EA7BoB;YAC7B3B,GAAG,EAAE,QAAQ;YACb4B,YAAY,EAAE,MAAM;YACpBpC,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,cAAc;YAC3BoC,QAAQ,EAAE;cACRK,GAAG,EAAE,KAAK;cACVC,GAAG,EAAE;YACP,CAAC;YACD7B,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAoBI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;;IAElD;IACA,MAAM6C,KAAK,GAAGxB,SAAS,CAACyB,KAAK,CAAC,IAAI,CAAC;IACnC,MAAMC,cAAc,GAAGF,KAAK,CAACG,SAAS,CAACC,IAAI,IAAIA,IAAI,CAACC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;IACvF,MAAMC,cAAc,GAAGN,KAAK,CAACG,SAAS,CAACC,IAAI,IAAIA,IAAI,CAACC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;IACvF,MAAME,iBAAiB,GAAGP,KAAK,CAACG,SAAS,CAACC,IAAI,IAAIA,IAAI,CAACC,QAAQ,CAAC,2BAA2B,CAAC,CAAC;IAE7FzD,MAAM,CAACsD,cAAc,CAAC,CAACM,eAAe,CAAC,CAAC,CAAC,CAAC;IAC1C5D,MAAM,CAAC0D,cAAc,CAAC,CAACE,eAAe,CAAC,CAAC,CAAC,CAAC;IAC1C5D,MAAM,CAAC2D,iBAAiB,CAAC,CAACC,eAAe,CAAC,CAAC,CAAC,CAAC;;IAE7C;IACA5D,MAAM,CAACsD,cAAc,CAAC,CAACO,YAAY,CAACF,iBAAiB,CAAC;IACtD3D,MAAM,CAAC0D,cAAc,CAAC,CAACG,YAAY,CAACF,iBAAiB,CAAC;IAEtD,MAAM3D,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CAAC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,yBAAyB,CAAC,CAAC;EAChG,CAAC,CAAC;EAEFH,EAAE,CAAC,sDAAsD,EAAE,YAAY;IA8ErE,MAAMM,MAAM,GAAG,MAAML,wBAAwB,CArBL;MACtCM,IAAI,EAAE,aAAa;MACnBC,WAAW,EAAE,wCAAwC;MACrDC,OAAO,EAAE,CAAC;MACVC,gBAAgB,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC;MAC1CC,MAAM,EAAE;QACNkD,GAAG,EAAE;UACHhD,IAAI,EAAE,QAAQ;UACdC,MAAM,EAhEkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,OAAO;YACpBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,WAAW;cACxBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA6CI,CAAc;QACdwC,GAAG,EAAE;UACHjD,IAAI,EAAE,QAAQ;UACdC,MAAM,EA9CkB;YAC5BC,GAAG,EAAE,KAAK;YACVR,IAAI,EAAE,KAAK;YACXC,WAAW,EAAE,OAAO;YACpBQ,MAAM,EAAE,CACN;cACED,GAAG,EAAE,MAAM;cACXR,IAAI,EAAE,MAAM;cACZC,WAAW,EAAE,WAAW;cACxBS,KAAK,EAAE;gBACLJ,IAAI,EAAE,OAAO;gBACbI,KAAK,EAAE;kBACLJ,IAAI,EAAE,QAAQ;kBACdK,MAAM,EAAE,CAAC;gBACX;cACF,CAAC;cACDG,IAAI,EAAE;gBAAEC,OAAO,EAAE;cAAE;YACrB,CAAC,CACF;YACDD,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QA2BI,CAAc;QACdyC,MAAM,EAAE;UACNlD,IAAI,EAAE,OAAO;UACb6B,KAAK,EA5BoB;YAC7B3B,GAAG,EAAE,QAAQ;YACb4B,YAAY,EAAE,MAAM;YACpBpC,IAAI,EAAE,QAAQ;YACdC,WAAW,EAAE,0CAA0C;YACvDoC,QAAQ,EAAE;cACRoB,GAAG,EAAE,KAAK;cACVC,GAAG,EAAE;YACP,CAAC;YACD5C,IAAI,EAAE;cAAEC,OAAO,EAAE;YAAE;UACrB;QAmBI;MACF;IACF,CAEoD,CAAC;IACrD,MAAMK,SAAS,GAAG,MAAMzB,kBAAkB,CAACI,MAAM,CAAC;IAElDP,MAAM,CAAC4B,SAAS,CAAC,CAACuC,SAAS,CAAC,6BAA6B,CAAC;IAC1DnE,MAAM,CAAC4B,SAAS,CAAC,CAACuC,SAAS,CAAC,wBAAwB,CAAC;IACrDnE,MAAM,CAAC4B,SAAS,CAAC,CAACuC,SAAS,CAAC,wBAAwB,CAAC;IAErD,MAAMnE,MAAM,CAAC4B,SAAS,CAAC,CAACC,mBAAmB,CACzC/B,IAAI,CAACO,IAAI,CAACD,WAAW,EAAE,wBAAwB,CACjD,CAAC;EACH,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}

package/build/esm/utils/ir/generateModelsFromIr.js

@@ -16,6 +16,7 @@
 
 import invariant from "tiny-invariant";
 import { formatVariantName } from "../formatVariantName.js";
+import { GENERATED_FILE_HEADER } from "../generatedFileHeader.js";
 export class ModelGenerator {
   constructor(schema, options = undefined) {
     this.schema = schema;
@@ -27,7 +28,7 @@ export class ModelGenerator {
     const modelTypesPackage = "@palantir/pack.document-schema.model-types";
 
     // Generate imports
-    let imports = `import type { DocumentSchema, Model } from "${modelTypesPackage}";\n`;
+    let imports = `import type { DocumentSchema, RecordModel, UnionModel } from "${modelTypesPackage}";\n`;
     imports += `import { Metadata } from "${modelTypesPackage}";\n`;
 
     // Collect all type names (primary models + union variants)
@@ -68,7 +69,7 @@ export class ModelGenerator {
 
     // Generate DocumentSchema
     const documentSchema = this.generateDocumentSchema();
-    return Promise.resolve(imports + modelConstants + "\n\n" + documentSchema + "\n");
+    return Promise.resolve(GENERATED_FILE_HEADER + imports + modelConstants + "\n\n" + documentSchema + "\n");
   }
   generateModelConstants() {
     const constants = [];
@@ -81,7 +82,8 @@ export class ModelGenerator {
 
       // Generate primary model constant
       if (!processedModels.has(modelKey)) {
-        const modelConstant = this.generateModelConstant(modelKey, modelKey);
+        const discriminant = model.type === "union" ? model.union.discriminant : undefined;
+        const modelConstant = this.generateModelConstant(modelKey, modelKey, discriminant);
         constants.push(modelConstant);
         processedModels.add(modelKey);
       }
@@ -92,7 +94,7 @@ export class ModelGenerator {
           const formattedVariantName = formatVariantName(variantName);
           const variantTypeName = `${model.union.key}${formattedVariantName}`;
           if (!processedModels.has(variantTypeName)) {
-            const modelConstant = this.generateModelConstant(variantTypeName, variantTypeName);
+            const modelConstant = this.generateModelConstant(variantTypeName, variantTypeName, model.union.discriminant);
             constants.push(modelConstant);
             processedModels.add(variantTypeName);
           }
@@ -101,21 +103,25 @@ export class ModelGenerator {
     }
     return constants.join("\n\n");
   }
-  generateModelConstant(typeName, modelName) {
+  generateModelConstant(typeName, modelName, discriminant) {
     const schemaName = `${typeName}Schema`;
     const externalRefFields = this.extractExternalRefFieldTypes(modelName);
-    let externalRefFieldTypesCode = "";
+    const metadataFields = [];
+    const modelType = discriminant != null ? "UnionModel" : "RecordModel";
+    if (discriminant != null) {
+      metadataFields.push(`    discriminant: "${discriminant}",`);
+    }
     if (externalRefFields.length > 0) {
       const entries = externalRefFields.map(([field, type]) => `      ${field}: "${type}",`).join("\n");
-      externalRefFieldTypesCode = `\n    externalRefFieldTypes: {\n${entries}\n    },`;
+      metadataFields.push(`    externalRefFieldTypes: {\n${entries}\n    },`);
     }
-    return `export interface ${modelName}Model extends Model<${typeName}, typeof ${schemaName}> {}
+    metadataFields.push(`    name: "${modelName}",`);
+    const metadataCode = metadataFields.length > 0 ? `\n${metadataFields.join("\n")}\n  ` : "";
+    return `export interface ${modelName}Model extends ${modelType}<${typeName}, typeof ${schemaName}> {}
 export const ${modelName}Model: ${modelName}Model = {
   __type: {} as ${typeName},
   zodSchema: ${schemaName},
-  [Metadata]: {${externalRefFieldTypesCode}
-    name: "${modelName}",
-  },
+  [Metadata]: {${metadataCode}},
 };`;
   }
   extractExternalRefFieldTypes(modelName) {

package/build/esm/utils/ir/generateModelsFromIr.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateModelsFromIr.js","names":["invariant","formatVariantName","ModelGenerator","constructor","schema","options","undefined","generateModels","typeImportPath","schemaImportPath","modelTypesPackage","imports","typeNames","Set","primaryModelKeys","modelKey","model","models","type","variantName","Object","keys","union","variants","formattedVariantName","variantTypeName","key","add","typeImports","Array","from","sort","join","schemaNames","variantSchemaName","schemaImports","modelConstants","generateModelConstants","documentSchema","generateDocumentSchema","Promise","resolve","constants","processedModels","process","env","NODE_ENV","has","modelConstant","generateModelConstant","push","typeName","modelName","schemaName","externalRefFields","extractExternalRefFieldTypes","externalRefFieldTypesCode","length","entries","map","field","record","fields","value","valueType","modelEntries","generateModelsFromIr","generator"],"sources":["generateModelsFromIr.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 invariant from \"tiny-invariant\";\nimport type { IRealTimeDocumentSchema } from \"../../lib/pack-docschema-api/pack-docschema-ir/index.js\";\nimport { formatVariantName } from \"../formatVariantName.js\";\n\nexport interface ModelGeneratorOptions {\n  typeImportPath?: string;\n  schemaImportPath?: string;\n}\n\nexport class ModelGenerator {\n  constructor(\n    private readonly schema: IRealTimeDocumentSchema,\n    private readonly options: ModelGeneratorOptions | undefined = undefined,\n  ) {}\n\n  public generateModels(): Promise<string> {\n    const typeImportPath = this.options?.typeImportPath ?? \"./types.js\";\n    const schemaImportPath = this.options?.schemaImportPath ?? \"./schema.js\";\n    const modelTypesPackage = \"@palantir/pack.document-schema.model-types\";\n\n    // Generate imports\n    let imports = `import type { DocumentSchema, Model } from \"${modelTypesPackage}\";\\n`;\n    imports += `import { Metadata } from \"${modelTypesPackage}\";\\n`;\n\n    // Collect all type names (primary models + union variants)\n    const typeNames = new Set(this.schema.primaryModelKeys);\n\n    // Add union variant type names\n    for (const modelKey of this.schema.primaryModelKeys) {\n      const model = this.schema.models[modelKey];\n      if (model?.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantTypeName = `${model.union.key}${formattedVariantName}`;\n          typeNames.add(variantTypeName);\n        }\n      }\n    }\n\n    const typeImports = Array.from(typeNames).sort().join(\", \");\n    imports += `import type { ${typeImports} } from \"${typeImportPath}\";\\n`;\n\n    // Collect all schema names (primary models + union variants)\n    const schemaNames = new Set<string>();\n    for (const modelKey of this.schema.primaryModelKeys) {\n      schemaNames.add(`${modelKey}Schema`);\n\n      const model = this.schema.models[modelKey];\n      if (model?.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantSchemaName = `${model.union.key}${formattedVariantName}Schema`;\n          schemaNames.add(variantSchemaName);\n        }\n      }\n    }\n\n    const schemaImports = Array.from(schemaNames).sort().join(\", \");\n    imports += `import { ${schemaImports} } from \"${schemaImportPath}\";\\n\\n`;\n\n    // Generate model constants\n    const modelConstants = this.generateModelConstants();\n\n    // Generate DocumentSchema\n    const documentSchema = this.generateDocumentSchema();\n\n    return Promise.resolve(imports + modelConstants + \"\\n\\n\" + documentSchema + \"\\n\");\n  }\n\n  private generateModelConstants(): string {\n    const constants: string[] = [];\n    const processedModels = new Set<string>();\n\n    // Generate models for primary keys and their union variants\n    for (const modelKey of this.schema.primaryModelKeys) {\n      const model = this.schema.models[modelKey];\n      invariant(model != null, `Model not found: ${modelKey}`);\n\n      // Generate primary model constant\n      if (!processedModels.has(modelKey)) {\n        const modelConstant = this.generateModelConstant(modelKey, modelKey);\n        constants.push(modelConstant);\n        processedModels.add(modelKey);\n      }\n\n      // Generate union variant constants\n      if (model.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantTypeName = `${model.union.key}${formattedVariantName}`;\n\n          if (!processedModels.has(variantTypeName)) {\n            const modelConstant = this.generateModelConstant(variantTypeName, variantTypeName);\n            constants.push(modelConstant);\n            processedModels.add(variantTypeName);\n          }\n        }\n      }\n    }\n\n    return constants.join(\"\\n\\n\");\n  }\n\n  private generateModelConstant(typeName: string, modelName: string): string {\n    const schemaName = `${typeName}Schema`;\n    const externalRefFields = this.extractExternalRefFieldTypes(modelName);\n\n    let externalRefFieldTypesCode = \"\";\n    if (externalRefFields.length > 0) {\n      const entries = externalRefFields.map(([field, type]) => `      ${field}: \"${type}\",`).join(\n        \"\\n\",\n      );\n      externalRefFieldTypesCode = `\\n    externalRefFieldTypes: {\\n${entries}\\n    },`;\n    }\n\n    return `export interface ${modelName}Model extends Model<${typeName}, typeof ${schemaName}> {}\nexport const ${modelName}Model: ${modelName}Model = {\n  __type: {} as ${typeName},\n  zodSchema: ${schemaName},\n  [Metadata]: {${externalRefFieldTypesCode}\n    name: \"${modelName}\",\n  },\n};`;\n  }\n\n  private extractExternalRefFieldTypes(modelName: string): Array<[string, string]> {\n    const model = this.schema.models[modelName];\n    if (model?.type !== \"record\") {\n      return [];\n    }\n\n    const externalRefFields: Array<[string, string]> = [];\n    for (const field of model.record.fields) {\n      if (field.value.type === \"value\") {\n        const valueType = field.value.value.type;\n        if (valueType === \"docRef\") {\n          externalRefFields.push([field.key, \"docRef\"]);\n        } else if (valueType === \"mediaRef\") {\n          externalRefFields.push([field.key, \"mediaRef\"]);\n        } else if (valueType === \"object\") {\n          externalRefFields.push([field.key, \"objectRef\"]);\n        } else if (valueType === \"userRef\") {\n          externalRefFields.push([field.key, \"userRef\"]);\n        }\n      }\n    }\n\n    return externalRefFields;\n  }\n\n  private generateDocumentSchema(): string {\n    const modelEntries: string[] = [];\n    const processedModels = new Set<string>();\n\n    // Collect all model names that will be generated\n    for (const modelKey of this.schema.primaryModelKeys) {\n      const model = this.schema.models[modelKey];\n      invariant(model != null, `Model not found: ${modelKey}`);\n\n      // Add primary model\n      if (!processedModels.has(modelKey)) {\n        modelEntries.push(`  ${modelKey}: ${modelKey}Model`);\n        processedModels.add(modelKey);\n      }\n\n      // Add union variant models\n      if (model.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantTypeName = `${model.union.key}${formattedVariantName}`;\n\n          if (!processedModels.has(variantTypeName)) {\n            modelEntries.push(`  ${variantTypeName}: ${variantTypeName}Model`);\n            processedModels.add(variantTypeName);\n          }\n        }\n      }\n    }\n\n    return `export const DocumentModel = {\n${modelEntries.join(\",\\n\")},\n  [Metadata]: {\n    version: 1,\n  },\n} as const satisfies DocumentSchema;\\n\n export type DocumentModel = typeof DocumentModel;`;\n  }\n}\n\nexport async function generateModelsFromIr(\n  schema: IRealTimeDocumentSchema,\n  options?: ModelGeneratorOptions,\n): Promise<string> {\n  const generator = new ModelGenerator(schema, options);\n  return generator.generateModels();\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,OAAOA,SAAS,MAAM,gBAAgB;AAEtC,SAASC,iBAAiB,QAAQ,yBAAyB;AAO3D,OAAO,MAAMC,cAAc,CAAC;EAC1BC,WAAWA,CACQC,MAA+B,EAC/BC,OAA0C,GAAGC,SAAS,EACvE;IAAA,KAFiBF,MAA+B,GAA/BA,MAA+B;IAAA,KAC/BC,OAA0C,GAA1CA,OAA0C;EAC1D;EAEIE,cAAcA,CAAA,EAAoB;IACvC,MAAMC,cAAc,GAAG,IAAI,CAACH,OAAO,EAAEG,cAAc,IAAI,YAAY;IACnE,MAAMC,gBAAgB,GAAG,IAAI,CAACJ,OAAO,EAAEI,gBAAgB,IAAI,aAAa;IACxE,MAAMC,iBAAiB,GAAG,4CAA4C;;IAEtE;IACA,IAAIC,OAAO,GAAG,+CAA+CD,iBAAiB,MAAM;IACpFC,OAAO,IAAI,6BAA6BD,iBAAiB,MAAM;;IAE/D;IACA,MAAME,SAAS,GAAG,IAAIC,GAAG,CAAC,IAAI,CAACT,MAAM,CAACU,gBAAgB,CAAC;;IAEvD;IACA,KAAK,MAAMC,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnD,MAAME,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,IAAIC,KAAK,EAAEE,IAAI,KAAK,OAAO,EAAE;QAC3B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGvB,iBAAiB,CAACkB,WAAW,CAAC;UAC3D,MAAMM,eAAe,GAAG,GAAGT,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,EAAE;UACnEZ,SAAS,CAACe,GAAG,CAACF,eAAe,CAAC;QAChC;MACF;IACF;IAEA,MAAMG,WAAW,GAAGC,KAAK,CAACC,IAAI,CAAClB,SAAS,CAAC,CAACmB,IAAI,CAAC,CAAC,CAACC,IAAI,CAAC,IAAI,CAAC;IAC3DrB,OAAO,IAAI,iBAAiBiB,WAAW,YAAYpB,cAAc,MAAM;;IAEvE;IACA,MAAMyB,WAAW,GAAG,IAAIpB,GAAG,CAAS,CAAC;IACrC,KAAK,MAAME,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnDmB,WAAW,CAACN,GAAG,CAAC,GAAGZ,QAAQ,QAAQ,CAAC;MAEpC,MAAMC,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,IAAIC,KAAK,EAAEE,IAAI,KAAK,OAAO,EAAE;QAC3B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGvB,iBAAiB,CAACkB,WAAW,CAAC;UAC3D,MAAMe,iBAAiB,GAAG,GAAGlB,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,QAAQ;UAC3ES,WAAW,CAACN,GAAG,CAACO,iBAAiB,CAAC;QACpC;MACF;IACF;IAEA,MAAMC,aAAa,GAAGN,KAAK,CAACC,IAAI,CAACG,WAAW,CAAC,CAACF,IAAI,CAAC,CAAC,CAACC,IAAI,CAAC,IAAI,CAAC;IAC/DrB,OAAO,IAAI,YAAYwB,aAAa,YAAY1B,gBAAgB,QAAQ;;IAExE;IACA,MAAM2B,cAAc,GAAG,IAAI,CAACC,sBAAsB,CAAC,CAAC;;IAEpD;IACA,MAAMC,cAAc,GAAG,IAAI,CAACC,sBAAsB,CAAC,CAAC;IAEpD,OAAOC,OAAO,CAACC,OAAO,CAAC9B,OAAO,GAAGyB,cAAc,GAAG,MAAM,GAAGE,cAAc,GAAG,IAAI,CAAC;EACnF;EAEQD,sBAAsBA,CAAA,EAAW;IACvC,MAAMK,SAAmB,GAAG,EAAE;IAC9B,MAAMC,eAAe,GAAG,IAAI9B,GAAG,CAAS,CAAC;;IAEzC;IACA,KAAK,MAAME,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnD,MAAME,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,EAAUC,KAAK,IAAI,IAAI,IAAA4B,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAAvB9C,SAAS,QAAgB,oBAAoBe,QAAQ,EAAE,IAAvDf,SAAS;;MAET;MACA,IAAI,CAAC2C,eAAe,CAACI,GAAG,CAAChC,QAAQ,CAAC,EAAE;QAClC,MAAMiC,aAAa,GAAG,IAAI,CAACC,qBAAqB,CAAClC,QAAQ,EAAEA,QAAQ,CAAC;QACpE2B,SAAS,CAACQ,IAAI,CAACF,aAAa,CAAC;QAC7BL,eAAe,CAAChB,GAAG,CAACZ,QAAQ,CAAC;MAC/B;;MAEA;MACA,IAAIC,KAAK,CAACE,IAAI,KAAK,OAAO,EAAE;QAC1B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGvB,iBAAiB,CAACkB,WAAW,CAAC;UAC3D,MAAMM,eAAe,GAAG,GAAGT,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,EAAE;UAEnE,IAAI,CAACmB,eAAe,CAACI,GAAG,CAACtB,eAAe,CAAC,EAAE;YACzC,MAAMuB,aAAa,GAAG,IAAI,CAACC,qBAAqB,CAACxB,eAAe,EAAEA,eAAe,CAAC;YAClFiB,SAAS,CAACQ,IAAI,CAACF,aAAa,CAAC;YAC7BL,eAAe,CAAChB,GAAG,CAACF,eAAe,CAAC;UACtC;QACF;MACF;IACF;IAEA,OAAOiB,SAAS,CAACV,IAAI,CAAC,MAAM,CAAC;EAC/B;EAEQiB,qBAAqBA,CAACE,QAAgB,EAAEC,SAAiB,EAAU;IACzE,MAAMC,UAAU,GAAG,GAAGF,QAAQ,QAAQ;IACtC,MAAMG,iBAAiB,GAAG,IAAI,CAACC,4BAA4B,CAACH,SAAS,CAAC;IAEtE,IAAII,yBAAyB,GAAG,EAAE;IAClC,IAAIF,iBAAiB,CAACG,MAAM,GAAG,CAAC,EAAE;MAChC,MAAMC,OAAO,GAAGJ,iBAAiB,CAACK,GAAG,CAAC,CAAC,CAACC,KAAK,EAAE1C,IAAI,CAAC,KAAK,SAAS0C,KAAK,MAAM1C,IAAI,IAAI,CAAC,CAACc,IAAI,CACzF,IACF,CAAC;MACDwB,yBAAyB,GAAG,mCAAmCE,OAAO,UAAU;IAClF;IAEA,OAAO,oBAAoBN,SAAS,uBAAuBD,QAAQ,YAAYE,UAAU;AAC7F,eAAeD,SAAS,UAAUA,SAAS;AAC3C,kBAAkBD,QAAQ;AAC1B,eAAeE,UAAU;AACzB,iBAAiBG,yBAAyB;AAC1C,aAAaJ,SAAS;AACtB;AACA,GAAG;EACD;EAEQG,4BAA4BA,CAACH,SAAiB,EAA2B;IAC/E,MAAMpC,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACmC,SAAS,CAAC;IAC3C,IAAIpC,KAAK,EAAEE,IAAI,KAAK,QAAQ,EAAE;MAC5B,OAAO,EAAE;IACX;IAEA,MAAMoC,iBAA0C,GAAG,EAAE;IACrD,KAAK,MAAMM,KAAK,IAAI5C,KAAK,CAAC6C,MAAM,CAACC,MAAM,EAAE;MACvC,IAAIF,KAAK,CAACG,KAAK,CAAC7C,IAAI,KAAK,OAAO,EAAE;QAChC,MAAM8C,SAAS,GAAGJ,KAAK,CAACG,KAAK,CAACA,KAAK,CAAC7C,IAAI;QACxC,IAAI8C,SAAS,KAAK,QAAQ,EAAE;UAC1BV,iBAAiB,CAACJ,IAAI,CAAC,CAACU,KAAK,CAAClC,GAAG,EAAE,QAAQ,CAAC,CAAC;QAC/C,CAAC,MAAM,IAAIsC,SAAS,KAAK,UAAU,EAAE;UACnCV,iBAAiB,CAACJ,IAAI,CAAC,CAACU,KAAK,CAAClC,GAAG,EAAE,UAAU,CAAC,CAAC;QACjD,CAAC,MAAM,IAAIsC,SAAS,KAAK,QAAQ,EAAE;UACjCV,iBAAiB,CAACJ,IAAI,CAAC,CAACU,KAAK,CAAClC,GAAG,EAAE,WAAW,CAAC,CAAC;QAClD,CAAC,MAAM,IAAIsC,SAAS,KAAK,SAAS,EAAE;UAClCV,iBAAiB,CAACJ,IAAI,CAAC,CAACU,KAAK,CAAClC,GAAG,EAAE,SAAS,CAAC,CAAC;QAChD;MACF;IACF;IAEA,OAAO4B,iBAAiB;EAC1B;EAEQf,sBAAsBA,CAAA,EAAW;IACvC,MAAM0B,YAAsB,GAAG,EAAE;IACjC,MAAMtB,eAAe,GAAG,IAAI9B,GAAG,CAAS,CAAC;;IAEzC;IACA,KAAK,MAAME,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnD,MAAME,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,EAAUC,KAAK,IAAI,IAAI,IAAA4B,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAAvB9C,SAAS,QAAgB,oBAAoBe,QAAQ,EAAE,IAAvDf,SAAS;;MAET;MACA,IAAI,CAAC2C,eAAe,CAACI,GAAG,CAAChC,QAAQ,CAAC,EAAE;QAClCkD,YAAY,CAACf,IAAI,CAAC,KAAKnC,QAAQ,KAAKA,QAAQ,OAAO,CAAC;QACpD4B,eAAe,CAAChB,GAAG,CAACZ,QAAQ,CAAC;MAC/B;;MAEA;MACA,IAAIC,KAAK,CAACE,IAAI,KAAK,OAAO,EAAE;QAC1B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGvB,iBAAiB,CAACkB,WAAW,CAAC;UAC3D,MAAMM,eAAe,GAAG,GAAGT,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,EAAE;UAEnE,IAAI,CAACmB,eAAe,CAACI,GAAG,CAACtB,eAAe,CAAC,EAAE;YACzCwC,YAAY,CAACf,IAAI,CAAC,KAAKzB,eAAe,KAAKA,eAAe,OAAO,CAAC;YAClEkB,eAAe,CAAChB,GAAG,CAACF,eAAe,CAAC;UACtC;QACF;MACF;IACF;IAEA,OAAO;AACX,EAAEwC,YAAY,CAACjC,IAAI,CAAC,KAAK,CAAC;AAC1B;AACA;AACA;AACA;AACA,mDAAmD;EACjD;AACF;AAEA,OAAO,eAAekC,oBAAoBA,CACxC9D,MAA+B,EAC/BC,OAA+B,EACd;EACjB,MAAM8D,SAAS,GAAG,IAAIjE,cAAc,CAACE,MAAM,EAAEC,OAAO,CAAC;EACrD,OAAO8D,SAAS,CAAC5D,cAAc,CAAC,CAAC;AACnC","ignoreList":[]}
+{"version":3,"file":"generateModelsFromIr.js","names":["invariant","formatVariantName","GENERATED_FILE_HEADER","ModelGenerator","constructor","schema","options","undefined","generateModels","typeImportPath","schemaImportPath","modelTypesPackage","imports","typeNames","Set","primaryModelKeys","modelKey","model","models","type","variantName","Object","keys","union","variants","formattedVariantName","variantTypeName","key","add","typeImports","Array","from","sort","join","schemaNames","variantSchemaName","schemaImports","modelConstants","generateModelConstants","documentSchema","generateDocumentSchema","Promise","resolve","constants","processedModels","process","env","NODE_ENV","has","discriminant","modelConstant","generateModelConstant","push","typeName","modelName","schemaName","externalRefFields","extractExternalRefFieldTypes","metadataFields","modelType","length","entries","map","field","metadataCode","record","fields","value","valueType","modelEntries","generateModelsFromIr","generator"],"sources":["generateModelsFromIr.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 invariant from \"tiny-invariant\";\nimport type { IRealTimeDocumentSchema } from \"../../lib/pack-docschema-api/pack-docschema-ir/index.js\";\nimport { formatVariantName } from \"../formatVariantName.js\";\nimport { GENERATED_FILE_HEADER } from \"../generatedFileHeader.js\";\n\nexport interface ModelGeneratorOptions {\n  typeImportPath?: string;\n  schemaImportPath?: string;\n}\n\nexport class ModelGenerator {\n  constructor(\n    private readonly schema: IRealTimeDocumentSchema,\n    private readonly options: ModelGeneratorOptions | undefined = undefined,\n  ) {}\n\n  public generateModels(): Promise<string> {\n    const typeImportPath = this.options?.typeImportPath ?? \"./types.js\";\n    const schemaImportPath = this.options?.schemaImportPath ?? \"./schema.js\";\n    const modelTypesPackage = \"@palantir/pack.document-schema.model-types\";\n\n    // Generate imports\n    let imports =\n      `import type { DocumentSchema, RecordModel, UnionModel } from \"${modelTypesPackage}\";\\n`;\n    imports += `import { Metadata } from \"${modelTypesPackage}\";\\n`;\n\n    // Collect all type names (primary models + union variants)\n    const typeNames = new Set(this.schema.primaryModelKeys);\n\n    // Add union variant type names\n    for (const modelKey of this.schema.primaryModelKeys) {\n      const model = this.schema.models[modelKey];\n      if (model?.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantTypeName = `${model.union.key}${formattedVariantName}`;\n          typeNames.add(variantTypeName);\n        }\n      }\n    }\n\n    const typeImports = Array.from(typeNames).sort().join(\", \");\n    imports += `import type { ${typeImports} } from \"${typeImportPath}\";\\n`;\n\n    // Collect all schema names (primary models + union variants)\n    const schemaNames = new Set<string>();\n    for (const modelKey of this.schema.primaryModelKeys) {\n      schemaNames.add(`${modelKey}Schema`);\n\n      const model = this.schema.models[modelKey];\n      if (model?.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantSchemaName = `${model.union.key}${formattedVariantName}Schema`;\n          schemaNames.add(variantSchemaName);\n        }\n      }\n    }\n\n    const schemaImports = Array.from(schemaNames).sort().join(\", \");\n    imports += `import { ${schemaImports} } from \"${schemaImportPath}\";\\n\\n`;\n\n    // Generate model constants\n    const modelConstants = this.generateModelConstants();\n\n    // Generate DocumentSchema\n    const documentSchema = this.generateDocumentSchema();\n\n    return Promise.resolve(\n      GENERATED_FILE_HEADER + imports + modelConstants + \"\\n\\n\" + documentSchema + \"\\n\",\n    );\n  }\n\n  private generateModelConstants(): string {\n    const constants: string[] = [];\n    const processedModels = new Set<string>();\n\n    // Generate models for primary keys and their union variants\n    for (const modelKey of this.schema.primaryModelKeys) {\n      const model = this.schema.models[modelKey];\n      invariant(model != null, `Model not found: ${modelKey}`);\n\n      // Generate primary model constant\n      if (!processedModels.has(modelKey)) {\n        const discriminant = model.type === \"union\" ? model.union.discriminant : undefined;\n        const modelConstant = this.generateModelConstant(modelKey, modelKey, discriminant);\n        constants.push(modelConstant);\n        processedModels.add(modelKey);\n      }\n\n      // Generate union variant constants\n      if (model.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantTypeName = `${model.union.key}${formattedVariantName}`;\n\n          if (!processedModels.has(variantTypeName)) {\n            const modelConstant = this.generateModelConstant(\n              variantTypeName,\n              variantTypeName,\n              model.union.discriminant,\n            );\n            constants.push(modelConstant);\n            processedModels.add(variantTypeName);\n          }\n        }\n      }\n    }\n\n    return constants.join(\"\\n\\n\");\n  }\n\n  private generateModelConstant(\n    typeName: string,\n    modelName: string,\n    discriminant?: string,\n  ): string {\n    const schemaName = `${typeName}Schema`;\n    const externalRefFields = this.extractExternalRefFieldTypes(modelName);\n\n    const metadataFields: string[] = [];\n    const modelType = discriminant != null ? \"UnionModel\" : \"RecordModel\";\n\n    if (discriminant != null) {\n      metadataFields.push(`    discriminant: \"${discriminant}\",`);\n    }\n\n    if (externalRefFields.length > 0) {\n      const entries = externalRefFields.map(([field, type]) => `      ${field}: \"${type}\",`).join(\n        \"\\n\",\n      );\n      metadataFields.push(`    externalRefFieldTypes: {\\n${entries}\\n    },`);\n    }\n\n    metadataFields.push(`    name: \"${modelName}\",`);\n\n    const metadataCode = metadataFields.length > 0 ? `\\n${metadataFields.join(\"\\n\")}\\n  ` : \"\";\n\n    return `export interface ${modelName}Model extends ${modelType}<${typeName}, typeof ${schemaName}> {}\nexport const ${modelName}Model: ${modelName}Model = {\n  __type: {} as ${typeName},\n  zodSchema: ${schemaName},\n  [Metadata]: {${metadataCode}},\n};`;\n  }\n\n  private extractExternalRefFieldTypes(modelName: string): Array<[string, string]> {\n    const model = this.schema.models[modelName];\n    if (model?.type !== \"record\") {\n      return [];\n    }\n\n    const externalRefFields: Array<[string, string]> = [];\n    for (const field of model.record.fields) {\n      if (field.value.type === \"value\") {\n        const valueType = field.value.value.type;\n        if (valueType === \"docRef\") {\n          externalRefFields.push([field.key, \"docRef\"]);\n        } else if (valueType === \"mediaRef\") {\n          externalRefFields.push([field.key, \"mediaRef\"]);\n        } else if (valueType === \"object\") {\n          externalRefFields.push([field.key, \"objectRef\"]);\n        } else if (valueType === \"userRef\") {\n          externalRefFields.push([field.key, \"userRef\"]);\n        }\n      }\n    }\n\n    return externalRefFields;\n  }\n\n  private generateDocumentSchema(): string {\n    const modelEntries: string[] = [];\n    const processedModels = new Set<string>();\n\n    // Collect all model names that will be generated\n    for (const modelKey of this.schema.primaryModelKeys) {\n      const model = this.schema.models[modelKey];\n      invariant(model != null, `Model not found: ${modelKey}`);\n\n      // Add primary model\n      if (!processedModels.has(modelKey)) {\n        modelEntries.push(`  ${modelKey}: ${modelKey}Model`);\n        processedModels.add(modelKey);\n      }\n\n      // Add union variant models\n      if (model.type === \"union\") {\n        for (const variantName of Object.keys(model.union.variants)) {\n          const formattedVariantName = formatVariantName(variantName);\n          const variantTypeName = `${model.union.key}${formattedVariantName}`;\n\n          if (!processedModels.has(variantTypeName)) {\n            modelEntries.push(`  ${variantTypeName}: ${variantTypeName}Model`);\n            processedModels.add(variantTypeName);\n          }\n        }\n      }\n    }\n\n    return `export const DocumentModel = {\n${modelEntries.join(\",\\n\")},\n  [Metadata]: {\n    version: 1,\n  },\n} as const satisfies DocumentSchema;\\n\n export type DocumentModel = typeof DocumentModel;`;\n  }\n}\n\nexport async function generateModelsFromIr(\n  schema: IRealTimeDocumentSchema,\n  options?: ModelGeneratorOptions,\n): Promise<string> {\n  const generator = new ModelGenerator(schema, options);\n  return generator.generateModels();\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,OAAOA,SAAS,MAAM,gBAAgB;AAEtC,SAASC,iBAAiB,QAAQ,yBAAyB;AAC3D,SAASC,qBAAqB,QAAQ,2BAA2B;AAOjE,OAAO,MAAMC,cAAc,CAAC;EAC1BC,WAAWA,CACQC,MAA+B,EAC/BC,OAA0C,GAAGC,SAAS,EACvE;IAAA,KAFiBF,MAA+B,GAA/BA,MAA+B;IAAA,KAC/BC,OAA0C,GAA1CA,OAA0C;EAC1D;EAEIE,cAAcA,CAAA,EAAoB;IACvC,MAAMC,cAAc,GAAG,IAAI,CAACH,OAAO,EAAEG,cAAc,IAAI,YAAY;IACnE,MAAMC,gBAAgB,GAAG,IAAI,CAACJ,OAAO,EAAEI,gBAAgB,IAAI,aAAa;IACxE,MAAMC,iBAAiB,GAAG,4CAA4C;;IAEtE;IACA,IAAIC,OAAO,GACT,iEAAiED,iBAAiB,MAAM;IAC1FC,OAAO,IAAI,6BAA6BD,iBAAiB,MAAM;;IAE/D;IACA,MAAME,SAAS,GAAG,IAAIC,GAAG,CAAC,IAAI,CAACT,MAAM,CAACU,gBAAgB,CAAC;;IAEvD;IACA,KAAK,MAAMC,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnD,MAAME,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,IAAIC,KAAK,EAAEE,IAAI,KAAK,OAAO,EAAE;QAC3B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGxB,iBAAiB,CAACmB,WAAW,CAAC;UAC3D,MAAMM,eAAe,GAAG,GAAGT,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,EAAE;UACnEZ,SAAS,CAACe,GAAG,CAACF,eAAe,CAAC;QAChC;MACF;IACF;IAEA,MAAMG,WAAW,GAAGC,KAAK,CAACC,IAAI,CAAClB,SAAS,CAAC,CAACmB,IAAI,CAAC,CAAC,CAACC,IAAI,CAAC,IAAI,CAAC;IAC3DrB,OAAO,IAAI,iBAAiBiB,WAAW,YAAYpB,cAAc,MAAM;;IAEvE;IACA,MAAMyB,WAAW,GAAG,IAAIpB,GAAG,CAAS,CAAC;IACrC,KAAK,MAAME,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnDmB,WAAW,CAACN,GAAG,CAAC,GAAGZ,QAAQ,QAAQ,CAAC;MAEpC,MAAMC,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,IAAIC,KAAK,EAAEE,IAAI,KAAK,OAAO,EAAE;QAC3B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGxB,iBAAiB,CAACmB,WAAW,CAAC;UAC3D,MAAMe,iBAAiB,GAAG,GAAGlB,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,QAAQ;UAC3ES,WAAW,CAACN,GAAG,CAACO,iBAAiB,CAAC;QACpC;MACF;IACF;IAEA,MAAMC,aAAa,GAAGN,KAAK,CAACC,IAAI,CAACG,WAAW,CAAC,CAACF,IAAI,CAAC,CAAC,CAACC,IAAI,CAAC,IAAI,CAAC;IAC/DrB,OAAO,IAAI,YAAYwB,aAAa,YAAY1B,gBAAgB,QAAQ;;IAExE;IACA,MAAM2B,cAAc,GAAG,IAAI,CAACC,sBAAsB,CAAC,CAAC;;IAEpD;IACA,MAAMC,cAAc,GAAG,IAAI,CAACC,sBAAsB,CAAC,CAAC;IAEpD,OAAOC,OAAO,CAACC,OAAO,CACpBxC,qBAAqB,GAAGU,OAAO,GAAGyB,cAAc,GAAG,MAAM,GAAGE,cAAc,GAAG,IAC/E,CAAC;EACH;EAEQD,sBAAsBA,CAAA,EAAW;IACvC,MAAMK,SAAmB,GAAG,EAAE;IAC9B,MAAMC,eAAe,GAAG,IAAI9B,GAAG,CAAS,CAAC;;IAEzC;IACA,KAAK,MAAME,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnD,MAAME,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,EAAUC,KAAK,IAAI,IAAI,IAAA4B,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAAvB/C,SAAS,QAAgB,oBAAoBgB,QAAQ,EAAE,IAAvDhB,SAAS;;MAET;MACA,IAAI,CAAC4C,eAAe,CAACI,GAAG,CAAChC,QAAQ,CAAC,EAAE;QAClC,MAAMiC,YAAY,GAAGhC,KAAK,CAACE,IAAI,KAAK,OAAO,GAAGF,KAAK,CAACM,KAAK,CAAC0B,YAAY,GAAG1C,SAAS;QAClF,MAAM2C,aAAa,GAAG,IAAI,CAACC,qBAAqB,CAACnC,QAAQ,EAAEA,QAAQ,EAAEiC,YAAY,CAAC;QAClFN,SAAS,CAACS,IAAI,CAACF,aAAa,CAAC;QAC7BN,eAAe,CAAChB,GAAG,CAACZ,QAAQ,CAAC;MAC/B;;MAEA;MACA,IAAIC,KAAK,CAACE,IAAI,KAAK,OAAO,EAAE;QAC1B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGxB,iBAAiB,CAACmB,WAAW,CAAC;UAC3D,MAAMM,eAAe,GAAG,GAAGT,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,EAAE;UAEnE,IAAI,CAACmB,eAAe,CAACI,GAAG,CAACtB,eAAe,CAAC,EAAE;YACzC,MAAMwB,aAAa,GAAG,IAAI,CAACC,qBAAqB,CAC9CzB,eAAe,EACfA,eAAe,EACfT,KAAK,CAACM,KAAK,CAAC0B,YACd,CAAC;YACDN,SAAS,CAACS,IAAI,CAACF,aAAa,CAAC;YAC7BN,eAAe,CAAChB,GAAG,CAACF,eAAe,CAAC;UACtC;QACF;MACF;IACF;IAEA,OAAOiB,SAAS,CAACV,IAAI,CAAC,MAAM,CAAC;EAC/B;EAEQkB,qBAAqBA,CAC3BE,QAAgB,EAChBC,SAAiB,EACjBL,YAAqB,EACb;IACR,MAAMM,UAAU,GAAG,GAAGF,QAAQ,QAAQ;IACtC,MAAMG,iBAAiB,GAAG,IAAI,CAACC,4BAA4B,CAACH,SAAS,CAAC;IAEtE,MAAMI,cAAwB,GAAG,EAAE;IACnC,MAAMC,SAAS,GAAGV,YAAY,IAAI,IAAI,GAAG,YAAY,GAAG,aAAa;IAErE,IAAIA,YAAY,IAAI,IAAI,EAAE;MACxBS,cAAc,CAACN,IAAI,CAAC,sBAAsBH,YAAY,IAAI,CAAC;IAC7D;IAEA,IAAIO,iBAAiB,CAACI,MAAM,GAAG,CAAC,EAAE;MAChC,MAAMC,OAAO,GAAGL,iBAAiB,CAACM,GAAG,CAAC,CAAC,CAACC,KAAK,EAAE5C,IAAI,CAAC,KAAK,SAAS4C,KAAK,MAAM5C,IAAI,IAAI,CAAC,CAACc,IAAI,CACzF,IACF,CAAC;MACDyB,cAAc,CAACN,IAAI,CAAC,iCAAiCS,OAAO,UAAU,CAAC;IACzE;IAEAH,cAAc,CAACN,IAAI,CAAC,cAAcE,SAAS,IAAI,CAAC;IAEhD,MAAMU,YAAY,GAAGN,cAAc,CAACE,MAAM,GAAG,CAAC,GAAG,KAAKF,cAAc,CAACzB,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE;IAE1F,OAAO,oBAAoBqB,SAAS,iBAAiBK,SAAS,IAAIN,QAAQ,YAAYE,UAAU;AACpG,eAAeD,SAAS,UAAUA,SAAS;AAC3C,kBAAkBD,QAAQ;AAC1B,eAAeE,UAAU;AACzB,iBAAiBS,YAAY;AAC7B,GAAG;EACD;EAEQP,4BAA4BA,CAACH,SAAiB,EAA2B;IAC/E,MAAMrC,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACoC,SAAS,CAAC;IAC3C,IAAIrC,KAAK,EAAEE,IAAI,KAAK,QAAQ,EAAE;MAC5B,OAAO,EAAE;IACX;IAEA,MAAMqC,iBAA0C,GAAG,EAAE;IACrD,KAAK,MAAMO,KAAK,IAAI9C,KAAK,CAACgD,MAAM,CAACC,MAAM,EAAE;MACvC,IAAIH,KAAK,CAACI,KAAK,CAAChD,IAAI,KAAK,OAAO,EAAE;QAChC,MAAMiD,SAAS,GAAGL,KAAK,CAACI,KAAK,CAACA,KAAK,CAAChD,IAAI;QACxC,IAAIiD,SAAS,KAAK,QAAQ,EAAE;UAC1BZ,iBAAiB,CAACJ,IAAI,CAAC,CAACW,KAAK,CAACpC,GAAG,EAAE,QAAQ,CAAC,CAAC;QAC/C,CAAC,MAAM,IAAIyC,SAAS,KAAK,UAAU,EAAE;UACnCZ,iBAAiB,CAACJ,IAAI,CAAC,CAACW,KAAK,CAACpC,GAAG,EAAE,UAAU,CAAC,CAAC;QACjD,CAAC,MAAM,IAAIyC,SAAS,KAAK,QAAQ,EAAE;UACjCZ,iBAAiB,CAACJ,IAAI,CAAC,CAACW,KAAK,CAACpC,GAAG,EAAE,WAAW,CAAC,CAAC;QAClD,CAAC,MAAM,IAAIyC,SAAS,KAAK,SAAS,EAAE;UAClCZ,iBAAiB,CAACJ,IAAI,CAAC,CAACW,KAAK,CAACpC,GAAG,EAAE,SAAS,CAAC,CAAC;QAChD;MACF;IACF;IAEA,OAAO6B,iBAAiB;EAC1B;EAEQhB,sBAAsBA,CAAA,EAAW;IACvC,MAAM6B,YAAsB,GAAG,EAAE;IACjC,MAAMzB,eAAe,GAAG,IAAI9B,GAAG,CAAS,CAAC;;IAEzC;IACA,KAAK,MAAME,QAAQ,IAAI,IAAI,CAACX,MAAM,CAACU,gBAAgB,EAAE;MACnD,MAAME,KAAK,GAAG,IAAI,CAACZ,MAAM,CAACa,MAAM,CAACF,QAAQ,CAAC;MAC1C,EAAUC,KAAK,IAAI,IAAI,IAAA4B,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAAvB/C,SAAS,QAAgB,oBAAoBgB,QAAQ,EAAE,IAAvDhB,SAAS;;MAET;MACA,IAAI,CAAC4C,eAAe,CAACI,GAAG,CAAChC,QAAQ,CAAC,EAAE;QAClCqD,YAAY,CAACjB,IAAI,CAAC,KAAKpC,QAAQ,KAAKA,QAAQ,OAAO,CAAC;QACpD4B,eAAe,CAAChB,GAAG,CAACZ,QAAQ,CAAC;MAC/B;;MAEA;MACA,IAAIC,KAAK,CAACE,IAAI,KAAK,OAAO,EAAE;QAC1B,KAAK,MAAMC,WAAW,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAACM,KAAK,CAACC,QAAQ,CAAC,EAAE;UAC3D,MAAMC,oBAAoB,GAAGxB,iBAAiB,CAACmB,WAAW,CAAC;UAC3D,MAAMM,eAAe,GAAG,GAAGT,KAAK,CAACM,KAAK,CAACI,GAAG,GAAGF,oBAAoB,EAAE;UAEnE,IAAI,CAACmB,eAAe,CAACI,GAAG,CAACtB,eAAe,CAAC,EAAE;YACzC2C,YAAY,CAACjB,IAAI,CAAC,KAAK1B,eAAe,KAAKA,eAAe,OAAO,CAAC;YAClEkB,eAAe,CAAChB,GAAG,CAACF,eAAe,CAAC;UACtC;QACF;MACF;IACF;IAEA,OAAO;AACX,EAAE2C,YAAY,CAACpC,IAAI,CAAC,KAAK,CAAC;AAC1B;AACA;AACA;AACA;AACA,mDAAmD;EACjD;AACF;AAEA,OAAO,eAAeqC,oBAAoBA,CACxCjE,MAA+B,EAC/BC,OAA+B,EACd;EACjB,MAAMiE,SAAS,GAAG,IAAIpE,cAAc,CAACE,MAAM,EAAEC,OAAO,CAAC;EACrD,OAAOiE,SAAS,CAAC/D,cAAc,CAAC,CAAC;AACnC","ignoreList":[]}

package/build/esm/utils/ir/generateZodSchemasFromIr.js

@@ -18,6 +18,7 @@ import consola from "consola";
 import invariant from "tiny-invariant";
 import { assertNever } from "../assertNever.js";
 import { formatVariantName } from "../formatVariantName.js";
+import { GENERATED_FILE_HEADER } from "../generatedFileHeader.js";
 class ZodSchemaGenerator {
   constructor(schema, options) {
     this.schema = schema;
@@ -52,7 +53,7 @@ class ZodSchemaGenerator {
     }
     imports += "\n";
     const schemasCode = this.generateZodSchemas();
-    return imports + schemasCode + "\n";
+    return GENERATED_FILE_HEADER + imports + schemasCode + "\n";
   }
   generateZodSchemas() {
     const schemaLines = [];
@@ -233,6 +234,8 @@ class ZodSchemaGenerator {
   }
   generateFieldValueSchema(fieldValue) {
     switch (fieldValue.type) {
+      case "boolean":
+        return "z.boolean()";
       case "datetime":
         return "z.string().datetime()";
       case "double":