Compare commits

..

No commits in common. "cc49b165b0586c02ed0c8f579fa796bdf3091e9d" and "47f3f8a6cf2f0d579ac58f68c98bb6fd0302b910" have entirely different histories.

220
bot.js
View File

@ -4,48 +4,44 @@ const crypto = require('hypercore-crypto');
const { Client, GatewayIntentBits, Partials, messageLink } = require("discord.js"); const { Client, GatewayIntentBits, Partials, messageLink } = require("discord.js");
const goodbye = require('graceful-goodbye') const goodbye = require('graceful-goodbye')
require('dotenv').config() require('dotenv').config()
const b4a = require('b4a')
function sleep(ms) {
return new Promise((resolve) => {
setTimeout(resolve, ms);
});
}
const client = new Client({ const client = new Client({
'intents': [ 'intents': [
GatewayIntentBits.DirectMessages, GatewayIntentBits.DirectMessages,
GatewayIntentBits.MessageContent, GatewayIntentBits.MessageContent,
GatewayIntentBits.GuildMessages, GatewayIntentBits.GuildMessages,
GatewayIntentBits.Guilds, GatewayIntentBits.Guilds,
GatewayIntentBits.GuildBans, GatewayIntentBits.GuildBans,
GatewayIntentBits.GuildMessages GatewayIntentBits.GuildMessages
], ],
'partials': [Partials.Channel, Partials.Message, Partials.MessageContent] 'partials': [Partials.Channel, Partials.Message, Partials.MessageContent]
}); });
const topicMap = new Map(); const topicMap = new Map();
const swarmMap = new Map(); goodbye(() => {
for(let [channel_id, topic_obj] of topicMap)
{
topic_obj.swarm.destroy();
}
})
client.on('ready', () => { client.on('ready', () => {
console.log(`Logged in as ${client.user.tag}!`); console.log(`Logged in as ${client.user.tag}!`);
}); });
client.on('messageCreate', msg => { client.on('messageCreate', msg => {
if (msg.content.startsWith("!") && !msg.author.bot) { if (msg.content.startsWith("!") && !msg.author.bot){
const args = msg.content.split(" "); const args = msg.content.split(" ");
const cmd = args.shift().slice(1); const cmd = args.shift().slice(1);
if (cmd === "current") { if (cmd === "current-topic") {
if (topicMap.has(msg.channel.id)) { if(topicMap.has(msg.channel.id))
let topic_obj = topicMap.get(msg.channel.id); {
msg.reply(`The current topic for this channel is: ${topic_obj.topic.toString('hex')}`); let topic_obj = topicMap.get(msg.channel.id);
msg.reply(`The current topic for this channel is: ${topic_obj.topic.toString('hex')}`);
} }
else { else{
msg.reply(`No topic is assigned for this channel`); msg.reply(`No topic is assigned for this channel`);
} }
} }
@ -57,143 +53,113 @@ client.on('messageCreate', msg => {
} }
let topic = crypto.randomBytes(32); let topic = crypto.randomBytes(32);
let channel_id = args[1]; let channel_id = args[1];
if (!channel_id) { if(!channel_id)
{
msg.reply("Please provide a channel_id to send this topic's messages") msg.reply("Please provide a channel_id to send this topic's messages")
return; return;
} }
//Check if the given channel_id exists or not. //Check if the given channel_id exists or not.
if (!client.channels.cache.get(channel_id)) { if(!client.channels.cache.get(channel_id))
{
msg.reply("The provided channel id does not exist."); msg.reply("The provided channel id does not exist.");
return; return;
} }
let swarm; let swarm;
let conns = []; let conns =[];
if (topicMap.has(channel_id)) { if(topicMap.has(channel_id))
{
topic_obj = topicMap.get(channel_id); topic_obj = topicMap.get(channel_id);
swarm = topic_obj.swarm; swarm = topic_obj.swarm;
conns = topic_obj.conns; conns = topic_obj.conns;
} }
else { else
{
swarm = new Hyperswarm(); swarm = new Hyperswarm();
topicMap.set(channel_id, { topic: topic, swarm: swarm, conns: conns }); topicMap.set(channel_id, {topic: topic, swarm: swarm, conns: conns});
// Join the new topic // Join the new topic
swarm.join(topic, { swarm.join(topic, {
lookup: true, lookup: true,
announce: true, announce: true,
timeout: 60000 timeout: 30000
});
swarm.on('connection', (conn, info) => {
console.log(`Connected to a peer`);
conns.push(conn);
conn.on('data', data => {
// Send the message from the topic to the Discord channel
client.channels.fetch(channel_id).then(channel => channel.send(data.toString()))
});
conn.on('close', () => clearInterval(checkInactiveInterval))
});
client.channels.fetch(channel_id).then(channel => {
msg.reply(`Successfully joined topic and sending messages to ${channel}`);
});
}
}
if (cmd === "set") {
let channel_id = msg.channel.id;
//Check if the given channel_id exists or not.
if (!client.channels.cache.get(channel_id)) {
msg.reply("The provided channel id does not exist.");
return;
}
let topic = crypto.randomBytes(32);
let swarm;
let conns = [];
if (topicMap.has(channel_id)) {
topic_obj = topicMap.get(channel_id);
conns = topic_obj.conns;
}
else {
topicMap.set(channel_id, { topic: topic, conns: conns });
}
if (!swarmMap.has(topic)) {
swarm = new Hyperswarm();
swarmMap.set(topic, swarm);
// Join the new topic
swarm.join(topic, {
lookup: true,
announce: true,
timeout: 60000
}); });
swarm.on('connection', (conn, info) => { swarm.on('connection', (conn, info) => {
console.log(`Connected to a peer for topic ${topic.toString('hex')}`); console.log(`Connected to a peer for topic ${topic.toString('hex')}`);
conns.push(conn); conns.push(conn);
conn.on('data', data => { conn.on('data', data => {
console.log(data.toString()) // Send the message from the topic to the Discord channel
if (data.toString().startsWith('CLOSED:')) { client.channels.fetch(channel_id).then(channel=>channel.send(data.toString()))
// Extract the key from the message string
const key = data.toString().split(':')[1].trim();
// Iterate through the conns array and remove the client with the matching key
console.log(`Removing peer ${key}`);
(async () => {
await sleep(5000)
conns = conns.filter(c => c !== conn);
conn.destroy();
})();
} else {
client.channels.fetch(channel_id).then(channel => channel.send(data.toString()))
}
}); });
}); });
client.channels.fetch(channel_id).then(channel => { client.channels.fetch(channel_id).then(channel => {
msg.reply(`Successfully joined topic and sending messages to ${channel}`); msg.reply(`Successfully joined topic and sending messages to ${channel}`);
}); });
} else {
swarm = swarmMap.get(topic);
swarm.join(topic, {
lookup: true,
announce: true,
timeout: 60000
});
client.channels.fetch(channel_id).then(channel => {
msg.reply(`Successfully joined topic and sending messages to ${channel}`);
});
} }
} }
if (cmd === "set-topic") {
let channel_id = msg.channel.id;
//Check if the given channel_id exists or not.
if(!client.channels.cache.get(channel_id))
{
msg.reply("The provided channel id does not exist.");
return;
}
let topic = crypto.randomBytes(32);
let swarm;
let conns =[];
if(topicMap.has(channel_id))
{
topic_obj = topicMap.get(channel_id);
topic_obj.swarm.leave(topic_obj.topic);
topic_obj.topic = topic;
swarm = topic_obj.swarm;
conns = topic_obj.conns;
}
else{
swarm = new Hyperswarm();
conns = [];
topicMap.set(channel_id, {topic: topic, swarm: swarm, conns: conns});
// Join the new topic
swarm.join(topic, {
lookup: true,
announce: true,
timeout: 30000
});
swarm.on('connection', (conn, info) => {
console.log(`Connected to a peer for topic ${topic.toString('hex')}`);
conns.push(conn);
conn.on('data', data => {
// Send the message from the topic to the Discord channel
client.channels.fetch(channel_id).then(channel=>channel.send(data.toString()))
});
});
}
msg.reply(`Successfully set topic ${topic.toString('hex')} for channel ${channel_id}`);
}
if (cmd === "leave-topic") { if (cmd === "leave-topic") {
let channel_id = msg.channel.id; let channel_id = msg.channel.id;
let topic_obj = topicMap.get(channel_id); let topic_obj = topicMap.get(channel_id);
if (topic_obj) { if (topic_obj) {
topic_obj.swarm.leave(topic_obj.topic); topic_obj.swarm.leave(topic_obj.topic);
topicMap.delete(channel_id); topicMap.delete(channel_id);
return msg.reply(`Successfully left topic ${topic_obj.topic.toString('hex')} for channel ${channel_id}`); return msg.reply(`Successfully left topic ${topic_obj.topic.toString('hex')} for channel ${channel_id}`);
} else { } else {
return msg.reply(`No topic is assigned for this channel`); return msg.reply(`No topic is assigned for this channel`);
}
}
}
// Check if message is sent in a channel with a topic assigned and send it to the topic
if(topicMap.has(msg.channel.id) && !msg.author.bot){
let topic_obj = topicMap.get(msg.channel.id);
let conns = topic_obj.conns;
for (const conn of conns) {
conn.write(`${msg.author.username}(Discord): ${msg.content}`);
} }
} }
});
} client.login(process.env.TOKEN);
// Check if message is sent in a channel with a topic assigned and send it to the topic
if (topicMap.has(msg.channel.id) && !msg.author.bot) {
let topic_obj = topicMap.get(msg.channel.id);
let conns = topic_obj.conns;
for (const conn of conns) {
conn.write(`${msg.author.username}(Discord): ${msg.content}`);
}
}
});
client.login(process.env.TOKEN);