From 299ae558ab0fe3180611b7b21260d10cf278e8b2 Mon Sep 17 00:00:00 2001
From: wobsoriano <sorianorobertc@gmail.com>
Date: Tue, 22 Aug 2023 10:48:44 -0700
Subject: [PATCH] fix: add missing useLazyQuery type

---
 playground/pages/index.vue |  2 +-
 src/client/index.ts        | 10 +++++-----
 src/client/types.ts        | 24 ++++++++++++++++++++----
 3 files changed, 26 insertions(+), 10 deletions(-)

diff --git a/playground/pages/index.vue b/playground/pages/index.vue
index dc208fb..f116ae6 100644
--- a/playground/pages/index.vue
+++ b/playground/pages/index.vue
@@ -17,7 +17,7 @@ const addTodo = async () => {
   }
 }
 
-const { data: todos, pending, error, refresh } = await $client.todo.getTodos.useQuery()
+const { data: todos, pending, error, refresh } = await $client.todo.getTodos.useLazyQuery()
 </script>
 
 <template>
diff --git a/src/client/index.ts b/src/client/index.ts
index c12b48a..3e01ae6 100644
--- a/src/client/index.ts
+++ b/src/client/index.ts
@@ -54,17 +54,17 @@ export function createNuxtProxyDecoration<TRouter extends AnyRouter> (name: stri
         controller = typeof AbortController !== 'undefined' ? new AbortController() : {} as AbortController
       }
 
-      if (lastArg === 'useLazyQuery') {
-        asyncDataOptions.lazy = true;
-      }
-
       const queryKey = customQueryKey || getQueryKey(path, unref(input))
+      const watch = isRef(input) ? [...(asyncDataOptions.watch || []), input] : asyncDataOptions.watch
+      const isLazy = lastArg === 'useLazyQuery' ? true : (asyncDataOptions.lazy || false)
+  
       return useAsyncData(queryKey, () => (client as any)[path].query(unref(input), {
         signal: controller?.signal,
         ...trpc
       }), {
         ...asyncDataOptions,
-        watch: isRef(input) ? [...(asyncDataOptions.watch || []), input] : asyncDataOptions.watch
+        watch,
+        lazy: isLazy
       })
     }
 
diff --git a/src/client/types.ts b/src/client/types.ts
index fdbb0b7..abedebe 100644
--- a/src/client/types.ts
+++ b/src/client/types.ts
@@ -53,10 +53,10 @@ type DecorateProcedure<
 > = TProcedure extends AnyQueryProcedure
   ? {
       useQuery: <
-      ResT = inferTransformedProcedureOutput<TProcedure>,
-      DataE = TRPCClientErrorLike<TProcedure>,
-      DataT = ResT,
-      PickKeys extends KeysOf<DataT> = KeysOf<DataT>,
+        ResT = inferTransformedProcedureOutput<TProcedure>,
+        DataE = TRPCClientErrorLike<TProcedure>,
+        DataT = ResT,
+        PickKeys extends KeysOf<DataT> = KeysOf<DataT>,
      >(
         input: MaybeRef<inferProcedureInput<TProcedure>>,
         opts?: AsyncDataOptions<ResT, DataT, PickKeys> & {
@@ -68,6 +68,22 @@ type DecorateProcedure<
           queryKey?: string
         },
       ) => AsyncData<PickFrom<DataT, PickKeys> | null, DataE>,
+      useLazyQuery: <
+        ResT = inferTransformedProcedureOutput<TProcedure>,
+        DataE = TRPCClientErrorLike<TProcedure>,
+        DataT = ResT,
+        PickKeys extends KeysOf<DataT> = KeysOf<DataT>,
+     >(
+        input: MaybeRef<inferProcedureInput<TProcedure>>,
+        opts?: Omit<AsyncDataOptions<ResT, DataT, PickKeys>, 'lazy'> & {
+          trpc?: TRPCRequestOptions
+          /**
+           * The custom unique key to use.
+           * @see https://nuxt.com/docs/api/composables/use-async-data#params
+           */
+          queryKey?: string
+        },
+      ) => AsyncData<PickFrom<DataT, PickKeys> | null, DataE>,
       query: Resolver<TProcedure>
     } : TProcedure extends AnyMutationProcedure ? {
       mutate: Resolver<TProcedure>