fix(zui): a ref identity is its uri (#384)

zui/src/deref.test.ts

@@ -120,9 +120,19 @@ describe('dereference', () => {
     const result = derefSchema.safeParse('astring')
     expect(result.success).toBe(true)
   })
+  test('should treat multiple references with the same uri as the same reference', () => {
+    const refSchema = z.object({
+      foo,
+      bar: z.ref('foo'),
+      baz: z.ref('foo'),
+    })
+    const derefSchema = refSchema.dereference(deref)
+    const result = derefSchema.safeParse({ foo: 'astring', bar: 'astring', baz: 'astring' })
+    expect(result.success).toBe(true)
+  })
 })
 
-describe('getRegerences', () => {
+describe('getReferences', () => {
   test('array', () => {
     const refSchema = z.array(bar)
     const refs = refSchema.getReferences()
@@ -204,4 +214,13 @@ describe('getRegerences', () => {
     const refs = refSchema.getReferences()
     expect(refs).toEqual(['foo', 'bar', 'baz'])
   })
+  test('should treat multiple references with the same uri as the same reference', () => {
+    const refSchema = z.object({
+      foo,
+      bar: z.ref('foo'),
+      baz: z.ref('foo'),
+    })
+    const refs = refSchema.getReferences()
+    expect(refs).toEqual(['foo'])
+  })
 })

zui/src/z/types/discriminatedUnion/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   ZodBranded,
   ZodCatch,
@@ -104,7 +105,7 @@ export class ZodDiscriminatedUnion<
   }
 
   getReferences(): string[] {
-    return this.options.flatMap((option) => option.getReferences())
+    return unique(this.options.flatMap((option) => option.getReferences()))
   }
 
   _parse(input: ParseInput): ParseReturnType<this['_output']> {

zui/src/z/types/function/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   defaultErrorMap,
   getErrorMap,
@@ -55,7 +56,7 @@ export class ZodFunction<Args extends ZodTuple<any, any>, Returns extends ZodTyp
   }
 
   getReferences(): string[] {
-    return [...this._def.args.getReferences(), ...this._def.returns.getReferences()]
+    return unique([...this._def.args.getReferences(), ...this._def.returns.getReferences()])
   }
 
   _parse(input: ParseInput): ParseReturnType<any> {

zui/src/z/types/intersection/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   ZodIssueCode,
   RawCreateParams,
@@ -85,7 +86,7 @@ export class ZodIntersection<T extends ZodTypeAny, U extends ZodTypeAny> extends
   }
 
   getReferences(): string[] {
-    return [...this._def.left.getReferences(), ...this._def.right.getReferences()]
+    return unique([...this._def.left.getReferences(), ...this._def.right.getReferences()])
   }
 
   _parse(input: ParseInput): ParseReturnType<this['_output']> {

zui/src/z/types/map/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   ZodIssueCode,
   ParseInputLazyPath,
@@ -45,7 +46,7 @@ export class ZodMap<Key extends ZodTypeAny = ZodTypeAny, Value extends ZodTypeAn
   }
 
   getReferences(): string[] {
-    return [...this._def.keyType.getReferences(), ...this._def.valueType.getReferences()]
+    return unique([...this._def.keyType.getReferences(), ...this._def.valueType.getReferences()])
   }
 
   _parse(input: ParseInput): ParseReturnType<this['_output']> {

zui/src/z/types/object/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   ZodArray,
   ZodEnum,
@@ -142,7 +143,7 @@ export class ZodObject<
     for (const key in shape) {
       refs.push(...shape[key]!.getReferences())
     }
-    return refs
+    return unique(refs)
   }
 
   _parse(input: ParseInput): ParseReturnType<this['_output']> {

zui/src/z/types/pipeline/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   ZodFirstPartyTypeKind,
   ZodType,
@@ -29,7 +30,7 @@ export class ZodPipeline<A extends ZodTypeAny, B extends ZodTypeAny> extends Zod
   }
 
   getReferences(): string[] {
-    return [...this._def.in.getReferences(), ...this._def.out.getReferences()]
+    return unique([...this._def.in.getReferences(), ...this._def.out.getReferences()])
   }
 
   _parse(input: ParseInput): ParseReturnType<any> {

zui/src/z/types/record/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   BRAND,
   ZodIssueCode,
@@ -59,7 +60,7 @@ export class ZodRecord<Key extends KeySchema = ZodString, Value extends ZodTypeA
   }
 
   getReferences(): string[] {
-    return [...this._def.keyType.getReferences(), ...this._def.valueType.getReferences()]
+    return unique([...this._def.keyType.getReferences(), ...this._def.valueType.getReferences()])
   }
 
   _parse(input: ParseInput): ParseReturnType<this['_output']> {

zui/src/z/types/tuple/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   ZodIssueCode,
   ParseInputLazyPath,
@@ -57,10 +58,10 @@ export class ZodTuple<
   }
 
   getReferences(): string[] {
-    return [
+    return unique([
       ...this._def.items.flatMap((item) => item.getReferences()),
       ...(this._def.rest ? this._def.rest.getReferences() : []),
-    ]
+    ])
   }
 
   _parse(input: ParseInput): ParseReturnType<this['_output']> {

zui/src/z/types/union/index.ts

@@ -1,3 +1,4 @@
+import { unique } from '../../utils'
 import {
   RawCreateParams,
   ZodFirstPartyTypeKind,
@@ -42,9 +43,11 @@ export class ZodUnion<T extends ZodUnionOptions> extends ZodType<
   }
 
   getReferences(): string[] {
-    return this._def.options.reduce<string[]>((acc, option) => {
-      return [...acc, ...option.getReferences()]
-    }, [])
+    return unique(
+      this._def.options.reduce<string[]>((acc, option) => {
+        return [...acc, ...option.getReferences()]
+      }, []),
+    )
   }
 
   _parse(input: ParseInput): ParseReturnType<this['_output']> {

zui/src/z/utils.ts

@@ -0,0 +1,3 @@
+export const unique = <T>(arr: T[]): T[] => {
+  return Array.from(new Set(arr))
+}