infisical/backend/src/db/migrations/20250414203701_add-notification-flag-service-token.ts

import { Knex } from "knex";

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

export async function up(knex: Knex): Promise<void> {
  const hasCol = await knex.schema.hasColumn(TableName.ServiceToken, "expiryNotificationSent");
  if (!hasCol) {
    await knex.schema.alterTable(TableName.ServiceToken, (t) => {
      t.boolean("expiryNotificationSent").defaultTo(false);
    });

    // Update only tokens where expiresAt is before current time
    await knex(TableName.ServiceToken)
      .whereRaw(`${TableName.ServiceToken}."expiresAt" < NOW()`)
      .whereNotNull("expiresAt")
      .update({ expiryNotificationSent: true });
  }
}

export async function down(knex: Knex): Promise<void> {
  const hasCol = await knex.schema.hasColumn(TableName.ServiceToken, "expiryNotificationSent");
  if (hasCol) {
    await knex.schema.alterTable(TableName.ServiceToken, (t) => {
      t.dropColumn("expiryNotificationSent");
    });
  }
}