try to reconnect after disconnect

authier-pm · Jan 25, 2024 · 4962f66 · 4962f66
1 parent c8e5836
commit 4962f66
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/web-extension/src/content-script/connectTRPC.ts b/web-extension/src/content-script/connectTRPC.ts
@@ -3,9 +3,28 @@ import { AppRouter } from '../background/chromeRuntimeListener'
 import { chromeLink } from '@capaj/trpc-browser/link'
 import browser from 'webextension-polyfill'
 
+const connectTRPC = () => {
+  const port = browser.runtime.connect()
+
+  trpc = createTRPCProxyClient<AppRouter>({
+    // @ts-expect-error
+    links: [chromeLink({ port })]
+  })
+
+  port.onDisconnect.addListener(() => {
+    connectTRPC() // recursively call to reconnect
+  })
+
+  return trpc
+}
+
 const port = browser.runtime.connect()
 
-export const trpc = createTRPCProxyClient<AppRouter>({
+export let trpc = createTRPCProxyClient<AppRouter>({
   // @ts-expect-error
   links: [chromeLink({ port })]
 })
+
+port.onDisconnect.addListener(() => {
+  connectTRPC() // recursively call to reconnect
+})