import { MigrationInterface, QueryRunner } from 'typeorm';

export class F4AlignAssetVersionFunctionChange1790000700000 implements MigrationInterface {
  name = 'F4AlignAssetVersionFunctionChange1790000700000';

  public async up(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(`
      ALTER TABLE asset_versions
      DROP CONSTRAINT IF EXISTS chk_asset_versions_change_type
    `);
    await queryRunner.query(`
      ALTER TABLE asset_versions
      ADD CONSTRAINT chk_asset_versions_change_type CHECK (change_type IN (
        'BASELINE', 'CREATED', 'UPDATED', 'CONTEXT_CHANGED', 'FUNCTION_CHANGED',
        'STATUS_CHANGED', 'OPERATIONAL_STATUS_CHANGED', 'REGISTRY_UPDATED',
        'GEOMETRY_UPDATED', 'GEOMETRY_REMOVED',
        'MEDIA_UPLOADED', 'MEDIA_UPDATED', 'MEDIA_REMOVED',
        'PROVENANCE_BASELINE', 'PROVENANCE_UPDATED', 'PROVENANCE_VERIFIED'
      ))
    `);
  }

  public async down(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(`
      UPDATE asset_versions
      SET change_type='UPDATED'
      WHERE change_type='FUNCTION_CHANGED'
    `);
    await queryRunner.query(`
      ALTER TABLE asset_versions
      DROP CONSTRAINT IF EXISTS chk_asset_versions_change_type
    `);
    await queryRunner.query(`
      ALTER TABLE asset_versions
      ADD CONSTRAINT chk_asset_versions_change_type CHECK (change_type IN (
        'BASELINE', 'CREATED', 'UPDATED', 'CONTEXT_CHANGED',
        'STATUS_CHANGED', 'OPERATIONAL_STATUS_CHANGED', 'REGISTRY_UPDATED',
        'GEOMETRY_UPDATED', 'GEOMETRY_REMOVED',
        'MEDIA_UPLOADED', 'MEDIA_UPDATED', 'MEDIA_REMOVED',
        'PROVENANCE_BASELINE', 'PROVENANCE_UPDATED', 'PROVENANCE_VERIFIED'
      ))
    `);
  }
}