add tests

Author: abhishekg999

src/utils.ts

@@ -58,18 +58,18 @@ export const mergeDeep = <
 		skipKeys?: string[]
 		override?: boolean
 		mergeArray?: boolean
-		seen?: WeakSet<object>
+		seen?: WeakMap<object, boolean>
 	}
 ): A & B => {
 	const skipKeys = options?.skipKeys
 	const override = options?.override ?? true
 	const mergeArray = options?.mergeArray ?? false
-	const seen = options?.seen ?? new WeakSet<object>()
+	const seen = options?.seen ?? new WeakMap<object, boolean>()
 
 	if (!isObject(target) || !isObject(source)) return target as A & B
 
-	if (seen.has(source)) return target as A & B
-	seen.add(source)
+	if (seen.get(source)) return target as A & B
+	seen.set(source, true)
 
 	for (const [key, value] of Object.entries(source)) {
 		if (
@@ -107,6 +107,8 @@ export const mergeDeep = <
 			} catch {}
 	}
 
+	seen.delete(source)
+
 	return target as A & B
 }
 export const mergeCookie = <const A extends Object, const B extends Object>(

test/units/merge-deep.test.ts

@@ -86,4 +86,78 @@ describe('mergeDeep', () => {
 			.decorate('db', Object.freeze({ hello: 'world' }))
 			.guard({}, (app) => app)
 	})
+
+	it('handle circular references', () => {
+		const a: {
+			x: number
+			toB?: typeof b
+		} = { x: 1 }
+		const b: {
+			y: number
+			toA?: typeof a
+		} = { y: 2 }
+
+		a.toB = b
+		b.toA = a
+
+		const target = {}
+		const source = { prop: a }
+
+		const result = mergeDeep(target, source)
+
+		expect(result.prop.x).toBe(1)
+		expect(result.prop.toB?.y).toBe(2)
+	})
+
+	it('handle shared references in different branches', () => {
+		const shared = { value: 123 }
+		const target = { x: {}, y: {} }
+		const source = { x: shared, y: shared }
+
+		const result = mergeDeep(target, source)
+
+		expect(result.x.value).toBe(123)
+		expect(result.y.value).toBe(123)
+	})
+
+	it('deduplicate plugin with circular decorators', async () => {
+		const a: {
+			x: number
+			toB?: typeof b
+		} = { x: 1 }
+		const b: {
+			y: number
+			toA?: typeof a
+		} = { y: 2 }
+		a.toB = b
+		b.toA = a
+
+		const complex = { a }
+
+		const Plugin = new Elysia({ name: 'Plugin', seed: 'seed' })
+			.decorate('dep', complex)
+			.as('scoped')
+
+		const ModuleA = new Elysia({ name: 'ModuleA' })
+			.use(Plugin)
+			.get('/moda/a', ({ dep }) => dep.a.x)
+			.get('/moda/b', ({ dep }) => dep.a.toB?.y)
+
+		const ModuleB = new Elysia({ name: 'ModuleB' })
+			.use(Plugin)
+			.get('/modb/a', ({ dep }) => dep.a.x)
+			.get('/modb/b', ({ dep }) => dep.a.toB?.y)
+
+		const app = new Elysia().use(ModuleA).use(ModuleB)
+
+		const resA = await app.handle(req('/moda/a')).then((x) => x.text())
+		const resB = await app.handle(req('/modb/a')).then((x) => x.text())
+		const resC = await app.handle(req('/moda/b')).then((x) => x.text())
+		const resD = await app.handle(req('/modb/b')).then((x) => x.text())
+
+		expect(resA).toBe('1')
+		expect(resB).toBe('1')
+		expect(resC).toBe('2')
+		expect(resD).toBe('2')
+	})
 })