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:
parent
10d757f588
commit
ab71481b8f
5 changed files with 59 additions and 13 deletions
|
@ -14,10 +14,6 @@ const MESSAGE_ARCHIVE_HEADER_FORMAT = trimLines(`
|
|||
`);
|
||||
const MESSAGE_ARCHIVE_MESSAGE_FORMAT =
|
||||
"[#{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 {
|
||||
protected archives: Repository<ArchiveEntry>;
|
||||
|
@ -35,6 +31,7 @@ export class GuildArchives extends BaseRepository {
|
|||
this.archives
|
||||
.createQueryBuilder()
|
||||
.where("guild_id = :guild_id", { guild_id: this.guildId })
|
||||
.andWhere("expires_at IS NOT NULL")
|
||||
.andWhere("expires_at <= NOW()")
|
||||
.delete()
|
||||
.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
|
||||
*/
|
||||
|
@ -81,11 +87,7 @@ export class GuildArchives extends BaseRepository {
|
|||
});
|
||||
});
|
||||
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);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
import { Entity, Column, PrimaryColumn, PrimaryGeneratedColumn } from "typeorm";
|
||||
import { Entity, Column, PrimaryGeneratedColumn } from "typeorm";
|
||||
|
||||
@Entity("archives")
|
||||
export class ArchiveEntry {
|
||||
|
|
21
src/migrations/1547392046629-AddIndexToArchivesExpiresAt.ts
Normal file
21
src/migrations/1547392046629-AddIndexToArchivesExpiresAt.ts
Normal 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"]
|
||||
})
|
||||
);
|
||||
}
|
||||
}
|
|
@ -6,12 +6,14 @@ import moment from "moment-timezone";
|
|||
import { CaseTypeColors } from "../data/CaseTypeColors";
|
||||
import { ZeppelinPlugin } from "./ZeppelinPlugin";
|
||||
import { GuildActions } from "../data/GuildActions";
|
||||
import { GuildArchives } from "../data/GuildArchives";
|
||||
|
||||
export class CasesPlugin extends ZeppelinPlugin {
|
||||
public static pluginName = 'cases';
|
||||
public static pluginName = "cases";
|
||||
|
||||
protected actions: GuildActions;
|
||||
protected cases: GuildCases;
|
||||
protected archives: GuildArchives;
|
||||
|
||||
getDefaultOptions() {
|
||||
return {
|
||||
|
@ -25,6 +27,7 @@ export class CasesPlugin extends ZeppelinPlugin {
|
|||
onLoad() {
|
||||
this.actions = GuildActions.getInstance(this.guildId);
|
||||
this.cases = GuildCases.getInstance(this.guildId);
|
||||
this.archives = GuildArchives.getInstance(this.guildId);
|
||||
|
||||
this.actions.register("createCase", args => {
|
||||
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) {
|
||||
try {
|
||||
await this.postCaseToCaseLogChannel(theCase.id);
|
||||
|
|
|
@ -2,6 +2,7 @@ import http, { ServerResponse } from "http";
|
|||
import { GlobalPlugin } from "knub";
|
||||
import { GuildArchives } from "../data/GuildArchives";
|
||||
import { sleep } from "../utils";
|
||||
import moment from "moment-timezone";
|
||||
|
||||
const DEFAULT_PORT = 9920;
|
||||
const archivesRegex = /^\/(spam-logs|archives)\/([a-z0-9\-]+)\/?$/i;
|
||||
|
@ -12,7 +13,7 @@ function notFound(res: ServerResponse) {
|
|||
}
|
||||
|
||||
export class LogServerPlugin extends GlobalPlugin {
|
||||
public static pluginName = 'log_server';
|
||||
public static pluginName = "log_server";
|
||||
|
||||
protected archives: GuildArchives;
|
||||
protected server: http.Server;
|
||||
|
@ -36,8 +37,21 @@ export class LogServerPlugin extends GlobalPlugin {
|
|||
const log = await this.archives.find(logId);
|
||||
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.end(log.body);
|
||||
res.end(body);
|
||||
}
|
||||
});
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue