From ff888171687b0016cd9344dfede52070bacd7a80 Mon Sep 17 00:00:00 2001
From: wobsoriano <sorianorobertc@gmail.com>
Date: Sun, 18 Dec 2022 22:17:37 -0800
Subject: [PATCH] feat: add convenience link wrappers

---
 package.json                      |  1 +
 playground/plugins/trpc-client.ts | 33 ++++------------
 pnpm-lock.yaml                    |  2 +
 src/client/index.ts               | 65 ++++++++++++++++++++++++++++++-
 4 files changed, 74 insertions(+), 27 deletions(-)

diff --git a/package.json b/package.json
index a13466e..bb3b417 100644
--- a/package.json
+++ b/package.json
@@ -39,6 +39,7 @@
   },
   "dependencies": {
     "h3": "^1.0.2",
+    "ofetch": "^1.0.0",
     "ohash": "^1.0.0",
     "ufo": "^1.0.0"
   },
diff --git a/playground/plugins/trpc-client.ts b/playground/plugins/trpc-client.ts
index 4b449a0..aac5a28 100644
--- a/playground/plugins/trpc-client.ts
+++ b/playground/plugins/trpc-client.ts
@@ -1,36 +1,19 @@
-import { httpBatchLink, loggerLink } from '@trpc/client'
+import { loggerLink } from '@trpc/client'
 import superjson from 'superjson'
-import { FetchError } from 'ofetch'
-import { createTRPCNuxtClient } from 'trpc-nuxt/client'
+import { createTRPCNuxtClient, httpBatchLink } from 'trpc-nuxt/client'
 import type { AppRouter } from '~~/server/trpc/routers'
 
 export default defineNuxtPlugin(() => {
-  const headers = useRequestHeaders()
   const client = createTRPCNuxtClient<AppRouter>({
     transformer: superjson,
     links: [
       // adds pretty logs to your console in development and logs errors in production
-      loggerLink({
-        enabled: opts =>
-          process.env.NODE_ENV === 'development' ||
-          (opts.direction === 'down' && opts.result instanceof Error)
-      }),
-      httpBatchLink({
-        url: '/api/trpc',
-        headers () {
-          return headers
-        },
-        fetch: (input, options) =>
-          globalThis.$fetch.raw(input.toString(), options)
-            .catch((e) => {
-              if (e instanceof FetchError && e.response) { return e.response }
-              throw e
-            })
-            .then(response => ({
-              ...response,
-              json: () => Promise.resolve(response._data)
-            }))
-      })
+      // loggerLink({
+      //   enabled: opts =>
+      //     process.env.NODE_ENV === 'development' ||
+      //     (opts.direction === 'down' && opts.result instanceof Error)
+      // }),
+      httpBatchLink()
     ]
   })
 
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index f8a9c09..b229986 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -14,12 +14,14 @@ importers:
       concurrently: ^7.5.0
       eslint: ^8.25.0
       h3: ^1.0.2
+      ofetch: ^1.0.0
       ohash: ^1.0.0
       tsup: 6.4.0
       typescript: ^4.7.4
       ufo: ^1.0.0
     dependencies:
       h3: 1.0.2
+      ofetch: 1.0.0
       ohash: 1.0.0
       ufo: 1.0.0
     devDependencies:
diff --git a/src/client/index.ts b/src/client/index.ts
index 73dbc40..2bd143c 100644
--- a/src/client/index.ts
+++ b/src/client/index.ts
@@ -1,10 +1,12 @@
-import { type CreateTRPCClientOptions, type inferRouterProxyClient, createTRPCProxyClient } from '@trpc/client'
+import { type CreateTRPCClientOptions, type inferRouterProxyClient, createTRPCProxyClient, httpLink as _httpLink, httpBatchLink as _httpBatchLink, HttpBatchLinkOptions } from '@trpc/client'
 import { type AnyRouter } from '@trpc/server'
 import { createFlatProxy, createRecursiveProxy } from '@trpc/server/shared'
 import { hash } from 'ohash'
+import { FetchError } from 'ofetch'
 import { type DecoratedProcedureRecord } from './types'
 // @ts-expect-error: Nuxt auto-imports
-import { getCurrentInstance, onScopeDispose, useAsyncData } from '#imports'
+import { getCurrentInstance, onScopeDispose, useAsyncData, useRequestHeaders } from '#imports'
+import { HTTPLinkOptions } from '@trpc/client/dist/links/internals/httpUtils'
 
 /**
  * Calculates the key used for `useAsyncData` call
@@ -64,3 +66,62 @@ export function createTRPCNuxtClient<TRouter extends AnyRouter> (opts: CreateTRP
 
   return decoratedClient
 }
+
+function customFetch(input: RequestInfo | URL, init?: RequestInit | undefined) {
+  return globalThis.$fetch.raw(input.toString(), init)
+    .catch((e) => {
+      if (e instanceof FetchError && e.response) { return e.response }
+      throw e
+    })
+    .then(response => ({
+      ...response,
+      json: () => Promise.resolve(response._data)
+    }))
+}
+
+/**
+ * This is a convenience wrapper around the original httpLink
+ * that replaces regular `fetch` with a `$fetch` from Nuxt.
+ *
+ * During server-side rendering, calling $fetch to fetch your internal API routes
+ * will directly call the relevant function (emulating the request),
+ * saving an additional API call.
+ *
+ * @see https://nuxt.com/docs/api/utils/dollarfetch
+ */
+export function httpLink<TRouter extends AnyRouter>(opts?: HTTPLinkOptions) {
+  const headers = useRequestHeaders()
+
+  return _httpLink<TRouter>({
+    url: '/api/trpc',
+    headers () {
+      return headers
+    },
+    fetch: customFetch,
+    ...opts,
+  })
+}
+
+
+/**
+ * This is a convenience wrapper around the original httpBatchLink
+ * that replaces regular `fetch` with a `$fetch` from Nuxt.
+ *
+ * During server-side rendering, calling $fetch to fetch your internal API routes
+ * will directly call the relevant function (emulating the request),
+ * saving an additional API call.
+ *
+ * @see https://nuxt.com/docs/api/utils/dollarfetch
+ */
+export function httpBatchLink<TRouter extends AnyRouter>(opts?: HttpBatchLinkOptions) {
+  const headers = useRequestHeaders()
+
+  return _httpBatchLink<TRouter>({
+    url: '/api/trpc',
+    headers () {
+      return headers
+    },
+    fetch: customFetch,
+    ...opts,
+  })
+}