@orpc/zod 0.0.0-next.f99e554 → 0.0.0-next.fb0d07c

package/dist/zod4/index.mjs

@@ -0,0 +1,660 @@
+import { isObject, guard, intercept } from '@orpc/shared';
+import { JSONSchemaFormat, JSONSchemaContentEncoding } from '@orpc/openapi';
+import { registry, globalRegistry } from 'zod/v4/core';
+
+class experimental_ZodSmartCoercionPlugin {
+  init(options) {
+    options.clientInterceptors ??= [];
+    options.clientInterceptors.unshift((options2) => {
+      const inputSchema = options2.procedure["~orpc"].inputSchema;
+      if (!inputSchema || inputSchema["~standard"].vendor !== "zod") {
+        return options2.next();
+      }
+      const coercedInput = this.#coerce(inputSchema, options2.input);
+      return options2.next({ ...options2, input: coercedInput });
+    });
+  }
+  #coerce(schema, value) {
+    switch (schema._zod.def.type) {
+      case "number": {
+        if (typeof value === "string") {
+          return this.#stringToNumber(value);
+        }
+        return value;
+      }
+      case "bigint": {
+        if (typeof value === "string") {
+          return this.#stringToBigInt(value);
+        }
+        return value;
+      }
+      case "boolean":
+      case "success": {
+        if (typeof value === "string") {
+          return this.#stringToBoolean(value);
+        }
+        return value;
+      }
+      case "date": {
+        if (typeof value === "string") {
+          return this.#stringToDate(value);
+        }
+        return value;
+      }
+      case "literal":
+      case "enum": {
+        const literal = schema;
+        if (!literal._zod.values.has(value) && typeof value === "string") {
+          const num = this.#stringToNumber(value);
+          if (literal._zod.values.has(num)) {
+            return num;
+          }
+          const bool = this.#stringToBoolean(value);
+          if (literal._zod.values.has(bool)) {
+            return bool;
+          }
+          const bigint = this.#stringToBigInt(value);
+          if (literal._zod.values.has(bigint)) {
+            return bigint;
+          }
+        }
+        return value;
+      }
+      case "array": {
+        const array = schema;
+        if (value === void 0) {
+          return [];
+        }
+        if (Array.isArray(value)) {
+          return value.map((v) => this.#coerce(array._zod.def.element, v));
+        }
+        return value;
+      }
+      case "tuple": {
+        const tuple = schema;
+        if (value === void 0) {
+          return [];
+        }
+        if (Array.isArray(value)) {
+          return value.map((v, i) => {
+            const s = tuple._zod.def.items[i] ?? tuple._zod.def.rest;
+            return s ? this.#coerce(s, v) : v;
+          });
+        }
+        return value;
+      }
+      case "set": {
+        const set = schema;
+        if (value === void 0) {
+          return /* @__PURE__ */ new Set();
+        }
+        if (Array.isArray(value)) {
+          return new Set(
+            value.map((v) => this.#coerce(set._zod.def.valueType, v))
+          );
+        }
+        if (value instanceof Set) {
+          return new Set(
+            Array.from(value).map((v) => this.#coerce(set._zod.def.valueType, v))
+          );
+        }
+        return value;
+      }
+      case "object":
+      case "interface": {
+        const object = schema;
+        if (value === void 0) {
+          return {};
+        }
+        if (isObject(value)) {
+          const newObj = {};
+          const keys = /* @__PURE__ */ new Set([
+            ...Object.keys(value),
+            ...Object.keys(object._zod.def.shape)
+          ]);
+          for (const k of keys) {
+            const s = object._zod.def.shape[k] ?? object._zod.def.catchall;
+            newObj[k] = s ? this.#coerce(s, value[k]) : value[k];
+          }
+          return newObj;
+        }
+        return value;
+      }
+      case "record": {
+        const record = schema;
+        if (value === void 0) {
+          return {};
+        }
+        if (isObject(value)) {
+          const newObj = {};
+          for (const [k, v] of Object.entries(value)) {
+            const key = this.#coerce(record._zod.def.keyType, k);
+            const val = this.#coerce(record._zod.def.valueType, v);
+            newObj[key] = val;
+          }
+          return newObj;
+        }
+        return value;
+      }
+      case "map": {
+        const map = schema;
+        if (value === void 0) {
+          return /* @__PURE__ */ new Map();
+        }
+        if (Array.isArray(value) && value.every((i) => Array.isArray(i) && i.length <= 2)) {
+          return new Map(
+            value.map(([k, v]) => [
+              this.#coerce(map._zod.def.keyType, k),
+              this.#coerce(map._zod.def.valueType, v)
+            ])
+          );
+        }
+        if (value instanceof Map) {
+          return new Map(
+            Array.from(value).map(([k, v]) => [
+              this.#coerce(map._zod.def.keyType, k),
+              this.#coerce(map._zod.def.valueType, v)
+            ])
+          );
+        }
+        return value;
+      }
+      case "union": {
+        const union = schema;
+        if (union._zod.def.options.length === 1) {
+          return this.#coerce(union._zod.def.options[0], value);
+        }
+        if (isObject(value)) {
+          for (const option of union._zod.def.options) {
+            if (option._zod.disc && this.#matchDiscriminators(value, option._zod.disc)) {
+              return this.#coerce(option, value);
+            }
+          }
+        }
+        return value;
+      }
+      case "intersection": {
+        const intersection = schema;
+        return this.#coerce(
+          intersection._zod.def.right,
+          this.#coerce(intersection._zod.def.left, value)
+        );
+      }
+      case "optional": {
+        const optional = schema;
+        if (value === void 0) {
+          return void 0;
+        }
+        return this.#coerce(optional._zod.def.innerType, value);
+      }
+      case "nonoptional": {
+        const nonoptional = schema;
+        return this.#coerce(nonoptional._zod.def.innerType, value);
+      }
+      case "nullable": {
+        const nullable = schema;
+        if (value === null) {
+          return null;
+        }
+        return this.#coerce(nullable._zod.def.innerType, value);
+      }
+      case "readonly": {
+        const readonly_ = schema;
+        return this.#coerce(readonly_._zod.def.innerType, value);
+      }
+      case "pipe": {
+        const pipe = schema;
+        return this.#coerce(pipe._zod.def.in, value);
+      }
+      case "default":
+      case "prefault": {
+        const default_ = schema;
+        if (value === void 0) {
+          return value;
+        }
+        return this.#coerce(default_._zod.def.innerType, value);
+      }
+      case "catch": {
+        const catch_ = schema;
+        return this.#coerce(catch_._zod.def.innerType, value);
+      }
+      case "lazy": {
+        const lazy = schema;
+        if (value !== void 0) {
+          return this.#coerce(lazy._zod.def.getter(), value);
+        }
+        return value;
+      }
+      default: {
+        schema._zod.def.type;
+        return value;
+      }
+    }
+  }
+  #stringToNumber(value) {
+    const num = Number(value);
+    return Number.isNaN(num) || num.toString() !== value ? value : num;
+  }
+  #stringToBigInt(value) {
+    return guard(() => BigInt(value)) ?? value;
+  }
+  #stringToBoolean(value) {
+    const lower = value.toLowerCase();
+    if (lower === "false" || lower === "off" || lower === "f") {
+      return false;
+    }
+    if (lower === "true" || lower === "on" || lower === "t") {
+      return true;
+    }
+    return value;
+  }
+  #stringToDate(value) {
+    const date = new Date(value);
+    if (!Number.isNaN(date.getTime()) && date.toISOString().startsWith(value)) {
+      return date;
+    }
+    return value;
+  }
+  /**
+   * This function is inspired from Zod, because it's not exported
+   * https://github.com/colinhacks/zod/blob/v4/packages/core/src/schemas.ts#L1903C1-L1921C2
+   */
+  #matchDiscriminators(input, discs) {
+    for (const [key, value] of discs) {
+      const data = input[key];
+      if (value.values.size && !value.values.has(data)) {
+        return false;
+      }
+      if (value.maps.length === 0) {
+        continue;
+      }
+      if (!isObject(data)) {
+        return false;
+      }
+      for (const m of value.maps) {
+        if (!this.#matchDiscriminators(data, m)) {
+          return false;
+        }
+      }
+    }
+    return true;
+  }
+}
+
+const experimental_JSON_SCHEMA_REGISTRY = registry();
+const experimental_JSON_SCHEMA_INPUT_REGISTRY = registry();
+const experimental_JSON_SCHEMA_OUTPUT_REGISTRY = registry();
+
+class experimental_ZodToJsonSchemaConverter {
+  maxLazyDepth;
+  anyJsonSchema;
+  unsupportedJsonSchema;
+  undefinedJsonSchema;
+  interceptors;
+  constructor(options = {}) {
+    this.maxLazyDepth = options.maxLazyDepth ?? 2;
+    this.anyJsonSchema = options.anyJsonSchema ?? {};
+    this.unsupportedJsonSchema = options.unsupportedJsonSchema ?? { not: {} };
+    this.undefinedJsonSchema = options.undefinedJsonSchema ?? { not: {} };
+    this.interceptors = options.interceptors ?? [];
+  }
+  condition(schema) {
+    return schema !== void 0 && schema["~standard"].vendor === "zod";
+  }
+  convert(schema, options) {
+    return this.#convert(schema, options, 0);
+  }
+  #convert(schema, options, lazyDepth, isHandledCustomJSONSchema = false) {
+    return intercept(
+      this.interceptors,
+      { schema, options, lazyDepth, isHandledCustomJSONSchema },
+      ({ schema: schema2, options: options2, lazyDepth: lazyDepth2, isHandledCustomJSONSchema: isHandledCustomJSONSchema2 }) => {
+        if (!isHandledCustomJSONSchema2) {
+          const customJSONSchema = this.#getCustomJsonSchema(schema2, options2);
+          if (customJSONSchema) {
+            const [required, json] = this.#convert(schema2, options2, lazyDepth2, true);
+            return [required, { ...json, ...customJSONSchema }];
+          }
+        }
+        switch (schema2._zod.def.type) {
+          case "string": {
+            const string = schema2;
+            const json = { type: "string" };
+            const { minimum, maximum, format, pattern, contentEncoding } = string._zod.bag;
+            if (typeof minimum === "number") {
+              json.minLength = minimum;
+            }
+            if (typeof maximum === "number") {
+              json.maxLength = maximum;
+            }
+            if (typeof contentEncoding === "string") {
+              json.contentEncoding = this.#handleContentEncoding(contentEncoding);
+            }
+            if (typeof format === "string" && format !== "regex" && json.contentEncoding === void 0) {
+              json.format = this.#handleStringFormat(format);
+            }
+            if (pattern instanceof RegExp && json.contentEncoding === void 0 && json.format === void 0) {
+              json.pattern = pattern.source;
+            }
+            if (format === "jwt" && json.contentEncoding === void 0 && json.format === void 0 && json.pattern === void 0) {
+              json.pattern = /^[\w-]+\.[\w-]+\.[\w-]+$/.source;
+            }
+            return [true, json];
+          }
+          case "number": {
+            const number = schema2;
+            const json = { type: "number" };
+            const { minimum, maximum, format, multipleOf, exclusiveMaximum, exclusiveMinimum } = number._zod.bag;
+            if (typeof format === "string" && format?.includes("int")) {
+              json.type = "integer";
+            }
+            if (typeof minimum === "number") {
+              json.minimum = minimum;
+            }
+            if (typeof maximum === "number") {
+              json.maximum = maximum;
+            }
+            if (typeof exclusiveMinimum === "number") {
+              json.exclusiveMinimum = exclusiveMinimum;
+            }
+            if (typeof exclusiveMaximum === "number") {
+              json.exclusiveMaximum = exclusiveMaximum;
+            }
+            if (typeof multipleOf === "number") {
+              json.multipleOf = multipleOf;
+            }
+            return [true, json];
+          }
+          case "boolean": {
+            return [true, { type: "boolean" }];
+          }
+          case "bigint": {
+            return [true, { type: "string", pattern: "^-?[0-9]+$" }];
+          }
+          case "date": {
+            return [true, { type: "string", format: JSONSchemaFormat.DateTime }];
+          }
+          case "null": {
+            return [true, { type: "null" }];
+          }
+          case "undefined":
+          case "void": {
+            return [false, this.undefinedJsonSchema];
+          }
+          case "any": {
+            return [false, this.anyJsonSchema];
+          }
+          case "unknown": {
+            return [false, this.anyJsonSchema];
+          }
+          case "never": {
+            return [true, this.unsupportedJsonSchema];
+          }
+          case "array": {
+            const array = schema2;
+            const json = { type: "array" };
+            const { minimum, maximum } = array._zod.bag;
+            if (typeof minimum === "number") {
+              json.minItems = minimum;
+            }
+            if (typeof maximum === "number") {
+              json.maxItems = maximum;
+            }
+            json.items = this.#handleArrayItemJsonSchema(this.#convert(array._zod.def.element, options2, lazyDepth2), options2);
+            return [true, json];
+          }
+          case "object": {
+            const object = schema2;
+            const json = { type: "object" };
+            for (const [key, value] of Object.entries(object._zod.def.shape)) {
+              const [itemRequired, itemJson] = this.#convert(value, options2, lazyDepth2);
+              json.properties ??= {};
+              json.properties[key] = itemJson;
+              if (itemRequired) {
+                json.required ??= [];
+                json.required.push(key);
+              }
+            }
+            if (object._zod.def.catchall) {
+              if (object._zod.def.catchall._zod.def.type === "never") {
+                json.additionalProperties = false;
+              } else {
+                const [_, addJson] = this.#convert(object._zod.def.catchall, options2, lazyDepth2);
+                json.additionalProperties = addJson;
+              }
+            }
+            return [true, json];
+          }
+          case "union": {
+            const union = schema2;
+            const anyOf = [];
+            let required = true;
+            for (const item of union._zod.def.options) {
+              const [itemRequired, itemJson] = this.#convert(item, options2, lazyDepth2);
+              if (!itemRequired) {
+                required = false;
+              }
+              if (options2.strategy === "input") {
+                if (itemJson !== this.undefinedJsonSchema && itemJson !== this.unsupportedJsonSchema) {
+                  anyOf.push(itemJson);
+                }
+              } else {
+                if (itemJson !== this.undefinedJsonSchema) {
+                  anyOf.push(itemJson);
+                }
+              }
+            }
+            return [required, anyOf.length === 1 ? anyOf[0] : { anyOf }];
+          }
+          case "intersection": {
+            const intersection = schema2;
+            const json = { allOf: [] };
+            let required = false;
+            for (const item of [intersection._zod.def.left, intersection._zod.def.right]) {
+              const [itemRequired, itemJson] = this.#convert(item, options2, lazyDepth2);
+              json.allOf.push(itemJson);
+              if (itemRequired) {
+                required = true;
+              }
+            }
+            return [required, json];
+          }
+          case "tuple": {
+            const tuple = schema2;
+            const json = { type: "array", prefixItems: [] };
+            for (const item of tuple._zod.def.items) {
+              json.prefixItems.push(this.#handleArrayItemJsonSchema(this.#convert(item, options2, lazyDepth2), options2));
+            }
+            if (tuple._zod.def.rest) {
+              json.items = this.#handleArrayItemJsonSchema(this.#convert(tuple._zod.def.rest, options2, lazyDepth2), options2);
+            }
+            const { minimum, maximum } = tuple._zod.bag;
+            if (typeof minimum === "number") {
+              json.minItems = minimum;
+            }
+            if (typeof maximum === "number") {
+              json.maxItems = maximum;
+            }
+            return [true, json];
+          }
+          case "record": {
+            const record = schema2;
+            const json = { type: "object" };
+            json.propertyNames = this.#convert(record._zod.def.keyType, options2, lazyDepth2)[1];
+            json.additionalProperties = this.#convert(record._zod.def.valueType, options2, lazyDepth2)[1];
+            return [true, json];
+          }
+          case "map": {
+            const map = schema2;
+            return [true, {
+              type: "array",
+              items: {
+                type: "array",
+                prefixItems: [
+                  this.#handleArrayItemJsonSchema(this.#convert(map._zod.def.keyType, options2, lazyDepth2), options2),
+                  this.#handleArrayItemJsonSchema(this.#convert(map._zod.def.valueType, options2, lazyDepth2), options2)
+                ],
+                maxItems: 2,
+                minItems: 2
+              }
+            }];
+          }
+          case "set": {
+            const set = schema2;
+            return [true, {
+              type: "array",
+              uniqueItems: true,
+              items: this.#handleArrayItemJsonSchema(this.#convert(set._zod.def.valueType, options2, lazyDepth2), options2)
+            }];
+          }
+          case "enum": {
+            const enum_ = schema2;
+            return [true, { enum: Object.values(enum_._zod.def.entries) }];
+          }
+          case "literal": {
+            const literal = schema2;
+            let required = true;
+            const values = /* @__PURE__ */ new Set();
+            for (const value of literal._zod.def.values) {
+              if (value === void 0) {
+                required = false;
+              } else {
+                values.add(typeof value === "bigint" ? value.toString() : value);
+              }
+            }
+            const json = values.size === 0 ? this.undefinedJsonSchema : values.size === 1 ? { const: values.values().next().value } : { enum: Array.from(values) };
+            return [required, json];
+          }
+          case "file": {
+            const file = schema2;
+            const oneOf = [];
+            const { mime } = file._zod.bag;
+            if (mime === void 0 || Array.isArray(mime) && mime.every((m) => typeof m === "string")) {
+              for (const type of mime ?? ["*/*"]) {
+                oneOf.push({
+                  type: "string",
+                  contentMediaType: type
+                });
+              }
+            }
+            return [true, oneOf.length === 1 ? oneOf[0] : { anyOf: oneOf }];
+          }
+          case "transform": {
+            return [false, this.anyJsonSchema];
+          }
+          case "nullable": {
+            const nullable = schema2;
+            const [required, json] = this.#convert(nullable._zod.def.innerType, options2, lazyDepth2);
+            return [required, { anyOf: [json, { type: "null" }] }];
+          }
+          case "nonoptional": {
+            const nonoptional = schema2;
+            const [, json] = this.#convert(nonoptional._zod.def.innerType, options2, lazyDepth2);
+            return [true, json];
+          }
+          case "success": {
+            return [true, { type: "boolean" }];
+          }
+          case "default":
+          case "prefault": {
+            const default_ = schema2;
+            const [, json] = this.#convert(default_._zod.def.innerType, options2, lazyDepth2);
+            return [false, {
+              ...json,
+              default: default_._zod.def.defaultValue
+            }];
+          }
+          case "catch": {
+            const catch_ = schema2;
+            return this.#convert(catch_._zod.def.innerType, options2, lazyDepth2);
+          }
+          case "nan": {
+            return [true, options2.strategy === "input" ? this.unsupportedJsonSchema : { type: "null" }];
+          }
+          case "pipe": {
+            const pipe = schema2;
+            return this.#convert(options2.strategy === "input" ? pipe._zod.def.in : pipe._zod.def.out, options2, lazyDepth2);
+          }
+          case "readonly": {
+            const readonly_ = schema2;
+            const [required, json] = this.#convert(readonly_._zod.def.innerType, options2, lazyDepth2);
+            return [required, { ...json, readOnly: true }];
+          }
+          case "template_literal": {
+            const templateLiteral = schema2;
+            return [true, {
+              type: "string",
+              pattern: templateLiteral._zod.pattern.source
+            }];
+          }
+          case "optional": {
+            const optional = schema2;
+            const [, json] = this.#convert(optional._zod.def.innerType, options2, lazyDepth2);
+            return [false, json];
+          }
+          case "lazy": {
+            const lazy = schema2;
+            if (lazyDepth2 >= this.maxLazyDepth) {
+              return [false, this.anyJsonSchema];
+            }
+            return this.#convert(lazy._zod.def.getter(), options2, lazyDepth2 + 1);
+          }
+          default: {
+            schema2._zod.def.type;
+            return [true, this.unsupportedJsonSchema];
+          }
+        }
+      }
+    );
+  }
+  #getCustomJsonSchema(schema, options) {
+    if (options.strategy === "input" && experimental_JSON_SCHEMA_INPUT_REGISTRY.has(schema)) {
+      return experimental_JSON_SCHEMA_INPUT_REGISTRY.get(schema);
+    }
+    if (options.strategy === "output" && experimental_JSON_SCHEMA_OUTPUT_REGISTRY.has(schema)) {
+      return experimental_JSON_SCHEMA_OUTPUT_REGISTRY.get(schema);
+    }
+    if (experimental_JSON_SCHEMA_REGISTRY.has(schema)) {
+      return experimental_JSON_SCHEMA_REGISTRY.get(schema);
+    }
+    const global = globalRegistry.get(schema);
+    if (global) {
+      return {
+        description: global.description,
+        examples: global.examples
+      };
+    }
+  }
+  #handleArrayItemJsonSchema([required, schema], options) {
+    if (required || options.strategy === "input" || schema.default !== void 0) {
+      return schema;
+    }
+    if (schema === this.undefinedJsonSchema) {
+      return { type: "null" };
+    }
+    return {
+      anyOf: [
+        // schema can contain { type: 'null' } so we should use anyOf instead of oneOf
+        schema,
+        { type: "null" }
+      ]
+    };
+  }
+  #handleStringFormat(format) {
+    if (format === "guid") {
+      return JSONSchemaFormat.UUID;
+    }
+    if (format === "url") {
+      return JSONSchemaFormat.URI;
+    }
+    if (format === "datetime") {
+      return JSONSchemaFormat.DateTime;
+    }
+    return Object.values(JSONSchemaFormat).includes(format) ? format : void 0;
+  }
+  #handleContentEncoding(contentEncoding) {
+    return Object.values(JSONSchemaContentEncoding).includes(contentEncoding) ? contentEncoding : void 0;
+  }
+}
+
+export { experimental_JSON_SCHEMA_INPUT_REGISTRY, experimental_JSON_SCHEMA_OUTPUT_REGISTRY, experimental_JSON_SCHEMA_REGISTRY, experimental_ZodSmartCoercionPlugin, experimental_ZodToJsonSchemaConverter };