@oneuptime/common 7.0.4006 → 7.0.4019

package/Server/Services/OnCallDutyPolicyService.ts

@@ -9,6 +9,18 @@ import OnCallDutyPolicyExecutionLog from "Common/Models/DatabaseModels/OnCallDut
 import DatabaseConfig from "../DatabaseConfig";
 import URL from "../../Types/API/URL";
 import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import OnCallDutyPolicySchedule from "../../Models/DatabaseModels/OnCallDutyPolicySchedule";
+import OnCallDutyPolicyScheduleService from "./OnCallDutyPolicyScheduleService";
+import TeamService from "./TeamService";
+import Team from "../../Models/DatabaseModels/Team";
+import OnCallDutyPolicyEscalationRuleUser from "../../Models/DatabaseModels/OnCallDutyPolicyEscalationRuleUser";
+import OnCallDutyPolicyEscalationRuleUserService from "./OnCallDutyPolicyEscalationRuleUserService";
+import { LIMIT_PER_PROJECT } from "../../Types/Database/LimitMax";
+import OnCallDutyPolicyEscalationRuleTeam from "../../Models/DatabaseModels/OnCallDutyPolicyEscalationRuleTeam";
+import OnCallDutyPolicyEscalationRuleTeamService from "./OnCallDutyPolicyEscalationRuleTeamService";
+import QueryHelper from "../Types/Database/QueryHelper";
+import OnCallDutyPolicyEscalationRuleSchedule from "../../Models/DatabaseModels/OnCallDutyPolicyEscalationRuleSchedule";
+import OnCallDutyPolicyEscalationRuleScheduleService from "./OnCallDutyPolicyEscalationRuleScheduleService";
 
 export class Service extends DatabaseService<OnCallDutyPolicy> {
   public constructor() {
@@ -100,5 +112,122 @@ export class Service extends DatabaseService<OnCallDutyPolicy> {
       },
     });
   }
+
+  public async getOnCallPoliciesWhereUserIsOnCallDuty(data: {
+    projectId: ObjectID;
+    userId: ObjectID;
+  }): Promise<{
+    escalationRulesByUser: Array<OnCallDutyPolicyEscalationRuleUser>;
+    escalationRulesByTeam: Array<OnCallDutyPolicyEscalationRuleTeam>;
+    escalationRulesBySchedule: Array<OnCallDutyPolicyEscalationRuleSchedule>;
+  }> {
+    // get all schedules where user is on call duty.
+    const onCallSchedules: Array<OnCallDutyPolicySchedule> =
+      await OnCallDutyPolicyScheduleService.getOnCallSchedulesWhereUserIsOnCallDuty(
+        data,
+      );
+
+    const teams: Array<Team> = await TeamService.getTeamsUserIsAPartOf({
+      userId: data.userId,
+      projectId: data.projectId,
+    });
+
+    // get escalationPolicies by user, team and schedule.
+    const escalationRulesByUser: Array<OnCallDutyPolicyEscalationRuleUser> =
+      await OnCallDutyPolicyEscalationRuleUserService.findBy({
+        query: {
+          userId: data.userId!,
+          projectId: data.projectId!,
+        },
+        select: {
+          onCallDutyPolicyEscalationRule: {
+            name: true,
+            _id: true,
+            order: true,
+          },
+          onCallDutyPolicy: {
+            name: true,
+            _id: true,
+          },
+        },
+        limit: LIMIT_PER_PROJECT,
+        skip: 0,
+        props: {
+          isRoot: true,
+        },
+      });
+
+    // do the same for teams.
+    const escalationRulesByTeam: Array<OnCallDutyPolicyEscalationRuleTeam> =
+      await OnCallDutyPolicyEscalationRuleTeamService.findBy({
+        query: {
+          teamId: QueryHelper.any(
+            teams.map((team: Team) => {
+              return team.id!;
+            }),
+          ),
+          projectId: data.projectId!,
+        },
+        select: {
+          onCallDutyPolicy: {
+            name: true,
+            _id: true,
+          },
+          onCallDutyPolicyEscalationRule: {
+            name: true,
+            _id: true,
+            order: true,
+          },
+          team: {
+            name: true,
+            _id: true,
+          },
+        },
+        limit: LIMIT_PER_PROJECT,
+        skip: 0,
+        props: {
+          isRoot: true,
+        },
+      });
+
+    // do the same for schedules.
+    const escalationRulesBySchedule: Array<OnCallDutyPolicyEscalationRuleSchedule> =
+      await OnCallDutyPolicyEscalationRuleScheduleService.findBy({
+        query: {
+          onCallDutyPolicyScheduleId: QueryHelper.any(
+            onCallSchedules.map((schedule: OnCallDutyPolicySchedule) => {
+              return schedule.id!;
+            }),
+          ),
+          projectId: data.projectId!,
+        },
+        select: {
+          onCallDutyPolicy: {
+            name: true,
+            _id: true,
+          },
+          onCallDutyPolicyEscalationRule: {
+            name: true,
+            _id: true,
+            order: true,
+          },
+          onCallDutyPolicySchedule: {
+            name: true,
+            _id: true,
+          },
+        },
+        limit: LIMIT_PER_PROJECT,
+        skip: 0,
+        props: {
+          isRoot: true,
+        },
+      });
+
+    return {
+      escalationRulesByUser: escalationRulesByUser,
+      escalationRulesByTeam: escalationRulesByTeam,
+      escalationRulesBySchedule: escalationRulesBySchedule,
+    };
+  }
 }
 export default new Service();

