@oneuptime/common 11.3.21 → 11.3.22

package/Server/Services/AlertReminderRuleService.ts

@@ -0,0 +1,253 @@
+import CreateBy from "../Types/Database/CreateBy";
+import { OnCreate, OnDelete, OnUpdate } from "../Types/Database/Hooks";
+import SortOrder from "../../Types/BaseDatabase/SortOrder";
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/AlertReminderRule";
+import Alert from "../../Models/DatabaseModels/Alert";
+import AlertSeverity from "../../Models/DatabaseModels/AlertSeverity";
+import AlertState from "../../Models/DatabaseModels/AlertState";
+import Label from "../../Models/DatabaseModels/Label";
+import AlertService from "./AlertService";
+import AlertStateService from "./AlertStateService";
+import QueryHelper from "../Types/Database/QueryHelper";
+import LIMIT_MAX, { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
+import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import logger, { LogAttributes } from "../Utils/Logger";
+import { IsBillingEnabled } from "../EnvironmentConfig";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+    if (IsBillingEnabled) {
+      this.hardDeleteItemsOlderThanInDays("createdAt", 3 * 365); // 3 years
+    }
+  }
+
+  @CaptureSpan()
+  protected override async onBeforeCreate(
+    createBy: CreateBy<Model>,
+  ): Promise<OnCreate<Model>> {
+    // Auto-assign order on creation
+    if (!createBy.data.order) {
+      const highestOrderRule: Model | null = await this.findOneBy({
+        query: {
+          projectId: createBy.data.projectId!,
+        },
+        select: {
+          order: true,
+        },
+        sort: {
+          order: SortOrder.Descending,
+        },
+        props: {
+          isRoot: true,
+        },
+      });
+
+      createBy.data.order = (highestOrderRule?.order || 0) + 1;
+    }
+
+    return {
+      createBy,
+      carryForward: null,
+    };
+  }
+
+  @CaptureSpan()
+  protected override async onCreateSuccess(
+    _onCreate: OnCreate<Model>,
+    createdItem: Model,
+  ): Promise<Model> {
+    if (createdItem.projectId) {
+      await this.refreshSchedulesForOpenAlerts(createdItem.projectId);
+    }
+
+    return createdItem;
+  }
+
+  @CaptureSpan()
+  protected override async onUpdateSuccess(
+    onUpdate: OnUpdate<Model>,
+    _updatedItemIds: Array<ObjectID>,
+  ): Promise<OnUpdate<Model>> {
+    const projectId: ObjectID | undefined = onUpdate.updateBy.props.tenantId as
+      | ObjectID
+      | undefined;
+
+    if (projectId) {
+      await this.refreshSchedulesForOpenAlerts(projectId);
+    }
+
+    return onUpdate;
+  }
+
+  @CaptureSpan()
+  protected override async onDeleteSuccess(
+    onDelete: OnDelete<Model>,
+    _itemIdsBeforeDelete: Array<ObjectID>,
+  ): Promise<OnDelete<Model>> {
+    const projectId: ObjectID | undefined = onDelete.deleteBy.props.tenantId as
+      | ObjectID
+      | undefined;
+
+    if (projectId) {
+      await this.refreshSchedulesForOpenAlerts(projectId);
+    }
+
+    return onDelete;
+  }
+
+  @CaptureSpan()
+  public async refreshSchedulesForOpenAlerts(
+    projectId: ObjectID,
+  ): Promise<void> {
+    try {
+      const unresolvedStates: Array<AlertState> =
+        await AlertStateService.findBy({
+          query: {
+            projectId: projectId,
+            isResolvedState: false,
+          },
+          select: {
+            _id: true,
+          },
+          limit: LIMIT_PER_PROJECT,
+          skip: 0,
+          props: {
+            isRoot: true,
+          },
+        });
+
+      const unresolvedStateIds: Array<ObjectID> = unresolvedStates
+        .map((state: AlertState) => {
+          return state.id!;
+        })
+        .filter(Boolean);
+
+      if (unresolvedStateIds.length === 0) {
+        return;
+      }
+
+      const openAlerts: Array<Alert> = await AlertService.findBy({
+        query: {
+          projectId: projectId,
+          currentAlertStateId: QueryHelper.any(unresolvedStateIds),
+        },
+        select: {
+          _id: true,
+        },
+        limit: LIMIT_MAX,
+        skip: 0,
+        props: {
+          isRoot: true,
+        },
+      });
+
+      for (const alert of openAlerts) {
+        try {
+          await AlertService.refreshReminderSchedule({
+            alertId: alert.id!,
+            projectId: projectId,
+          });
+        } catch (error) {
+          logger.error(
+            `Failed to refresh reminder schedule for alert ${alert.id}: ${error}`,
+            { projectId: projectId?.toString() } as LogAttributes,
+          );
+        }
+      }
+    } catch (error) {
+      logger.error(
+        `Failed to refresh reminder schedules for open alerts: ${error}`,
+        { projectId: projectId?.toString() } as LogAttributes,
+      );
+    }
+  }
+
+  @CaptureSpan()
+  public async findMatchingRule(data: {
+    projectId: ObjectID;
+    alertSeverityId?: ObjectID | undefined;
+    labelIds?: Array<ObjectID> | undefined;
+  }): Promise<Model | null> {
+    // Get all enabled rules sorted by order. First matching rule wins.
+    const rules: Array<Model> = await this.findBy({
+      query: {
+        projectId: data.projectId,
+        isEnabled: true,
+      },
+      sort: {
+        order: SortOrder.Ascending,
+      },
+      select: {
+        _id: true,
+        name: true,
+        order: true,
+        reminderIntervalInMinutes: true,
+        stopRemindersOnState: true,
+        alertSeverities: {
+          _id: true,
+        },
+        labels: {
+          _id: true,
+        },
+      },
+      limit: 100,
+      skip: 0,
+      props: {
+        isRoot: true,
+      },
+    });
+
+    for (const rule of rules) {
+      if (rule.alertSeverities && rule.alertSeverities.length > 0) {
+        if (!data.alertSeverityId) {
+          continue;
+        }
+
+        const severityIds: Array<string> = rule.alertSeverities.map(
+          (severity: AlertSeverity) => {
+            return severity.id?.toString() || "";
+          },
+        );
+
+        if (!severityIds.includes(data.alertSeverityId.toString())) {
+          continue;
+        }
+      }
+
+      if (rule.labels && rule.labels.length > 0) {
+        if (!data.labelIds || data.labelIds.length === 0) {
+          continue;
+        }
+
+        const ruleLabelIds: Array<string> = rule.labels.map((label: Label) => {
+          return label.id?.toString() || "";
+        });
+
+        const hasMatchingLabel: boolean = data.labelIds.some(
+          (labelId: ObjectID) => {
+            return ruleLabelIds.includes(labelId.toString());
+          },
+        );
+
+        if (!hasMatchingLabel) {
+          continue;
+        }
+      }
+
+      // Rule with no severities and no labels matches all alerts.
+      logger.debug(
+        `Alert reminder rule ${rule.name || rule.id} matched for project ${data.projectId}`,
+        { projectId: data.projectId?.toString() } as LogAttributes,
+      );
+
+      return rule;
+    }
+
+    return null;
+  }
+}
+
+export default new Service();

