Merge pull request 'Rework ChatBot' (#1) from MiTask/LinkUp-P2P-Chat:main into main

Reviewed-on: snxraven/LinkUp-P2P-Chat#1

chatBot/bot.js

@@ -1,14 +1,10 @@
 import fs from 'fs';
 import path from 'path';
-import chatBot from './includes/chatBot.js'; // Adjust the import path as necessary
+import Client from './includes/client.js'; // Adjust the import path as necessary
-import 'dotenv/config'
+import 'dotenv/config';
-
-// Generate a Buffer from the hexadecimal topic value
-const topicHex = process.env.LINKUP_ROOM_ID;
-const topicBuffer = Buffer.from(topicHex, 'hex');
 
 // Create a new instance of the chatBot class with a valid botName
-const botName = process.env.BOT_NAME;; // Replace 'MyBot' with the desired bot name
+const botName = process.env.BOT_NAME; // Replace 'MyBot' with the desired bot name
 
 // Load commands from the 'commands' directory
 const commandsDir = path.join(new URL('./commands/', import.meta.url).pathname);
@@ -38,7 +34,11 @@ async function loadCommands() {
 }
 
 loadCommands().then(commands => {
-  const onMessageReceived = (peer, message) => {
+  // Create the chatBot instance with the defined onMessageReceived function
+  const bot = new Client(process.env.LINKUP_ROOM_ID, botName);
+
+  // We use Event Emitter here to handle new messages
+  bot.on('onMessage', (peer, message) => {
     if (message.type == "icon") return;
     console.log(message);
 
@@ -57,10 +57,7 @@ loadCommands().then(commands => {
         commandHandler.handler(bot, args, message);
       }
     }
-  };
+  })
-
-  // Create the chatBot instance with the defined onMessageReceived function
-  const bot = new chatBot(topicBuffer, botName, onMessageReceived);
   bot.joinChatRoom();
 
   // Wait for 2 seconds for the bot to connect

chatBot/includes/client.js

@@ -1,17 +1,18 @@
 import Hyperswarm from 'hyperswarm';
+import EventEmitter from 'node:events'
 
-class chatBot {
+class Client extends EventEmitter {
-  constructor(topicBuffer, botName, onMessageReceived) {
+  constructor(chatRoomID, botName) {
-    this.topicBuffer = topicBuffer;
+    super();
+    this.topicBuffer = Buffer.from(chatRoomID, 'hex');
     this.botName = botName;
     this.swarm = new Hyperswarm();
-    this.onMessageReceived = onMessageReceived;
     this.setupSwarm();
   }
 
   setupSwarm() {
     this.swarm.on('connection', (peer) => {
-      peer.on('data', message => this.onMessageReceived(peer, JSON.parse(message.toString())));
+      peer.on('data', message => this.emit("onMessage", peer, JSON.parse(message.toString())));
       peer.on('error', e => console.log(`Connection error: ${e}`));
     });
 
@@ -41,4 +42,4 @@ class chatBot {
   }
 }
 
-export default chatBot;
+export default Client;

package.json

@@ -22,6 +22,7 @@
   "dependencies": {
     "b4a": "^1.6.6",
     "corestore": "^6.18.2",
+    "dotenv": "^16.4.5",
     "electron": "^30.0.8",
     "hypercore-crypto": "^3.4.1",
     "hyperdrive": "^11.8.1",