package/Server/Services/TeamService.ts

@@ -6,12 +6,48 @@ import LIMIT_MAX from "../../Types/Database/LimitMax";
 import BadDataException from "../../Types/Exception/BadDataException";
 import Model from "Common/Models/DatabaseModels/Team";
 import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import ObjectID from "../../Types/ObjectID";
+import TeamMember from "../../Models/DatabaseModels/TeamMember";
+import TeamMemberService from "./TeamMemberService";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
     super(Model);
   }
 
+  public async getTeamsUserIsAPartOf(data: {
+    userId: ObjectID;
+    projectId: ObjectID;
+  }): Promise<Array<Model>> {
+    const teamMembers: Array<TeamMember> = await TeamMemberService.findBy({
+      query: {
+        userId: data.userId,
+        projectId: data.projectId,
+      },
+      select: {
+        team: {
+          name: true,
+          _id: true,
+        },
+      },
+      limit: LIMIT_MAX,
+      skip: 0,
+      props: {
+        isRoot: true,
+      },
+    });
+
+    const teams: Array<Model> = [];
+
+    for (const teamMember of teamMembers) {
+      if (teamMember.team) {
+        teams.push(teamMember.team);
+      }
+    }
+
+    return teams;
+  }
+
   @CaptureSpan()
   protected override async onBeforeUpdate(
     updateBy: UpdateBy<Model>,

package/Server/Services/UserNotificationSettingService.ts

@@ -197,266 +197,143 @@ export class Service extends DatabaseService<UserNotificationSetting> {
     userId: ObjectID,
     projectId: ObjectID,
   ): Promise<void> {
-    const probeOwnerAddedNotificationEvent: PositiveNumber = await this.countBy(
-      {
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_PROBE_OWNER_ADDED_NOTIFICATION,
-        },
-        props: {
-          isRoot: true,
-        },
-      },
+    await this.addProbeOwnerNotificationSettings(userId, projectId);
+    await this.addIncidentNotificationSettings(userId, projectId);
+    await this.addMonitorNotificationSettings(userId, projectId);
+    await this.addOnCallNotificationSettings(userId, projectId);
+    await this.addAlertNotificationSettings(userId, projectId);
+  }
+
+  private async addProbeOwnerNotificationSettings(
+    userId: ObjectID,
+    projectId: ObjectID,
+  ): Promise<void> {
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_PROBE_OWNER_ADDED_NOTIFICATION,
     );
 
-    if (probeOwnerAddedNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_PROBE_OWNER_ADDED_NOTIFICATION;
-      item.alertByEmail = true;
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_PROBE_STATUS_CHANGED_OWNER_NOTIFICATION,
+    );
+  }
 
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
+  private async addIncidentNotificationSettings(
+    userId: ObjectID,
+    projectId: ObjectID,
+  ): Promise<void> {
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_INCIDENT_CREATED_OWNER_NOTIFICATION,
+    );
 
-    const probeStatusChangedNotificationEvent: PositiveNumber =
-      await this.countBy({
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_PROBE_STATUS_CHANGED_OWNER_NOTIFICATION,
-        },
-        props: {
-          isRoot: true,
-        },
-      });
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_INCIDENT_STATE_CHANGED_OWNER_NOTIFICATION,
+    );
+  }
 
-    if (probeStatusChangedNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_PROBE_STATUS_CHANGED_OWNER_NOTIFICATION;
-      item.alertByEmail = true;
+  private async addMonitorNotificationSettings(
+    userId: ObjectID,
+    projectId: ObjectID,
+  ): Promise<void> {
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_MONITOR_STATUS_CHANGED_OWNER_NOTIFICATION,
+    );
 
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_MONITOR_NOTIFICATION_WHEN_NO_PROBES_ARE_MONITORING_THE_MONITOR,
+    );
 
