Bass Boost implemented

src/js/offscreen.js

@@ -1,6 +1,7 @@
 console.log("Offscreen is running");
 let audioContext;
 const gainNodes = new Map(); // Change the variable name to 'gainNodes'
+const biquadFilters = new Map(); // Add a new map for biquad filters
 
 chrome.runtime.onMessage.addListener(async (msg) => {
   console.log("[OFFSCREEN] Message received from WORKER");
@@ -51,4 +52,64 @@ chrome.runtime.onMessage.addListener(async (msg) => {
       }
     }
 
-});
+    if (msg.type === 'lowshelf-start'){
+      if (gainNodes.has(msg.tabId) && biquadFilters.has(msg.tabId)){
+        console.log('[OFFSCREEN] ERROR found gain node in lowshelf-start ');
+      }    
+      else{
+        console.log('[OFFSCREEN] Creating new gain node in lowshelf-start');
+
+        // creating a new gainNode
+        const media = await navigator.mediaDevices.getUserMedia({
+          audio: {
+            mandatory: {
+              chromeMediaSource: "tab",
+              chromeMediaSourceId: msg.data,
+            },
+          },
+        });
+
+        // Continue to play the captured audio to the user
+        const output = new AudioContext();
+        const source = output.createMediaStreamSource(media);
+
+        // Create gainNode
+        const gainNode = output.createGain();
+        //gainNode.gain.value = 0;
+
+        // Create a biquad filter
+        const biquadFilter = output.createBiquadFilter();
+        biquadFilter.type = "lowshelf";
+        biquadFilter.frequency.value = 200; // Cutoff frequency of 200 Hz
+        biquadFilter.gain.value = 6; // Boost frequencies below the cutoff by 6 dB
+
+        // Connect nodes
+        source.connect(gainNode);
+        gainNode.connect(biquadFilter);
+        biquadFilter.connect(output.destination);
+
+        // Save the gain node & biquadFilter for future changes
+        gainNodes.set(msg.tabId, gainNode);
+        biquadFilters.set(msg.tabId, biquadFilter);
+
+      }
+
+    }
+
+    if (msg.type === 'lowshelf'){
+
+      console.log('[OFFSCREEN-lowshelf] lowshelf entered');
+
+      if(gainNodes.has(msg.tabId) && biquadFilters.has(msg.tabId)){
+        console.log('[OFFSCREEN-lowshelf] Found gain node');
+        const biquadFilter = biquadFilters.get(msg.tabId);
+        biquadFilter.type = "lowshelf";
+        biquadFilter.frequency.value = 200;
+        biquadFilter.gain.value = 6;
+      }
+      else{
+        console.log('[OFFSCREEN] lowshelf entered');
+      }
+    }
+
+});

src/js/popup.js

@@ -23,7 +23,7 @@ document.addEventListener('DOMContentLoaded', async () => {
     // Currently for testing purposes
     bassBoost.addEventListener('click', async () => {
       console.log("[POPUP] Bass Boost clicked");
-      await chrome.runtime.sendMessage({ type: 'testGet'});
+      await chrome.runtime.sendMessage({ type: 'lowshelf-worker'});
     });
 
     // Clear storage for testing purposes

src/js/worker.js

@@ -56,17 +56,13 @@ chrome.runtime.onMessage.addListener(async (msg) => {
 
     case "lowshelf-worker":
       console.log("[SERVICE-WORKER] Lowshelf message received");
-      await lowshelf(); 
-
+      var currTab = await getCurrentTab();
       await lowshelf(currTab.id);
       break;
   }
 });
 
-// Adjust bass frequencies with lowshelf filter
-async function lowshelf(tabId){
-  console.log("[SERVICE-WORKER] Lowshelf function called");
-}
+
 
 async function saveTabLevel(tabId, level){
   let items = await chrome.storage.local.get('levels');
@@ -130,24 +126,46 @@ async function createOffscreenDocument(){
     }
 }
 
+
+// Adjust bass frequencies with lowshelf filter
+async function lowshelf(tabId){
+  console.log("[SERVICE-WORKER] Lowshelf function called");
+  await createOffscreenDocument();
+  let tabIdS = tabId.toString();
+
+  if (await containsTab(tabIdS)){
+    chrome.runtime.sendMessage({ type: 'lowshelf', target: 'offscreen', tabId: tabId});
+  }
+  else{
+    console.log("[SERVICE-WORKER] tab not found in activeStreams W/ tabId: ", tabId);
+    const streamId = await chrome.tabCapture.getMediaStreamId({
+      targetTabId: tabId
+    });
+    // Send the stream ID to the offscreen document to start recording
+    chrome.runtime.sendMessage({ type: 'lowshelf-start', target: 'offscreen', data: streamId, tabId: tabId});
+    await saveTabLevel(tabIdS, 100);
+
+  }
+}
+
+
 async function updateTabVolume(tabId, volume){
   await createOffscreenDocument();
   let tabIdS = tabId.toString();
   let volumeS = volume.toString();
   // Tab already exits
   if(await containsTab(tabIdS)){
-    console.log("[WORKER] tab found in activeStreams W/ tabId: ", tabId);
+    console.log("[WORKER] tab found in activeStreams W/ tabId: ", tabId, " volume: ", volume);
     chrome.runtime.sendMessage({ type: 'adjust-level', target: 'offscreen', tabId: tabId, level: volume});
     await saveTabLevel(tabIdS, volumeS);
   }
   // Tab doesn't exist
   else{
-    console.log("[WORKER] tab not found in activeStreams W/ tabId: ", tabId);
+    console.log("[WORKER] tab not found in activeStreams W/ tabId: ", tabId, " volume: ", volume);
     // Get a MediaStream for the Active Tab
     const streamId = await chrome.tabCapture.getMediaStreamId({
       targetTabId: tabId
     });
-
     // Send the stream ID to the offscreen document to start recording
     chrome.runtime.sendMessage({ type: 'start-recording', target: 'offscreen', data: streamId, tabId: tabId, level: volume});
     await saveTabLevel(tabIdS, volumeS);