Complete refactor (#1)
This commit is contained in:
jacob janzen
GitHub
parent
da5c9f9f0b
commit
6007a8bbfa
3 changed files with 213 additions and 123 deletions
207
app.js
207
app.js
|
|
@ -1,130 +1,91 @@
|
||||||
import 'dotenv/config';
|
import "dotenv/config";
|
||||||
import express from 'express';
|
import express from "express";
|
||||||
import cron from 'node-cron';
|
|
||||||
import {
|
import {
|
||||||
InteractionType,
|
InteractionType,
|
||||||
InteractionResponseType,
|
InteractionResponseType,
|
||||||
verifyKeyMiddleware,
|
verifyKeyMiddleware,
|
||||||
} from 'discord-interactions';
|
} from "discord-interactions";
|
||||||
import { Client, GatewayIntentBits, ActivityType } from 'discord.js';
|
import { Client, GatewayIntentBits, ActivityType } from "discord.js";
|
||||||
import fs from 'node:fs';
|
import { pet, schedule_message, unschedule_message } from "./command_impls.js";
|
||||||
import { v4 as uuidv4 } from 'uuid';
|
import { MessageSchedule } from "./message-scheduler.js";
|
||||||
|
|
||||||
const client = new Client ({ intents: [GatewayIntentBits.Guilds] });
|
class State {
|
||||||
var jobs = {};
|
constructor() {
|
||||||
const job_crons = {};
|
this.job_db = "jobs.json";
|
||||||
client.on('ready', () => {
|
this.client = new Client({ intents: [GatewayIntentBits.Guilds] });
|
||||||
console.log(`Logged in as ${client.user.tag}!`);
|
}
|
||||||
fs.readFile('jobs.json', 'utf8', (err,data) => {
|
}
|
||||||
jobs = err ? {} : JSON.parse(data);
|
|
||||||
|
|
||||||
Object.keys(jobs).forEach((job) => {
|
function handle_application_command(state, data, channel_id) {
|
||||||
job_crons[job] = cron.schedule(jobs[job].crontab, () => {
|
const { name, options } = data;
|
||||||
client.channels.cache.get(jobs[job].channel_id).send(jobs[job].message);
|
|
||||||
}, {
|
switch (name) {
|
||||||
scheduled: true,
|
case "schedule_message":
|
||||||
timezone: process.env.TIMEZONE
|
return schedule_message(
|
||||||
});
|
state,
|
||||||
|
options[0].value,
|
||||||
|
channel_id,
|
||||||
|
options[1].value,
|
||||||
|
);
|
||||||
|
|
||||||
|
case "unschedule_message":
|
||||||
|
return unschedule_message(state, options[0].value);
|
||||||
|
|
||||||
|
case "pet":
|
||||||
|
return pet(state);
|
||||||
|
|
||||||
|
default:
|
||||||
|
console.error(`unknown command: ${name}`);
|
||||||
|
return res.status(400).json({ error: "unknown command" });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function main() {
|
||||||
|
const state = new State();
|
||||||
|
|
||||||
|
state.client.on("ready", () => {
|
||||||
|
console.log(`Logged in as ${state.client.user.tag}!`);
|
||||||
|
state.schedule = new MessageSchedule(state);
|
||||||
|
state.client.user.setPresence({
|
||||||
|
activities: [
|
||||||
|
{
|
||||||
|
name: "sily",
|
||||||
|
type: ActivityType.Playing,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
status: "idle",
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
client.user.setPresence({
|
|
||||||
activities: [{
|
state.client.login(process.env.DISCORD_TOKEN);
|
||||||
name: 'sily',
|
|
||||||
type: ActivityType.Playing,
|
const app = express();
|
||||||
}],
|
const PORT = process.env.PORT || 3000;
|
||||||
status: "idle"
|
|
||||||
|
app.post(
|
||||||
|
"/",
|
||||||
|
verifyKeyMiddleware(process.env.PUBLIC_KEY),
|
||||||
|
async (req, res) => {
|
||||||
|
const { type, data, channel_id } = req.body;
|
||||||
|
|
||||||
|
state.res = res;
|
||||||
|
switch (type) {
|
||||||
|
case InteractionType.PING:
|
||||||
|
return res.send({ type: InteractionResponseType.PONG });
|
||||||
|
case InteractionType.APPLICATION_COMMAND:
|
||||||
|
return handle_application_command(state, data, channel_id);
|
||||||
|
default:
|
||||||
|
console.error("unknown interaction type", type);
|
||||||
|
return res
|
||||||
|
.status(400)
|
||||||
|
.json({ error: "unknown interaction type" });
|
||||||
|
}
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
app.listen(PORT, () => {
|
||||||
|
console.log("Listening on port", PORT);
|
||||||
});
|
});
|
||||||
});
|
}
|
||||||
|
|
||||||
client.login(process.env.DISCORD_TOKEN);
|
main();
|
||||||
|
|
||||||
const app = express();
|
|
||||||
const PORT = process.env.PORT || 3000;
|
|
||||||
|
|
||||||
app.post('/', verifyKeyMiddleware(process.env.PUBLIC_KEY), async function (req, res) {
|
|
||||||
const { type, data } = req.body;
|
|
||||||
|
|
||||||
if (type === InteractionType.PING) {
|
|
||||||
return res.send({ type: InteractionResponseType.PONG });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (type === InteractionType.APPLICATION_COMMAND) {
|
|
||||||
const { name, options, guild_id } = data;
|
|
||||||
|
|
||||||
if (name === 'schedule_message') {
|
|
||||||
const message = options[0].value;
|
|
||||||
const crontab = options[1].value;
|
|
||||||
const valid = cron.validate(crontab);
|
|
||||||
const id = uuidv4();
|
|
||||||
|
|
||||||
const content = valid ? `registered message: "${message}" with cron: "${crontab}" and id: "${id}"` : 'invalid cron';
|
|
||||||
|
|
||||||
if (valid) {
|
|
||||||
jobs[id] = {
|
|
||||||
message: message,
|
|
||||||
crontab: crontab,
|
|
||||||
channel_id: req.body.channel_id,
|
|
||||||
};
|
|
||||||
job_crons[id] = cron.schedule(crontab, () => {
|
|
||||||
client.channels.cache.get(req.body.channel_id).send(message);
|
|
||||||
}, {
|
|
||||||
scheduled: true,
|
|
||||||
timezone: process.env.TIMEZONE
|
|
||||||
});
|
|
||||||
|
|
||||||
const json = JSON.stringify(jobs);
|
|
||||||
fs.writeFile('jobs.json', json, 'utf8', err => {
|
|
||||||
if (err) {
|
|
||||||
console.error(err);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return res.send({
|
|
||||||
type: InteractionResponseType.CHANNEL_MESSAGE_WITH_SOURCE,
|
|
||||||
data: {
|
|
||||||
content: content,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
} else if (name === 'unschedule_message') {
|
|
||||||
const id = options[0].value;
|
|
||||||
const content = id in jobs ? `stopped job ${id}` : `no such job ${id}`;
|
|
||||||
if (id in jobs) {
|
|
||||||
job_crons[id].stop();
|
|
||||||
delete jobs[id]
|
|
||||||
delete job_crons[id];
|
|
||||||
|
|
||||||
const json = JSON.stringify(jobs);
|
|
||||||
fs.writeFile('jobs.json', json, 'utf8', err => {
|
|
||||||
if (err) {
|
|
||||||
console.error(err);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return res.send({
|
|
||||||
type: InteractionResponseType.CHANNEL_MESSAGE_WITH_SOURCE,
|
|
||||||
data: {
|
|
||||||
content: content,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
} else if (name === 'pet') {
|
|
||||||
return res.send({
|
|
||||||
type: InteractionResponseType.CHANNEL_MESSAGE_WITH_SOURCE,
|
|
||||||
data: {
|
|
||||||
content: '[purring noises]',
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
console.error(`unknown command: ${name}`);
|
|
||||||
return res.status(400).json({ error: 'unknown command' });
|
|
||||||
}
|
|
||||||
|
|
||||||
console.error('unknown interaction type', type);
|
|
||||||
return res.status(400).json({ error: 'unknown interaction type' });
|
|
||||||
});
|
|
||||||
|
|
||||||
app.listen(PORT, () => {
|
|
||||||
console.log('Listening on port', PORT);
|
|
||||||
});
|
|
||||||
|
|
|
||||||
37
command_impls.js
Normal file
37
command_impls.js
Normal file
|
|
@ -0,0 +1,37 @@
|
||||||
|
import { InteractionResponseType } from "discord-interactions";
|
||||||
|
import { Message } from "./message-scheduler.js";
|
||||||
|
|
||||||
|
function send(state, content) {
|
||||||
|
return state.res.send({
|
||||||
|
type: InteractionResponseType.CHANNEL_MESSAGE_WITH_SOURCE,
|
||||||
|
data: {
|
||||||
|
content: content,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
export function schedule_message(state, msg, channel, crontab) {
|
||||||
|
const message = new Message(msg, channel, crontab);
|
||||||
|
|
||||||
|
const schedule_valid = state.schedule.schedule(message);
|
||||||
|
|
||||||
|
return send(
|
||||||
|
state,
|
||||||
|
schedule_valid
|
||||||
|
? `registered message: "${msg}" with cron: "${crontab}" and id: "${message.id}"`
|
||||||
|
: "invalid cron",
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function unschedule_message(state, id) {
|
||||||
|
const unschedule_valid = state.schedule.unschedule(id);
|
||||||
|
|
||||||
|
return send(
|
||||||
|
state,
|
||||||
|
unschedule_valid ? `stopped job ${id}` : `no such job ${id}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function pet(state) {
|
||||||
|
return send(state, "[purring noises]");
|
||||||
|
}
|
||||||
92
message-scheduler.js
Normal file
92
message-scheduler.js
Normal file
|
|
@ -0,0 +1,92 @@
|
||||||
|
import "dotenv/config";
|
||||||
|
import cron from "node-cron";
|
||||||
|
import { v4 as uuidv4 } from "uuid";
|
||||||
|
import fs from "node:fs";
|
||||||
|
|
||||||
|
export class Message {
|
||||||
|
constructor(message, channel_id, crontab, repeat = true, id = null) {
|
||||||
|
this.id = id ? id : uuidv4();
|
||||||
|
this.repeat = repeat;
|
||||||
|
|
||||||
|
this.channel_id = channel_id;
|
||||||
|
this.message = message;
|
||||||
|
this.crontab = crontab;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export class MessageSchedule {
|
||||||
|
constructor(state) {
|
||||||
|
this.state = state;
|
||||||
|
this.crons = {};
|
||||||
|
|
||||||
|
// read db file if there is one and start up all jobs
|
||||||
|
fs.readFile(state.job_db, "utf8", (err, data) => {
|
||||||
|
this.jobs = err ? {} : JSON.parse(data);
|
||||||
|
|
||||||
|
Object.keys(this.jobs).forEach((job) => {
|
||||||
|
// fix incompatibility with older storage version
|
||||||
|
const repeat = this.jobs[job].repeat
|
||||||
|
? this.jobs[job].repeat
|
||||||
|
: true;
|
||||||
|
this.jobs[job].id = job;
|
||||||
|
this.jobs[job].repeat = repeat;
|
||||||
|
|
||||||
|
this.schedule(this.jobs[job]);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
save() {
|
||||||
|
const json = JSON.stringify(this.jobs);
|
||||||
|
fs.writeFile(this.state.job_db, json, "utf8", (err) => {
|
||||||
|
if (err) console.log(err);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
schedule(message) {
|
||||||
|
const valid = cron.validate(message.crontab);
|
||||||
|
|
||||||
|
if (valid) {
|
||||||
|
this.jobs[message.id] = message;
|
||||||
|
|
||||||
|
this.crons[message.id] = cron.schedule(
|
||||||
|
message.crontab,
|
||||||
|
() => {
|
||||||
|
this.state.client.channels.cache
|
||||||
|
.get(message.channel_id)
|
||||||
|
.send(message.message);
|
||||||
|
if (!message.repeat) {
|
||||||
|
this.unschedule(message.id);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
scheduled: true,
|
||||||
|
timezone: process.env.TIMEZONE,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
this.save();
|
||||||
|
}
|
||||||
|
|
||||||
|
return valid;
|
||||||
|
}
|
||||||
|
|
||||||
|
unschedule(id) {
|
||||||
|
var valid = true;
|
||||||
|
if (id in this.jobs) {
|
||||||
|
delete this.jobs[id];
|
||||||
|
} else {
|
||||||
|
valid = false;
|
||||||
|
}
|
||||||
|
if (id in this.crons) {
|
||||||
|
this.crons[id].stop();
|
||||||
|
delete this.crons[id];
|
||||||
|
} else {
|
||||||
|
valid = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.save();
|
||||||
|
|
||||||
|
return valid;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Add table
Reference in a new issue