@oneuptime/common 9.2.16 → 9.2.17

package/Server/Infrastructure/Postgres/SchemaMigrations/1765633554715-MigrationName.ts

@@ -0,0 +1,75 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class MigrationName1765633554715 implements MigrationInterface {
+  public name = "MigrationName1765633554715";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `CREATE TABLE "ServiceCatalogCodeRepository" ("_id" uuid NOT NULL DEFAULT uuid_generate_v4(), "createdAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(), "updatedAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(), "deletedAt" TIMESTAMP WITH TIME ZONE, "version" integer NOT NULL, "projectId" uuid NOT NULL, "serviceCatalogId" uuid NOT NULL, "codeRepositoryId" uuid NOT NULL, "servicePathInRepository" character varying, "createdByUserId" uuid, "deletedByUserId" uuid, CONSTRAINT "PK_1a0f992cb6c55f48d9fae93369a" PRIMARY KEY ("_id"))`,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_41a90624dfbb954e3b50c93dfc" ON "ServiceCatalogCodeRepository" ("projectId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_4638e464dc7cab644c7926674b" ON "ServiceCatalogCodeRepository" ("serviceCatalogId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_05b6c65746b28329def0accba9" ON "ServiceCatalogCodeRepository" ("codeRepositoryId") `,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD CONSTRAINT "FK_41a90624dfbb954e3b50c93dfc1" FOREIGN KEY ("projectId") REFERENCES "Project"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD CONSTRAINT "FK_4638e464dc7cab644c7926674b8" FOREIGN KEY ("serviceCatalogId") REFERENCES "ServiceCatalog"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD CONSTRAINT "FK_05b6c65746b28329def0accba94" FOREIGN KEY ("codeRepositoryId") REFERENCES "CodeRepository"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD CONSTRAINT "FK_c838ecb46958f5de7a699f50c53" FOREIGN KEY ("createdByUserId") REFERENCES "User"("_id") ON DELETE SET NULL ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD CONSTRAINT "FK_1aae6b3a023a36f7c004749afe4" FOREIGN KEY ("deletedByUserId") REFERENCES "User"("_id") ON DELETE SET NULL ON UPDATE NO ACTION`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP CONSTRAINT "FK_1aae6b3a023a36f7c004749afe4"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP CONSTRAINT "FK_c838ecb46958f5de7a699f50c53"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP CONSTRAINT "FK_05b6c65746b28329def0accba94"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP CONSTRAINT "FK_4638e464dc7cab644c7926674b8"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP CONSTRAINT "FK_41a90624dfbb954e3b50c93dfc1"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_05b6c65746b28329def0accba9"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_4638e464dc7cab644c7926674b"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_41a90624dfbb954e3b50c93dfc"`,
+    );
+    await queryRunner.query(`DROP TABLE "ServiceCatalogCodeRepository"`);
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1765801357168-MigrationName.ts

@@ -0,0 +1,32 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+import CaptureSpan from "../../../Utils/Telemetry/CaptureSpan";
+
+export class MigrationName1765801357168 implements MigrationInterface {
+  public name = "MigrationName1765801357168";
+
+  @CaptureSpan()
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD "enableAutomaticImprovements" boolean NOT NULL DEFAULT true`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD "maxOpenPullRequests" integer NOT NULL DEFAULT 3`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" ADD "restrictedImprovementActions" jsonb`,
+    );
+  }
+
+  @CaptureSpan()
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP COLUMN "restrictedImprovementActions"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP COLUMN "maxOpenPullRequests"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "ServiceCatalogCodeRepository" DROP COLUMN "enableAutomaticImprovements"`,
+    );
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1765810218488-MigrationName.ts

@@ -0,0 +1,69 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class MigrationName1765810218488 implements MigrationInterface {
+  public name = "MigrationName1765810218488";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "aiCurrentBalanceInUSDCents" integer NOT NULL DEFAULT '0'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "autoAiRechargeByBalanceInUSD" integer NOT NULL DEFAULT '20'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "autoRechargeAiWhenCurrentBalanceFallsInUSD" integer NOT NULL DEFAULT '10'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "enableAi" boolean NOT NULL DEFAULT true`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "enableAutoRechargeAiBalance" boolean NOT NULL DEFAULT false`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "lowAiBalanceNotificationSentToOwners" boolean NOT NULL DEFAULT false`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "failedAiBalanceChargeNotificationSentToOwners" boolean NOT NULL DEFAULT false`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" ADD "notEnabledAiNotificationSentToOwners" boolean NOT NULL DEFAULT false`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "notEnabledAiNotificationSentToOwners"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "failedAiBalanceChargeNotificationSentToOwners"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "lowAiBalanceNotificationSentToOwners"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "enableAutoRechargeAiBalance"`,
+    );
+    await queryRunner.query(`ALTER TABLE "Project" DROP COLUMN "enableAi"`);
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "autoRechargeAiWhenCurrentBalanceFallsInUSD"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "autoAiRechargeByBalanceInUSD"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "Project" DROP COLUMN "aiCurrentBalanceInUSDCents"`,
+    );
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1765830758857-MigrationName.ts

