Merge pull request #31 from cloudoptlab/fix/2.x/fix-downloading-protection-switch

Fix download protection switch issue

Author: T-baby

src/background/download.ts

@@ -2,15 +2,27 @@ import * as coreConfig from '../core/config'
 import * as grade from '../core/grade'
 import * as notification from '../core/notification'
 import * as i18n from '../core/i18n'
+import * as message from './message'
 
-let listenerAdded: boolean = false
 let downloadSafetyEnabled: boolean = false
 
-export async function refresh() {
+async function refreshConfig(): Promise<void> {
     const config = await coreConfig.get()
     downloadSafetyEnabled = config.safeDownload
+}
+
+export async function initialize() {
+    refreshConfig()
+    message.addListener({
+        type: 'refresh-config',
+        callback: async (msg, sender, sendResponse) => {
+            refreshConfig()
+            sendResponse({})
+            return true
+        },
+    })
 
-    if (!listenerAdded && chrome.downloads) {
+    if (chrome.downloads) {
         chrome.downloads.onCreated.addListener(async (downloadItem) => {
             if (!downloadSafetyEnabled) {
                 return
@@ -27,8 +39,7 @@ export async function refresh() {
                 chrome.downloads.cancel(downloadItem.id)
             }
         })
-        listenerAdded = true
     }
 }
 
-refresh()
+initialize()