import { MigrationInterface, QueryRunner } from "typeorm"

export class AddCompanyFkRelations1763545890403 implements MigrationInterface {
  name = "AddCompanyFkRelations1763545890403"

  public async up(queryRunner: QueryRunner): Promise<void> {
    // FK for company_profiles → companies
    await queryRunner.query(`
            ALTER TABLE "company_profiles"
            ADD CONSTRAINT "FK_company_profiles_company_id"
            FOREIGN KEY ("company_id") REFERENCES "companies"("id")
            ON DELETE NO ACTION ON UPDATE NO ACTION
        `)

    // FK for company_subscriptions → companies
    await queryRunner.query(`
            ALTER TABLE "company_subscriptions"
            ADD CONSTRAINT "FK_company_subscriptions_company_id"
            FOREIGN KEY ("company_id") REFERENCES "companies"("id")
            ON DELETE NO ACTION ON UPDATE NO ACTION
        `)
  }

  public async down(queryRunner: QueryRunner): Promise<void> {
    await queryRunner.query(`
            ALTER TABLE "company_subscriptions"
            DROP CONSTRAINT "FK_company_subscriptions_company_id"
        `)

    await queryRunner.query(`
            ALTER TABLE "company_profiles"
            DROP CONSTRAINT "FK_company_profiles_company_id"
        `)
  }
}