import { MigrationInterface, QueryRunner } from "typeorm"

export class AddCityInTeamMemberAndFleetEntity1761718578265 implements MigrationInterface {
  name = "AddCityInTeamMemberAndFleetEntity1761718578265"

  public async up(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(
      `ALTER TABLE "team_members" ADD "working_city_location" integer`,
    )
    await queryRunner.query(
      `ALTER TABLE "fleet_operations" ADD "vehicle_city_location_id" integer`,
    )
    await queryRunner.query(
      `ALTER TABLE "team_members" ADD CONSTRAINT "FK_d0fc999a9afd145266e010cab49" FOREIGN KEY ("working_city_location") REFERENCES "cities"("id") ON DELETE NO ACTION ON UPDATE NO ACTION`,
    )
    await queryRunner.query(
      `ALTER TABLE "fleet_operations" ADD CONSTRAINT "FK_c0ea62edd4c0c5dfb5da3bb4f26" FOREIGN KEY ("vehicle_city_location_id") REFERENCES "cities"("id") ON DELETE NO ACTION ON UPDATE NO ACTION`,
    )
  }

  public async down(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(
      `ALTER TABLE "fleet_operations" DROP CONSTRAINT "FK_c0ea62edd4c0c5dfb5da3bb4f26"`,
    )
    await queryRunner.query(
      `ALTER TABLE "team_members" DROP CONSTRAINT "FK_d0fc999a9afd145266e010cab49"`,
    )
    await queryRunner.query(
      `ALTER TABLE "fleet_operations" DROP COLUMN "vehicle_city_location_id"`,
    )
    await queryRunner.query(
      `ALTER TABLE "team_members" DROP COLUMN "working_city_location"`,
    )
  }
}
