Merge pull request #1264 from ueokande/fix-establish-connection-issue

Search a content from frames successfully loaded

1 files changed, 23 insertions, 0 deletions

diff --git a/src/background/infrastructures/FindPortListener.ts b/src/background/infrastructures/FindPortListener.ts
new file mode 100644
index 0000000..ca82439
--- /dev/null
+++ b/src/background/infrastructures/FindPortListener.ts
@@ -0,0 +1,23 @@
+import { injectable } from "tsyringe";
+
+type OnConnectFunc = (port: browser.runtime.Port) => void;
+type OnDisconnectFunc = (port: browser.runtime.Port) => void;
+
+@injectable()
+export default class FindPortListener {
+  constructor(
+    private readonly onConnect: OnConnectFunc,
+    private readonly onDisconnect: OnDisconnectFunc
+  ) {}
+
+  run(): void {
+    browser.runtime.onConnect.addListener((port) => {
+      if (port.name !== "vimvixen-find") {
+        return;
+      }
+
+      port.onDisconnect.addListener(this.onDisconnect);
+      this.onConnect(port);
+    });
+  }
+}