From c32fdf78939e617cc180ef9e7d1b18573205b881 Mon Sep 17 00:00:00 2001
From: wobsoriano <sorianorobertc@gmail.com>
Date: Mon, 15 May 2023 09:00:04 -0700
Subject: [PATCH] feat: allow providing a transform output type

---
 playground/pages/index.vue |  9 +++++++++
 src/client/types.ts        | 10 ++++++----
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/playground/pages/index.vue b/playground/pages/index.vue
index dc208fb..b9116e2 100644
--- a/playground/pages/index.vue
+++ b/playground/pages/index.vue
@@ -1,4 +1,13 @@
 <script setup lang="ts">
+import { TRPCClientError, TRPCClientErrorLike } from '@trpc/client'
+import type { inferRouterOutputs } from '@trpc/server'
+import type { AppRouter } from '@/server/trpc/routers'
+
+type RouterOutput = inferRouterOutputs<AppRouter>
+type GetTodosOutput = RouterOutput['todo']['getTodos']
+
+type ErrorOutput = TRPCClientError<AppRouter>
+
 const { $client } = useNuxtApp()
 
 const addTodo = async () => {
diff --git a/src/client/types.ts b/src/client/types.ts
index a42c670..359857b 100644
--- a/src/client/types.ts
+++ b/src/client/types.ts
@@ -50,12 +50,14 @@ type DecorateProcedure<
 > = TProcedure extends AnyQueryProcedure
   ? {
       useQuery: <
-      TData = inferTransformedProcedureOutput<TProcedure>,
-      PickKeys extends KeysOf<TData> = KeysOf<TData>,
+      ResT = inferTransformedProcedureOutput<TProcedure>,
+      DataE = TRPCClientErrorLike<TProcedure>,
+      DataT = ResT,
+      PickKeys extends KeysOf<DataT> = KeysOf<DataT>,
      >(
         input: inferProcedureInput<TProcedure>,
-        opts?: AsyncDataOptions<TData, TData, PickKeys> & { trpc?: TRPCRequestOptions },
-      ) => AsyncData<PickFrom<TData, PickKeys>, TRPCClientErrorLike<TProcedure>>,
+        opts?: AsyncDataOptions<ResT, DataT, PickKeys> & { trpc?: TRPCRequestOptions },
+      ) => AsyncData<PickFrom<DataT, PickKeys>, DataE>,
       query: Resolver<TProcedure>
     } : TProcedure extends AnyMutationProcedure ? {
       mutate: Resolver<TProcedure>