Refactor content script

[rbdr/junction] / extension / content_script.js

diff --git a/extension/content_script.js b/extension/content_script.js
index a9d713d74150aa9208533f8c24beaab4e1bb4e56..721176e8330e19ea21168656053634ff4457c3d2 100644 (file)
--- a/extension/content_script.js
+++ b/extension/content_script.js
@@ -1,84 +1,92 @@
-(() => {
-
-  const io = require('socket.io-client');
-
-  const internals = {
-
-    kSocketUrl: 'http://unlimited.pizza:8000/',
-
-    port: null,
-    socket: null,
-    peers: 0,
-
-    onMessage(message) {
-      internals[message.action](message.data);
-    },
-
-    async joinAudioCall(data) {
-
-      try {
-        const mediaStream = await navigator.mediaDevices.getUserMedia({ audio: true });
-
-        internals.socket = io(socketUrl);
-
-        internals.socket.on('connect', function() {
-
-          console.log("Connected to signaling server");
-          internals.socket.emit('join', {
-            'url': currentUrl,
-          });
+import { io } from "socket.io-client";
+import { initializeTada } from "./tada";
+import {
+  addPeer,
+  addIceCandidate,
+  answerPeerOffer,
+  processPeerAnswer,
+  removePeer,
+  resetPeers,
+} from "./peers";
+import { startMedia, stopMedia } from "./media";
+
+const internals = {
+  kSocketUrl: "https://junction.tranquil.services",
+  kIceServers: [{ urls: "stun:stun.l.google.com:19302" }],
+
+  port: null,
+  socket: null,
+
+  onMessage(message) {
+    internals[message.action](message.data);
+  },
+
+  onError(error) {
+    // TODO: How do we want to handle errors?
+    console.error(error.stack || error);
+  },
+
+  async joinAudioCall({ currentUrl, tada }) {
+    try {
+      const mediaStream = await startMedia();
+      const playTada = initializeTada(tada);
+
+      const socket = (internals.socket = io(internals.kSocketUrl, {
+        transports: ["websocket"],
+      }));
+
+      socket.on("error", internals.handleError);
+      socket.on("connect_error", internals.handleError);
+
+      socket.on("connect", () => {
+        console.debug("Connected to signaling server, group: ", currentUrl);
+        playTada();
+        socket.emit("join", {
+          room: currentUrl,
         });
-
-        internals.socket.on('disconnect', function() {
-
-          console.log("disconnected from signaling server");
+      });
+
+      socket.on("disconnect", () => {
+        console.debug("disconnected from signaling server");
+      });
+
+      socket.on("addPeer", ({ peerId }) => {
+        addPeer({
+          peerId,
+          mediaStream,
+          onOffer: (data) => socket.emit("relayOffer", data),
+          socket
         });
-
-        internals.socket.on('addPeer', function(data) {
-
-          console.log(data);
-          internals.peers++;
-          console.log(`There are now ${internals.peers} participants`);
-        });
-
-        internals.socket.on('removePeer', function() {
-
-          internals.peers--;
-          console.log(`There are now ${internals.peers} participants`);
-        });
-
-        internals.createAudioElement(data.tada);
-      }
-      catch (err) {
-
-        internals.port.postMessage({
-          action: 'error'
-        });
-        internals.port.disconnect();
-        internals.createAudioElement(data.tada);
-      }
-    },
-
-    hangUp() {
-      document.querySelectorAll('.junction-call-audio').forEach((audioElement) => audioElement.remove());
-      internals.socket.close();
+        playTada();
+      });
+
+      socket.on("offerReceived", async ({ peerId, offer }) => {
+        socket.emit("relayAnswer", await answerPeerOffer({ peerId, offer }));
+      });
+
+      socket.on("answerReceived", (data) => processPeerAnswer(data));
+      socket.on("ICECandidateReceived", (data) => addIceCandidate(data));
+      socket.on("removePeer", (data) => removePeer(data));
+    } catch (err) {
+      internals.port.postMessage({
+        action: "error",
+      });
       internals.port.disconnect();
-    },
+    }
+  },
 
-    createAudioElement(source, type = 'audio/wav') {
+  hangUp() {
+    resetPeers();
+    stopMedia();
+    internals.socket.close();
+    internals.port.disconnect();
+  },
+};
 
-      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);
-    }
-  };
+internals.port = chrome.runtime.connect({ name: "content" });
+internals.port.onMessage.addListener(internals.onMessage);
 
-  internals.port = chrome.runtime.connect({ name:"content" });
-  internals.port.onMessage.addListener(internals.onMessage);
-})();
+console.debug("Content Script Loaded");
 
 // Indicates to the background script that we executed correctly
 true;