arthur/trpc-nuxt
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/trpc-nuxt.git synced 2026-01-14 12:14:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: option to select headers to pass to useRequestHeaders in http links

Browse Source
This commit is contained in:
wobsoriano
2022-12-18 22:35:36 -08:00
parent a5cb0c754e
commit 09e0a6c479
1 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/client/links.ts
View File

@@ -1,11 +1,11 @@
import { httpLink as _httpLink, httpBatchLink as _httpBatchLink, HttpBatchLinkOptions } from '@trpc/client' import { httpLink as _httpLink, httpBatchLink as _httpBatchLink } from '@trpc/client'
import { type AnyRouter } from '@trpc/server' import { type AnyRouter } from '@trpc/server'
import { FetchError } from 'ofetch' import { FetchError } from 'ofetch'
// @ts-expect-error: Nuxt auto-imports // @ts-expect-error: Nuxt auto-imports
import { useRequestHeaders } from '#imports' import { useRequestHeaders } from '#imports'
import { type HTTPLinkOptions } from '@trpc/client/dist/links/internals/httpUtils' import { type HTTPLinkOptions as _HTTPLinkOptions } from '@trpc/client/dist/links/internals/httpUtils'
function customFetch(input: RequestInfo | URL, init?: RequestInit | undefined) { function customFetch(input: RequestInfo | URL, init?: RequestInit) {
return globalThis.$fetch.raw(input.toString(), init) return globalThis.$fetch.raw(input.toString(), init)
.catch((e) => { .catch((e) => {
if (e instanceof FetchError && e.response) { return e.response } if (e instanceof FetchError && e.response) { return e.response }
@@ -17,9 +17,17 @@ function customFetch(input: RequestInfo | URL, init?: RequestInit | undefined) {
})) }))
} }
export interface HTTPLinkOptions extends _HTTPLinkOptions {
/**
* Select headers to pass to `useRequestHeaders`.
*/
pickHeaders?: string[]
}
/** /**
* This is a convenience wrapper around the original httpLink * This is a convenience wrapper around the original httpLink
* that replaces regular `fetch` with a `$fetch` from Nuxt. * that replaces regular `fetch` with a `$fetch` from Nuxt. It
* also sets the default headers based on `useRequestHeaders` values.
* *
* During server-side rendering, calling $fetch to fetch your internal API routes * During server-side rendering, calling $fetch to fetch your internal API routes
* will directly call the relevant function (emulating the request), * will directly call the relevant function (emulating the request),
@@ -28,7 +36,7 @@ function customFetch(input: RequestInfo | URL, init?: RequestInit | undefined) {
* @see https://nuxt.com/docs/api/utils/dollarfetch * @see https://nuxt.com/docs/api/utils/dollarfetch
*/ */
export function httpLink<TRouter extends AnyRouter>(opts?: HTTPLinkOptions) { export function httpLink<TRouter extends AnyRouter>(opts?: HTTPLinkOptions) {
const headers = useRequestHeaders() const headers = useRequestHeaders(opts?.pickHeaders)
return _httpLink<TRouter>({ return _httpLink<TRouter>({
url: '/api/trpc', url: '/api/trpc',
@@ -40,10 +48,15 @@ export function httpLink<TRouter extends AnyRouter>(opts?: HTTPLinkOptions) {
}) })
} }
export interface HttpBatchLinkOptions extends HTTPLinkOptions {
maxURLLength?: number;
}
/** /**
* This is a convenience wrapper around the original httpBatchLink * This is a convenience wrapper around the original httpBatchLink
* that replaces regular `fetch` with a `$fetch` from Nuxt. * that replaces regular `fetch` with a `$fetch` from Nuxt. It
* also sets the default headers based on `useRequestHeaders` values.
* *
* During server-side rendering, calling $fetch to fetch your internal API routes * During server-side rendering, calling $fetch to fetch your internal API routes
* will directly call the relevant function (emulating the request), * will directly call the relevant function (emulating the request),
@@ -52,7 +65,7 @@ export function httpLink<TRouter extends AnyRouter>(opts?: HTTPLinkOptions) {
* @see https://nuxt.com/docs/api/utils/dollarfetch * @see https://nuxt.com/docs/api/utils/dollarfetch
*/ */
export function httpBatchLink<TRouter extends AnyRouter>(opts?: HttpBatchLinkOptions) { export function httpBatchLink<TRouter extends AnyRouter>(opts?: HttpBatchLinkOptions) {
const headers = useRequestHeaders() const headers = useRequestHeaders(opts?.pickHeaders)
return _httpBatchLink<TRouter>({ return _httpBatchLink<TRouter>({
url: '/api/trpc', url: '/api/trpc',