1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
import { useMemo } from "react";
import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
import { createTRPCClient, httpBatchLink } from "@trpc/client";
import superjson from "superjson";
import type { AppRouter } from "@karakeep/trpc/routers/_app";
import { TRPCProvider } from "../trpc";
interface Settings {
apiKey?: string;
address: string;
customHeaders?: Record<string, string>;
}
function getTRPCClient(settings: Settings) {
return createTRPCClient<AppRouter>({
links: [
httpBatchLink({
url: `${settings.address}/api/trpc`,
maxURLLength: 14000,
headers() {
return {
Authorization: settings.apiKey
? `Bearer ${settings.apiKey}`
: undefined,
...settings.customHeaders,
};
},
transformer: superjson,
}),
],
});
}
export function TRPCSettingsProvider({
settings,
children,
}: {
settings: Settings;
children: React.ReactNode;
}) {
const queryClient = useMemo(() => new QueryClient(), [settings]);
const trpcClient = useMemo(() => getTRPCClient(settings), [settings]);
return (
<QueryClientProvider client={queryClient}>
<TRPCProvider trpcClient={trpcClient} queryClient={queryClient}>
{children}
</TRPCProvider>
</QueryClientProvider>
);
}
|