package/Server/Services/AlertService.ts

@@ -50,6 +50,8 @@ import Label from "../../Models/DatabaseModels/Label";
 import LabelService from "./LabelService";
 import AlertSeverity from "../../Models/DatabaseModels/AlertSeverity";
 import AlertSeverityService from "./AlertSeverityService";
+import AlertReminderRuleService from "./AlertReminderRuleService";
+import AlertReminderRule from "../../Models/DatabaseModels/AlertReminderRule";
 import WorkspaceType from "../../Types/Workspace/WorkspaceType";
 import NotificationRuleWorkspaceChannel from "../../Types/Workspace/NotificationRules/NotificationRuleWorkspaceChannel";
 import AlertWorkspaceMessages from "../Utils/Workspace/WorkspaceMessages/Alert";
@@ -121,6 +123,64 @@ export class Service extends DatabaseService<Model> {
     return false;
   }
 
+  @CaptureSpan()
+  public async refreshReminderSchedule(data: {
+    alertId: ObjectID;
+    projectId: ObjectID;
+  }): Promise<void> {
+    const alert: Model | null = await this.findOneById({
+      id: data.alertId,
+      select: {
+        enableReminders: true,
+        alertSeverityId: true,
+        labels: {
+          _id: true,
+        },
+      },
+      props: {
+        isRoot: true,
+      },
+    });
+
+    if (!alert) {
+      return;
+    }
+
+    let nextReminderNotificationAt: Date | null = null;
+
+    if (alert.enableReminders !== false) {
+      const matchingRule: AlertReminderRule | null =
+        await AlertReminderRuleService.findMatchingRule({
+          projectId: data.projectId,
+          alertSeverityId: alert.alertSeverityId,
+          labelIds: alert.labels?.map((label: Label) => {
+            return label.id!;
+          }),
+        });
+
+      if (
+        matchingRule &&
+        matchingRule.reminderIntervalInMinutes &&
+        !(await this.isAlertResolved({ alertId: data.alertId }))
+      ) {
+        nextReminderNotificationAt = OneUptimeDate.addRemoveMinutes(
+          OneUptimeDate.getCurrentDate(),
+          matchingRule.reminderIntervalInMinutes,
+        );
+      }
+    }
+
+    await this.updateOneById({
+      id: data.alertId,
+      data: {
+        nextReminderNotificationAt: nextReminderNotificationAt,
+      },
+      props: {
+        isRoot: true,
+      },
+    });
+  }
+
   @CaptureSpan()
   public async acknowledgeAlert(
     alertId: ObjectID,
@@ -479,6 +539,25 @@ export class Service extends DatabaseService<Model> {
           return Promise.resolve();
         }
       })
