mirror of
https://github.com/ZeppelinBot/Zeppelin.git
synced 2025-03-18 06:51:51 +00:00
73 lines
2 KiB
TypeScript
73 lines
2 KiB
TypeScript
![]() |
import uuid from "uuid/v4"; // tslint:disable-line
|
||
|
import moment from "moment-timezone";
|
||
|
import knex from "../knex";
|
||
|
import SpamLog from "../models/SpamLog";
|
||
|
import { Message } from "eris";
|
||
|
import { formatTemplateString, stripObjectToScalars, trimLines } from "../utils";
|
||
|
|
||
|
const EXPIRY_DAYS = 7;
|
||
|
const MESSAGE_FORMAT =
|
||
|
"[{timestamp}] [{message.id}] {user.username}#{user.discriminator}: {message.content}{attachments}";
|
||
|
|
||
|
function cleanExpiredLogs() {
|
||
|
knex("spam_logs")
|
||
|
.where("expires_at", "<=", knex.raw("NOW()"))
|
||
|
.delete();
|
||
|
}
|
||
|
|
||
|
cleanExpiredLogs();
|
||
|
setInterval(cleanExpiredLogs, 1000 * 60 * 60); // Clean expired logs every hour
|
||
|
|
||
|
export class GuildSpamLogs {
|
||
|
protected guildId: string;
|
||
|
|
||
|
constructor(guildId) {
|
||
|
this.guildId = guildId;
|
||
|
}
|
||
|
|
||
|
async find(id: string): Promise<SpamLog> {
|
||
|
const result = await knex("spam_logs")
|
||
|
.where("id", id)
|
||
|
.first();
|
||
|
|
||
|
return result ? new SpamLog(result) : null;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @return ID of the created spam log entry
|
||
|
*/
|
||
|
async createFromMessages(messages: Message[], header: string = null) {
|
||
|
const lines = messages.map(msg => {
|
||
|
return formatTemplateString(MESSAGE_FORMAT, {
|
||
|
user: stripObjectToScalars(msg.author),
|
||
|
message: stripObjectToScalars(msg),
|
||
|
timestamp: moment(msg.timestamp).format("YYYY-MM-DD HH:mm:ss zz"),
|
||
|
attachments: msg.attachments.length
|
||
|
? ` (message contained ${msg.attachments.length} attachment(s))`
|
||
|
: ""
|
||
|
});
|
||
|
});
|
||
|
|
||
|
const id = uuid();
|
||
|
const now = moment().format("YYYY-MM-DD HH:mm:ss zz");
|
||
|
const expiresAt = moment().add(EXPIRY_DAYS, "days");
|
||
|
|
||
|
const body = trimLines(`
|
||
|
Log file generated on ${now}. Expires at ${expiresAt.format("YYYY-MM-DD HH:mm:ss zz")}.${
|
||
|
header ? "\n" + header : ""
|
||
|
}
|
||
|
|
||
|
${lines.join("\n")}
|
||
|
`);
|
||
|
|
||
|
await knex("spam_logs").insert({
|
||
|
id,
|
||
|
guild_id: this.guildId,
|
||
|
body,
|
||
|
expires_at: expiresAt.format("YYYY-MM-DD HH:mm:ss")
|
||
|
});
|
||
|
|
||
|
return id;
|
||
|
}
|
||
|
}
|