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

export async function up({ db, payload, req }: MigrateUpArgs): Promise<void> {
  await db.execute(sql`
   CREATE TYPE "public"."enum_npc_staffing_labor_classification" AS ENUM('general', 'doctor', 'mechanic', 'engineer', 'security', 'other');
  ALTER TABLE "npc_staffing" ALTER COLUMN "npc_id" DROP NOT NULL;
  ALTER TABLE "npc_staffing" ADD COLUMN "headcount" numeric DEFAULT 1;
  ALTER TABLE "npc_staffing" ADD COLUMN "labor_classification" "enum_npc_staffing_labor_classification";`)
}

export async function down({ db, payload, req }: MigrateDownArgs): Promise<void> {
  await db.execute(sql`
   ALTER TABLE "npc_staffing" ALTER COLUMN "npc_id" SET NOT NULL;
  ALTER TABLE "npc_staffing" DROP COLUMN "headcount";
  ALTER TABLE "npc_staffing" DROP COLUMN "labor_classification";
  DROP TYPE "public"."enum_npc_staffing_labor_classification";`)
}