Reorganize project. Add folder for shared code between backend/dashboard. Switch from jest to ava for tests.

backend/src/data/GuildArchives.ts

@@ -0,0 +1,100 @@
+import uuid from "uuid/v4"; // tslint:disable-line
+import moment from "moment-timezone";
+import { ArchiveEntry } from "./entities/ArchiveEntry";
+import { getRepository, Repository } from "typeorm";
+import { BaseGuildRepository } from "./BaseGuildRepository";
+import { trimLines } from "../utils";
+import { SavedMessage } from "./entities/SavedMessage";
+import { Channel, Guild, User } from "eris";
+import { renderTemplate } from "../templateFormatter";
+
+const DEFAULT_EXPIRY_DAYS = 30;
+
+const MESSAGE_ARCHIVE_HEADER_FORMAT = trimLines(`
+  Server: {guild.name} ({guild.id})
+`);
+const MESSAGE_ARCHIVE_MESSAGE_FORMAT =
+  "[#{channel.name}] [{user.id}] [{timestamp}] {user.username}#{user.discriminator}: {content}{attachments}";
+
+export class GuildArchives extends BaseGuildRepository {
+  protected archives: Repository<ArchiveEntry>;
+
+  constructor(guildId) {
+    super(guildId);
+    this.archives = getRepository(ArchiveEntry);
+
+    // Clean expired archives at start and then every hour
+    this.deleteExpiredArchives();
+    setInterval(() => this.deleteExpiredArchives(), 1000 * 60 * 60);
+  }
+
+  private deleteExpiredArchives() {
+    this.archives
+      .createQueryBuilder()
+      .where("guild_id = :guild_id", { guild_id: this.guildId })
+      .andWhere("expires_at IS NOT NULL")
+      .andWhere("expires_at <= NOW()")
+      .delete()
+      .execute();
+  }
+
+  async find(id: string): Promise<ArchiveEntry> {
+    return this.archives.findOne({
+      where: { id },
+      relations: this.getRelations(),
+    });
+  }
+
+  async makePermanent(id: string): Promise<void> {
+    await this.archives.update(
+      { id },
+      {
+        expires_at: null,
+      },
+    );
+  }
+
+  /**
+   * @returns ID of the created entry
+   */
+  async create(body: string, expiresAt: moment.Moment = null): Promise<string> {
+    if (!expiresAt) {
+      expiresAt = moment().add(DEFAULT_EXPIRY_DAYS, "days");
+    }
+
+    const result = await this.archives.insert({
+      guild_id: this.guildId,
+      body,
+      expires_at: expiresAt.format("YYYY-MM-DD HH:mm:ss"),
+    });
+
+    return result.identifiers[0].id;
+  }
+
+  async createFromSavedMessages(savedMessages: SavedMessage[], guild: Guild, expiresAt = null) {
+    if (expiresAt == null) expiresAt = moment().add(DEFAULT_EXPIRY_DAYS, "days");
+
+    const headerStr = await renderTemplate(MESSAGE_ARCHIVE_HEADER_FORMAT, { guild });
+    const msgLines = [];
+    for (const msg of savedMessages) {
+      const channel = guild.channels.get(msg.channel_id);
+      const user = { ...msg.data.author, id: msg.user_id };
+
+      const line = await renderTemplate(MESSAGE_ARCHIVE_MESSAGE_FORMAT, {
+        id: msg.id,
+        timestamp: moment(msg.posted_at).format("YYYY-MM-DD HH:mm:ss"),
+        content: msg.data.content,
+        user,
+        channel,
+      });
+      msgLines.push(line);
+    }
+    const messagesStr = msgLines.join("\n");
+
+    return this.create([headerStr, messagesStr].join("\n\n"), expiresAt);
+  }
+
+  getUrl(baseUrl, archiveId) {
+    return baseUrl ? `${baseUrl}/archives/${archiveId}` : `Archive ID: ${archiveId}`;
+  }
+}