From c75bb10e6b8f7d4ae54d702f875b00e4f27a84c1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Mon, 31 Mar 2025 18:22:18 +0200
Subject: [PATCH] QueryClient work

---
 packages/extension-query/src/QueryClient.ts | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/packages/extension-query/src/QueryClient.ts b/packages/extension-query/src/QueryClient.ts
index 4e1b416..692b2b6 100644
--- a/packages/extension-query/src/QueryClient.ts
+++ b/packages/extension-query/src/QueryClient.ts
@@ -20,7 +20,11 @@ export interface ServiceProps<EH, HandledE> {
 }
 
 export interface ServiceResult<Self, EH, HandledE> extends Context.TagClass<Self, typeof id, QueryClient<HandledE>> {
-    readonly Live: Layer.Layer<Self, never, EH>
+    readonly Live: Layer.Layer<
+        Self,
+        never,
+        EH extends ErrorHandler.DefaultErrorHandler ? never : EH
+    >
 }
 
 export const Service = <Self>() => (
@@ -31,9 +35,17 @@ export const Service = <Self>() => (
         props?: ServiceProps<EH, HandledE>
     ): ServiceResult<Self, EH, HandledE> => {
         const TagClass = Context.Tag(id)() as ServiceResult<Self, EH, HandledE>
+
         (TagClass as Mutable<typeof TagClass>).Live = Layer.effect(TagClass, Effect.Do.pipe(
-            Effect.bind("errorHandler", () => (props?.ErrorHandler as Effect.Effect<ErrorHandler.ErrorHandler<HandledE>, never, EH>) ?? ErrorHandler.DefaultErrorHandler as Effect.Effect<ErrorHandler.ErrorHandler<HandledE>, never, EH>)
+            Effect.bind("errorHandler", () =>
+                (props?.ErrorHandler ?? ErrorHandler.DefaultErrorHandler) as Effect.Effect<
+                    ErrorHandler.ErrorHandler<HandledE>,
+                    never,
+                    EH extends ErrorHandler.DefaultErrorHandler ? never : EH
+                >
+            )
         ))
+
         return TagClass
     }
 )