115 lines
3.8 KiB
TypeScript
115 lines
3.8 KiB
TypeScript
import { MigrationInterface, QueryRunner, Table, TableColumn } from "typeorm";
|
|
|
|
export class MoveStarboardsToConfig1573248462469 implements MigrationInterface {
|
|
public async up(queryRunner: QueryRunner): Promise<any> {
|
|
// Edge case: Since we're dropping the primary key temporarily, we need to disable the sql_require_primary_key setting if it's enabled
|
|
// This is restored at the end of the migration
|
|
const sqlRequirePrimaryKey = (await queryRunner.query("SELECT @@sql_require_primary_key AS value"))[0].value;
|
|
await queryRunner.query("SET SESSION sql_require_primary_key=0");
|
|
|
|
// Create a new column for the channel's id
|
|
const chanid_column = new TableColumn({
|
|
name: "starboard_channel_id",
|
|
type: "bigint",
|
|
unsigned: true,
|
|
});
|
|
await queryRunner.addColumn("starboard_messages", chanid_column);
|
|
|
|
// Since we are removing the guild_id with the starboards table, we might want it here
|
|
const guid_column = new TableColumn({
|
|
name: "guild_id",
|
|
type: "bigint",
|
|
unsigned: true,
|
|
});
|
|
await queryRunner.addColumn("starboard_messages", guid_column);
|
|
|
|
// Migrate the old starboard_id to the new starboard_channel_id
|
|
await queryRunner.query(`
|
|
UPDATE starboard_messages AS sm
|
|
JOIN starboards AS sb
|
|
ON sm.starboard_id = sb.id
|
|
SET sm.starboard_channel_id = sb.channel_id, sm.guild_id = sb.guild_id;
|
|
`);
|
|
|
|
// Drop the starboard_id column as it is now obsolete
|
|
await queryRunner.dropColumn("starboard_messages", "starboard_id");
|
|
// Set new Primary Key
|
|
await queryRunner.dropPrimaryKey("starboard_messages");
|
|
await queryRunner.createPrimaryKey("starboard_messages", ["starboard_message_id"]);
|
|
// Finally, drop the starboards channel as it is now obsolete
|
|
await queryRunner.dropTable("starboards", true);
|
|
|
|
await queryRunner.query(`SET SESSION sql_require_primary_key=${sqlRequirePrimaryKey}`);
|
|
}
|
|
|
|
public async down(queryRunner: QueryRunner): Promise<any> {
|
|
const sqlRequirePrimaryKey = (await queryRunner.query("SELECT @@sql_require_primary_key AS value"))[0].value;
|
|
await queryRunner.query("SET SESSION sql_require_primary_key=0");
|
|
|
|
await queryRunner.dropColumn("starboard_messages", "starboard_channel_id");
|
|
await queryRunner.dropColumn("starboard_messages", "guild_id");
|
|
|
|
const sbId = new TableColumn({
|
|
name: "starboard_id",
|
|
type: "int",
|
|
unsigned: true,
|
|
});
|
|
await queryRunner.addColumn("starboard_messages", sbId);
|
|
|
|
await queryRunner.dropPrimaryKey("starboard_messages");
|
|
await queryRunner.createPrimaryKey("starboard_messages", ["starboard_id", "message_id"]);
|
|
|
|
await queryRunner.createTable(
|
|
new Table({
|
|
name: "starboards",
|
|
columns: [
|
|
{
|
|
name: "id",
|
|
type: "int",
|
|
unsigned: true,
|
|
isGenerated: true,
|
|
generationStrategy: "increment",
|
|
isPrimary: true,
|
|
},
|
|
{
|
|
name: "guild_id",
|
|
type: "bigint",
|
|
unsigned: true,
|
|
},
|
|
{
|
|
name: "channel_id",
|
|
type: "bigint",
|
|
unsigned: true,
|
|
},
|
|
{
|
|
name: "channel_whitelist",
|
|
type: "text",
|
|
isNullable: true,
|
|
default: null,
|
|
},
|
|
{
|
|
name: "emoji",
|
|
type: "varchar",
|
|
length: "64",
|
|
},
|
|
{
|
|
name: "reactions_required",
|
|
type: "smallint",
|
|
unsigned: true,
|
|
},
|
|
],
|
|
indices: [
|
|
{
|
|
columnNames: ["guild_id", "emoji"],
|
|
},
|
|
{
|
|
columnNames: ["guild_id", "channel_id"],
|
|
isUnique: true,
|
|
},
|
|
],
|
|
}),
|
|
);
|
|
|
|
await queryRunner.query(`SET SESSION sql_require_primary_key=${sqlRequirePrimaryKey}`);
|
|
}
|
|
}
|