From 35b98d82f75161103691de37c3bfc1aa6e1011c0 Mon Sep 17 00:00:00 2001
From: Nathan Bierema <nbierema@gmail.com>
Date: Sat, 10 Aug 2024 17:23:29 -0400
Subject: [PATCH] Keep connection from content script to background alive

---
 extension/src/background/store/apiMiddleware.ts | 3 ++-
 extension/src/contentScript/index.ts            | 4 ++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/extension/src/background/store/apiMiddleware.ts b/extension/src/background/store/apiMiddleware.ts
index ac072f0e..54c52ae6 100644
--- a/extension/src/background/store/apiMiddleware.ts
+++ b/extension/src/background/store/apiMiddleware.ts
@@ -601,7 +601,8 @@ function onConnect<S, A extends Action<string>>(port: chrome.runtime.Port) {
     id = getId(port.sender!);
     if (port.sender!.frameId) id = `${id}-${port.sender!.frameId}`;
     connections.tab[id] = port;
-    listener = (msg: ContentScriptToBackgroundMessage<S, A>) => {
+    listener = (msg: ContentScriptToBackgroundMessage<S, A> | 'heartbeat') => {
+      if (msg === 'heartbeat') return;
       if (msg.name === 'INIT_INSTANCE') {
         if (typeof id === 'number') {
           chrome.pageAction.show(id);
diff --git a/extension/src/contentScript/index.ts b/extension/src/contentScript/index.ts
index 123bd3a3..dd0e96ff 100644
--- a/extension/src/contentScript/index.ts
+++ b/extension/src/contentScript/index.ts
@@ -318,3 +318,7 @@ function handleMessages<S, A extends Action<string>>(
 }
 
 window.addEventListener('message', handleMessages, false);
+
+setInterval(() => {
+  bg?.postMessage('heartbeat');
+}, 15000);