@@ -0,0 +1,111 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class MigrationName1765830758857 implements MigrationInterface {
+  public name = "MigrationName1765830758857";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `CREATE TABLE "LlmLog" ("_id" uuid NOT NULL DEFAULT uuid_generate_v4(), "createdAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(), "updatedAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(), "deletedAt" TIMESTAMP WITH TIME ZONE, "version" integer NOT NULL, "projectId" uuid NOT NULL, "llmProviderId" uuid, "llmProviderName" character varying(100), "llmType" character varying(100), "modelName" character varying(100), "isGlobalProvider" boolean NOT NULL DEFAULT false, "inputTokens" integer NOT NULL DEFAULT '0', "outputTokens" integer NOT NULL DEFAULT '0', "totalTokens" integer NOT NULL DEFAULT '0', "costInUSDCents" integer NOT NULL DEFAULT '0', "wasBilled" boolean NOT NULL DEFAULT false, "status" character varying(100) NOT NULL, "statusMessage" character varying(500), "feature" character varying(100), "requestPrompt" text, "responsePreview" text, "incidentId" uuid, "alertId" uuid, "scheduledMaintenanceId" uuid, "userId" uuid, "requestStartedAt" TIMESTAMP WITH TIME ZONE, "requestCompletedAt" TIMESTAMP WITH TIME ZONE, "durationMs" integer, "deletedByUserId" uuid, CONSTRAINT "PK_807b7f4578f9dcbb1f7aeeb94f8" PRIMARY KEY ("_id"))`,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_c3c061a924f368e2cc68a23308" ON "LlmLog" ("projectId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_bfd15354697dc30fedf7a96976" ON "LlmLog" ("llmProviderId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_159d2b07c02788dcac8575bf4a" ON "LlmLog" ("incidentId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_83696b0732c0a3601a9d5d7afe" ON "LlmLog" ("alertId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_19d277440e9b9e3ed4fa46c227" ON "LlmLog" ("scheduledMaintenanceId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_5c6c985581a3f85d84a2987daa" ON "LlmLog" ("userId") `,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmProvider" ADD "costPerMillionTokensInUSDCents" integer NOT NULL DEFAULT '0'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD CONSTRAINT "FK_c3c061a924f368e2cc68a233083" FOREIGN KEY ("projectId") REFERENCES "Project"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD CONSTRAINT "FK_bfd15354697dc30fedf7a96976e" FOREIGN KEY ("llmProviderId") REFERENCES "LlmProvider"("_id") ON DELETE SET NULL ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD CONSTRAINT "FK_159d2b07c02788dcac8575bf4a6" FOREIGN KEY ("incidentId") REFERENCES "Incident"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD CONSTRAINT "FK_83696b0732c0a3601a9d5d7afe1" FOREIGN KEY ("alertId") REFERENCES "Alert"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD CONSTRAINT "FK_19d277440e9b9e3ed4fa46c227a" FOREIGN KEY ("scheduledMaintenanceId") REFERENCES "ScheduledMaintenance"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD CONSTRAINT "FK_5c6c985581a3f85d84a2987daae" FOREIGN KEY ("userId") REFERENCES "User"("_id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD CONSTRAINT "FK_bbe2bdcf251d6ef1ea43b666370" FOREIGN KEY ("deletedByUserId") REFERENCES "User"("_id") ON DELETE SET NULL ON UPDATE NO ACTION`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" DROP CONSTRAINT "FK_bbe2bdcf251d6ef1ea43b666370"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" DROP CONSTRAINT "FK_5c6c985581a3f85d84a2987daae"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" DROP CONSTRAINT "FK_19d277440e9b9e3ed4fa46c227a"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" DROP CONSTRAINT "FK_83696b0732c0a3601a9d5d7afe1"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" DROP CONSTRAINT "FK_159d2b07c02788dcac8575bf4a6"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" DROP CONSTRAINT "FK_bfd15354697dc30fedf7a96976e"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" DROP CONSTRAINT "FK_c3c061a924f368e2cc68a233083"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmProvider" DROP COLUMN "costPerMillionTokensInUSDCents"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_5c6c985581a3f85d84a2987daa"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_19d277440e9b9e3ed4fa46c227"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_83696b0732c0a3601a9d5d7afe"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_159d2b07c02788dcac8575bf4a"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_bfd15354697dc30fedf7a96976"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_c3c061a924f368e2cc68a23308"`,
+    );
+    await queryRunner.query(`DROP TABLE "LlmLog"`);
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/1765834537501-MigrationName.ts

@@ -0,0 +1,39 @@
+import { MigrationInterface, QueryRunner } from "typeorm";
+
+export class MigrationName1765834537501 implements MigrationInterface {
+  public name = "MigrationName1765834537501";
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(`ALTER TABLE "LlmLog" DROP COLUMN "inputTokens"`);
+    await queryRunner.query(`ALTER TABLE "LlmLog" DROP COLUMN "outputTokens"`);
+    await queryRunner.query(`ALTER TABLE "LlmLog" DROP COLUMN "totalTokens"`);
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD "totalTokens" integer NOT NULL DEFAULT '0'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type":"Recurring","value":{"intervalType":"Day","intervalCount":{"_type":"PositiveNumber","value":1}}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type":"RestrictionTimes","value":{"restictionType":"None","dayRestrictionTimes":null,"weeklyRestrictionTimes":[]}}'`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "restrictionTimes" SET DEFAULT '{"_type": "RestrictionTimes", "value": {"restictionType": "None", "dayRestrictionTimes": null, "weeklyRestrictionTimes": []}}'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "OnCallDutyPolicyScheduleLayer" ALTER COLUMN "rotation" SET DEFAULT '{"_type": "Recurring", "value": {"intervalType": "Day", "intervalCount": {"_type": "PositiveNumber", "value": 1}}}'`,
+    );
+    await queryRunner.query(`ALTER TABLE "LlmLog" DROP COLUMN "totalTokens"`);
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD "totalTokens" integer NOT NULL DEFAULT '0'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD "outputTokens" integer NOT NULL DEFAULT '0'`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "LlmLog" ADD "inputTokens" integer NOT NULL DEFAULT '0'`,
+    );
+  }
+}

