import { MigrationInterface, QueryRunner } from "typeorm";

export class guildChannelOrdering1696420827239 implements MigrationInterface {
	name = "guildChannelOrdering1696420827239";

	public async up(queryRunner: QueryRunner): Promise<void> {
		const guilds = await queryRunner.query(
			`SELECT id FROM guilds`,
			undefined,
			true,
		);

		await queryRunner.query(
			`ALTER TABLE guilds ADD channel_ordering text NOT NULL DEFAULT '[]'`,
		);

		for (const guild_id of guilds.records.map((x) => x.id)) {
			const channels: Array<{ position: number; id: string }> = (
				await queryRunner.query(
					`SELECT id, position FROM channels WHERE guild_id = $1`,
					[guild_id],
					true,
				)
			).records;

			channels.sort((a, b) => a.position - b.position);

			await queryRunner.query(
				`UPDATE guilds SET channel_ordering = $1 WHERE id = $2`,
				[JSON.stringify(channels.map((x) => x.id)), guild_id],
			);
		}

		await queryRunner.query(`ALTER TABLE channels DROP COLUMN position`);
	}

	public async down(queryRunner: QueryRunner): Promise<void> {
		await queryRunner.query(`ALTER TABLE channels ADD position integer NOT NULL DEFAULT 0`);

		const guilds = await queryRunner.query(
			`SELECT id, channel_ordering FROM guilds`,
			undefined,
			true,
		);

		for (const guild of guilds.records) {
			const channel_ordering: string[] = JSON.parse(guild.channel_ordering);

			for (let i = 0; i < channel_ordering.length; i++) {
				const channel_id = channel_ordering[i];
				await queryRunner.query(
					`UPDATE channels SET position = $1 WHERE id = $2`,
					[i, channel_id],
				);
			}
		}

		await queryRunner.query(
			`ALTER TABLE guilds DROP COLUMN channel_ordering`,
		);
	}
}