From f419474536cc702c18df8ded25baae8b3a54c8cf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Mon, 14 Jul 2025 21:12:54 +0200
Subject: [PATCH] Fix

---
 packages/effect-fc/src/Component.ts      | 15 +++++++++++----
 packages/example/src/routes/dev/memo.tsx |  3 +--
 2 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/packages/effect-fc/src/Component.ts b/packages/effect-fc/src/Component.ts
index 482b227..af91d9a 100644
--- a/packages/effect-fc/src/Component.ts
+++ b/packages/effect-fc/src/Component.ts
@@ -87,19 +87,26 @@ export interface Memoized<P> {
     readonly propsAreEqual?: Equivalence.Equivalence<P>
 }
 
-export const memo: {
+export const memo = <T extends Component<any, any, any>>(
+    self: ExcludeKeys<T, keyof Memoized<Component.Props<T>>>
+): T & Memoized<Component.Props<T>> => Object.setPrototypeOf(
+    { ...self, memo: true },
+    Object.getPrototypeOf(self),
+)
+
+export const memoWithEquivalence: {
     <T extends Component<any, any, any>>(
-        propsAreEqual?: Equivalence.Equivalence<Component.Props<T>>
+        propsAreEqual: Equivalence.Equivalence<Component.Props<T>>
     ): (
         self: ExcludeKeys<T, keyof Memoized<Component.Props<T>>>
     ) => T & Memoized<Component.Props<T>>
     <T extends Component<any, any, any>>(
         self: ExcludeKeys<T, keyof Memoized<Component.Props<T>>>,
-        propsAreEqual?: Equivalence.Equivalence<Component.Props<T>>,
+        propsAreEqual: Equivalence.Equivalence<Component.Props<T>>,
     ): T & Memoized<Component.Props<T>>
 } = Function.dual(2, <T extends Component<any, any, any>>(
     self: ExcludeKeys<T, keyof Memoized<Component.Props<T>>>,
-    propsAreEqual?: Equivalence.Equivalence<Component.Props<T>>,
+    propsAreEqual: Equivalence.Equivalence<Component.Props<T>>,
 ): T & Memoized<Component.Props<T>> => Object.setPrototypeOf(
     { ...self, memo: true, propsAreEqual },
     Object.getPrototypeOf(self),
diff --git a/packages/example/src/routes/dev/memo.tsx b/packages/example/src/routes/dev/memo.tsx
index adeb13d..f58e6ad 100644
--- a/packages/example/src/routes/dev/memo.tsx
+++ b/packages/example/src/routes/dev/memo.tsx
@@ -33,8 +33,7 @@ const SubComponent = Component.make(function* SubComponent() {
     return <Text>{id}</Text>
 })
 
-const MemoizedSubComponent = Component.memo()(SubComponent)
-console.log(MemoizedSubComponent.memo)
+const MemoizedSubComponent = Component.memo(SubComponent)
 
 export const Route = createFileRoute("/dev/memo")({
     component: RouteComponent,