import { MigrateUpArgs, MigrateDownArgs, sql } from '@payloadcms/db-postgres';

export async function up({ db }: MigrateUpArgs): Promise<void> {
  await db.execute(sql`
    ALTER TABLE "media" ADD COLUMN IF NOT EXISTS "sizes_tablet_url" varchar;
    ALTER TABLE "media" ADD COLUMN IF NOT EXISTS "sizes_tablet_width" numeric;
    ALTER TABLE "media" ADD COLUMN IF NOT EXISTS "sizes_tablet_height" numeric;
    ALTER TABLE "media" ADD COLUMN IF NOT EXISTS "sizes_tablet_mime_type" varchar;
    ALTER TABLE "media" ADD COLUMN IF NOT EXISTS "sizes_tablet_filesize" numeric;
    ALTER TABLE "media" ADD COLUMN IF NOT EXISTS "sizes_tablet_filename" varchar;
  `);

  await db.execute(sql`
    CREATE INDEX IF NOT EXISTS "media_sizes_tablet_sizes_tablet_filename_idx" ON "media" USING btree ("sizes_tablet_filename");
  `);
}

export async function down({ db }: MigrateDownArgs): Promise<void> {
  await db.execute(sql`
    DROP INDEX IF EXISTS "media_sizes_tablet_sizes_tablet_filename_idx";
  `);

  await db.execute(sql`
    ALTER TABLE "media" DROP COLUMN IF EXISTS "sizes_tablet_filename";
    ALTER TABLE "media" DROP COLUMN IF EXISTS "sizes_tablet_filesize";
    ALTER TABLE "media" DROP COLUMN IF EXISTS "sizes_tablet_mime_type";
    ALTER TABLE "media" DROP COLUMN IF EXISTS "sizes_tablet_height";
    ALTER TABLE "media" DROP COLUMN IF EXISTS "sizes_tablet_width";
    ALTER TABLE "media" DROP COLUMN IF EXISTS "sizes_tablet_url";
  `);
}