From 34df85f8b97cfada184e47561a982776d9a57b68 Mon Sep 17 00:00:00 2001
From: wanhose <hello@wanhose.dev>
Date: Tue, 8 Oct 2024 01:46:47 +0200
Subject: [PATCH] feat(browser-extension): switch to warn mode after reporting
 an issue

---
 packages/browser-extension/src/scripts/background.js | 9 ++++++---
 packages/browser-extension/src/scripts/dialog.js     | 3 +++
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/packages/browser-extension/src/scripts/background.js b/packages/browser-extension/src/scripts/background.js
index a68ddf5..6b55bb7 100644
--- a/packages/browser-extension/src/scripts/background.js
+++ b/packages/browser-extension/src/scripts/background.js
@@ -294,7 +294,10 @@ browser.runtime.onMessage.addListener((message, sender, callback) => {
       }
       break;
     case 'ENABLE_ICON':
-      if (isPage && tabId !== undefined) enableIcon(hostname, tabId);
+      if (isPage && tabId !== undefined) {
+        enableIcon(hostname, tabId).then(callback);
+        return true;
+      }
       break;
     case 'ENABLE_POPUP':
       if (isPage && tabId !== undefined) {
@@ -344,8 +347,8 @@ browser.runtime.onMessage.addListener((message, sender, callback) => {
       }
       break;
     case 'UPDATE_STORE':
-      triggerStoreUpdate(hostname, message.state);
-      break;
+      triggerStoreUpdate(hostname, message.state).then(callback);
+      return true;
     default:
       break;
   }
diff --git a/packages/browser-extension/src/scripts/dialog.js b/packages/browser-extension/src/scripts/dialog.js
index 211ac15..c9af5d6 100644
--- a/packages/browser-extension/src/scripts/dialog.js
+++ b/packages/browser-extension/src/scripts/dialog.js
@@ -219,7 +219,10 @@ async function submitButtonClickHandler(event) {
   const submitView = dialog?.getElementsByClassName('report-dialog-submit-view')[0];
   const userAgent = window.navigator.userAgent;
   const issueUrl = await dispatch({ userAgent, reason: reasonText, url: urlText, type: 'REPORT' });
+  const issue = { expiresIn: Date.now() + 8 * 60 * 60 * 1000, flags: ['bug'], url: issueUrl };
 
+  await dispatch({ hostname, state: { issue }, type: 'UPDATE_STORE' });
+  await dispatch({ hostname, type: 'ENABLE_ICON' });
   formView?.setAttribute('hidden', 'true');
   issueButton?.addEventListener('click', () => window.open(issueUrl, '_blank'));
   submitView?.removeAttribute('hidden');