-    const incidentCreatedNotificationEvent: PositiveNumber = await this.countBy(
-      {
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_INCIDENT_CREATED_OWNER_NOTIFICATION,
-        },
-        props: {
-          isRoot: true,
-        },
-      },
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_MONITOR_NOTIFICATION_WHEN_PORBE_STATUS_CHANGES,
     );
+  }
 
-    if (incidentCreatedNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_INCIDENT_CREATED_OWNER_NOTIFICATION;
-      item.alertByEmail = true;
+  public async addOnCallNotificationSettings(
+    userId: ObjectID,
+    projectId: ObjectID,
+  ): Promise<void> {
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_WHEN_USER_IS_ON_CALL_ROSTER,
+    );
 
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_WHEN_USER_IS_NEXT_ON_CALL_ROSTER,
+    );
+
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_WHEN_USER_IS_ADDED_TO_ON_CALL_POLICY,
+    );
+
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_WHEN_USER_IS_REMOVED_FROM_ON_CALL_POLICY,
+    );
+
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_WHEN_USER_IS_NO_LONGER_ACTIVE_ON_ON_CALL_ROSTER,
+    );
+  }
+
+  private async addAlertNotificationSettings(
+    userId: ObjectID,
+    projectId: ObjectID,
+  ): Promise<void> {
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_ALERT_CREATED_OWNER_NOTIFICATION,
+    );
 
