fix: score column empty on message filter

friedPotat0 · Jul 7, 2022 · 18cceec · 18cceec
1 parent b37b3b6
commit 18cceec
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/background/background.js b/src/background/background.js
@@ -113,6 +113,17 @@ async function onMessageDisplayed(tab, message) {
   }
 }
 
+/**
+ * Fired when the selected messages change in any mail tab.
+ * @param {Tab} tab
+ * @param {MessageList} selectedMessages
+ */
+async function onSelectedMessagesChanged(tab, selectedMessages) {
+  const spamScores = messenger.SpamScores
+  const win = await messenger.windows.getCurrent()
+  spamScores.repaint(win.id)
+}
+
 /**
  * Fired when the displayed folder changes in any mail tab
  * @param {Tab} tab
@@ -166,6 +177,7 @@ const init = async () => {
   // Add Listeners
   messenger.messageDisplay.onMessageDisplayed.addListener(onMessageDisplayed)
   messenger.mailTabs.onDisplayedFolderChanged.addListener(onDisplayedFolderChanged)
+  messenger.mailTabs.onSelectedMessagesChanged.addListener(onSelectedMessagesChanged)
 
   const win = await messenger.windows.getCurrent()
   spamScores.repaint(win.id)