From 50fca3c0d3870ab2ebb96833cdc4cc4af176b891 Mon Sep 17 00:00:00 2001
From: wanhose <hello@wanhose.dev>
Date: Sat, 1 Jun 2024 10:22:28 +0200
Subject: [PATCH] fix(browser-extension): icon disabled and no count due
 pageshow not setting up the extension

---
 packages/browser-extension/src/scripts/content.js | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/packages/browser-extension/src/scripts/content.js b/packages/browser-extension/src/scripts/content.js
index 43ccf82..e8132e3 100644
--- a/packages/browser-extension/src/scripts/content.js
+++ b/packages/browser-extension/src/scripts/content.js
@@ -324,6 +324,10 @@ async function setup(skipReadyStateHack) {
     skips = data?.skips;
     tokens = data?.tokens;
 
+    if (count > 0) {
+      dispatch({ type: 'SET_BADGE', value: `${count}` });
+    }
+
     // 2023-06-13: hack to force clean when data request takes too long and there are no changes later
     if (document.readyState === 'complete' && !skipReadyStateHack) {
       window.dispatchEvent(new Event(setupEventName));
@@ -402,6 +406,7 @@ window.addEventListener('load', () => {
  */
 window.addEventListener('pageshow', (event) => {
   if (document.visibilityState === 'visible' && event.persisted) {
+    setup(true);
     window.dispatchEvent(new Event(setupEventName));
   }
 });