@oneuptime/common 9.4.12 → 9.5.0

package/Server/Services/ProjectService.ts

@@ -14,11 +14,13 @@ import { OnCreate, OnDelete, OnFind, OnUpdate } from "../Types/Database/Hooks";
 import QueryHelper from "../Types/Database/QueryHelper";
 import UpdateBy from "../Types/Database/UpdateBy";
 import logger from "../Utils/Logger";
+import Errors from "../Utils/Errors";
 import AccessTokenService from "./AccessTokenService";
 import BillingService from "./BillingService";
 import DatabaseService from "./DatabaseService";
 import IncidentSeverityService from "./IncidentSeverityService";
 import IncidentStateService from "./IncidentStateService";
+import IncidentRoleService from "./IncidentRoleService";
 import MailService from "./MailService";
 import MonitorStatusService from "./MonitorStatusService";
 import NotificationService from "./NotificationService";
@@ -36,9 +38,13 @@ import SubscriptionPlan, {
 import SubscriptionStatus from "../../Types/Billing/SubscriptionStatus";
 import {
   Black,
+  Blue500,
+  Gray500,
   Green,
   Moroon500,
+  Purple500,
   Red,
+  Teal500,
   Yellow,
   Yellow500,
 } from "../../Types/BrandColors";
@@ -48,10 +54,12 @@ import OneUptimeDate from "../../Types/Date";
 import EmailTemplateType from "../../Types/Email/EmailTemplateType";
 import BadDataException from "../../Types/Exception/BadDataException";
 import NotAuthorizedException from "../../Types/Exception/NotAuthorizedException";
+import IconProp from "../../Types/Icon/IconProp";
 import ObjectID from "../../Types/ObjectID";
 import Permission from "../../Types/Permission";
 import IncidentSeverity from "../../Models/DatabaseModels/IncidentSeverity";
 import IncidentState from "../../Models/DatabaseModels/IncidentState";
+import IncidentRole from "../../Models/DatabaseModels/IncidentRole";
 import MonitorStatus from "../../Models/DatabaseModels/MonitorStatus";
 import Model from "../../Models/DatabaseModels/Project";
 import PromoCode from "../../Models/DatabaseModels/PromoCode";
@@ -387,6 +395,7 @@ export class ProjectService extends DatabaseService<Model> {
         paymentProviderSubscriptionSeats: true,
         paymentProviderPlanId: true,
         trialEndsAt: true,
+        paymentProviderCustomerId: true,
       },
       props: {
         isRoot: true,
@@ -407,6 +416,19 @@ export class ProjectService extends DatabaseService<Model> {
       );
     }
 
+    // Check if customer has payment methods before attempting to change plan
+    if (!project.paymentProviderCustomerId) {
+      throw new BadDataException("Payment Provider customer not found");
+    }
+
+    const hasPaymentMethods: boolean = await BillingService.hasPaymentMethods(
+      project.paymentProviderCustomerId,
+    );
+
+    if (!hasPaymentMethods) {
+      throw new BadDataException(Errors.BillingService.NO_PAYMENTS_METHODS);
+    }
+
     const plan: SubscriptionPlan | undefined =
       SubscriptionPlan.getSubscriptionPlanById(
         params.paymentProviderPlanId,
@@ -689,6 +711,7 @@ export class ProjectService extends DatabaseService<Model> {
     createdItem = await this.addDefaultIncidentState(createdItem);
     createdItem = await this.addDefaultScheduledMaintenanceState(createdItem);
     createdItem = await this.addDefaultAlertState(createdItem);
+    createdItem = await this.addDefaultIncidentRoles(createdItem);
 
     if (NotificationSlackWebhookOnCreateProject) {
       // fetch project again.
@@ -928,6 +951,73 @@ export class ProjectService extends DatabaseService<Model> {
     return createdItem;
   }
 
+  public async addDefaultIncidentRoles(createdItem: Model): Promise<Model> {
+    let incidentCommander: IncidentRole = new IncidentRole();
+    incidentCommander.name = "Incident Commander";
+    incidentCommander.description =
+      "Primary decision maker during an incident. Responsible for coordinating the response and making final decisions.";
+    incidentCommander.color = Purple500;
+    incidentCommander.roleIcon = IconProp.ShieldCheck;
+    incidentCommander.projectId = createdItem.id!;
+    incidentCommander.isPrimaryRole = true;
+    incidentCommander.isDeleteable = false;
+
+    incidentCommander = await IncidentRoleService.create({
+      data: incidentCommander,
+      props: {
+        isRoot: true,
+      },
+    });
+
+    let responder: IncidentRole = new IncidentRole();
+    responder.name = "Responder";
+    responder.description =
+      "Active participant in incident resolution. Performs hands-on work to resolve the incident.";
+    responder.color = Blue500;
+    responder.roleIcon = IconProp.Wrench;
+    responder.projectId = createdItem.id!;
+
+    responder = await IncidentRoleService.create({
+      data: responder,
+      props: {
+        isRoot: true,
+      },
+    });
+
+    let communicationsLead: IncidentRole = new IncidentRole();
+    communicationsLead.name = "Communications Lead";
+    communicationsLead.description =
+      "Handles stakeholder communication and status updates during an incident.";
+    communicationsLead.color = Teal500;
+    communicationsLead.roleIcon = IconProp.Announcement;
+    communicationsLead.projectId = createdItem.id!;
+
+    communicationsLead = await IncidentRoleService.create({
+      data: communicationsLead,
+      props: {
+        isRoot: true,
+      },
+    });
+
+    let observer: IncidentRole = new IncidentRole();
+    observer.name = "Observer";
+    observer.description =
+      "Read-only participant who monitors the incident without active involvement.";
+    observer.color = Gray500;
+    observer.roleIcon = IconProp.Activity;
+    observer.projectId = createdItem.id!;
+    observer.canAssignMultipleUsers = true;
+
+    observer = await IncidentRoleService.create({
+      data: observer,
+      props: {
+        isRoot: true,
+      },
+    });
+
+    return createdItem;
+  }
+
   private async addDefaultMonitorStatus(createdItem: Model): Promise<Model> {
     let operationalStatus: MonitorStatus = new MonitorStatus();
     operationalStatus.name = "Operational";

package/Server/Services/ProjectUserProfileService.ts

@@ -0,0 +1,9 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/ProjectUserProfile";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+  }
+}
+export default new Service();

package/Server/Services/SmsService.ts

@@ -29,6 +29,7 @@ export class SmsService extends BaseService {
       incidentId?: ObjectID | undefined;
       alertId?: ObjectID | undefined;
       alertEpisodeId?: ObjectID | undefined;
+      incidentEpisodeId?: ObjectID | undefined;
       scheduledMaintenanceId?: ObjectID | undefined;
       statusPageId?: ObjectID | undefined;
       statusPageAnnouncementId?: ObjectID | undefined;

package/Server/Services/TeamMemberCustomFieldService.ts

@@ -0,0 +1,9 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/TeamMemberCustomField";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+  }
+}
+export default new Service();

package/Server/Services/UserNotificationRuleService.ts

@@ -56,6 +56,8 @@ import AlertEpisode from "../../Models/DatabaseModels/AlertEpisode";
 import AlertEpisodeService from "./AlertEpisodeService";
 import AlertEpisodeMember from "../../Models/DatabaseModels/AlertEpisodeMember";
 import AlertEpisodeMemberService from "./AlertEpisodeMemberService";
+import IncidentEpisode from "../../Models/DatabaseModels/IncidentEpisode";
+import IncidentEpisodeService from "./IncidentEpisodeService";
 import WorkspaceNotificationRule from "../../Models/DatabaseModels/WorkspaceNotificationRule";
 import WorkspaceNotificationRuleService from "./WorkspaceNotificationRuleService";
 import PushNotificationService from "./PushNotificationService";
@@ -79,6 +81,7 @@ export class Service extends DatabaseService<Model> {
       triggeredByIncidentId?: ObjectID | undefined;
       triggeredByAlertId?: ObjectID | undefined;
       triggeredByAlertEpisodeId?: ObjectID | undefined;
+      triggeredByIncidentEpisodeId?: ObjectID | undefined;
       userNotificationEventType: UserNotificationEventType;
       onCallPolicyExecutionLogId?: ObjectID | undefined;
       onCallPolicyId: ObjectID | undefined;
@@ -212,6 +215,11 @@ export class Service extends DatabaseService<Model> {
         options.triggeredByAlertEpisodeId;
     }
 
+    if (options.triggeredByIncidentEpisodeId) {
+      logTimelineItem.triggeredByIncidentEpisodeId =
+        options.triggeredByIncidentEpisodeId;
+    }
+
     if (options.onCallDutyPolicyExecutionLogTimelineId) {
       logTimelineItem.onCallDutyPolicyExecutionLogTimelineId =
         options.onCallDutyPolicyExecutionLogTimelineId;
@@ -312,9 +320,41 @@ export class Service extends DatabaseService<Model> {
       });
     }
 
-    if (!incident && !alert && !alertEpisode) {
+    let incidentEpisode: IncidentEpisode | null = null;
+
+    if (
+      options.userNotificationEventType ===
+        UserNotificationEventType.IncidentEpisodeCreated &&
+      options.triggeredByIncidentEpisodeId
+    ) {
+      incidentEpisode = await IncidentEpisodeService.findOneById({
+        id: options.triggeredByIncidentEpisodeId!,
+        props: {
+          isRoot: true,
+        },
+        select: {
+          _id: true,
+          title: true,
+          description: true,
+          projectId: true,
+          project: {
+            name: true,
+          },
+          currentIncidentState: {
+            name: true,
+          },
+          incidentSeverity: {
+            name: true,
+          },
+          episodeNumber: true,
+          rootCause: true,
+        },
+      });
+    }
+
+    if (!incident && !alert && !alertEpisode && !incidentEpisode) {
       throw new BadDataException(
-        "Incident, Alert, or Alert Episode not found.",
+        "Incident, Alert, Alert Episode, or Incident Episode not found.",
       );
     }
 
@@ -1969,6 +2009,7 @@ export class Service extends DatabaseService<Model> {
       triggeredByIncidentId?: ObjectID | undefined;
       triggeredByAlertId?: ObjectID | undefined;
       triggeredByAlertEpisodeId?: ObjectID | undefined;
+      triggeredByIncidentEpisodeId?: ObjectID | undefined;
       userNotificationEventType: UserNotificationEventType;
       onCallPolicyExecutionLogId?: ObjectID | undefined;
       onCallPolicyId: ObjectID | undefined;
@@ -1998,6 +2039,11 @@ export class Service extends DatabaseService<Model> {
         options.triggeredByAlertEpisodeId;
     }
 
+    if (options.triggeredByIncidentEpisodeId) {
+      userOnCallLog.triggeredByIncidentEpisodeId =
+        options.triggeredByIncidentEpisodeId;
+    }
+
     userOnCallLog.userNotificationEventType = options.userNotificationEventType;
 
     if (options.onCallPolicyExecutionLogId) {

package/Server/Services/UserNotificationSettingService.ts

@@ -50,6 +50,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
     incidentId?: ObjectID | undefined;
     alertId?: ObjectID | undefined;
     alertEpisodeId?: ObjectID | undefined;
+    incidentEpisodeId?: ObjectID | undefined;
     scheduledMaintenanceId?: ObjectID | undefined;
     statusPageId?: ObjectID | undefined;
     statusPageAnnouncementId?: ObjectID | undefined;
@@ -115,6 +116,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
               incidentId: data.incidentId,
               alertId: data.alertId,
               alertEpisodeId: data.alertEpisodeId,
+              incidentEpisodeId: data.incidentEpisodeId,
               scheduledMaintenanceId: data.scheduledMaintenanceId,
               statusPageId: data.statusPageId,
               statusPageAnnouncementId: data.statusPageAnnouncementId,
@@ -161,6 +163,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
               incidentId: data.incidentId,
               alertId: data.alertId,
               alertEpisodeId: data.alertEpisodeId,
+              incidentEpisodeId: data.incidentEpisodeId,
               scheduledMaintenanceId: data.scheduledMaintenanceId,
               statusPageId: data.statusPageId,
               statusPageAnnouncementId: data.statusPageAnnouncementId,
@@ -214,6 +217,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
               incidentId: data.incidentId,
               alertId: data.alertId,
               alertEpisodeId: data.alertEpisodeId,
+              incidentEpisodeId: data.incidentEpisodeId,
               scheduledMaintenanceId: data.scheduledMaintenanceId,
               statusPageId: data.statusPageId,
               statusPageAnnouncementId: data.statusPageAnnouncementId,
@@ -259,6 +263,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
               incidentId: data.incidentId,
               alertId: data.alertId,
               alertEpisodeId: data.alertEpisodeId,
+              incidentEpisodeId: data.incidentEpisodeId,
               scheduledMaintenanceId: data.scheduledMaintenanceId,
               statusPageId: data.statusPageId,
               statusPageAnnouncementId: data.statusPageAnnouncementId,
@@ -346,6 +351,7 @@ export class Service extends DatabaseService<UserNotificationSetting> {
     await this.addOnCallNotificationSettings(userId, projectId);
     await this.addAlertNotificationSettings(userId, projectId);
     await this.addAlertEpisodeNotificationSettings(userId, projectId);
+    await this.addIncidentEpisodeNotificationSettings(userId, projectId);
   }
 
   private async addProbeOwnerNotificationSettings(
@@ -474,6 +480,23 @@ export class Service extends DatabaseService<UserNotificationSetting> {
     );
   }
 
+  private async addIncidentEpisodeNotificationSettings(
+    userId: ObjectID,
+    projectId: ObjectID,
+  ): Promise<void> {
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_INCIDENT_EPISODE_CREATED_OWNER_NOTIFICATION,
+    );
+
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_INCIDENT_EPISODE_STATE_CHANGED_OWNER_NOTIFICATION,
+    );
+  }
+
   private async addNotificationSettingIfNotExists(
     userId: ObjectID,
     projectId: ObjectID,

package/Server/Services/UserOnCallLogService.ts

@@ -21,6 +21,8 @@ import Alert from "../../Models/DatabaseModels/Alert";
 import AlertService from "./AlertService";
 import AlertEpisode from "../../Models/DatabaseModels/AlertEpisode";
 import AlertEpisodeService from "./AlertEpisodeService";
+import IncidentEpisode from "../../Models/DatabaseModels/IncidentEpisode";
+import IncidentEpisodeService from "./IncidentEpisodeService";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
@@ -206,6 +208,34 @@ export class Service extends DatabaseService<Model> {
       });
     }
 
+    // get rule count for incident episodes.
+    let incidentEpisode: IncidentEpisode | null = null;
+    if (createdItem.triggeredByIncidentEpisodeId) {
+      incidentEpisode = await IncidentEpisodeService.findOneById({
+        id: createdItem.triggeredByIncidentEpisodeId,
+        props: {
+          isRoot: true,
+        },
+        select: {
+          incidentSeverityId: true,
+        },
+      });
+
+      ruleCount = await UserNotificationRuleService.countBy({
+        query: {
+          userId: createdItem.userId!,
+          projectId: createdItem.projectId!,
+          ruleType: notificationRuleType,
+          incidentSeverityId: incidentEpisode?.incidentSeverityId as ObjectID,
+        },
+        skip: 0,
+        limit: LIMIT_PER_PROJECT,
+        props: {
+          isRoot: true,
+        },
+      });
+    }
+
     if (ruleCount.toNumber() === 0) {
       // update this item to be processed.
       await this.updateOneById({
@@ -239,6 +269,7 @@ export class Service extends DatabaseService<Model> {
     /*
      * find immediate notification rule and alert the user.
      * Determine the alertSeverityId - can come from alert or alertEpisode
+     * Determine the incidentSeverityId - can come from incident or incidentEpisode
      */
     const alertSeverityIdForQuery: ObjectID | undefined =
       alert && alert.alertSeverityId
@@ -247,6 +278,13 @@ export class Service extends DatabaseService<Model> {
           ? (alertEpisode.alertSeverityId as ObjectID)
           : undefined;
 
+    const incidentSeverityIdForQuery: ObjectID | undefined =
+      incident && incident.incidentSeverityId
+        ? (incident.incidentSeverityId as ObjectID)
+        : incidentEpisode && incidentEpisode.incidentSeverityId
+          ? (incidentEpisode.incidentSeverityId as ObjectID)
+          : undefined;
+
     const immediateNotificationRule: Array<UserNotificationRule> =
       await UserNotificationRuleService.findBy({
         query: {
@@ -254,10 +292,7 @@ export class Service extends DatabaseService<Model> {
           projectId: createdItem.projectId!,
           notifyAfterMinutes: 0,
           ruleType: notificationRuleType,
-          incidentSeverityId:
-            incident && incident.incidentSeverityId
-              ? (incident?.incidentSeverityId as ObjectID)
-              : undefined,
+          incidentSeverityId: incidentSeverityIdForQuery,
           alertSeverityId: alertSeverityIdForQuery,
         },
         select: {
@@ -279,6 +314,8 @@ export class Service extends DatabaseService<Model> {
           triggeredByIncidentId: createdItem.triggeredByIncidentId,
           triggeredByAlertId: createdItem.triggeredByAlertId,
           triggeredByAlertEpisodeId: createdItem.triggeredByAlertEpisodeId,
+          triggeredByIncidentEpisodeId:
+            createdItem.triggeredByIncidentEpisodeId,
           userNotificationEventType: createdItem.userNotificationEventType!,
           onCallPolicyExecutionLogId:
             createdItem.onCallDutyPolicyExecutionLogId,
@@ -339,6 +376,13 @@ export class Service extends DatabaseService<Model> {
       return NotificationRuleType.ON_CALL_EXECUTED_ALERT_EPISODE;
     }
 
+    if (
+      userNotificationEventType ===
+      UserNotificationEventType.IncidentEpisodeCreated
+    ) {
+      return NotificationRuleType.ON_CALL_EXECUTED_INCIDENT_EPISODE;
+    }
+
     // Invalid user notification event type.
     throw new BadDataException("Invalid user notification event type.");
   }

package/Server/Services/WhatsAppService.ts

@@ -27,6 +27,7 @@ export class WhatsAppService extends BaseService {
       incidentId?: ObjectID | undefined;
       alertId?: ObjectID | undefined;
       alertEpisodeId?: ObjectID | undefined;
+      incidentEpisodeId?: ObjectID | undefined;
       scheduledMaintenanceId?: ObjectID | undefined;
       statusPageId?: ObjectID | undefined;
       statusPageAnnouncementId?: ObjectID | undefined;

package/Server/Services/WorkspaceNotificationLogService.ts

@@ -21,6 +21,8 @@ export interface WorkspaceLogData {
   // Relations to resources (optional)
   incidentId?: ObjectID | undefined;
   alertId?: ObjectID | undefined;
+  alertEpisodeId?: ObjectID | undefined;
+  incidentEpisodeId?: ObjectID | undefined;
   scheduledMaintenanceId?: ObjectID | undefined;
   userId?: ObjectID | undefined;
   teamId?: ObjectID | undefined;
@@ -75,6 +77,12 @@ export class Service extends DatabaseService<Model> {
     if (data.alertId) {
       log.alertId = data.alertId;
     }
+    if (data.alertEpisodeId) {
+      log.alertEpisodeId = data.alertEpisodeId;
+    }
+    if (data.incidentEpisodeId) {
+      log.incidentEpisodeId = data.incidentEpisodeId;
+    }
     if (data.scheduledMaintenanceId) {
       log.scheduledMaintenanceId = data.scheduledMaintenanceId;
     }
@@ -221,6 +229,8 @@ export class Service extends DatabaseService<Model> {
       // Optional resource associations
       incidentId?: ObjectID;
       alertId?: ObjectID;
+      alertEpisodeId?: ObjectID;
+      incidentEpisodeId?: ObjectID;
       scheduledMaintenanceId?: ObjectID;
       onCallDutyPolicyId?: ObjectID;
       statusPageId?: ObjectID;
@@ -256,6 +266,12 @@ export class Service extends DatabaseService<Model> {
     if (data.alertId) {
       logData.alertId = data.alertId;
     }
+    if (data.alertEpisodeId) {
+      logData.alertEpisodeId = data.alertEpisodeId;
+    }
+    if (data.incidentEpisodeId) {
+      logData.incidentEpisodeId = data.incidentEpisodeId;
+    }
     if (data.scheduledMaintenanceId) {
       logData.scheduledMaintenanceId = data.scheduledMaintenanceId;
     }

package/Server/Services/WorkspaceNotificationRuleService.ts

@@ -50,6 +50,8 @@ import WorkspaceNotificationLogService from "./WorkspaceNotificationLogService";
 import WorkspaceNotificationStatus from "../../Types/Workspace/WorkspaceNotificationStatus";
 import WorkspaceNotificationActionType from "../../Types/Workspace/WorkspaceNotificationActionType";
 import ExceptionMessages from "../../Types/Exception/ExceptionMessages";
+import IncidentEpisode from "../../Models/DatabaseModels/IncidentEpisode";
+import IncidentEpisodeService from "./IncidentEpisodeService";
 
 export interface MessageBlocksByWorkspaceType {
   workspaceType: WorkspaceType;
@@ -60,6 +62,7 @@ export interface NotificationFor {
   incidentId?: ObjectID | undefined;
   alertId?: ObjectID | undefined;
   alertEpisodeId?: ObjectID | undefined;
+  incidentEpisodeId?: ObjectID | undefined;
   scheduledMaintenanceId?: ObjectID | undefined;
   monitorId?: ObjectID | undefined;
   onCallDutyPolicyId?: ObjectID | undefined;
@@ -749,6 +752,14 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
       return NotificationRuleEventType.Incident;
     }
 
+    if (notificationFor.alertEpisodeId) {
+      return NotificationRuleEventType.AlertEpisode;
+    }
+
+    if (notificationFor.incidentEpisodeId) {
+      return NotificationRuleEventType.IncidentEpisode;
+    }
+
     if (notificationFor.monitorId) {
       return NotificationRuleEventType.Monitor;
     }
@@ -1949,6 +1960,12 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
         [NotificationRuleConditionCheckOn.AlertEpisodeSeverity]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeState]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeLabels]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeTitle]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeDescription]:
+          undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeState]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeLabels]: undefined,
       };
     }
 
@@ -2031,6 +2048,12 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
         [NotificationRuleConditionCheckOn.AlertEpisodeSeverity]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeState]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeLabels]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeTitle]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeDescription]:
+          undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeState]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeLabels]: undefined,
       };
     }
 
@@ -2119,6 +2142,12 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
         [NotificationRuleConditionCheckOn.AlertEpisodeSeverity]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeState]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeLabels]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeTitle]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeDescription]:
