feat(zod): support readOnly, propertyNames in JSON schema (#211)

* feat(zod): support the readOnly flag in JSON schemas

* feat(zod): support the propertyNames in JSON schemas

Author: dinwwwh

packages/zod/src/converter.test.ts

@@ -296,7 +296,8 @@ const processedCases: SchemaTestCase[] = [
   },
   {
     schema: z.number().readonly(),
-    input: [true, { type: 'number' }],
+    input: [true, { type: 'number', readOnly: true }],
+    ignoreZodToJsonSchema: true,
   },
 ]
 
@@ -399,6 +400,11 @@ const edgeCases: SchemaTestCase[] = [
     schema: z.object({ value: z.string() }).catchall(z.number()),
     input: [true, { type: 'object', properties: { value: { type: 'string' } }, required: ['value'], additionalProperties: { type: 'number' } }],
   },
+  {
+    schema: z.record(z.number(), z.string()),
+    input: [true, { type: 'object', additionalProperties: { type: 'string' }, propertyNames: { type: 'number' } }],
+    ignoreZodToJsonSchema: true,
+  },
 ]
 
 describe.each([

packages/zod/src/converter.ts

@@ -413,7 +413,12 @@ export class ZodToJsonSchemaConverter implements ConditionalSchemaConverter {
 
         const json: JSONSchema = { type: 'object' }
 
-        // WARN: ignore keyType
+        const keyTypeName = this.#getZodTypeName(schema_._def.keyType._def)
+
+        if (keyTypeName !== ZodFirstPartyTypeKind.ZodString) {
+          const [_, keyJson] = this.convert(schema_._def.keyType, options, lazyDepth, false, false)
+          json.propertyNames = keyJson
+        }
 
         const [_, itemJson] = this.convert(schema_._def.valueType, options, lazyDepth, false, false)
 
@@ -524,7 +529,9 @@ export class ZodToJsonSchemaConverter implements ConditionalSchemaConverter {
 
       case ZodFirstPartyTypeKind.ZodReadonly: {
         const schema_ = schema as ZodReadonly<ZodTypeAny>
-        return this.convert(schema_._def.innerType, options, lazyDepth, false, false)
+        const [required, json] = this.convert(schema_._def.innerType, options, lazyDepth, false, false)
+
+        return [required, { ...json, readOnly: true }]
       }
 
       case ZodFirstPartyTypeKind.ZodDefault: {