From 2faf2b6f78a6b7a99d3c04b3be7aacbb43c8b1b6 Mon Sep 17 00:00:00 2001
From: Raven Scott <raven@ssh.surf>
Date: Thu, 13 Jun 2024 21:44:30 -0400
Subject: [PATCH] Moving to MultiSwarm with proper connection tracking per
 topic

---
 app.js | 92 +++++++++++++++++++++++++++-------------------------------
 1 file changed, 42 insertions(+), 50 deletions(-)

diff --git a/app.js b/app.js
index bf83784..eef5d48 100644
--- a/app.js
+++ b/app.js
@@ -13,7 +13,6 @@ const drive = new Hyperdrive(store);
 
 await drive.ready();
 
-let swarm;
 let activeRooms = [];
 const eventEmitter = new EventEmitter();
 
@@ -36,10 +35,25 @@ function getRandomPort() {
   return Math.floor(Math.random() * (65535 - 49152 + 1)) + 49152;
 }
 
+function currentTopic() {
+  return document.querySelector('#chat-room-topic').innerText;
+}
+
+function getCurrentPeerCount() {
+  const topic = currentTopic();
+  const room = activeRooms.find(room => room.topic === topic);
+  return room ? room.swarm.connections.size : 0;
+}
+
+function updatePeerCount() {
+  const peerCountElement = document.querySelector('#peers-count');
+  if (peerCountElement) {
+    peerCountElement.textContent = getCurrentPeerCount(); // Display the actual peer count
+  }
+}
+
 async function initialize() {
   try {
-    swarm = new Hyperswarm();
-
     servePort = getRandomPort();
     const serve = new ServeDrive({ port: servePort, get: ({ key, filename, version }) => drive });
     await serve.ready();
@@ -63,10 +77,6 @@ async function initialize() {
       handleIncomingMessage(messageObj);
     });
 
-    swarm.on('connection', handleConnection);
-    swarm.on('error', (err) => console.error('Swarm error:', err));
-    swarm.on('close', () => console.log('Swarm closed'));
-
     document.addEventListener("DOMContentLoaded", (event) => {
       hljs.highlightAll();
     });
@@ -109,7 +119,7 @@ function setupEventListeners() {
   }
   if (removeRoomBtn) {
     removeRoomBtn.addEventListener('click', () => {
-      const topic = document.querySelector('#chat-room-topic').innerText;
+      const topic = currentTopic();
       leaveRoom(topic);
     });
   }
@@ -162,30 +172,18 @@ function handleIncomingMessage(messageObj) {
 }
 
 function handleConnection(connection, info) {
-  updatePeerCount();
-  console.log('Peer connected, current peer count:', swarm.connections.size);
-
-  // Send the current user's icon to the new peer
-  drive.get(`/icons/${config.userName}.png`).then(iconBuffer => {
-    if (iconBuffer) {
-      const iconMessage = JSON.stringify({
-        type: 'icon',
-        username: config.userName,
-        avatar: b4a.toString(iconBuffer, 'base64'),
-      });
-      connection.write(iconMessage);
-    }
-  });
-
+  console.log('New connection', info);
   connection.on('data', (data) => {
     const messageObj = JSON.parse(data.toString());
     eventEmitter.emit('onMessage', messageObj);
   });
 
   connection.on('close', () => {
+    console.log('Connection closed', info);
     updatePeerCount();
-    console.log('Peer disconnected, current peer count:', swarm.connections.size);
   });
+
+  updatePeerCount();
 }
 
 function setupTalkButton() {
@@ -211,7 +209,7 @@ function setupTalkButton() {
       const arrayBuffer = await audioBlob.arrayBuffer();
       const buffer = new Uint8Array(arrayBuffer);
 
-      const topic = document.querySelector('#chat-room-topic').innerText;
+      const topic = currentTopic();
       const filePath = `/audio/${Date.now()}.webm`;
       await drive.put(filePath, buffer);
 
@@ -228,7 +226,7 @@ function setupTalkButton() {
 
       console.log('Sending audio message:', audioMessage); // Debugging log
 
-      const peers = [...swarm.connections];
+      const peers = [...activeRooms.find(room => room.topic === topic).swarm.connections];
       for (const peer of peers) {
         peer.write(JSON.stringify(audioMessage));
       }
@@ -315,10 +313,15 @@ async function joinSwarm(topicBuffer) {
   const topic = b4a.toString(topicBuffer, 'hex');
   if (!activeRooms.some(room => room.topic === topic)) {
     try {
+      const swarm = new Hyperswarm();
       const discovery = swarm.join(topicBuffer, { client: true, server: true });
       await discovery.flushed();
 
-      activeRooms.push({ topic, discovery });
+      swarm.on('connection', (connection, info) => {
+        handleConnection(connection, info);
+      });
+
+      activeRooms.push({ topic, swarm, discovery });
 
       console.log('Joined room:', topic); // Debugging log
 
@@ -385,8 +388,7 @@ function exitEditMode(roomItem, input, topic) {
     }
 
     // Check if the edited room is the current room in view
-    const currentTopic = document.querySelector('#chat-room-topic').innerText;
-    if (currentTopic === topic) {
+    if (currentTopic() === topic) {
       const chatRoomName = document.querySelector('#chat-room-name');
       if (chatRoomName) {
         chatRoomName.innerText = newAlias;
@@ -430,7 +432,7 @@ function leaveRoom(topic) {
   const roomIndex = activeRooms.findIndex(room => room.topic === topic);
   if (roomIndex !== -1) {
     const room = activeRooms[roomIndex];
-    room.discovery.destroy();
+    room.swarm.destroy();
     activeRooms.splice(roomIndex, 1);
   }
 
@@ -455,7 +457,7 @@ function sendMessage(e) {
   const message = document.querySelector('#message').value;
   document.querySelector('#message').value = '';
 
-  const topic = document.querySelector('#chat-room-topic').innerText;
+  const topic = currentTopic();
 
   console.log('Sending message:', message); // Debugging log
 
@@ -470,7 +472,7 @@ function sendMessage(e) {
     timestamp: Date.now()
   });
 
-  const peers = [...swarm.connections];
+  const peers = [...activeRooms.find(room => room.topic === topic).swarm.connections];
   for (const peer of peers) {
     peer.write(messageObj);
   }
@@ -486,7 +488,7 @@ async function handleFileInput(event) {
       await drive.put(filePath, buffer);
       const fileUrl = `http://localhost:${servePort}${filePath}`;
 
-      const topic = document.querySelector('#chat-room-topic').innerText;
+      const topic = currentTopic();
 
       const fileMessage = {
         type: 'file',
@@ -500,7 +502,7 @@ async function handleFileInput(event) {
 
       console.log('Sending file message:', fileMessage); // Debugging log
 
-      const peers = [...swarm.connections];
+      const peers = [...activeRooms.find(room => room.topic === topic).swarm.connections];
       for (const peer of peers) {
         peer.write(JSON.stringify(fileMessage));
       }
@@ -550,19 +552,11 @@ function addFileMessage(from, fileName, fileUrl, fileType, avatar, topic) {
   $div.appendChild($content);
 
   // Only render the message if it's for the current room
-  const currentTopic = document.querySelector('#chat-room-topic').innerText;
-  if (currentTopic === topic) {
+  if (currentTopic() === topic) {
     document.querySelector('#messages').appendChild($div);
     scrollToBottom();
   } else {
-    console.log(`Message for topic ${topic} not rendered because current topic is ${currentTopic}`); // Debugging log
-  }
-}
-
-function updatePeerCount() {
-  const peerCountElement = document.querySelector('#peers-count');
-  if (peerCountElement) {
-    peerCountElement.textContent = swarm.connections.size; // Display the actual peer count
+    console.log(`Message for topic ${topic} not rendered because current topic is ${currentTopic()}`); // Debugging log
   }
 }
 
@@ -583,8 +577,7 @@ function onMessageAdded(from, message, avatar, topic) {
   addMessageToStore(topic, messageObj);
 
   // Only render messages for the current room
-  const currentTopic = document.querySelector('#chat-room-topic').innerText;
-  if (currentTopic === topic) {
+  if (currentTopic() === topic) {
     const $div = document.createElement('div');
     $div.classList.add('message');
 
@@ -624,7 +617,7 @@ function onMessageAdded(from, message, avatar, topic) {
     document.querySelector('#messages').appendChild($div);
     scrollToBottom();
   } else {
-    console.log(`Message for topic ${topic} not rendered because current topic is ${currentTopic}`); // Debugging log
+    console.log(`Message for topic ${topic} not rendered because current topic is ${currentTopic()}`); // Debugging log
   }
 }
 
@@ -658,12 +651,11 @@ function addAudioMessage(from, audioUrl, avatar, topic) {
   $div.appendChild($content);
 
   // Only render the message if it's for the current room
-  const currentTopic = document.querySelector('#chat-room-topic').innerText;
-  if (currentTopic === topic) {
+  if (currentTopic() === topic) {
     document.querySelector('#messages').appendChild($div);
     scrollToBottom();
   } else {
-    console.log(`Message for topic ${topic} not rendered because current topic is ${currentTopic}`); // Debugging log
+    console.log(`Message for topic ${topic} not rendered because current topic is ${currentTopic()}`); // Debugging log
   }
 }