1 files changed, 10 insertions, 14 deletions
diff --git a/extension/background.js b/extension/background.js
index 020a2be..9b7db26 100644
--- a/extension/background.js
+++ b/extension/background.js
@@ -26,18 +26,14 @@ port.onDisconnect.addListener(() => {
 });
 
 browser.runtime.onMessage.addListener(({hostname}) => {
-    const map = {
-        'scratch.mit.edu': ['mantis'],
-        'stackoverflow.com': ['bass', 'main'],
-        'www.youtube.com': []
-    };
-
-    const mode = map[hostname];
-
-    if (mode) {
-        console.log('BGM:', mode);
-        port.postMessage(mode.map(track => ({track, volume: 100})));
-    } else {
-        console.log('No BGM found for ' + location.hostname);
-    }
+    browser.storage.sync.get('siteSettings').then(({siteSettings}) => {
+        const mode = siteSettings[hostname];
+
+        if (mode) {
+            console.log('BGM:', mode);
+            port.postMessage(mode.map(track => ({track, volume: 100})));
+        } else {
+            console.log('No BGM found for ' + hostname);
+        }
+    });
 });