Update all versions

[rbdr/junction] / extension / peers.js

diff --git a/extension/peers.js b/extension/peers.js
index 51fd036b63053e68dafa49c6a20c7034d26918d8..fced1085277c85649c6f551fbc84ecba440af091 100644 (file)
--- a/extension/peers.js
+++ b/extension/peers.js
@@ -2,46 +2,104 @@ const internals = {
   peers: {},
 
   createAudioElement(source) {
   peers: {},
 
   createAudioElement(source) {

+    const audioElement = document.createElement("audio");
+    audioElement.setAttribute("class", "junction-call-audio");
+    audioElement.autoplay = "autoplay";
+    audioElement.srcObject = source;

 
 

-    const audioElement = document.createElement('audio');
-    audioElement.setAttribute('class', 'junction-call-audio');
-    audioElement.autoplay = 'autoplay';
+    document.querySelector("body").appendChild(audioElement);

 
 

-    // WE WILL NOT LOSE TADA SUPPORT
-    if (typeof source === 'string') {
-      audioElement.src = source;
+    return audioElement;
+  },
+};
+
+export function addPeer({
+  peerId,
+  shouldCreateOffer,
+  mediaStream,
+  onOffer,
+  socket,
+}) {
+  const peerConnection = new RTCPeerConnection(
+    { iceServers: internals.kIceServers },
+    { optional: [{ DtlsSrtpKeyAgreement: true }] },
+  );
+
+  internals.peers[peerId] = peerConnection;
+  mediaStream.getTracks().forEach((track) => {
+    peerConnection.addTrack(track, mediaStream);
+  });
+
+  peerConnection.onicecandidate = (event) => {
+    if (event.candidate) {
+      socket.emit("relayICECandidate", {
+        peerId: peerId,
+        candidate: event.candidate,
+      });

     }
     }

-    else {
-      audioElement.srcObject = source;
+  };
+
+  const remoteStream = new MediaStream();
+  peerConnection.ontrack = (event) => {
+    remoteStream.addTrack(event.track);
+    const remoteAudioElement = new Audio();
+    remoteAudioElement.srcObject = remoteStream;
+    remoteAudioElement.play();
+  };
+
+  peerConnection.onnegotiationneeded = async () => {
+    if (shouldCreateOffer) {
+      console.debug("Creating RTC offer to ", peerId);
+      const offer = await peerConnection.createOffer();
+      await peerConnection.setLocalDescription(offer);
+
+      onOffer({ peerId, offer });

     }
     }

+  };

 
 

-    document.querySelector('body').appendChild(audioElement);
+  console.info(`There are now ${countPeers()} participants`);
+}

 
 

-    return audioElement;
-  }
-};
+export function removePeer({ peerId }) {
+  delete internals.peers[peerId];
+  console.info(`There are now ${countPeers()} participants`);
+}

 
 

-export default {
-  add(id, source) {
+export async function answerPeerOffer({ peerId, offer }) {
+  console.info(`Answering peer ${peerId}`);
+  const peerConnection = internals.peers[peerId];

 
 

-    internals.peers[id] && this.remove(id);
-    internals.peers[id] = internals.createAudioElement(source)
-  },
+  const remoteDescription = new RTCSessionDescription(offer);
+  await peerConnection.setRemoteDescription(remoteDescription);

 
 

-  remove(id) {
+  const answer = await peerConnection.createAnswer();
+  await peerConnection.setLocalDescription(answer);

 
 

-    internals.peers[id] && internals.peers[id].remove();
-    delete internals.peers[id];
-  },
+  return { peerId, answer };
+}

 
 

-  count() {
+export async function processPeerAnswer({ peerId, answer }) {
+  console.info(`Processing answer for peer ${peerId}`);
+  const peerConnection = internals.peers[peerId];
+  const remoteDescription = new RTCSessionDescription(answer);
+  await peerConnection.setRemoteDescription(remoteDescription);
+}

 
 

-    return Object.keys(internals.peers).length;
-  },
+export async function addIceCandidate({ peerId, candidate }) {
+  console.info(`Adding ICE candidate for peer ${peerId}`);
+  const peerConnection = internals.peers[peerId];
+  console.info(peerConnection.signalingState);
+  const iceCandidate = new RTCIceCandidate(candidate);
+  await peerConnection.addIceCandidate(iceCandidate);
+}

 
 

-  reset() {
+export function countPeers() {
+  return Object.keys(internals.peers).length;
+}

 
 

-    internals.peers = {};
-    document.querySelectorAll('.junction-call-audio').forEach((audioElement) => audioElement.remove());
-  }
-};
+export function resetPeers() {
+  internals.peers = {};
+  document
+    .querySelectorAll(".junction-call-audio")
+    .forEach((audioElement) => audioElement.remove());
+}