fix(chrome): fix chrome ext disconnect (#5)

unadlib · Jan 2, 2025 · 742c994 · 742c994
1 parent 6607495
commit 742c994
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/examples/browser-extension/src/background/index.ts b/examples/browser-extension/src/background/index.ts
@@ -37,6 +37,10 @@ transport.onConnect((id) => {
   console.log('connect:', id);
 });
 
+transport.onDisconnect((id) => {
+  console.log('disconnect:', id);
+});
+
 // @ts-ignore
 transport.listen('contentToBg', async (a, b) => a);
 

diff --git a/src/transports/browserExtensionsTransport.ts b/src/transports/browserExtensionsTransport.ts
@@ -163,6 +163,9 @@ export abstract class BrowserExtensionsMainTransport<
               this.ports.delete(id);
             }
           });
+          this._onDisconnectCallback.forEach((callback) => {
+            callback(id);
+          });
         });
 
         // @ts-ignore