infisical/backend/src/db/migrations/20240624172027_default-saml-ldap-org.ts

import { Knex } from "knex";

import { TableName } from "../schemas";

const DEFAULT_AUTH_ORG_ID_FIELD = "defaultAuthOrgId";

export async function up(knex: Knex): Promise<void> {
  const hasDefaultOrgColumn = await knex.schema.hasColumn(TableName.SuperAdmin, DEFAULT_AUTH_ORG_ID_FIELD);

  await knex.schema.alterTable(TableName.SuperAdmin, (t) => {
    if (!hasDefaultOrgColumn) {
      t.uuid(DEFAULT_AUTH_ORG_ID_FIELD).nullable();
      t.foreign(DEFAULT_AUTH_ORG_ID_FIELD).references("id").inTable(TableName.Organization).onDelete("SET NULL");
    }
  });
}

export async function down(knex: Knex): Promise<void> {
  const hasDefaultOrgColumn = await knex.schema.hasColumn(TableName.SuperAdmin, DEFAULT_AUTH_ORG_ID_FIELD);

  await knex.schema.alterTable(TableName.SuperAdmin, (t) => {
    if (hasDefaultOrgColumn) {
      t.dropForeign([DEFAULT_AUTH_ORG_ID_FIELD]);
      t.dropColumn(DEFAULT_AUTH_ORG_ID_FIELD);
    }
  });
}