From 2d2834af56b74f2d346d072d5846246c48264147 Mon Sep 17 00:00:00 2001
From: Jimmyzm <me@jimmycn.com>
Date: Fri, 15 Mar 2024 17:57:25 +0800
Subject: [PATCH] fix: add generic parameter to fix type error when using
 `default` option.

---
 src/client/types.ts | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/src/client/types.ts b/src/client/types.ts
index d19fc3c..df3a30f 100644
--- a/src/client/types.ts
+++ b/src/client/types.ts
@@ -60,9 +60,10 @@ export type DecorateProcedure<
         DataE = TRPCClientErrorLike<TProcedure>,
         DataT = ResT,
         PickKeys extends KeysOf<DataT> = KeysOf<DataT>,
+        DefaultT = null,
      >(
         input: MaybeRefOrGetter<inferProcedureInput<TProcedure>>,
-        opts?: AsyncDataOptions<ResT, DataT, PickKeys> & {
+        opts?: AsyncDataOptions<ResT, DataT, PickKeys, DefaultT> & {
           trpc?: TRPCRequestOptions
           /**
            * The custom unique key to use.
@@ -70,15 +71,16 @@ export type DecorateProcedure<
            */
           queryKey?: string
         },
-      ) => AsyncData<PickFrom<DataT, PickKeys> | null, DataE>,
+      ) => AsyncData<PickFrom<DataT, PickKeys> | DefaultT, DataE>,
       useLazyQuery: <
         ResT = inferTransformedProcedureOutput<TProcedure>,
         DataE = TRPCClientErrorLike<TProcedure>,
         DataT = ResT,
         PickKeys extends KeysOf<DataT> = KeysOf<DataT>,
+        DefaultT = null,
      >(
         input: MaybeRefOrGetter<inferProcedureInput<TProcedure>>,
-        opts?: Omit<AsyncDataOptions<ResT, DataT, PickKeys>, 'lazy'> & {
+        opts?: Omit<AsyncDataOptions<ResT, DataT, PickKeys, DefaultT>, 'lazy'> & {
           trpc?: TRPCRequestOptions
           /**
            * The custom unique key to use.
@@ -86,7 +88,7 @@ export type DecorateProcedure<
            */
           queryKey?: string
         },
-      ) => AsyncData<PickFrom<DataT, PickKeys> | null, DataE>,
+      ) => AsyncData<PickFrom<DataT, PickKeys> | DefaultT, DataE>,
       query: Resolver<TProcedure>
     } : TProcedure extends AnyMutationProcedure ? {
       mutate: Resolver<TProcedure>
@@ -95,11 +97,12 @@ export type DecorateProcedure<
         DataE = TRPCClientErrorLike<TProcedure>,
         DataT = ResT,
         PickKeys extends KeysOf<DataT> = KeysOf<DataT>,
+        DefaultT = null,
      >(
-        opts?: Omit<AsyncDataOptions<ResT, DataT, PickKeys>, 'lazy'> & {
+        opts?: Omit<AsyncDataOptions<ResT, DataT, PickKeys, DefaultT>, 'lazy'> & {
           trpc?: TRPCRequestOptions
         },
-      ) => AsyncData<PickFrom<DataT, PickKeys> | null, DataE> & {
+      ) => AsyncData<PickFrom<DataT, PickKeys> | DefaultT, DataE> & {
         /**
          * The function to call to trigger the mutation.
          */