+      .then(async () => {
+        // Schedule reminder notifications for alert if a matching reminder rule exists
+        try {
+          if (createdItem.projectId && createdItem.id) {
+            await this.refreshReminderSchedule({
+              alertId: createdItem.id,
+              projectId: createdItem.projectId,
+            });
+          }
+        } catch (error) {
+          logger.error(
+            `Reminder scheduling failed in AlertService.onCreateSuccess: ${error}`,
+            {
+              projectId: createdItem.projectId?.toString(),
+              alertId: createdItem.id?.toString(),
+            } as LogAttributes,
+          );
+        }
+      })
       .catch((error: Error) => {
         logger.error(
           `Critical error in AlertService sequential operations: ${error}`,
@@ -1093,6 +1172,33 @@ ${alertSeverity.name}
           }
         }
 
+        // Re-evaluate reminder schedule when severity or labels change or reminders are toggled
+        if (
+          (onUpdate.updateBy.data.alertSeverity &&
+            (onUpdate.updateBy.data.alertSeverity as any)._id) ||
+          (onUpdate.updateBy.data.labels &&
+            Array.isArray(onUpdate.updateBy.data.labels)) ||
+          Object.prototype.hasOwnProperty.call(
+            onUpdate.updateBy.data,
+            "enableReminders",
+          )
+        ) {
+          try {
+            await this.refreshReminderSchedule({
+              alertId: alertId,
+              projectId: projectId,
+            });
+          } catch (reminderError) {
+            logger.error(
+              `Reminder rescheduling failed in AlertService.onUpdateSuccess: ${reminderError}`,
+              {
+                projectId: projectId?.toString(),
+                alertId: alertId?.toString(),
+              } as LogAttributes,
+            );
+          }
+        }
+
         if (shouldAddAlertFeed) {
           await AlertFeedService.createAlertFeedItem({
             alertId: alertId,

package/Server/Services/IncidentReminderRuleService.ts

@@ -0,0 +1,256 @@
+import CreateBy from "../Types/Database/CreateBy";
+import { OnCreate, OnDelete, OnUpdate } from "../Types/Database/Hooks";
+import SortOrder from "../../Types/BaseDatabase/SortOrder";
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/IncidentReminderRule";
+import Incident from "../../Models/DatabaseModels/Incident";
+import IncidentSeverity from "../../Models/DatabaseModels/IncidentSeverity";
+import Label from "../../Models/DatabaseModels/Label";
+import IncidentState from "../../Models/DatabaseModels/IncidentState";
+import IncidentService from "./IncidentService";
+import IncidentStateService from "./IncidentStateService";
+import QueryHelper from "../Types/Database/QueryHelper";
+import LIMIT_MAX, { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
+import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import logger, { LogAttributes } from "../Utils/Logger";
+import { IsBillingEnabled } from "../EnvironmentConfig";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+    if (IsBillingEnabled) {
+      this.hardDeleteItemsOlderThanInDays("createdAt", 3 * 365); // 3 years
+    }
+  }
+
+  @CaptureSpan()
+  protected override async onBeforeCreate(
+    createBy: CreateBy<Model>,
+  ): Promise<OnCreate<Model>> {
+    // Auto-assign order on creation
+    if (!createBy.data.order) {
+      const highestOrderRule: Model | null = await this.findOneBy({
+        query: {
+          projectId: createBy.data.projectId!,
+        },
+        select: {
+          order: true,
+        },
+        sort: {
+          order: SortOrder.Descending,
+        },
+        props: {
+          isRoot: true,
+        },
+      });
+
+      createBy.data.order = (highestOrderRule?.order || 0) + 1;
+    }
+
+    return {
+      createBy,
+      carryForward: null,
+    };
+  }
+
+  @CaptureSpan()
+  protected override async onCreateSuccess(
+    _onCreate: OnCreate<Model>,
+    createdItem: Model,
+  ): Promise<Model> {
+    if (createdItem.projectId) {
+      await this.refreshSchedulesForOpenIncidents(createdItem.projectId);
+    }
+
+    return createdItem;
+  }
+
+  @CaptureSpan()
+  protected override async onUpdateSuccess(
+    onUpdate: OnUpdate<Model>,
+    _updatedItemIds: Array<ObjectID>,
+  ): Promise<OnUpdate<Model>> {
+    const projectId: ObjectID | undefined = onUpdate.updateBy.props.tenantId as
+      | ObjectID
+      | undefined;
+
+    if (projectId) {
+      await this.refreshSchedulesForOpenIncidents(projectId);
+    }
+
+    return onUpdate;
+  }
+
+  @CaptureSpan()
+  protected override async onDeleteSuccess(
+    onDelete: OnDelete<Model>,
+    _itemIdsBeforeDelete: Array<ObjectID>,
+  ): Promise<OnDelete<Model>> {
+    const projectId: ObjectID | undefined = onDelete.deleteBy.props.tenantId as
+      | ObjectID
+      | undefined;
+
+    if (projectId) {
+      await this.refreshSchedulesForOpenIncidents(projectId);
+    }
+
+    return onDelete;
+  }
+
+  @CaptureSpan()
+  public async refreshSchedulesForOpenIncidents(
+    projectId: ObjectID,
+  ): Promise<void> {
+    try {
+      const unresolvedStates: Array<IncidentState> =
+        await IncidentStateService.findBy({
+          query: {
+            projectId: projectId,
+            isResolvedState: false,
+          },
+          select: {
+            _id: true,
+          },
+          limit: LIMIT_PER_PROJECT,
+          skip: 0,
+          props: {
+            isRoot: true,
+          },
+        });
+
+      const unresolvedStateIds: Array<ObjectID> = unresolvedStates
+        .map((state: IncidentState) => {
+          return state.id!;
+        })
+        .filter(Boolean);
+
+      if (unresolvedStateIds.length === 0) {
+        return;
+      }
+
+      const openIncidents: Array<Incident> = await IncidentService.findBy({
+        query: {
+          projectId: projectId,
+          currentIncidentStateId: QueryHelper.any(unresolvedStateIds),
+        },
+        select: {
+          _id: true,
+        },
+        limit: LIMIT_MAX,
+        skip: 0,
+        props: {
+          isRoot: true,
+        },
+      });
+
+      for (const incident of openIncidents) {
+        try {
+          await IncidentService.refreshReminderSchedule({
+            incidentId: incident.id!,
+            projectId: projectId,
+          });
+        } catch (error) {
+          logger.error(
+            `Failed to refresh reminder schedule for incident ${incident.id}: ${error}`,
+            { projectId: projectId?.toString() } as LogAttributes,
+          );
+        }
+      }
+    } catch (error) {
+      logger.error(
+        `Failed to refresh reminder schedules for open incidents: ${error}`,
+        { projectId: projectId?.toString() } as LogAttributes,
+      );
+    }
+  }
+
+  @CaptureSpan()
+  public async findMatchingRule(data: {
+    projectId: ObjectID;
+    incidentSeverityId?: ObjectID | undefined;
+    labelIds?: Array<ObjectID> | undefined;
+  }): Promise<Model | null> {
+    // Get all enabled rules sorted by order. First matching rule wins.
+    const rules: Array<Model> = await this.findBy({
+      query: {
+        projectId: data.projectId,
+        isEnabled: true,
+      },
+      sort: {
+        order: SortOrder.Ascending,
+      },
+      select: {
+        _id: true,
+        name: true,
+        order: true,
+        reminderIntervalInMinutes: true,
+        stopRemindersOnState: true,
+        incidentSeverities: {
+          _id: true,
+        },
+        labels: {
+          _id: true,
+        },
+      },
+      limit: 100,
+      skip: 0,
+      props: {
+        isRoot: true,
+      },
+    });
+
+    for (const rule of rules) {
+      if (rule.incidentSeverities && rule.incidentSeverities.length > 0) {
+        if (!data.incidentSeverityId) {
+          continue;
+        }
+
+        const severityIds: Array<string> = rule.incidentSeverities.map(
+          (severity: IncidentSeverity) => {
+            return severity.id?.toString() || "";
+          },
+        );
+
+        if (!severityIds.includes(data.incidentSeverityId.toString())) {
+          continue;
+        }
+      }
+
+      if (rule.labels && rule.labels.length > 0) {
+        if (!data.labelIds || data.labelIds.length === 0) {
+          continue;
+        }
+
+        const ruleLabelIds: Array<string> = rule.labels.map((label: Label) => {
+          return label.id?.toString() || "";
+        });
+
+        const hasMatchingLabel: boolean = data.labelIds.some(
+          (labelId: ObjectID) => {
+            return ruleLabelIds.includes(labelId.toString());
+          },
+        );
+
+        if (!hasMatchingLabel) {
+          continue;
+        }
+      }
+
+      /*
+       * Rule matched: (no severities OR severity matches) AND
+       * (no labels OR labels intersect). A rule with neither matches all.
+       */
+      logger.debug(
+        `Incident reminder rule ${rule.name || rule.id} matched for project ${data.projectId}`,
+        { projectId: data.projectId?.toString() } as LogAttributes,
+      );
+
+      return rule;
+    }
+
+    return null;
+  }
+}
+
+export default new Service();