edr-platform/apps/edr-freight-api/src/migrations/3010000000000-AddBookingTransitAssignee.ts

import { MigrationInterface, QueryRunner } from 'typeorm';

/**
 * Transit-assignee handshake on the SHIPMENT, not the contract.
 *
 * Clearance runs per booking now, so the ask GL Ethiopia raises before filing a
 * customs declaration ("who handles this shipment in Djibouti?") and Djibouti's
 * answer belong on the booking. The contract-cycle columns added by
 * 2950000000000 stay for the legacy contract-level cycles.
 */
export class AddBookingTransitAssignee3010000000000 implements MigrationInterface {
  public async up(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(`
      ALTER TABLE freight.bookings
        ADD COLUMN IF NOT EXISTS transit_assignee_requested_at timestamptz NULL,
        ADD COLUMN IF NOT EXISTS transit_assignee_request_note text NULL,
        ADD COLUMN IF NOT EXISTS transit_assignee_name text NULL,
        ADD COLUMN IF NOT EXISTS transit_assignee_assigned_at timestamptz NULL
    `);
  }

  public async down(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(`
      ALTER TABLE freight.bookings
        DROP COLUMN IF EXISTS transit_assignee_requested_at,
        DROP COLUMN IF EXISTS transit_assignee_request_note,
        DROP COLUMN IF EXISTS transit_assignee_name,
        DROP COLUMN IF EXISTS transit_assignee_assigned_at
    `);
  }
}