diff --git a/chatBot/bot.js b/chatBot/bot.js index 1c356f2..9f9e0a9 100644 --- a/chatBot/bot.js +++ b/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 'dotenv/config' - -// Generate a Buffer from the hexadecimal topic value -const topicHex = process.env.LINKUP_ROOM_ID; -const topicBuffer = Buffer.from(topicHex, 'hex'); +import Client from './includes/client.js'; // Adjust the import path as necessary +import 'dotenv/config'; // 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 diff --git a/chatBot/includes/chatBot.js b/chatBot/includes/client.js similarity index 79% rename from chatBot/includes/chatBot.js rename to chatBot/includes/client.js index 82ab798..1d4369a 100644 --- a/chatBot/includes/chatBot.js +++ b/chatBot/includes/client.js @@ -1,17 +1,18 @@ import Hyperswarm from 'hyperswarm'; +import EventEmitter from 'node:events' -class chatBot { - constructor(topicBuffer, botName, onMessageReceived) { - this.topicBuffer = topicBuffer; +class Client extends EventEmitter { + constructor(chatRoomID, botName) { + 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; diff --git a/package.json b/package.json index 446d262..1edae13 100644 --- a/package.json +++ b/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",