import { MigrationInterface, QueryRunner } from "typeorm"

export class AddFleetTypeId1755081913545 implements MigrationInterface {
  name = "AddFleetTypeId1755081913545"

  public async up(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(
      `ALTER TABLE "trip_fleet_assignments" ADD "fleet_type_id" integer`,
    )
    await queryRunner.query(
      `ALTER TABLE "trip_fleet_assignments" ADD "trip_completed_at" TIMESTAMP`,
    )
    await queryRunner.query(
      `ALTER TABLE "trip_fleet_assignments" ADD CONSTRAINT "FK_0bb15fd4cd73e133513b02e574b" FOREIGN KEY ("fleet_type_id") REFERENCES "vehicle_types"("id") ON DELETE NO ACTION ON UPDATE NO ACTION`,
    )
  }

  public async down(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(
      `ALTER TABLE "trip_fleet_assignments" DROP CONSTRAINT "FK_0bb15fd4cd73e133513b02e574b"`,
    )
    await queryRunner.query(
      `ALTER TABLE "trip_fleet_assignments" DROP COLUMN "trip_completed_at"`,
    )
    await queryRunner.query(
      `ALTER TABLE "trip_fleet_assignments" DROP COLUMN "fleet_type_id"`,
    )
  }
}