2024-06-03 19:23:14 -04:00
|
|
|
import Hyperswarm from 'hyperswarm';
|
2024-06-08 15:34:25 -04:00
|
|
|
import EventEmitter from 'node:events'
|
2024-06-03 19:23:14 -04:00
|
|
|
|
2024-06-08 15:34:25 -04:00
|
|
|
class Client extends EventEmitter {
|
2024-06-09 03:51:15 -04:00
|
|
|
constructor(botName) {
|
2024-06-08 15:34:25 -04:00
|
|
|
super();
|
2024-06-09 03:51:15 -04:00
|
|
|
if(!botName) return console.error("BotName is not defined!");
|
2024-06-03 19:23:14 -04:00
|
|
|
this.botName = botName;
|
|
|
|
this.swarm = new Hyperswarm();
|
|
|
|
this.setupSwarm();
|
|
|
|
}
|
|
|
|
|
|
|
|
setupSwarm() {
|
|
|
|
this.swarm.on('connection', (peer) => {
|
2024-06-08 16:31:24 -04:00
|
|
|
peer.on('data', message => this.emit('onMessage', peer, JSON.parse(message.toString())));
|
2024-06-08 16:29:51 -04:00
|
|
|
peer.on('error', e => {
|
2024-06-08 16:31:24 -04:00
|
|
|
this.emit('onError', e);
|
2024-06-08 16:29:51 -04:00
|
|
|
console.error(`Connection error: ${e}`)
|
|
|
|
});
|
2024-06-03 19:23:14 -04:00
|
|
|
});
|
|
|
|
|
|
|
|
this.swarm.on('update', () => {
|
|
|
|
console.log(`Peers count: ${this.swarm.connections.size}`);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2024-06-09 03:51:15 -04:00
|
|
|
joinChatRoom(chatRoomID) {
|
|
|
|
this.discovery = this.swarm.join(Buffer.from(chatRoomID, 'hex'), {client: true, server: true});
|
2024-06-03 19:23:14 -04:00
|
|
|
this.discovery.flushed().then(() => {
|
|
|
|
console.log(`Bot ${this.botName} joined the chat room.`);
|
2024-06-08 16:36:00 -04:00
|
|
|
this.emit('onBotJoinRoom');
|
2024-06-03 19:23:14 -04:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
sendMessage(message) {
|
|
|
|
console.log('Bot name:', this.botName);
|
|
|
|
const timestamp = Date.now(); // Generate timestamp
|
|
|
|
const peers = [...this.swarm.connections];
|
2024-06-08 16:31:24 -04:00
|
|
|
const data = JSON.stringify({name: this.botName, message, timestamp}); // Include timestamp
|
2024-06-03 19:23:14 -04:00
|
|
|
for (const peer of peers) peer.write(data);
|
|
|
|
}
|
|
|
|
|
|
|
|
destroy() {
|
|
|
|
this.swarm.destroy();
|
|
|
|
console.log(`Bot ${this.botName} disconnected.`);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-06-08 16:31:24 -04:00
|
|
|
export default Client;
|