+          undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeState]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeLabels]: undefined,
       };
     }
 
@@ -2185,6 +2214,12 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
         [NotificationRuleConditionCheckOn.AlertEpisodeSeverity]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeState]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeLabels]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeTitle]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeDescription]:
+          undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeState]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeLabels]: undefined,
       };
     }
 
@@ -2250,6 +2285,87 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
         [NotificationRuleConditionCheckOn.AlertEpisodeSeverity]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeState]: undefined,
         [NotificationRuleConditionCheckOn.AlertEpisodeLabels]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeTitle]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeDescription]:
+          undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeState]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeLabels]: undefined,
+      };
+    }
+
+    // Handle Incident Episode
+    if (data.notificationFor.incidentEpisodeId) {
+      logger.debug("Fetching incident episode details for ID:");
+      logger.debug(data.notificationFor.incidentEpisodeId);
+
+      const incidentEpisode: IncidentEpisode | null =
+        await IncidentEpisodeService.findOneById({
+          id: data.notificationFor.incidentEpisodeId,
+          select: {
+            title: true,
+            description: true,
+            incidentSeverity: true,
+            currentIncidentState: true,
+            labels: true,
+          },
+          props: {
+            isRoot: true,
+          },
+        });
+
+      if (!incidentEpisode) {
+        logger.debug("Incident Episode not found for ID:");
+        logger.debug(data.notificationFor.incidentEpisodeId);
+        throw new BadDataException("Incident Episode ID not found");
+      }
+
+      logger.debug("Incident Episode details retrieved:");
+      logger.debug(incidentEpisode);
+
+      return {
+        [NotificationRuleConditionCheckOn.MonitorName]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentTitle]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentDescription]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentState]: undefined,
+        [NotificationRuleConditionCheckOn.MonitorType]: undefined,
+        [NotificationRuleConditionCheckOn.MonitorStatus]: undefined,
+        [NotificationRuleConditionCheckOn.AlertTitle]: undefined,
+        [NotificationRuleConditionCheckOn.AlertDescription]: undefined,
+        [NotificationRuleConditionCheckOn.AlertSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.AlertState]: undefined,
+        [NotificationRuleConditionCheckOn.ScheduledMaintenanceTitle]: undefined,
+        [NotificationRuleConditionCheckOn.ScheduledMaintenanceDescription]:
+          undefined,
+        [NotificationRuleConditionCheckOn.ScheduledMaintenanceState]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentLabels]: undefined,
+        [NotificationRuleConditionCheckOn.AlertLabels]: undefined,
+        [NotificationRuleConditionCheckOn.MonitorLabels]: undefined,
+        [NotificationRuleConditionCheckOn.ScheduledMaintenanceLabels]:
+          undefined,
+        [NotificationRuleConditionCheckOn.Monitors]: undefined,
+        [NotificationRuleConditionCheckOn.OnCallDutyPolicyName]: undefined,
+        [NotificationRuleConditionCheckOn.OnCallDutyPolicyDescription]:
+          undefined,
+        [NotificationRuleConditionCheckOn.OnCallDutyPolicyLabels]: undefined,
+        [NotificationRuleConditionCheckOn.AlertEpisodeTitle]: undefined,
+        [NotificationRuleConditionCheckOn.AlertEpisodeDescription]: undefined,
+        [NotificationRuleConditionCheckOn.AlertEpisodeSeverity]: undefined,
+        [NotificationRuleConditionCheckOn.AlertEpisodeState]: undefined,
+        [NotificationRuleConditionCheckOn.AlertEpisodeLabels]: undefined,
+        [NotificationRuleConditionCheckOn.IncidentEpisodeTitle]:
+          incidentEpisode.title || "",
+        [NotificationRuleConditionCheckOn.IncidentEpisodeDescription]:
+          incidentEpisode.description || "",
+        [NotificationRuleConditionCheckOn.IncidentEpisodeSeverity]:
+          incidentEpisode.incidentSeverity?._id?.toString() || "",
+        [NotificationRuleConditionCheckOn.IncidentEpisodeState]:
+          incidentEpisode.currentIncidentState?._id?.toString() || "",
+        [NotificationRuleConditionCheckOn.IncidentEpisodeLabels]:
+          incidentEpisode.labels?.map((label: Label) => {
+            return label._id?.toString() || "";
+          }) || [],
       };
     }