Add discovery filtering

2 files changed, 21 insertions, 0 deletions

diff --git a/src/util/entities/Guild.ts b/src/util/entities/Guild.ts

index 56740ab5..aac3daf0 100644
--- a/src/util/entities/Guild.ts
+++ b/src/util/entities/Guild.ts
@@ -298,6 +298,12 @@ export class Guild extends BaseClass {
     @Column({ select: false, type: "simple-array" })
     channel_ordering: string[];
 
+    @Column()
+    discovery_weight: number = 0;
+
+    @Column()
+    discovery_excluded: boolean = false;
+
     static async createGuild(body: {
         name?: string;
         icon?: string | null;
diff --git a/src/util/migration/postgres/1770748070808-GuildDiscoveryHoisting.ts b/src/util/migration/postgres/1770748070808-GuildDiscoveryHoisting.ts
new file mode 100644

index 00000000..de878388
--- /dev/null
+++ b/src/util/migration/postgres/1770748070808-GuildDiscoveryHoisting.ts
@@ -0,0 +1,15 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class GuildDiscoveryHoisting1770748070808 implements MigrationInterface {
+    name = "GuildDiscoveryHoisting1770748070808";
+
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`ALTER TABLE "guilds" ADD "discovery_weight" integer NOT NULL`);
+        await queryRunner.query(`ALTER TABLE "guilds" ADD "discovery_excluded" boolean NOT NULL`);
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`ALTER TABLE "guilds" DROP COLUMN "discovery_excluded"`);
+        await queryRunner.query(`ALTER TABLE "guilds" DROP COLUMN "discovery_weight"`);
+    }
+}