@fluidframework/tree-agent 2.74.0-365691 → 2.74.0-370705

package/lib/renderZodTypeScript.js

@@ -0,0 +1,267 @@
+/*!
+ * Copyright (c) Microsoft Corporation and contributors. All rights reserved.
+ * Licensed under the MIT License.
+ */
+/* eslint-disable @typescript-eslint/no-unsafe-argument */
+import { UsageError } from "@fluidframework/telemetry-utils/internal";
+import { ObjectNodeSchema } from "@fluidframework/tree/alpha";
+import { z } from "zod";
+/**
+ * Converts Zod schema definitions into TypeScript declaration text.
+ */
+export function renderZodTypeScript(zodType, getFriendlyName, instanceOfLookup) {
+    let result = "";
+    let startOfLine = true;
+    let indent = 0;
+    appendType(zodType);
+    return result;
+    function appendType(type, minPrecedence = 2 /* TypePrecedence.Object */) {
+        const shouldParenthesize = getTypePrecendece(type) < minPrecedence;
+        if (shouldParenthesize) {
+            append("(");
+        }
+        appendTypeDefinition(type);
+        if (shouldParenthesize) {
+            append(")");
+        }
+    }
+    function append(s) {
+        if (startOfLine) {
+            result += "    ".repeat(indent);
+            startOfLine = false;
+        }
+        result += s;
+    }
+    function appendNewLine() {
+        append("\n");
+        startOfLine = true;
+    }
+    function appendTypeDefinition(type) {
+        switch (getTypeKind(type)) {
+            case z.ZodFirstPartyTypeKind.ZodString: {
+                append("string");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodNumber: {
+                append("number");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodBoolean: {
+                append("boolean");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodDate: {
+                append("Date");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodUndefined: {
+                append("undefined");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodNull: {
+                append("null");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodUnknown: {
+                append("unknown");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodArray: {
+                appendArrayType(type);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodObject: {
+                appendObjectType(type);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodUnion: {
+                appendUnionOrIntersectionTypes(type._def.options, 0 /* TypePrecedence.Union */);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodDiscriminatedUnion: {
+                appendUnionOrIntersectionTypes([...type._def.options.values()], 0 /* TypePrecedence.Union */);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodIntersection: {
+                appendUnionOrIntersectionTypes([
+                    type._def.left,
+                    type._def.right,
+                ], 1 /* TypePrecedence.Intersection */);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodTuple: {
+                appendTupleType(type);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodRecord: {
+                appendRecordType(type);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodMap: {
+                appendMapType(type);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodLiteral: {
+                appendLiteral(type._def.value);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodEnum: {
+                append(type._def.values.map((value) => JSON.stringify(value)).join(" | "));
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodOptional: {
+                appendUnionOrIntersectionTypes([type._def.innerType, z.undefined()], 0 /* TypePrecedence.Union */);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodReadonly: {
+                appendReadonlyType(type);
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodEffects: {
+                const schema = instanceOfLookup.get(type);
+                if (schema === undefined) {
+                    throw new UsageError(`Unsupported zod effects type when formatting helper types: ${getTypeKind(type)}`);
+                }
+                append(getFriendlyName(schema));
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodVoid: {
+                append("void");
+                return;
+            }
+            case z.ZodFirstPartyTypeKind.ZodLazy: {
+                appendType(type._def.getter());
+                return;
+            }
+            default: {
+                throw new UsageError(`Unsupported type when formatting helper types: ${getTypeKind(type)}`);
+            }
+        }
+    }
+    function appendArrayType(arrayType) {
+        appendType(arrayType._def.type, 2 /* TypePrecedence.Object */);
+        append("[]");
+    }
+    function appendObjectType(objectType) {
+        append("{");
+        appendNewLine();
+        indent++;
+        for (const [name, entry] of Object.entries(objectType._def.shape())) {
+            let propertyType = entry;
+            append(name);
+            if (getTypeKind(propertyType) === z.ZodFirstPartyTypeKind.ZodOptional) {
+                append("?");
+                propertyType = propertyType._def.innerType;
+            }
+            append(": ");
+            appendType(propertyType);
+            append(";");
+            appendNewLine();
+        }
+        indent--;
+        append("}");
+    }
+    function appendUnionOrIntersectionTypes(types, minPrecedence) {
+        let first = true;
+        for (const innerType of types) {
+            if (!first) {
+                append(minPrecedence === 1 /* TypePrecedence.Intersection */ ? " & " : " | ");
+            }
+            appendType(innerType, minPrecedence);
+            first = false;
+        }
+    }
+    function appendTupleType(tupleType) {
+        append("[");
+        let first = true;
+        for (const innerType of tupleType._def
+            .items) {
+            if (!first) {
+                append(", ");
+            }
+            if (getTypeKind(innerType) === z.ZodFirstPartyTypeKind.ZodOptional) {
+                appendType(innerType._def.innerType, 2 /* TypePrecedence.Object */);
+                append("?");
+            }
+            else {
+                appendType(innerType);
+            }
+            first = false;
+        }
+        const rest = tupleType._def.rest;
+        if (rest !== null) {
+            if (!first) {
+                append(", ");
+            }
+            append("...");
+            appendType(rest, 2 /* TypePrecedence.Object */);
+            append("[]");
+        }
+        append("]");
+    }
+    function appendRecordType(recordType) {
+        append("Record<");
+        appendType(recordType._def.keyType);
+        append(", ");
+        appendType(recordType._def.valueType);
+        append(">");
+    }
+    function appendMapType(mapType) {
+        append("Map<");
+        appendType(mapType._def.keyType);
+        append(", ");
+        appendType(mapType._def.valueType);
+        append(">");
+    }
+    function appendLiteral(value) {
+        append(typeof value === "string" || typeof value === "number" || typeof value === "boolean"
+            ? JSON.stringify(value)
+            : "any");
+    }
+    function appendReadonlyType(readonlyType) {
+        append("Readonly<");
+        appendType(readonlyType._def.innerType);
+        append(">");
+    }
+}
+function getTypeKind(type) {
+    return type._def.typeName;
+}
+var TypePrecedence;
+(function (TypePrecedence) {
+    TypePrecedence[TypePrecedence["Union"] = 0] = "Union";
+    TypePrecedence[TypePrecedence["Intersection"] = 1] = "Intersection";
+    TypePrecedence[TypePrecedence["Object"] = 2] = "Object";
+})(TypePrecedence || (TypePrecedence = {}));
+function getTypePrecendece(type) {
+    switch (getTypeKind(type)) {
+        case z.ZodFirstPartyTypeKind.ZodEnum:
+        case z.ZodFirstPartyTypeKind.ZodUnion:
+        case z.ZodFirstPartyTypeKind.ZodDiscriminatedUnion: {
+            return 0 /* TypePrecedence.Union */;
+        }
+        case z.ZodFirstPartyTypeKind.ZodIntersection: {
+            return 1 /* TypePrecedence.Intersection */;
+        }
+        default: {
+            return 2 /* TypePrecedence.Object */;
+        }
+    }
+}
+/**
+ * Create a Zod schema for a SharedTree schema class.
+ * @alpha
+ */
+export function instanceOf(schema) {
+    if (!(schema instanceof ObjectNodeSchema)) {
+        throw new UsageError(`${schema.identifier} must be an instance of ObjectNodeSchema.`);
+    }
+    const effect = z.instanceof(schema);
+    instanceOfs.set(effect, schema);
+    return effect;
+}
+/**
+ * A lookup from Zod instanceOf schemas to their corresponding ObjectNodeSchema.
+ */
+export const instanceOfs = new WeakMap();
+//# sourceMappingURL=renderZodTypeScript.js.map

package/lib/renderZodTypeScript.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"renderZodTypeScript.js","sourceRoot":"","sources":["../src/renderZodTypeScript.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,0DAA0D;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,0CAA0C,CAAC;AAEtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAClC,OAAqB,EACrB,eAAmD,EACnD,gBAAyD;IAEzD,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,WAAW,GAAG,IAAI,CAAC;IACvB,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,UAAU,CAAC,OAAO,CAAC,CAAC;IACpB,OAAO,MAAM,CAAC;IAEd,SAAS,UAAU,CAAC,IAAkB,EAAE,aAAa,gCAAwB;QAC5E,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC;QACnE,IAAI,kBAAkB,EAAE,CAAC;YACxB,MAAM,CAAC,GAAG,CAAC,CAAC;QACb,CAAC;QACD,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC3B,IAAI,kBAAkB,EAAE,CAAC;YACxB,MAAM,CAAC,GAAG,CAAC,CAAC;QACb,CAAC;IACF,CAAC;IAED,SAAS,MAAM,CAAC,CAAS;QACxB,IAAI,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAChC,WAAW,GAAG,KAAK,CAAC;QACrB,CAAC;QACD,MAAM,IAAI,CAAC,CAAC;IACb,CAAC;IAED,SAAS,aAAa;QACrB,MAAM,CAAC,IAAI,CAAC,CAAC;QACb,WAAW,GAAG,IAAI,CAAC;IACpB,CAAC;IAED,SAAS,oBAAoB,CAAC,IAAkB;QAC/C,QAAQ,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,KAAK,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACjB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACjB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CAAC;gBACzC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAClB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;gBACtC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACf,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC,CAAC;gBAC3C,MAAM,CAAC,WAAW,CAAC,CAAC;gBACpB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;gBACtC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACf,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CAAC;gBACzC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAClB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACvC,eAAe,CAAC,IAAI,CAAC,CAAC;gBACtB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxC,gBAAgB,CAAC,IAAI,CAAC,CAAC;gBACvB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACvC,8BAA8B,CAC5B,IAAI,CAAC,IAAsB,CAAC,OAAO,+BAEpC,CAAC;gBACF,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBACpD,8BAA8B,CAC7B,CAAC,GAAI,IAAI,CAAC,IAA2C,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,+BAEvE,CAAC;gBACF,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,eAAe,CAAC,CAAC,CAAC;gBAC9C,8BAA8B,CAC7B;oBACE,IAAI,CAAC,IAA6B,CAAC,IAAI;oBACvC,IAAI,CAAC,IAA6B,CAAC,KAAK;iBACzC,sCAED,CAAC;gBACF,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACvC,eAAe,CAAC,IAAI,CAAC,CAAC;gBACtB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxC,gBAAgB,CAAC,IAAI,CAAC,CAAC;gBACvB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,CAAC;gBACpB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CAAC;gBACzC,aAAa,CAAE,IAAI,CAAC,IAAwB,CAAC,KAAK,CAAC,CAAC;gBACpD,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;gBACtC,MAAM,CACJ,IAAI,CAAC,IAAqB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CACpF,CAAC;gBACF,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,CAAC;gBAC1C,8BAA8B,CAC7B,CAAE,IAAI,CAAC,IAAyB,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,+BAE1D,CAAC;gBACF,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,CAAC;gBAC1C,kBAAkB,CAAC,IAAI,CAAC,CAAC;gBACzB,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CAAC;gBACzC,MAAM,MAAM,GAAG,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC1C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC1B,MAAM,IAAI,UAAU,CACnB,8DAA8D,WAAW,CAAC,IAAI,CAAC,EAAE,CACjF,CAAC;gBACH,CAAC;gBACD,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC;gBAChC,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;gBACtC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACf,OAAO;YACR,CAAC;YACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;gBACtC,UAAU,CAAE,IAAI,CAAC,IAAqB,CAAC,MAAM,EAAE,CAAC,CAAC;gBACjD,OAAO;YACR,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACT,MAAM,IAAI,UAAU,CACnB,kDAAkD,WAAW,CAAC,IAAI,CAAC,EAAE,CACrE,CAAC;YACH,CAAC;QACF,CAAC;IACF,CAAC;IAED,SAAS,eAAe,CAAC,SAAuB;QAC/C,UAAU,CAAE,SAAS,CAAC,IAAsB,CAAC,IAAI,gCAAwB,CAAC;QAC1E,MAAM,CAAC,IAAI,CAAC,CAAC;IACd,CAAC;IAED,SAAS,gBAAgB,CAAC,UAAwB;QACjD,MAAM,CAAC,GAAG,CAAC,CAAC;QACZ,aAAa,EAAE,CAAC;QAChB,MAAM,EAAE,CAAC;QACT,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAE,UAAU,CAAC,IAAuB,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC;YACzF,IAAI,YAAY,GAAG,KAAK,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,CAAC;YACb,IAAI,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,WAAW,EAAE,CAAC;gBACvE,MAAM,CAAC,GAAG,CAAC,CAAC;gBACZ,YAAY,GAAI,YAAY,CAAC,IAAyB,CAAC,SAAS,CAAC;YAClE,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,CAAC;YACb,UAAU,CAAC,YAAY,CAAC,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,CAAC;YACZ,aAAa,EAAE,CAAC;QACjB,CAAC;QACD,MAAM,EAAE,CAAC;QACT,MAAM,CAAC,GAAG,CAAC,CAAC;IACb,CAAC;IAED,SAAS,8BAA8B,CACtC,KAA8B,EAC9B,aAA6B;QAE7B,IAAI,KAAK,GAAG,IAAI,CAAC;QACjB,KAAK,MAAM,SAAS,IAAI,KAAK,EAAE,CAAC;YAC/B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACZ,MAAM,CAAC,aAAa,wCAAgC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACvE,CAAC;YACD,UAAU,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;YACrC,KAAK,GAAG,KAAK,CAAC;QACf,CAAC;IACF,CAAC;IAED,SAAS,eAAe,CAAC,SAAuB;QAC/C,MAAM,CAAC,GAAG,CAAC,CAAC;QACZ,IAAI,KAAK,GAAG,IAAI,CAAC;QACjB,KAAK,MAAM,SAAS,IAAK,SAAS,CAAC,IAAkD;aACnF,KAAK,EAAE,CAAC;YACT,IAAI,CAAC,KAAK,EAAE,CAAC;gBACZ,MAAM,CAAC,IAAI,CAAC,CAAC;YACd,CAAC;YACD,IAAI,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,WAAW,EAAE,CAAC;gBACpE,UAAU,CAAE,SAAS,CAAC,IAAyB,CAAC,SAAS,gCAAwB,CAAC;gBAClF,MAAM,CAAC,GAAG,CAAC,CAAC;YACb,CAAC;iBAAM,CAAC;gBACP,UAAU,CAAC,SAAS,CAAC,CAAC;YACvB,CAAC;YACD,KAAK,GAAG,KAAK,CAAC;QACf,CAAC;QACD,MAAM,IAAI,GAAI,SAAS,CAAC,IAAyD,CAAC,IAAI,CAAC;QACvF,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YACnB,IAAI,CAAC,KAAK,EAAE,CAAC;gBACZ,MAAM,CAAC,IAAI,CAAC,CAAC;YACd,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,CAAC;YACd,UAAU,CAAC,IAAI,gCAAwB,CAAC;YACxC,MAAM,CAAC,IAAI,CAAC,CAAC;QACd,CAAC;QACD,MAAM,CAAC,GAAG,CAAC,CAAC;IACb,CAAC;IAED,SAAS,gBAAgB,CAAC,UAAwB;QACjD,MAAM,CAAC,SAAS,CAAC,CAAC;QAClB,UAAU,CAAE,UAAU,CAAC,IAAuB,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,CAAC;QACb,UAAU,CAAE,UAAU,CAAC,IAAuB,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,CAAC,GAAG,CAAC,CAAC;IACb,CAAC;IAED,SAAS,aAAa,CAAC,OAAqB;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC;QACf,UAAU,CAAE,OAAO,CAAC,IAAoB,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,CAAC,IAAI,CAAC,CAAC;QACb,UAAU,CAAE,OAAO,CAAC,IAAoB,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,GAAG,CAAC,CAAC;IACb,CAAC;IAED,SAAS,aAAa,CAAC,KAAc;QACpC,MAAM,CACL,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS;YACnF,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YACvB,CAAC,CAAC,KAAK,CACR,CAAC;IACH,CAAC;IAED,SAAS,kBAAkB,CAAC,YAA0B;QACrD,MAAM,CAAC,WAAW,CAAC,CAAC;QACpB,UAAU,CAAE,YAAY,CAAC,IAAyB,CAAC,SAAS,CAAC,CAAC;QAC9D,MAAM,CAAC,GAAG,CAAC,CAAC;IACb,CAAC;AACF,CAAC;AAED,SAAS,WAAW,CAAC,IAAe;IACnC,OAAQ,IAAI,CAAC,IAA6D,CAAC,QAAQ,CAAC;AACrF,CAAC;AAED,IAAW,cAIV;AAJD,WAAW,cAAc;IACxB,qDAAS,CAAA;IACT,mEAAgB,CAAA;IAChB,uDAAU,CAAA;AACX,CAAC,EAJU,cAAc,KAAd,cAAc,QAIxB;AAED,SAAS,iBAAiB,CAAC,IAAe;IACzC,QAAQ,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3B,KAAK,CAAC,CAAC,qBAAqB,CAAC,OAAO,CAAC;QACrC,KAAK,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC;QACtC,KAAK,CAAC,CAAC,qBAAqB,CAAC,qBAAqB,CAAC,CAAC,CAAC;YACpD,oCAA4B;QAC7B,CAAC;QACD,KAAK,CAAC,CAAC,qBAAqB,CAAC,eAAe,CAAC,CAAC,CAAC;YAC9C,2CAAmC;QACpC,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;YACT,qCAA6B;QAC9B,CAAC;IACF,CAAC;AACF,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,UAAU,CACzB,MAAS;IAET,IAAI,CAAC,CAAC,MAAM,YAAY,gBAAgB,CAAC,EAAE,CAAC;QAC3C,MAAM,IAAI,UAAU,CAAC,GAAG,MAAM,CAAC,UAAU,2CAA2C,CAAC,CAAC;IACvF,CAAC;IACD,MAAM,MAAM,GAAG,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACpC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,OAAO,MAAM,CAAC;AACf,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG,IAAI,OAAO,EAAkC,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\n/* eslint-disable @typescript-eslint/no-unsafe-argument */\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport type { TreeNodeSchema, TreeNodeSchemaClass } from \"@fluidframework/tree/alpha\";\nimport { ObjectNodeSchema } from \"@fluidframework/tree/alpha\";\nimport { z } from \"zod\";\n\n/**\n * Converts Zod schema definitions into TypeScript declaration text.\n */\nexport function renderZodTypeScript(\n\tzodType: z.ZodTypeAny,\n\tgetFriendlyName: (schema: TreeNodeSchema) => string,\n\tinstanceOfLookup: WeakMap<z.ZodTypeAny, ObjectNodeSchema>,\n): string {\n\tlet result = \"\";\n\tlet startOfLine = true;\n\tlet indent = 0;\n\n\tappendType(zodType);\n\treturn result;\n\n\tfunction appendType(type: z.ZodTypeAny, minPrecedence = TypePrecedence.Object): void {\n\t\tconst shouldParenthesize = getTypePrecendece(type) < minPrecedence;\n\t\tif (shouldParenthesize) {\n\t\t\tappend(\"(\");\n\t\t}\n\t\tappendTypeDefinition(type);\n\t\tif (shouldParenthesize) {\n\t\t\tappend(\")\");\n\t\t}\n\t}\n\n\tfunction append(s: string): void {\n\t\tif (startOfLine) {\n\t\t\tresult += \"    \".repeat(indent);\n\t\t\tstartOfLine = false;\n\t\t}\n\t\tresult += s;\n\t}\n\n\tfunction appendNewLine(): void {\n\t\tappend(\"\\n\");\n\t\tstartOfLine = true;\n\t}\n\n\tfunction appendTypeDefinition(type: z.ZodTypeAny): void {\n\t\tswitch (getTypeKind(type)) {\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodString: {\n\t\t\t\tappend(\"string\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodNumber: {\n\t\t\t\tappend(\"number\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodBoolean: {\n\t\t\t\tappend(\"boolean\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodDate: {\n\t\t\t\tappend(\"Date\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodUndefined: {\n\t\t\t\tappend(\"undefined\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodNull: {\n\t\t\t\tappend(\"null\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodUnknown: {\n\t\t\t\tappend(\"unknown\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodArray: {\n\t\t\t\tappendArrayType(type);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodObject: {\n\t\t\t\tappendObjectType(type);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodUnion: {\n\t\t\t\tappendUnionOrIntersectionTypes(\n\t\t\t\t\t(type._def as z.ZodUnionDef).options,\n\t\t\t\t\tTypePrecedence.Union,\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodDiscriminatedUnion: {\n\t\t\t\tappendUnionOrIntersectionTypes(\n\t\t\t\t\t[...(type._def as z.ZodDiscriminatedUnionDef<string>).options.values()],\n\t\t\t\t\tTypePrecedence.Union,\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodIntersection: {\n\t\t\t\tappendUnionOrIntersectionTypes(\n\t\t\t\t\t[\n\t\t\t\t\t\t(type._def as z.ZodIntersectionDef).left,\n\t\t\t\t\t\t(type._def as z.ZodIntersectionDef).right,\n\t\t\t\t\t],\n\t\t\t\t\tTypePrecedence.Intersection,\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodTuple: {\n\t\t\t\tappendTupleType(type);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodRecord: {\n\t\t\t\tappendRecordType(type);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodMap: {\n\t\t\t\tappendMapType(type);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodLiteral: {\n\t\t\t\tappendLiteral((type._def as z.ZodLiteralDef).value);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodEnum: {\n\t\t\t\tappend(\n\t\t\t\t\t(type._def as z.ZodEnumDef).values.map((value) => JSON.stringify(value)).join(\" | \"),\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodOptional: {\n\t\t\t\tappendUnionOrIntersectionTypes(\n\t\t\t\t\t[(type._def as z.ZodOptionalDef).innerType, z.undefined()],\n\t\t\t\t\tTypePrecedence.Union,\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodReadonly: {\n\t\t\t\tappendReadonlyType(type);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodEffects: {\n\t\t\t\tconst schema = instanceOfLookup.get(type);\n\t\t\t\tif (schema === undefined) {\n\t\t\t\t\tthrow new UsageError(\n\t\t\t\t\t\t`Unsupported zod effects type when formatting helper types: ${getTypeKind(type)}`,\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t\tappend(getFriendlyName(schema));\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodVoid: {\n\t\t\t\tappend(\"void\");\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase z.ZodFirstPartyTypeKind.ZodLazy: {\n\t\t\t\tappendType((type._def as z.ZodLazyDef).getter());\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tdefault: {\n\t\t\t\tthrow new UsageError(\n\t\t\t\t\t`Unsupported type when formatting helper types: ${getTypeKind(type)}`,\n\t\t\t\t);\n\t\t\t}\n\t\t}\n\t}\n\n\tfunction appendArrayType(arrayType: z.ZodTypeAny): void {\n\t\tappendType((arrayType._def as z.ZodArrayDef).type, TypePrecedence.Object);\n\t\tappend(\"[]\");\n\t}\n\n\tfunction appendObjectType(objectType: z.ZodTypeAny): void {\n\t\tappend(\"{\");\n\t\tappendNewLine();\n\t\tindent++;\n\t\tfor (const [name, entry] of Object.entries((objectType._def as z.ZodObjectDef).shape())) {\n\t\t\tlet propertyType = entry;\n\t\t\tappend(name);\n\t\t\tif (getTypeKind(propertyType) === z.ZodFirstPartyTypeKind.ZodOptional) {\n\t\t\t\tappend(\"?\");\n\t\t\t\tpropertyType = (propertyType._def as z.ZodOptionalDef).innerType;\n\t\t\t}\n\t\t\tappend(\": \");\n\t\t\tappendType(propertyType);\n\t\t\tappend(\";\");\n\t\t\tappendNewLine();\n\t\t}\n\t\tindent--;\n\t\tappend(\"}\");\n\t}\n\n\tfunction appendUnionOrIntersectionTypes(\n\t\ttypes: readonly z.ZodTypeAny[],\n\t\tminPrecedence: TypePrecedence,\n\t): void {\n\t\tlet first = true;\n\t\tfor (const innerType of types) {\n\t\t\tif (!first) {\n\t\t\t\tappend(minPrecedence === TypePrecedence.Intersection ? \" & \" : \" | \");\n\t\t\t}\n\t\t\tappendType(innerType, minPrecedence);\n\t\t\tfirst = false;\n\t\t}\n\t}\n\n\tfunction appendTupleType(tupleType: z.ZodTypeAny): void {\n\t\tappend(\"[\");\n\t\tlet first = true;\n\t\tfor (const innerType of (tupleType._def as z.ZodTupleDef<z.ZodTupleItems, z.ZodType>)\n\t\t\t.items) {\n\t\t\tif (!first) {\n\t\t\t\tappend(\", \");\n\t\t\t}\n\t\t\tif (getTypeKind(innerType) === z.ZodFirstPartyTypeKind.ZodOptional) {\n\t\t\t\tappendType((innerType._def as z.ZodOptionalDef).innerType, TypePrecedence.Object);\n\t\t\t\tappend(\"?\");\n\t\t\t} else {\n\t\t\t\tappendType(innerType);\n\t\t\t}\n\t\t\tfirst = false;\n\t\t}\n\t\tconst rest = (tupleType._def as z.ZodTupleDef<z.ZodTupleItems, z.ZodType | null>).rest;\n\t\tif (rest !== null) {\n\t\t\tif (!first) {\n\t\t\t\tappend(\", \");\n\t\t\t}\n\t\t\tappend(\"...\");\n\t\t\tappendType(rest, TypePrecedence.Object);\n\t\t\tappend(\"[]\");\n\t\t}\n\t\tappend(\"]\");\n\t}\n\n\tfunction appendRecordType(recordType: z.ZodTypeAny): void {\n\t\tappend(\"Record<\");\n\t\tappendType((recordType._def as z.ZodRecordDef).keyType);\n\t\tappend(\", \");\n\t\tappendType((recordType._def as z.ZodRecordDef).valueType);\n\t\tappend(\">\");\n\t}\n\n\tfunction appendMapType(mapType: z.ZodTypeAny): void {\n\t\tappend(\"Map<\");\n\t\tappendType((mapType._def as z.ZodMapDef).keyType);\n\t\tappend(\", \");\n\t\tappendType((mapType._def as z.ZodMapDef).valueType);\n\t\tappend(\">\");\n\t}\n\n\tfunction appendLiteral(value: unknown): void {\n\t\tappend(\n\t\t\ttypeof value === \"string\" || typeof value === \"number\" || typeof value === \"boolean\"\n\t\t\t\t? JSON.stringify(value)\n\t\t\t\t: \"any\",\n\t\t);\n\t}\n\n\tfunction appendReadonlyType(readonlyType: z.ZodTypeAny): void {\n\t\tappend(\"Readonly<\");\n\t\tappendType((readonlyType._def as z.ZodReadonlyDef).innerType);\n\t\tappend(\">\");\n\t}\n}\n\nfunction getTypeKind(type: z.ZodType): z.ZodFirstPartyTypeKind {\n\treturn (type._def as z.ZodTypeDef & { typeName: z.ZodFirstPartyTypeKind }).typeName;\n}\n\nconst enum TypePrecedence {\n\tUnion = 0,\n\tIntersection = 1,\n\tObject = 2,\n}\n\nfunction getTypePrecendece(type: z.ZodType): TypePrecedence {\n\tswitch (getTypeKind(type)) {\n\t\tcase z.ZodFirstPartyTypeKind.ZodEnum:\n\t\tcase z.ZodFirstPartyTypeKind.ZodUnion:\n\t\tcase z.ZodFirstPartyTypeKind.ZodDiscriminatedUnion: {\n\t\t\treturn TypePrecedence.Union;\n\t\t}\n\t\tcase z.ZodFirstPartyTypeKind.ZodIntersection: {\n\t\t\treturn TypePrecedence.Intersection;\n\t\t}\n\t\tdefault: {\n\t\t\treturn TypePrecedence.Object;\n\t\t}\n\t}\n}\n\n/**\n * Create a Zod schema for a SharedTree schema class.\n * @alpha\n */\nexport function instanceOf<T extends TreeNodeSchemaClass>(\n\tschema: T,\n): z.ZodType<InstanceType<T>, z.ZodTypeDef, InstanceType<T>> {\n\tif (!(schema instanceof ObjectNodeSchema)) {\n\t\tthrow new UsageError(`${schema.identifier} must be an instance of ObjectNodeSchema.`);\n\t}\n\tconst effect = z.instanceof(schema);\n\tinstanceOfs.set(effect, schema);\n\treturn effect;\n}\n\n/**\n * A lookup from Zod instanceOf schemas to their corresponding ObjectNodeSchema.\n */\nexport const instanceOfs = new WeakMap<z.ZodTypeAny, ObjectNodeSchema>();\n"]}

package/lib/typeGeneration.d.ts

@@ -3,11 +3,9 @@
  * Licensed under the MIT License.
  */
 import type { ImplicitFieldSchema, SimpleTreeSchema } from "@fluidframework/tree/internal";
-import { type ZodTypeAny } from "zod";
+import { type SchemaTypeScriptRenderResult } from "./renderSchemaTypeScript.js";
 /**
- * TODO
+ * Generates TypeScript declarations for the schemas reachable from the provided root field schema.
  */
-export declare function generateEditTypesForPrompt(rootSchema: ImplicitFieldSchema, schema: SimpleTreeSchema): {
-    domainTypes: Record<string, ZodTypeAny>;
-};
+export declare function generateEditTypesForPrompt(rootSchema: ImplicitFieldSchema, schema: SimpleTreeSchema): SchemaTypeScriptRenderResult;
 //# sourceMappingURL=typeGeneration.d.ts.map

package/lib/typeGeneration.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"typeGeneration.d.ts","sourceRoot":"","sources":["../src/typeGeneration.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAeH,OAAO,KAAK,EACX,mBAAmB,EAGnB,gBAAgB,EAEhB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAK,KAAK,UAAU,EAAE,MAAM,KAAK,CAAC;AAkCzC;;GAEG;AACH,wBAAgB,0BAA0B,CACzC,UAAU,EAAE,mBAAmB,EAC/B,MAAM,EAAE,gBAAgB,GACtB;IACF,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;CACxC,CAqCA"}
+{"version":3,"file":"typeGeneration.d.ts","sourceRoot":"","sources":["../src/typeGeneration.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,KAAK,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAK3F,OAAO,EAEN,KAAK,4BAA4B,EACjC,MAAM,6BAA6B,CAAC;AAKrC;;GAEG;AACH,wBAAgB,0BAA0B,CACzC,UAAU,EAAE,mBAAmB,EAC/B,MAAM,EAAE,gBAAgB,GACtB,4BAA4B,CAI9B"}

package/lib/typeGeneration.js

@@ -2,273 +2,40 @@
  * Copyright (c) Microsoft Corporation and contributors. All rights reserved.
  * Licensed under the MIT License.
  */
-import { assert, unreachableCase } from "@fluidframework/core-utils/internal";
-import { UsageError } from "@fluidframework/telemetry-utils/internal";
-import { ArrayNodeSchema, FieldKind, getSimpleSchema, MapNodeSchema, NodeKind, ObjectNodeSchema, RecordNodeSchema, ValueSchema, walkFieldSchema, } from "@fluidframework/tree/internal";
-import { z } from "zod";
-import { FunctionWrapper, getExposedMethods } from "./methodBinding.js";
+import { getSimpleSchema, walkFieldSchema } from "@fluidframework/tree/internal";
+import { getExposedMethods, isBindableSchema } from "./methodBinding.js";
 import { getExposedProperties } from "./propertyBinding.js";
-import { fail, getOrCreate, hasAtLeastTwo, llmDefault, mapIterable, tryGetSingleton, } from "./utils.js";
-/**
- *
- * TODO: Add a prompt suggestion API!
- *
- * TODO: Handle rate limit errors.
- *
- * TODO: Pass descriptions from schema metadata to the generated TS types that we put in the prompt
- *
- * TODO make the Ids be "Vector-2" instead of "Vector2" (or else it gets weird when you have a type called "Vector2")
- */
-/**
- * Cache used to prevent repeatedly generating the same Zod validation objects for the same {@link SimpleTreeSchema} as generate propts for repeated calls to an LLM
- */
+import { renderSchemaTypeScript, } from "./renderSchemaTypeScript.js";
+import { getOrCreate } from "./utils.js";
 const promptSchemaCache = new WeakMap();
 /**
- * TODO
+ * Generates TypeScript declarations for the schemas reachable from the provided root field schema.
  */
 export function generateEditTypesForPrompt(rootSchema, schema) {
-    return getOrCreate(promptSchemaCache, schema, () => {
-        const allSchemas = new Set();
-        const objectTypeSchemas = new Set();
-        walkFieldSchema(rootSchema, {
-            node: (n) => {
-                allSchemas.add(n);
-                if (n instanceof ObjectNodeSchema ||
-                    n instanceof MapNodeSchema ||
-                    n instanceof ArrayNodeSchema ||
-                    n instanceof RecordNodeSchema) {
-                    objectTypeSchemas.add(n);
-                    const exposedMethods = getExposedMethods(n);
-                    for (const t of exposedMethods.referencedTypes) {
-                        allSchemas.add(t);
-                        objectTypeSchemas.add(t);
+    return getOrCreate(promptSchemaCache, schema, () => buildPromptSchemaDescription(rootSchema));
+}
+function buildPromptSchemaDescription(rootSchema) {
+    const bindableSchemas = new Map();
+    walkFieldSchema(rootSchema, {
+        node: (node) => {
+            if (isBindableSchema(node)) {
+                bindableSchemas.set(node.identifier, node);
+                const exposedMethods = getExposedMethods(node);
+                for (const referenced of exposedMethods.referencedTypes) {
+                    if (isBindableSchema(referenced)) {
+                        bindableSchemas.set(referenced.identifier, referenced);
                     }
-                    const exposedProperties = getExposedProperties(n);
-                    for (const t of exposedProperties.referencedTypes) {
-                        allSchemas.add(t);
-                        objectTypeSchemas.add(t);
-                    }
-                }
-            },
-        });
-        const nodeSchemas = [...objectTypeSchemas.values()];
-        const bindableSchemas = new Map(nodeSchemas.map((nodeSchema) => [nodeSchema.identifier, nodeSchema]));
-        return generateEditTypes([...allSchemas.values()].map((s) => getSimpleSchema(s)), new Map(), bindableSchemas);
-    });
-}
-/**
- * Generates a set of ZOD validation objects for the various types of data that can be put into the provided {@link SimpleTreeSchema}
- * and then uses those sets to generate an all-encompassing ZOD object for each type of {@link TreeEdit} that can validate any of the types of data that can be put into the tree.
- *
- * @returns a Record of schema names to Zod validation objects, and the name of the root schema used to encompass all of the other schemas.
- *
- * @remarks The return type of this function is designed to work with Typechat's createZodJsonValidator as well as be used as the JSON schema for OpenAi's structured output response format.
- */
-function generateEditTypes(schemas, objectCache, bindableSchemas) {
-    const domainTypes = {};
-    for (const schema of schemas) {
-        for (const name of schema.definitions.keys()) {
-            // If this does overwrite anything in domainTypes, it is guaranteed to be overwritten with an identical value due to the getOrCreate
-            domainTypes[name] = getOrCreateType(schema.definitions, name, objectCache, bindableSchemas);
-        }
-    }
-    return {
-        domainTypes,
-    };
-}
-function getBoundMembersForBindable(bindableSchema) {
-    const memberTypes = [];
-    const methods = getExposedMethods(bindableSchema);
-    for (const [name, method] of Object.entries(methods.methods)) {
-        const zodFunction = z.instanceof(FunctionWrapper);
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any
-        zodFunction.method = method;
-        memberTypes.push([name, zodFunction]);
-    }
-    const props = getExposedProperties(bindableSchema);
-    for (const [name, prop] of Object.entries(props.properties)) {
-        const schema = prop.schema;
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any
-        schema.property = prop;
-        memberTypes.push([name, prop.schema]);
-    }
-    return {
-        members: memberTypes,
-        referencedTypes: new Set([...props.referencedTypes, ...methods.referencedTypes]),
-    };
-}
-function getBoundMembers(definition, bindableSchemas) {
-    const bindableSchema = bindableSchemas.get(definition) ?? fail("unknown definition");
-    return getBoundMembersForBindable(bindableSchema).members;
-}
-function addBindingIntersectionIfNeeded(typeString, zodTypeBound, definition, simpleNodeSchema, bindableSchemas) {
-    let zodType = zodTypeBound;
-    let description = simpleNodeSchema.metadata?.description ?? "";
-    const boundMembers = getBoundMembers(definition, bindableSchemas);
-    const methods = {};
-    const properties = {};
-    let hasProperties = false;
-    let hasMethods = false;
-    if (boundMembers.length > 0) {
-        for (const [name, zodMember] of boundMembers) {
-            const kind = 
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/no-unsafe-member-access
-            zodMember.method === undefined
-                ? // eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/no-unsafe-member-access
-                    zodMember.property === undefined
-                        ? "Unknown"
-                        : "Property"
-                : "Method";
-            if (kind === "Unknown") {
-                throw new UsageError(`Unrecognized bound member ${name} in schema ${definition}`);
-            }
-            if (kind === "Method") {
-                if (methods[name] !== undefined) {
-                    throw new UsageError(`${kind} ${name} conflicts with field of the same name in schema ${definition}`);
-                }
-                methods[name] = zodMember;
-            }
-            if (kind === "Property") {
-                if (properties[name] !== undefined) {
-                    throw new UsageError(`${kind} ${name} conflicts with field of the same name in schema ${definition}`);
                 }
-                properties[name] = zodMember;
-            }
-        }
-        hasMethods = Object.keys(methods).length > 0 ? true : false;
-        hasProperties = Object.keys(properties).length > 0 ? true : false;
-        if (hasMethods) {
-            zodType = z.intersection(zodType, z.object(methods));
-        }
-        if (hasProperties) {
-            zodType = z.intersection(zodType, z.object(properties));
-        }
-    }
-    let note = "";
-    if (hasMethods && hasProperties) {
-        note = `Note: this ${typeString} has custom user-defined methods and properties directly on it.`;
-    }
-    else if (hasMethods) {
-        note = `Note: this ${typeString} has custom user-defined methods directly on it.`;
-    }
-    else if (hasProperties) {
-        note = `Note: this ${typeString} has custom user-defined properties directly on it.`;
-    }
-    if (note !== "") {
-        description = description === "" ? note : `${description} - ${note}`;
-    }
-    return zodType.describe(description);
-}
-function getOrCreateType(definitionMap, definition, objectCache, bindableSchemas) {
-    const simpleNodeSchema = definitionMap.get(definition) ?? fail("Unexpected definition");
-    return getOrCreate(objectCache, simpleNodeSchema, () => {
-        // Handle recursive types: temporarily create a zod "lazy" type that can be referenced by a recursive call to getOrCreateType.
-        let type;
-        objectCache.set(simpleNodeSchema, z.lazy(() => type ?? fail("Recursive type used before creation")));
-        switch (simpleNodeSchema.kind) {
-            case NodeKind.Object: {
-                // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
-                const properties = Object.fromEntries([...simpleNodeSchema.fields]
-                    .map(([key, field]) => {
-                    return [
-                        key,
-                        getOrCreateTypeForField(definitionMap, field, objectCache, bindableSchemas),
-                    ];
-                })
-                    .filter(([, value]) => value !== undefined));
-                for (const [name, zodMember] of getBoundMembers(definition, bindableSchemas)) {
-                    // eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/no-unsafe-member-access
-                    const kind = zodMember.method === undefined ? "Property" : "Method";
-                    if (properties[name] !== undefined) {
-                        throw new UsageError(`${kind} ${name} conflicts with field of the same name in schema ${definition}`);
+                const exposedProperties = getExposedProperties(node);
+                for (const referenced of exposedProperties.referencedTypes) {
+                    if (isBindableSchema(referenced)) {
+                        bindableSchemas.set(referenced.identifier, referenced);
                     }
-                    properties[name] = zodMember;
                 }
-                return (type = z
-                    .object(properties)
-                    .describe(simpleNodeSchema.metadata?.description ?? ""));
-            }
-            case NodeKind.Map: {
-                const zodType = z.map(z.string(), getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
-                return (type = addBindingIntersectionIfNeeded("map", zodType, definition, simpleNodeSchema, bindableSchemas));
-            }
-            case NodeKind.Record: {
-                const zodType = z.record(getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
-                return (type = addBindingIntersectionIfNeeded("record", zodType, definition, simpleNodeSchema, bindableSchemas));
             }
-            case NodeKind.Array: {
-                const zodType = z.array(getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
-                return (type = addBindingIntersectionIfNeeded("array", zodType, definition, simpleNodeSchema, bindableSchemas));
-            }
-            case NodeKind.Leaf: {
-                switch (simpleNodeSchema.leafKind) {
-                    case ValueSchema.Boolean: {
-                        return (type = z.boolean());
-                    }
-                    case ValueSchema.Number: {
-                        return (type = z.number());
-                    }
-                    case ValueSchema.String: {
-                        return (type = z.string());
-                    }
-                    case ValueSchema.Null: {
-                        return (type = z.null());
-                    }
-                    default: {
-                        throw new Error(`Unsupported leaf kind ${NodeKind[simpleNodeSchema.leafKind]}.`);
-                    }
-                }
-            }
-            default: {
-                return unreachableCase(simpleNodeSchema, "Unknown node kind");
-            }
-        }
+        },
     });
-}
-function getOrCreateTypeForField(definitionMap, fieldSchema, objectCache, bindableSchemas) {
-    const customMetadata = fieldSchema.metadata.custom;
-    const getDefault = customMetadata?.[llmDefault];
-    if (getDefault !== undefined) {
-        if (typeof getDefault !== "function") {
-            throw new UsageError(`Expected value of ${llmDefault.description} property to be a function, but got ${typeof getDefault}`);
-        }
-        if (fieldSchema.kind !== FieldKind.Optional) {
-            throw new UsageError(`The ${llmDefault.description} property is only permitted on optional fields.`);
-        }
-    }
-    const field = getTypeForAllowedTypes(definitionMap, new Set(fieldSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas).describe(getDefault === undefined
-        ? (fieldSchema.metadata?.description ?? "")
-        : "Do not populate this field. It will be automatically supplied by the system after insertion.");
-    switch (fieldSchema.kind) {
-        case FieldKind.Required: {
-            return field;
-        }
-        case FieldKind.Optional: {
-            return field.optional();
-        }
-        case FieldKind.Identifier: {
-            return field
-                .optional()
-                .describe("This is an ID automatically generated by the system. Do not supply it when constructing a new object.");
-        }
-        default: {
-            throw new Error(`Unsupported field kind ${NodeKind[fieldSchema.kind]}.`);
-        }
-    }
-}
-function getTypeForAllowedTypes(definitionMap, allowedTypes, objectCache, bindableSchemas) {
-    const single = tryGetSingleton(allowedTypes);
-    if (single === undefined) {
-        const types = [
-            ...mapIterable(allowedTypes, (name) => {
-                return getOrCreateType(definitionMap, name, objectCache, bindableSchemas);
-            }),
-        ];
-        assert(hasAtLeastTwo(types), 0xa7e /* Expected at least two types */);
-        return z.union(types);
-    }
-    else {
-        return getOrCreateType(definitionMap, single, objectCache, bindableSchemas);
-    }
+    const simpleTreeSchema = getSimpleSchema(rootSchema);
+    return renderSchemaTypeScript(simpleTreeSchema.definitions, bindableSchemas);
 }
 //# sourceMappingURL=typeGeneration.js.map