apps/browser-extension/src/utils/providers.tsx


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

import { useEffect, useState } from "react";
import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
import { httpBatchLink } from "@trpc/client";
import superjson from "superjson";

import usePluginSettings, { getPluginSettings } from "./settings";
import { api } from "./trpc";

function getTRPCClient(address: string) {
  return api.createClient({
    links: [
      httpBatchLink({
        url: `${address}/api/trpc`,
        async headers() {
          const settings = await getPluginSettings();
          return {
            Authorization: `Bearer ${settings.apiKey}`,
          };
        },
        transformer: superjson,
      }),
    ],
  });
}

export function Providers({ children }: { children: React.ReactNode }) {
  const { settings } = usePluginSettings();
  const [queryClient] = useState(() => new QueryClient());

  const [trpcClient, setTrpcClient] = useState<
    ReturnType<typeof getTRPCClient>
  >(getTRPCClient(settings.address));

  useEffect(() => {
    setTrpcClient(getTRPCClient(settings.address));
  }, [settings.address]);

  return (
    <api.Provider
      key={settings.address}
      client={trpcClient}
      queryClient={queryClient}
    >
      <QueryClientProvider client={queryClient}>{children}</QueryClientProvider>
    </api.Provider>
  );
}