MoSiren/IEM-Alert
1
0
Fork 0
forked from ChrisChrome/weather-bot
Code Pull requests Activity

Add colors package to make logs nice to read

Browse source
This commit is contained in:
Christopher Cookman 2024-05-08 08:41:29 -06:00
parent f0b45c190b
commit 6008c0f9aa
Signed by: ChrisChrome
GPG key ID: A023A26E42C33A42
3 changed files with 33 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
index.js
View file

@ -4,6 +4,7 @@ const { client, xml } = require("@xmpp/client");
const fetch = require("node-fetch"); const fetch = require("node-fetch");
const html = require("html-entities") const html = require("html-entities")
const Discord = require("discord.js"); const Discord = require("discord.js");
const colors = require("colors");
const sqlite3 = require("sqlite3").verbose(); const sqlite3 = require("sqlite3").verbose();
// Setup Discord // Setup Discord
const discord = new Discord.Client({ const discord = new Discord.Client({
@ -22,9 +23,9 @@ const rest = new REST({
// Setup SQlite DB // Setup SQlite DB
const db = new sqlite3.Database("channels.db", (err) => { const db = new sqlite3.Database("channels.db", (err) => {
if (err) { if (err) {
console.error(err.message); console.log(`${colors.red("[ERROR]")} Error connecting to database: ${err.message}`);
} }
console.log("Connected to the channels database."); console.log(`${colors.cyan("[INFO]")} Connected to the database`);
// Create tables if they dont exist // Create tables if they dont exist
db.run(`CREATE TABLE IF NOT EXISTS channels (channelid TEXT, iemchannel TEXT, custommessage TEXT)`); db.run(`CREATE TABLE IF NOT EXISTS channels (channelid TEXT, iemchannel TEXT, custommessage TEXT)`);
}); });
@ -85,7 +86,7 @@ const xmpp = client({
//debug(xmpp, true); //debug(xmpp, true);
xmpp.on("error", (err) => { xmpp.on("error", (err) => {
console.log(`ERR: ${err}`); console.log(`${colors.red("[ERROR]")} XMPP Error: ${err}. Trying to reconnect...`);
setTimeout(() => { setTimeout(() => {
xmpp.stop().then(() => { xmpp.stop().then(() => {
start(); start();
@ -94,7 +95,7 @@ xmpp.on("error", (err) => {
}); });
xmpp.on("offline", () => { xmpp.on("offline", () => {
console.log("offline"); console.log(`${colors.yellow("[WARN]")} XMPP offline, trying to reconnect...`);
setTimeout(() => { setTimeout(() => {
xmpp.stop().then(() => { xmpp.stop().then(() => {
start(); start();
@ -104,7 +105,7 @@ xmpp.on("offline", () => {
xmpp.on("stanza", (stanza) => { xmpp.on("stanza", (stanza) => {
if (config.debug >= 2) console.log(stanza.toString()); if (config.debug >= 2) console.log(`${colors.magenta("[DEBUG]")} Stanza: ${stanza.toString()}`);
// Stops spam from getting old messages // Stops spam from getting old messages
if (startup) return; if (startup) return;
// Get new messages and log them, ignore old messages // Get new messages and log them, ignore old messages
@ -126,11 +127,11 @@ xmpp.on("stanza", (stanza) => {
const now = new Date(); const now = new Date();
const diff = (now - product_id.timestamp) / 1000 / 60; const diff = (now - product_id.timestamp) / 1000 / 60;
if (diff > 3) return; if (diff > 3) return;
if (config.debug >= 1) console.log(`New message from ${fromChannel}`); if (config.debug >= 1) console.log(`${colors.magenta("[DEBUG]")} New message from ${fromChannel}`);
messages++; messages++;
// Handle NTFY // Handle NTFY
if (config.ntfy.enabled) { if (config.ntfy.enabled) {
if (config.debug >= 1) console.log(`Sending NTFY for ${config.ntfy.prefix}${fromChannel}`) if (config.debug >= 1) console.log(`${colors.magenta("[DEBUG]")} Sending NTFY for ${config.ntfy.prefix}${fromChannel}`)
ntfyBody = { ntfyBody = {
"topic": `${config.ntfy.prefix}${fromChannel}`, "topic": `${config.ntfy.prefix}${fromChannel}`,
"message": bodyData.string, "message": bodyData.string,
@ -149,7 +150,7 @@ xmpp.on("stanza", (stanza) => {
'Authorization': `Bearer ${config.ntfy.token}` 'Authorization': `Bearer ${config.ntfy.token}`
} }
}).then((res) => { }).then((res) => {
if (config.debug >= 1) console.log(res.status) if (config.debug >= 1) console.log(`${colors.magenta("[DEBUG]")} NTFY sent for ${config.ntfy.prefix}${fromChannel} with status ${res.status}`);
}).catch((err) => { }).catch((err) => {
console.error(err) console.error(err)
@ -185,7 +186,7 @@ xmpp.on("stanza", (stanza) => {
if (err) { if (err) {
console.error(err.message); console.error(err.message);
} }
console.log(`Deleted channel ${row.channelid} from database`) console.log(`${colors.cyan("[INFO]")} Deleted channel ${row.channelid} from database`);
}); });
}; };
channel.send({ channel.send({
@ -216,7 +217,7 @@ xmpp.on("stanza", (stanza) => {
).then((msg) => { ).then((msg) => {
if (msg.channel.type === Discord.ChannelType.GuildAnnouncement) msg.crosspost(); if (msg.channel.type === Discord.ChannelType.GuildAnnouncement) msg.crosspost();
}).catch((err) => { }).catch((err) => {
console.log(`Failed to send message to ${row.channelid}, ${err}`); console.log(`${colors.red("[ERROR]")} Failed to send message to channel ${row.channelid}: ${err.message}`);
}); });
}); });
}); });
@ -234,21 +235,21 @@ xmpp.on("online", async (address) => {
// Join all channels // Join all channels
config.iem.channels.forEach((channel => { config.iem.channels.forEach((channel => {
console.log(`Joining ${channel.jid.split("@")[0]}:${channel.name}`) console.log(`${colors.cyan("[INFO]")} Joining ${channel.jid.split("@")[0]}:${channel.name}`)
xmpp.send(xml("presence", { to: `${channel.jid}/${channel.jid.split("@")[0]}` })); xmpp.send(xml("presence", { to: `${channel.jid}/${channel.jid.split("@")[0]}` }));
})) }))
console.log("online as", address.toString()); console.log(`${colors.cyan("[INFO]")} Connected to XMPP server as ${address.toString()}`);
setTimeout(() => { setTimeout(() => {
startup = false; startup = false;
console.log("Startup complete, forwarding messages now"); console.log(`${colors.cyan("[INFO]")} Startup complete, listening for messages...`);
}, 1000) }, 1000)
}); });
const start = () => { const start = () => {
xmpp.start().catch((err) => { xmpp.start().catch((err) => {
console.error(`start failed, ${err}\nGonna try again in 5 seconds...`); console.log(`${colors.red("[ERROR]")} XMPP failed to start: ${err}. Trying again in 5 seconds...`);
setTimeout(() => { setTimeout(() => {
start(); start();
}, 5000); }, 5000);
@ -260,7 +261,7 @@ const start = () => {
// START DISCORD // START DISCORD
discord.on('ready', async () => { discord.on('ready', async () => {
console.log(`Logged in as ${discord.user.tag}!`); console.log(`${colors.cyan("[INFO]")} Logged in as ${discord.user.tag}`);
// Do slash command stuff // Do slash command stuff
const commands = [ const commands = [
{ {
@ -514,11 +515,11 @@ discord.on("interactionCreate", async (interaction) => {
}); });
process.on("unhandledRejection", (error) => { process.on("unhandledRejection", (error) => {
console.error("Unhandled promise rejection:", error); console.log(`${colors.red("[ERROR]")} Unhandled Rejection: ${error.message}`);
}); });
process.on("uncaughtException", (error) => { process.on("uncaughtException", (error) => {
console.error("Uncaught exception:", error); console.log(`${colors.red("[ERROR]")} Uncaught Exception: ${error.message}`);
}); });
// Login to discord // Login to discord

14
package-lock.json generated
View file

@ -11,6 +11,7 @@
"dependencies": { "dependencies": {
"@xmpp/client": "^0.13.1", "@xmpp/client": "^0.13.1",
"@xmpp/debug": "^0.13.0", "@xmpp/debug": "^0.13.0",
"colors": "^1.4.0",
"discord.js": "^14.15.2", "discord.js": "^14.15.2",
"html-entities": "^2.5.2", "html-entities": "^2.5.2",
"sqlite3": "^5.1.7", "sqlite3": "^5.1.7",
@ -1523,6 +1524,14 @@
"color-support": "bin.js" "color-support": "bin.js"
} }
}, },
"node_modules/colors": {
"version": "1.4.0",
"resolved": "https://registry.npmjs.org/colors/-/colors-1.4.0.tgz",
"integrity": "sha512-a+UqTh4kgZg/SlGvfbzDHpgRu7AAQOmmqRHJnxhRZICKFUT91brVhNNt58CMWU9PsBbv3PDCZUHbVxuDiH2mtA==",
"engines": {
"node": ">=0.1.90"
}
},
"node_modules/commander": { "node_modules/commander": {
"version": "4.1.1", "version": "4.1.1",
"resolved": "https://registry.npmjs.org/commander/-/commander-4.1.1.tgz", "resolved": "https://registry.npmjs.org/commander/-/commander-4.1.1.tgz",
@ -5528,6 +5537,11 @@
"integrity": "sha512-qiBjkpbMLO/HL68y+lh4q0/O1MZFj2RX6X/KmMa3+gJD3z+WwI1ZzDHysvqHGS3mP6mznPckpXmw1nI9cJjyRg==", "integrity": "sha512-qiBjkpbMLO/HL68y+lh4q0/O1MZFj2RX6X/KmMa3+gJD3z+WwI1ZzDHysvqHGS3mP6mznPckpXmw1nI9cJjyRg==",
"optional": true "optional": true
}, },
"colors": {
"version": "1.4.0",
"resolved": "https://registry.npmjs.org/colors/-/colors-1.4.0.tgz",
"integrity": "sha512-a+UqTh4kgZg/SlGvfbzDHpgRu7AAQOmmqRHJnxhRZICKFUT91brVhNNt58CMWU9PsBbv3PDCZUHbVxuDiH2mtA=="
},
"commander": { "commander": {
"version": "4.1.1", "version": "4.1.1",
"resolved": "https://registry.npmjs.org/commander/-/commander-4.1.1.tgz", "resolved": "https://registry.npmjs.org/commander/-/commander-4.1.1.tgz",

1
package.json
View file

@ -11,6 +11,7 @@
"dependencies": { "dependencies": {
"@xmpp/client": "^0.13.1", "@xmpp/client": "^0.13.1",
"@xmpp/debug": "^0.13.0", "@xmpp/debug": "^0.13.0",
"colors": "^1.4.0",
"discord.js": "^14.15.2", "discord.js": "^14.15.2",
"html-entities": "^2.5.2", "html-entities": "^2.5.2",
"sqlite3": "^5.1.7", "sqlite3": "^5.1.7",