From ad9217999ad9b639b95b4bd4a590550c253cbb8c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Sat, 9 Aug 2025 00:57:11 +0200
Subject: [PATCH] Cleanup

---
 packages/effect-fc/src/Memoized.ts | 16 ++++++++--------
 packages/effect-fc/src/Suspense.ts | 26 ++++++++++++--------------
 2 files changed, 20 insertions(+), 22 deletions(-)

diff --git a/packages/effect-fc/src/Memoized.ts b/packages/effect-fc/src/Memoized.ts
index 10d17e5..86cab06 100644
--- a/packages/effect-fc/src/Memoized.ts
+++ b/packages/effect-fc/src/Memoized.ts
@@ -23,24 +23,24 @@ const MemoizedProto = Object.freeze({
 
 export const isMemoized = (u: unknown): u is Memoized<unknown> => Predicate.hasProperty(u, TypeId)
 
-export const memo = <T extends Component.Component<any, any, any>>(
+export const memo = <T extends Component.Component<any, any, any, any>>(
     self: T
-): T & Memoized<Component.FunctionComponent.Props<Component.Component.FunctionComponent<T>>> => Object.setPrototypeOf(
+): T & Memoized<Component.Component.Props<T>> => Object.setPrototypeOf(
     Object.assign(function() {}, self),
     Object.freeze({ ...Object.getPrototypeOf(self), ...MemoizedProto }),
 )
 
 export const withOptions: {
-    <T extends Component.Component<any, any, any> & Memoized<any>>(
-        options: Partial<Memoized.Options<Component.FunctionComponent.Props<Component.Component.FunctionComponent<T>>>>
+    <T extends Component.Component<any, any, any, any> & Memoized<any>>(
+        options: Partial<Memoized.Options<Component.Component.Props<T>>>
     ): (self: T) => T
-    <T extends Component.Component<any, any, any> & Memoized<any>>(
+    <T extends Component.Component<any, any, any, any> & Memoized<any>>(
         self: T,
-        options: Partial<Memoized.Options<Component.FunctionComponent.Props<Component.Component.FunctionComponent<T>>>>,
+        options: Partial<Memoized.Options<Component.Component.Props<T>>>,
     ): T
-} = Function.dual(2, <T extends Component.Component<any, any, any> & Memoized<any>>(
+} = Function.dual(2, <T extends Component.Component<any, any, any, any> & Memoized<any>>(
     self: T,
-    options: Partial<Memoized.Options<Component.FunctionComponent.Props<Component.Component.FunctionComponent<T>>>>,
+    options: Partial<Memoized.Options<Component.Component.Props<T>>>,
 ): T => Object.setPrototypeOf(
     Object.assign(function() {}, self, options),
     Object.getPrototypeOf(self),
diff --git a/packages/effect-fc/src/Suspense.ts b/packages/effect-fc/src/Suspense.ts
index 45212cf..e89c539 100644
--- a/packages/effect-fc/src/Suspense.ts
+++ b/packages/effect-fc/src/Suspense.ts
@@ -1,6 +1,6 @@
 import { Effect, Function, Predicate, Runtime, Scope } from "effect"
 import * as React from "react"
-import type * as Component from "./Component.js"
+import * as Component from "./Component.js"
 
 
 export const TypeId: unique symbol = Symbol.for("effect-fc/Suspense")
@@ -22,8 +22,8 @@ export namespace Suspense {
 const SuspenseProto = Object.freeze({
     [TypeId]: TypeId,
 
-    makeFunctionComponent<F extends Component.FunctionComponent, E, R>(
-        this: Component.Component<F, E, R> & Suspense,
+    makeFunctionComponent<P, A extends React.ReactNode, E, R>(
+        this: Component.Component<P, A, E, R> & Suspense,
         runtimeRef: React.RefObject<Runtime.Runtime<Exclude<R, Scope.Scope>>>,
         scope: Scope.Scope,
     ) {
@@ -31,7 +31,7 @@ const SuspenseProto = Object.freeze({
 
         return ({ fallback, name, ...props }: Suspense.Props) => {
             const promise = Runtime.runPromise(runtimeRef.current)(
-                Effect.provideService(this.body(props as Component.FunctionComponent.Props<F>), Scope.Scope, scope)
+                Effect.provideService(this.body(props as P), Scope.Scope, scope)
             )
 
             return React.createElement(
@@ -46,18 +46,16 @@ const SuspenseProto = Object.freeze({
 
 export const isSuspense = (u: unknown): u is Suspense => Predicate.hasProperty(u, TypeId)
 
-export const suspense = <T extends Component.Component<any, any, any>>(
+export const suspense = <T extends Component.Component<any, any, any, any>>(
     self: T
 ): (
     & Omit<T, keyof Component.Component.AsComponent<T>>
     & Component.Component<
-        Component.Component.FunctionComponent<T> extends (...args: readonly [infer P, ...infer Args]) => infer A
-            ? A extends React.ReactNode
-                ? (...args: readonly [props: P & Suspense.Props, ...Args]) => A
-                : never
-            : never,
+        Component.Component.Props<T> & Suspense.Props,
+        Component.Component.Success<T>,
         Component.Component.Error<T>,
-        Component.Component.Context<T>>
+        Component.Component.Context<T>
+    >
     & Suspense
 ) => Object.setPrototypeOf(
     Object.assign(function() {}, self),
@@ -65,14 +63,14 @@ export const suspense = <T extends Component.Component<any, any, any>>(
 )
 
 export const withOptions: {
-    <T extends Component.Component<any, any, any> & Suspense>(
+    <T extends Component.Component<any, any, any, any> & Suspense>(
         options: Partial<Suspense.Options>
     ): (self: T) => T
-    <T extends Component.Component<any, any, any> & Suspense>(
+    <T extends Component.Component<any, any, any, any> & Suspense>(
         self: T,
         options: Partial<Suspense.Options>,
     ): T
-} = Function.dual(2, <T extends Component.Component<any, any, any> & Suspense>(
+} = Function.dual(2, <T extends Component.Component<any, any, any, any> & Suspense>(
     self: T,
     options: Partial<Suspense.Options>,
 ): T => Object.setPrototypeOf(