move getType to _def.getType

src/index.ts

@@ -27,7 +27,7 @@ const callGetType = (
 	let type: ReturnType<GetTypeFunction> | undefined
 
 	// this must be called before accessing 'type'
-	if (zod.getType) type = zod.getType(ts, identifier, options)
+	if (zod._def.getType) type = zod._def.getType(ts, identifier, options)
 	return type
 }
 

src/types.ts

@@ -23,4 +23,4 @@ export type GetTypeFunction = (
 	options: RequiredZodToTsOptions,
 ) => ts.Identifier | ts.TypeNode
 
-export type GetType = { getType?: GetTypeFunction }
+export type GetType = { _def: { getType?: GetTypeFunction } }

src/utils.ts

@@ -38,7 +38,7 @@ export const printNode = (node: ts.Node, printerOptions?: ts.PrinterOptions) =>
 }
 
 export const withGetType = <T extends ZodTypeAny & GetType>(schema: T, getType: GetTypeFunction): T => {
-	schema.getType = getType
+	schema._def.getType = getType
 	return schema
 }
 

test/describe.test.ts

@@ -0,0 +1,28 @@
+import { describe, expect, it } from 'vitest'
+import { z } from 'zod'
+import { withGetType, zodToTs } from '../src'
+import { printNodeTest } from './utils'
+
+describe('z.describe()', () => {
+	it('supports describing schema after withGetType', () => {
+		const Enum = z.nativeEnum({
+			ONE: 1,
+			TWO: 2,
+		})
+
+		withGetType(Enum, ts => ts.factory.createIdentifier('Enum'))
+
+		const schema = z.object({
+			key: Enum.describe('Comment for key'),
+		})
+
+		const { node } = zodToTs(schema)
+
+		expect(printNodeTest(node)).toMatchInlineSnapshot(`
+			"{
+			    /** Comment for key */
+			    key: Enum;
+			}"
+		`)
+	})
+})

test/example2.ts

@@ -0,0 +1,20 @@
+import { z } from 'zod'
+import { printNode, withGetType, zodToTs } from '../src'
+
+const Enum = z.nativeEnum({
+	ONE: 1,
+	TWO: 2,
+})
+
+withGetType(Enum, ts => ts.factory.createIdentifier('Enum'))
+
+const schema = z.object({
+	key: Enum.describe('Comment for key'),
+})
+
+const { node } = zodToTs(schema, undefined, { resolveNativeEnums: true })
+console.log(printNode(node))
+// {
+//     /** Comment for key */
+//     key: unknown;
+// }