import { MongoMigration } from '../mongo_migrator.js'

export default class CreateMigrationRoutesCollection extends MongoMigration {
  async up(): Promise<void> {
    const collection = this.db.collection('migration_routes')
    await collection.createIndex({ identifier: 1 }, { unique: true, name: 'idx_identifier' })
    await collection.createIndex({ productClass: 1 }, { name: 'idx_product_class' })
    await collection.createIndex(
      { productClass: 1, sourceVersionPattern: 1 },
      { name: 'idx_product_class_source_version' }
    )
  }

  async down(): Promise<void> {
    await this.db.dropCollection('migration_routes').catch(() => {})
  }
}