-    const alertCreatedNotificationEvent: PositiveNumber = await this.countBy({
+    await this.addNotificationSettingIfNotExists(
+      userId,
+      projectId,
+      NotificationSettingEventType.SEND_ALERT_STATE_CHANGED_OWNER_NOTIFICATION,
+    );
+  }
+
+  private async addNotificationSettingIfNotExists(
+    userId: ObjectID,
+    projectId: ObjectID,
+    eventType: NotificationSettingEventType,
+  ): Promise<void> {
+    const existingNotification: PositiveNumber = await this.countBy({
       query: {
         userId,
         projectId,
-        eventType:
-          NotificationSettingEventType.SEND_ALERT_CREATED_OWNER_NOTIFICATION,
+        eventType,
       },
       props: {
         isRoot: true,
       },
     });
 
-    if (alertCreatedNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_ALERT_CREATED_OWNER_NOTIFICATION;
-      item.alertByEmail = true;
-
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
-
-    // check monitor state changed notification
-    const monitorStateChangedNotificationEvent: PositiveNumber =
-      await this.countBy({
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_MONITOR_STATUS_CHANGED_OWNER_NOTIFICATION,
-        },
-        props: {
-          isRoot: true,
-        },
-      });
-
-    if (monitorStateChangedNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_MONITOR_STATUS_CHANGED_OWNER_NOTIFICATION;
-      item.alertByEmail = true;
-
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
-
-    // SEND_MONITOR_NOTIFICATION_WHEN_NO_PROBES_ARE_MONITORING_THE_MONITOR
-
-    const monitorNoProbesNotificationEvent: PositiveNumber = await this.countBy(
-      {
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_MONITOR_NOTIFICATION_WHEN_NO_PROBES_ARE_MONITORING_THE_MONITOR,
-        },
-        props: {
-          isRoot: true,
-        },
-      },
-    );
-
-    if (monitorNoProbesNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_MONITOR_NOTIFICATION_WHEN_NO_PROBES_ARE_MONITORING_THE_MONITOR;
-      item.alertByEmail = true;
-
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
-
-    // SEND_MONITOR_NOTIFICATION_WHEN_PORBE_STATUS_CHANGES
-
-    const monitorProbeStatusChangedNotificationEvent: PositiveNumber =
-      await this.countBy({
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_MONITOR_NOTIFICATION_WHEN_PORBE_STATUS_CHANGES,
-        },
-        props: {
-          isRoot: true,
-        },
-      });
-
-    if (monitorProbeStatusChangedNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_MONITOR_NOTIFICATION_WHEN_PORBE_STATUS_CHANGES;
-      item.alertByEmail = true;
-
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
-
-    // check incident state changed notification
-    const incidentStateChangedNotificationEvent: PositiveNumber =
-      await this.countBy({
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_INCIDENT_STATE_CHANGED_OWNER_NOTIFICATION,
-        },
-        props: {
-          isRoot: true,
-        },
-      });
-
-    if (incidentStateChangedNotificationEvent.toNumber() === 0) {
-      const item: UserNotificationSetting = new UserNotificationSetting();
-      item.userId = userId;
-      item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_INCIDENT_STATE_CHANGED_OWNER_NOTIFICATION;
-      item.alertByEmail = true;
-
-      await this.create({
-        data: item,
-        props: {
-          isRoot: true,
-        },
-      });
-    }
-
-    // check alert state changed notification
-    const alertStateChangedNotificationEvent: PositiveNumber =
-      await this.countBy({
-        query: {
-          userId,
-          projectId,
-          eventType:
-            NotificationSettingEventType.SEND_ALERT_STATE_CHANGED_OWNER_NOTIFICATION,
-        },
-        props: {
-          isRoot: true,
-        },
-      });
-
-    if (alertStateChangedNotificationEvent.toNumber() === 0) {
+    if (existingNotification.toNumber() === 0) {
       const item: UserNotificationSetting = new UserNotificationSetting();
       item.userId = userId;
       item.projectId = projectId;
-      item.eventType =
-        NotificationSettingEventType.SEND_ALERT_STATE_CHANGED_OWNER_NOTIFICATION;
+      item.eventType = eventType;
       item.alertByEmail = true;
 
       await this.create({

package/Server/Services/UserService.ts

@@ -34,6 +34,7 @@ import UserTwoFactorAuthService from "./UserTwoFactorAuthService";
 import BadDataException from "../../Types/Exception/BadDataException";
 import Name from "../../Types/Name";
 import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import Timezone from "../../Types/Timezone";
 
 export class Service extends DatabaseService<Model> {
   public constructor() {
@@ -364,6 +365,26 @@ export class Service extends DatabaseService<Model> {
       props: props,
     });
   }
+
+  public async getTimezoneForUser(userId: ObjectID): Promise<Timezone | null> {
+    const user: Model | null = await this.findOneBy({
+      query: {
+        _id: userId,
+      },
+      select: {
+        timezone: true,
+      },
+      props: {
+        isRoot: true,
+      },
+    });
+
+    if (!user) {
+      return null;
+    }
+
+    return user.timezone || null;
+  }
 }
 
 export default new Service();

package/Server/Utils/Workspace/Slack/Actions/Alert.ts

@@ -503,7 +503,7 @@ export default class SlackAlertActions {
         // send a message to the channel visible to user, that the alert has already been Resolved.
         const markdwonPayload: WorkspacePayloadMarkdown = {
           _type: "WorkspacePayloadMarkdown",
-          text: `@${slackUsername}, unfortunately you cannot execute the on call policy for **[Alert ${alertNumber?.toString()}](${await AlertService.getAlertLinkInDashboard(slackRequest.projectId!, alertId)})**. It has already been resolved.`,
+          text: `@${slackUsername}, unfortunately you cannot execute the on-call policy for **[Alert ${alertNumber?.toString()}](${await AlertService.getAlertLinkInDashboard(slackRequest.projectId!, alertId)})**. It has already been resolved.`,
         };
 
         await SlackUtil.sendDirectMessageToUser({
@@ -529,7 +529,7 @@ export default class SlackAlertActions {
       const onCallPolicyString: string =
         data.slackRequest.viewValues["onCallPolicy"].toString();
 
-      // get the on call policy id.
+      // get the on-call policy id.
       const onCallPolicyId: ObjectID = new ObjectID(onCallPolicyString);
 
       await OnCallDutyPolicyService.executePolicy(onCallPolicyId, {

package/Server/Utils/Workspace/Slack/Actions/Incident.ts

@@ -400,7 +400,7 @@ export default class SlackIncidentActions {
       blocks.push(monitorStatusDropdown);
     }
 
-    // add on call policy dropdown.
+    // add on-call policy dropdown.
 
     const onCallPolicies: Array<OnCallDutyPolicy> =
       await OnCallDutyPolicyService.findBy({
@@ -957,7 +957,7 @@ export default class SlackIncidentActions {
         // send a message to the channel visible to user, that the incident has already been Resolved.
         const markdwonPayload: WorkspacePayloadMarkdown = {
           _type: "WorkspacePayloadMarkdown",
-          text: `@${slackUsername}, unfortunately you cannot execute the on call policy for **[Incident ${incidentNumber?.toString()}](${await IncidentService.getIncidentLinkInDashboard(slackRequest.projectId!, incidentId)})**. It has already been resolved.`,
+          text: `@${slackUsername}, unfortunately you cannot execute the on-call policy for **[Incident ${incidentNumber?.toString()}](${await IncidentService.getIncidentLinkInDashboard(slackRequest.projectId!, incidentId)})**. It has already been resolved.`,
         };
 
         await SlackUtil.sendDirectMessageToUser({
@@ -983,7 +983,7 @@ export default class SlackIncidentActions {
       const onCallPolicyString: string =
         data.slackRequest.viewValues["onCallPolicy"].toString();
 
-      // get the on call policy id.
+      // get the on-call policy id.
       const onCallPolicyId: ObjectID = new ObjectID(onCallPolicyString);
 
       await OnCallDutyPolicyService.executePolicy(onCallPolicyId, {