infisical/backend/src/db/migrations/20240830142938_native-slack-integration.ts

import { Knex } from "knex";

import { TableName } from "../schemas";
import { createOnUpdateTrigger, dropOnUpdateTrigger } from "../utils";

export async function up(knex: Knex): Promise<void> {
  if (!(await knex.schema.hasTable(TableName.WorkflowIntegrations))) {
    await knex.schema.createTable(TableName.WorkflowIntegrations, (tb) => {
      tb.uuid("id", { primaryKey: true }).defaultTo(knex.fn.uuid());
      tb.string("integration").notNullable();
      tb.string("slug").notNullable();
      tb.uuid("orgId").notNullable();
      tb.foreign("orgId").references("id").inTable(TableName.Organization).onDelete("CASCADE");
      tb.string("description");
      tb.unique(["orgId", "slug"]);
      tb.timestamps(true, true, true);
    });

    await createOnUpdateTrigger(knex, TableName.WorkflowIntegrations);
  }

  if (!(await knex.schema.hasTable(TableName.SlackIntegrations))) {
    await knex.schema.createTable(TableName.SlackIntegrations, (tb) => {
      tb.uuid("id", { primaryKey: true }).notNullable();
      tb.foreign("id").references("id").inTable(TableName.WorkflowIntegrations).onDelete("CASCADE");
      tb.string("teamId").notNullable();
      tb.string("teamName").notNullable();
      tb.string("slackUserId").notNullable();
      tb.string("slackAppId").notNullable();
      tb.binary("encryptedBotAccessToken").notNullable();
      tb.string("slackBotId").notNullable();
      tb.string("slackBotUserId").notNullable();
      tb.timestamps(true, true, true);
    });

    await createOnUpdateTrigger(knex, TableName.SlackIntegrations);
  }

  if (!(await knex.schema.hasTable(TableName.ProjectSlackConfigs))) {
    await knex.schema.createTable(TableName.ProjectSlackConfigs, (tb) => {
      tb.uuid("id", { primaryKey: true }).defaultTo(knex.fn.uuid());
      tb.string("projectId").notNullable().unique();
      tb.foreign("projectId").references("id").inTable(TableName.Project).onDelete("CASCADE");
      tb.uuid("slackIntegrationId").notNullable();
      tb.foreign("slackIntegrationId").references("id").inTable(TableName.SlackIntegrations).onDelete("CASCADE");
      tb.boolean("isAccessRequestNotificationEnabled").notNullable().defaultTo(false);
      tb.string("accessRequestChannels").notNullable().defaultTo("");
      tb.boolean("isSecretRequestNotificationEnabled").notNullable().defaultTo(false);
      tb.string("secretRequestChannels").notNullable().defaultTo("");
      tb.timestamps(true, true, true);
    });

    await createOnUpdateTrigger(knex, TableName.ProjectSlackConfigs);
  }

  const doesSuperAdminHaveSlackClientId = await knex.schema.hasColumn(TableName.SuperAdmin, "encryptedSlackClientId");
  const doesSuperAdminHaveSlackClientSecret = await knex.schema.hasColumn(
    TableName.SuperAdmin,
    "encryptedSlackClientSecret"
  );

  await knex.schema.alterTable(TableName.SuperAdmin, (tb) => {
    if (!doesSuperAdminHaveSlackClientId) {
      tb.binary("encryptedSlackClientId");
    }
    if (!doesSuperAdminHaveSlackClientSecret) {
      tb.binary("encryptedSlackClientSecret");
    }
  });
}

export async function down(knex: Knex): Promise<void> {
  await knex.schema.dropTableIfExists(TableName.ProjectSlackConfigs);
  await dropOnUpdateTrigger(knex, TableName.ProjectSlackConfigs);

  await knex.schema.dropTableIfExists(TableName.SlackIntegrations);
  await dropOnUpdateTrigger(knex, TableName.SlackIntegrations);

  await knex.schema.dropTableIfExists(TableName.WorkflowIntegrations);
  await dropOnUpdateTrigger(knex, TableName.WorkflowIntegrations);

  const doesSuperAdminHaveSlackClientId = await knex.schema.hasColumn(TableName.SuperAdmin, "encryptedSlackClientId");
  const doesSuperAdminHaveSlackClientSecret = await knex.schema.hasColumn(
    TableName.SuperAdmin,
    "encryptedSlackClientSecret"
  );

  await knex.schema.alterTable(TableName.SuperAdmin, (tb) => {
    if (doesSuperAdminHaveSlackClientId) {
      tb.dropColumn("encryptedSlackClientId");
    }
    if (doesSuperAdminHaveSlackClientSecret) {
      tb.dropColumn("encryptedSlackClientSecret");
    }
  });
}