118 lines
3.3 KiB
TypeScript
118 lines
3.3 KiB
TypeScript
import { MigrationInterface, QueryRunner, Table, TableColumn } from "typeorm";
|
|
|
|
export class MoveStarboardsToConfig1573248462469 implements MigrationInterface {
|
|
public async up(queryRunner: QueryRunner): Promise<any> {
|
|
// Create a new column for the channel's id
|
|
await queryRunner.addColumn(
|
|
"starboard_messages",
|
|
new TableColumn({
|
|
name: "starboard_channel_id",
|
|
type: "bigint",
|
|
unsigned: true,
|
|
}),
|
|
);
|
|
|
|
// Since we are removing the guild_id with the starboards table, we might want it here
|
|
await queryRunner.addColumn(
|
|
"starboard_messages",
|
|
new TableColumn({
|
|
name: "guild_id",
|
|
type: "bigint",
|
|
unsigned: true,
|
|
}),
|
|
);
|
|
|
|
// 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;
|
|
`);
|
|
|
|
// Set new Primary Key
|
|
await queryRunner.query(`
|
|
ALTER TABLE starboard_messages
|
|
DROP PRIMARY KEY,
|
|
ADD PRIMARY KEY(\`starboard_message_id\`);
|
|
`);
|
|
// Drop the starboard_id column as it is now obsolete
|
|
// We can't use queyrRunner.dropColumn() here because TypeORM helpfully thinks that
|
|
// starboard_id is still part of the primary key and tries to drop the PK first
|
|
await queryRunner.query("ALTER TABLE starboard_messages DROP COLUMN starboard_id");
|
|
|
|
// Finally, drop the starboards table as it is now obsolete
|
|
await queryRunner.dropTable("starboards", true);
|
|
}
|
|
|
|
public async down(queryRunner: QueryRunner): Promise<any> {
|
|
await queryRunner.dropColumn("starboard_messages", "starboard_channel_id");
|
|
await queryRunner.dropColumn("starboard_messages", "guild_id");
|
|
|
|
await queryRunner.addColumn(
|
|
"starboard_messages",
|
|
new TableColumn({
|
|
name: "starboard_id",
|
|
type: "int",
|
|
unsigned: true,
|
|
}),
|
|
);
|
|
|
|
await queryRunner.query(`
|
|
ALTER TABLE starboard_messages
|
|
DROP PRIMARY KEY,
|
|
ADD PRIMARY KEY(\`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,
|
|
},
|
|
],
|
|
}),
|
|
);
|
|
}
|
|
}
|