45 lines
1.1 KiB
TypeScript
45 lines
1.1 KiB
TypeScript
|
import http, { ServerResponse } from "http";
|
||
|
|
import { GlobalPlugin } from "knub";
|
||
|
|
import { GuildSpamLogs } from "../data/GuildSpamLogs";
|
||
|
|
|
||
|
|
const DEFAULT_PORT = 9920;
|
||
|
|
const logUrlRegex = /^\/spam-logs\/([a-z0-9\-]+)\/?$/i;
|
||
|
|
|
||
|
|
function notFound(res: ServerResponse) {
|
||
|
|
res.statusCode = 404;
|
||
|
|
res.end("Not Found");
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* A global plugin that allows bot owners to control the bot
|
||
|
|
*/
|
||
|
|
export class LogServerPlugin extends GlobalPlugin {
|
||
|
|
protected spamLogs: GuildSpamLogs;
|
||
|
|
protected server: http.Server;
|
||
|
|
|
||
|
|
onLoad() {
|
||
|
|
this.spamLogs = new GuildSpamLogs(null);
|
||
|
|
|
||
|
|
this.server = http.createServer(async (req, res) => {
|
||
|
|
const logId = req.url.match(logUrlRegex);
|
||
|
|
if (!logId) return notFound(res);
|
||
|
|
|
||
|
|
if (logId) {
|
||
|
|
const log = await this.spamLogs.find(logId[1]);
|
||
|
|
if (!log) return notFound(res);
|
||
|
|
|
||
|
|
res.setHeader("Content-Type", "text/plain; charset=UTF-8");
|
||
|
|
res.end(log.body);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
this.server.listen(this.configValue("port", DEFAULT_PORT));
|
||
|
|
}
|
||
|
|
|
||
|
|
async onUnload() {
|
||
|
|
return new Promise(resolve => {
|
||
|
|
this.server.close(() => resolve());
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|