3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-15 05:41:51 +00:00

Allow making archives permanent. Make archives permanent when attached to a case. Add log file metadata dynamically when served (instead of in the body directly). Add missing index on archives.expires_at.

This commit is contained in:
Dragory 2019-01-13 17:29:26 +02:00
parent 10d757f588
commit ab71481b8f
5 changed files with 59 additions and 13 deletions

View file

@ -14,10 +14,6 @@ const MESSAGE_ARCHIVE_HEADER_FORMAT = trimLines(`
`); `);
const MESSAGE_ARCHIVE_MESSAGE_FORMAT = const MESSAGE_ARCHIVE_MESSAGE_FORMAT =
"[#{channel.name}] [{user.id}] [{timestamp}] {user.username}#{user.discriminator}: {content}{attachments}"; "[#{channel.name}] [{user.id}] [{timestamp}] {user.username}#{user.discriminator}: {content}{attachments}";
const MESSAGE_ARCHIVE_FOOTER_FORMAT = trimLines(`
Log file generated on {timestamp}
Expires at {expires}
`);
export class GuildArchives extends BaseRepository { export class GuildArchives extends BaseRepository {
protected archives: Repository<ArchiveEntry>; protected archives: Repository<ArchiveEntry>;
@ -35,6 +31,7 @@ export class GuildArchives extends BaseRepository {
this.archives this.archives
.createQueryBuilder() .createQueryBuilder()
.where("guild_id = :guild_id", { guild_id: this.guildId }) .where("guild_id = :guild_id", { guild_id: this.guildId })
.andWhere("expires_at IS NOT NULL")
.andWhere("expires_at <= NOW()") .andWhere("expires_at <= NOW()")
.delete() .delete()
.execute(); .execute();
@ -47,6 +44,15 @@ export class GuildArchives extends BaseRepository {
}); });
} }
async makePermanent(id: string): Promise<void> {
await this.archives.update(
{ id },
{
expires_at: null
}
);
}
/** /**
* @returns ID of the created entry * @returns ID of the created entry
*/ */
@ -81,11 +87,7 @@ export class GuildArchives extends BaseRepository {
}); });
}); });
const messagesStr = msgLines.join("\n"); const messagesStr = msgLines.join("\n");
const footerStr = formatTemplateString(MESSAGE_ARCHIVE_FOOTER_FORMAT, {
timestamp: moment().format("YYYY-MM-DD [at] HH:mm:ss (Z)"),
expires: expiresAt.format("YYYY-MM-DD [at] HH:mm:ss (Z)")
});
return this.create([headerStr, messagesStr, footerStr].join("\n\n"), expiresAt); return this.create([headerStr, messagesStr].join("\n\n"), expiresAt);
} }
} }

View file

@ -1,4 +1,4 @@
import { Entity, Column, PrimaryColumn, PrimaryGeneratedColumn } from "typeorm"; import { Entity, Column, PrimaryGeneratedColumn } from "typeorm";
@Entity("archives") @Entity("archives")
export class ArchiveEntry { export class ArchiveEntry {

View file

@ -0,0 +1,21 @@
import { MigrationInterface, QueryRunner, TableColumn, TableIndex } from "typeorm";
export class AddIndexToArchivesExpiresAt1547392046629 implements MigrationInterface {
public async up(queryRunner: QueryRunner): Promise<any> {
await queryRunner.createIndex(
"archives",
new TableIndex({
columnNames: ["expires_at"]
})
);
}
public async down(queryRunner: QueryRunner): Promise<any> {
await queryRunner.dropIndex(
"archives",
new TableIndex({
columnNames: ["expires_at"]
})
);
}
}

View file

@ -6,12 +6,14 @@ import moment from "moment-timezone";
import { CaseTypeColors } from "../data/CaseTypeColors"; import { CaseTypeColors } from "../data/CaseTypeColors";
import { ZeppelinPlugin } from "./ZeppelinPlugin"; import { ZeppelinPlugin } from "./ZeppelinPlugin";
import { GuildActions } from "../data/GuildActions"; import { GuildActions } from "../data/GuildActions";
import { GuildArchives } from "../data/GuildArchives";
export class CasesPlugin extends ZeppelinPlugin { export class CasesPlugin extends ZeppelinPlugin {
public static pluginName = 'cases'; public static pluginName = "cases";
protected actions: GuildActions; protected actions: GuildActions;
protected cases: GuildCases; protected cases: GuildCases;
protected archives: GuildArchives;
getDefaultOptions() { getDefaultOptions() {
return { return {
@ -25,6 +27,7 @@ export class CasesPlugin extends ZeppelinPlugin {
onLoad() { onLoad() {
this.actions = GuildActions.getInstance(this.guildId); this.actions = GuildActions.getInstance(this.guildId);
this.cases = GuildCases.getInstance(this.guildId); this.cases = GuildCases.getInstance(this.guildId);
this.archives = GuildArchives.getInstance(this.guildId);
this.actions.register("createCase", args => { this.actions.register("createCase", args => {
return this.createCase( return this.createCase(
@ -135,6 +138,12 @@ export class CasesPlugin extends ZeppelinPlugin {
}); });
} }
const archiveLinkMatch = body && body.match(/\/archives\/([a-zA-Z0-9\-]+)/);
if (archiveLinkMatch) {
const archiveId = archiveLinkMatch[1];
this.archives.makePermanent(archiveId);
}
if ((!automatic || this.configValue("log_automatic_actions")) && postInCaseLogOverride !== false) { if ((!automatic || this.configValue("log_automatic_actions")) && postInCaseLogOverride !== false) {
try { try {
await this.postCaseToCaseLogChannel(theCase.id); await this.postCaseToCaseLogChannel(theCase.id);

View file

@ -2,6 +2,7 @@ import http, { ServerResponse } from "http";
import { GlobalPlugin } from "knub"; import { GlobalPlugin } from "knub";
import { GuildArchives } from "../data/GuildArchives"; import { GuildArchives } from "../data/GuildArchives";
import { sleep } from "../utils"; import { sleep } from "../utils";
import moment from "moment-timezone";
const DEFAULT_PORT = 9920; const DEFAULT_PORT = 9920;
const archivesRegex = /^\/(spam-logs|archives)\/([a-z0-9\-]+)\/?$/i; const archivesRegex = /^\/(spam-logs|archives)\/([a-z0-9\-]+)\/?$/i;
@ -12,7 +13,7 @@ function notFound(res: ServerResponse) {
} }
export class LogServerPlugin extends GlobalPlugin { export class LogServerPlugin extends GlobalPlugin {
public static pluginName = 'log_server'; public static pluginName = "log_server";
protected archives: GuildArchives; protected archives: GuildArchives;
protected server: http.Server; protected server: http.Server;
@ -36,8 +37,21 @@ export class LogServerPlugin extends GlobalPlugin {
const log = await this.archives.find(logId); const log = await this.archives.find(logId);
if (!log) return notFound(res); if (!log) return notFound(res);
let body = log.body;
// Add some metadata at the end of the log file (but only if it doesn't already have it directly in the body)
if (log.body.indexOf("Log file generated on") === -1) {
const createdAt = moment(log.created_at).format("YYYY-MM-DD [at] HH:mm:ss [(+00:00)]");
body += `\n\nLog file generated on ${createdAt}`;
if (log.expires_at !== null) {
const expiresAt = moment(log.expires_at).format("YYYY-MM-DD [at] HH:mm:ss [(+00:00)]");
body += `\nExpires at ${expiresAt}`;
}
}
res.setHeader("Content-Type", "text/plain; charset=UTF-8"); res.setHeader("Content-Type", "text/plain; charset=UTF-8");
res.end(log.body); res.end(body);
} }
}); });