Use peer and media in content script

1 files changed, 14 insertions, 19 deletions

diff --git a/extension/content_script.js b/extension/content_script.js
index 84a2fd3..70c085f 100644
--- a/extension/content_script.js
+++ b/extension/content_script.js
@@ -1,6 +1,10 @@
+'use strict';
+
 (() => {
 
   const io = require('socket.io-client');
+  const Peers = require('./peers');
+  const Media = require('./media');
 
   const internals = {
 
@@ -16,8 +20,10 @@
 
     async joinAudioCall(data) {
 
+      internals.tada = data.tada; // Keeping for fun
+
       try {
-        const mediaStream = await navigator.mediaDevices.getUserMedia({ audio: true });
+        const mediaStream = await Media.start();
 
         internals.socket = io(internals.kSocketUrl, {
           transports: ['websocket']
@@ -49,17 +55,15 @@
         internals.socket.on('addPeer', function(data) {
 
           console.log(data);
-          internals.peers++;
-          console.log(`There are now ${internals.peers} participants`);
+          Peers.add('id-'+Peers.count(), internals.tada);
+          console.log(`There are now ${Peers.count()} participants`);
         });
 
         internals.socket.on('removePeer', function() {
 
-          internals.peers--;
-          console.log(`There are now ${internals.peers} participants`);
+          Peers.remove('id-'+(Peers.count() - 1)); // This is only for testing, don't use count to remove ids.
+          console.log(`There are now ${Peers.count()} participants`);
         });
-
-        internals.createAudioElement(data.tada);
       }
       catch (err) {
 
@@ -67,24 +71,15 @@
           action: 'error'
         });
         internals.port.disconnect();
-        internals.createAudioElement(data.tada);
       }
     },
 
     hangUp() {
-      document.querySelectorAll('.junction-call-audio').forEach((audioElement) => audioElement.remove());
+
+      Peers.reset();
+      Media.stop();
       internals.socket.close();
       internals.port.disconnect();
-    },
-
-    createAudioElement(source, type = 'audio/wav') {
-
-      const audioElement = document.createElement('audio');
-      audioElement.setAttribute('class', 'junction-call-audio');
-      audioElement.src = source;
-      audioElement.autoplay = 'autoplay';
-      audioElement.type = type;
-      document.querySelector('body').appendChild(audioElement);
     }
   };