package/Server/Infrastructure/Postgres/SchemaMigrations/Index.ts

@@ -196,6 +196,12 @@ import { MigrationName1765477339178 } from "./1765477339178-MigrationName";
 import { MigrationName1765540325149 } from "./1765540325149-MigrationName";
 import { MigrationName1765540549739 } from "./1765540549739-MigrationName";
 import { MigrationName1765544010078 } from "./1765544010078-MigrationName";
+import { MigrationName1765580181582 } from "./1765580181582-MigrationName";
+import { MigrationName1765633554715 } from "./1765633554715-MigrationName";
+import { MigrationName1765801357168 } from "./1765801357168-MigrationName";
+import { MigrationName1765810218488 } from "./1765810218488-MigrationName";
+import { MigrationName1765830758857 } from "./1765830758857-MigrationName";
+import { MigrationName1765834537501 } from "./1765834537501-MigrationName";
 
 export default [
   InitialMigration,
@@ -396,4 +402,10 @@ export default [
   MigrationName1765540325149,
   MigrationName1765540549739,
   MigrationName1765544010078,
+  MigrationName1765580181582,
+  MigrationName1765633554715,
+  MigrationName1765801357168,
+  MigrationName1765810218488,
+  MigrationName1765830758857,
+  MigrationName1765834537501,
 ];

package/Server/Services/AIBillingService.ts

@@ -0,0 +1,247 @@
+import {
+  IsBillingEnabled,
+  NotificationSlackWebhookOnSubscriptionUpdate,
+} from "../EnvironmentConfig";
+import logger from "../Utils/Logger";
+import BaseService from "./BaseService";
+import BillingService from "./BillingService";
+import ProjectService from "./ProjectService";
+import BadDataException from "../../Types/Exception/BadDataException";
+import ObjectID from "../../Types/ObjectID";
+import Project from "../../Models/DatabaseModels/Project";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import SlackUtil from "../Utils/Workspace/Slack/Slack";
+import URL from "../../Types/API/URL";
+import Exception from "../../Types/Exception/Exception";
+
+export class AIBillingService extends BaseService {
+  public constructor() {
+    super();
+  }
+
+  @CaptureSpan()
+  public async rechargeBalance(
+    projectId: ObjectID,
+    amountInUSD: number,
+  ): Promise<number> {
+    if (!IsBillingEnabled) {
+      throw new BadDataException("Billing is not enabled");
+    }
+
+    const project: Project | null = await ProjectService.findOneById({
+      id: projectId,
+      select: {
+        aiCurrentBalanceInUSDCents: true,
+        enableAutoRechargeAiBalance: true,
+        enableAi: true,
+        autoAiRechargeByBalanceInUSD: true,
+        autoRechargeAiWhenCurrentBalanceFallsInUSD: true,
+        paymentProviderCustomerId: true,
+        name: true,
+        failedAiBalanceChargeNotificationSentToOwners: true,
+      },
+      props: {
+        isRoot: true,
+      },
+    });
+
+    if (!project) {
+      return 0;
+    }
+
+    try {
+      if (
+        !(await BillingService.hasPaymentMethods(
+          project.paymentProviderCustomerId!,
+        ))
+      ) {
+        if (!project.failedAiBalanceChargeNotificationSentToOwners) {
+          await ProjectService.updateOneById({
+            data: {
+              failedAiBalanceChargeNotificationSentToOwners: true,
+            },
+            id: project.id!,
+            props: {
+              isRoot: true,
+            },
+          });
+          await ProjectService.sendEmailToProjectOwners(
+            project.id!,
+            "ACTION REQUIRED: AI Balance Recharge Failed for project - " +
+              (project.name || ""),
+            `We have tried to recharge your AI balance for project - ${
+              project.name || ""
+            } and failed. We could not find a payment method for the project. Please add a payment method in Project Settings.`,
+          );
+        }
+        throw new BadDataException(
+          "No payment methods found for the project. Please add a payment method in Project Settings to continue.",
+        );
+      }
+
+      // recharge balance
+      const updatedAmount: number = Math.floor(
+        (project.aiCurrentBalanceInUSDCents || 0) + amountInUSD * 100,
+      );
+
+      // If the recharge is successful, then update the project balance.
+      await BillingService.generateInvoiceAndChargeCustomer(
+        project.paymentProviderCustomerId!,
+        "AI Balance Recharge",
+        amountInUSD,
+      );
+
+      await ProjectService.updateOneById({
+        data: {
+          aiCurrentBalanceInUSDCents: updatedAmount,
+          failedAiBalanceChargeNotificationSentToOwners: false, // reset this flag
+          lowAiBalanceNotificationSentToOwners: false, // reset this flag
+          notEnabledAiNotificationSentToOwners: false,
+        },
+        id: project.id!,
+        props: {
+          isRoot: true,
+        },
+      });
+
+      await ProjectService.sendEmailToProjectOwners(
+        project.id!,
+        "AI Balance Recharge Successful for project - " + (project.name || ""),
+        `We have successfully recharged your AI balance for project - ${
+          project.name || ""
+        } by ${amountInUSD} USD. Your current balance is ${
+          updatedAmount / 100
+        } USD.`,
+      );
+
+      // Send Slack notification for balance refill
+      this.sendBalanceRefillSlackNotification({
+        project: project,
+        amountInUSD: amountInUSD,
+        currentBalanceInUSD: updatedAmount / 100,
+      }).catch((error: Exception) => {
+        logger.error(
+          "Error sending slack message for AI balance refill: " + error,
+        );
+      });
+
+      project.aiCurrentBalanceInUSDCents = updatedAmount;
+
+      return updatedAmount;
+    } catch (err) {
+      await ProjectService.updateOneById({
+        data: {
+          failedAiBalanceChargeNotificationSentToOwners: true,
+        },
+        id: project.id!,
+        props: {
+          isRoot: true,
+        },
+      });
+      await ProjectService.sendEmailToProjectOwners(
+        project.id!,
+        "ACTION REQUIRED: AI Balance Recharge Failed for project - " +
+          (project.name || ""),
+        `We have tried to recharge your AI balance for project - ${
+          project.name || ""
+        } and failed. Please make sure your payment method is up to date and has sufficient balance. You can add new payment methods in Project Settings.`,
+      );
+      logger.error(err);
+      throw err;
+    }
+  }
+
+  @CaptureSpan()
+  public async rechargeIfBalanceIsLow(
+    projectId: ObjectID,
+    options?: {
+      autoAiRechargeByBalanceInUSD: number;
+      autoRechargeAiWhenCurrentBalanceFallsInUSD: number;
+      enableAutoRechargeAiBalance: boolean;
+    },
+  ): Promise<number> {
+    let project: Project | null = null;
+    if (projectId && IsBillingEnabled) {
+      // check payment methods.
+
+      project = await ProjectService.findOneById({
+        id: projectId,
+        select: {
+          aiCurrentBalanceInUSDCents: true,
+          enableAutoRechargeAiBalance: true,
+          autoAiRechargeByBalanceInUSD: true,
+          autoRechargeAiWhenCurrentBalanceFallsInUSD: true,
+        },
+        props: {
+          isRoot: true,
+        },
+      });
+
+      const autoRechargeAiWhenCurrentBalanceFallsInUSD: number =
+        options?.autoRechargeAiWhenCurrentBalanceFallsInUSD ||
+        project?.autoRechargeAiWhenCurrentBalanceFallsInUSD ||
+        0;
+      const autoAiRechargeByBalanceInUSD: number =
+        options?.autoAiRechargeByBalanceInUSD ||
+        project?.autoAiRechargeByBalanceInUSD ||
+        0;
+
+      const enableAutoRechargeAiBalance: boolean = options
+        ? options.enableAutoRechargeAiBalance
+        : project?.enableAutoRechargeAiBalance || false;
+
+      if (!project) {
+        return 0;
+      }
+
+      if (
+        enableAutoRechargeAiBalance &&
+        autoAiRechargeByBalanceInUSD &&
+        autoRechargeAiWhenCurrentBalanceFallsInUSD
+      ) {
+        if (
+          (project.aiCurrentBalanceInUSDCents || 0) / 100 <
+          autoRechargeAiWhenCurrentBalanceFallsInUSD
+        ) {
+          const updatedAmount: number = await this.rechargeBalance(
+            projectId,
+            autoAiRechargeByBalanceInUSD,
+          );
+          project.aiCurrentBalanceInUSDCents = updatedAmount;
+        }
+      }
+    }
+
+    return project?.aiCurrentBalanceInUSDCents || 0;
+  }
+
+  @CaptureSpan()
+  private async sendBalanceRefillSlackNotification(data: {
+    project: Project;
+    amountInUSD: number;
+    currentBalanceInUSD: number;
+  }): Promise<void> {
+    const { project, amountInUSD, currentBalanceInUSD } = data;
+
+    if (NotificationSlackWebhookOnSubscriptionUpdate) {
+      const slackMessage: string = `*AI Balance Refilled:*
+*Project Name:* ${project.name?.toString() || "N/A"}
+*Project ID:* ${project.id?.toString() || "N/A"}
+*Refill Amount:* $${amountInUSD} USD
+*Current Balance:* $${currentBalanceInUSD} USD
+
+${project.createdOwnerName && project.createdOwnerEmail ? `*Project Created By:* ${project.createdOwnerName.toString()} (${project.createdOwnerEmail.toString()})` : ""}`;
+
+      SlackUtil.sendMessageToChannelViaIncomingWebhook({
+        url: URL.fromString(NotificationSlackWebhookOnSubscriptionUpdate),
+        text: slackMessage,
+      }).catch((error: Exception) => {
+        logger.error(
+          "Error sending slack message for AI balance refill: " + error,
+        );
+      });
+    }
+  }
+}
+
+export default new AIBillingService();