@oneuptime/common 7.0.4007 → 7.0.4026

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/Services/WorkspaceNotificationRuleService.ts

@@ -264,6 +264,87 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     }
   }
 
+  @CaptureSpan()
+  public async archiveWorkspaceChannels(data: {
+    projectId: ObjectID;
+    notificationFor: NotificationFor;
+    sendMessageBeforeArchiving: WorkspacePayloadMarkdown;
+  }): Promise<void> {
+    const workspaceTypes: Array<WorkspaceType> = Service.getAllWorkspaceTypes();
+
+    for (const workspaceType of workspaceTypes) {
+      const notificationRules: Array<WorkspaceNotificationRule> =
+        await this.getMatchingNotificationRules({
+          projectId: data.projectId,
+          notificationFor: data.notificationFor,
+          workspaceType: workspaceType,
+          notificationRuleEventType: this.getNotificationRuleEventType(
+            data.notificationFor,
+          ),
+        });
+
+      // check if any of these rules have archive channel to true.
+      let shouldArchiveChannel: boolean = false;
+
+      for (const notificationRule of notificationRules) {
+        const rule: CreateChannelNotificationRule =
+          notificationRule.notificationRule as CreateChannelNotificationRule;
+        if (rule && rule.archiveChannelAutomatically) {
+          shouldArchiveChannel = true;
+          break;
+        }
+      }
+
+      if (!shouldArchiveChannel) {
+        continue; // check next workspace type.
+      }
+
+      const channels: Array<WorkspaceChannel> =
+        await this.getWorkspaceChannelsByNotificationFor({
+          projectId: data.projectId,
+          notificationFor: data.notificationFor,
+          workspaceType: workspaceType,
+        });
+
+      const channelIds: Array<string> = channels.map(
+        (channel: WorkspaceChannel) => {
+          return channel.id;
+        },
+      );
+
+      // get project auth token.
+      const projectAuth: WorkspaceProjectAuthToken | null =
+        await WorkspaceProjectAuthTokenService.findOneBy({
+          query: {
+            projectId: data.projectId,
+            workspaceType: workspaceType,
+          },
+          select: {
+            authToken: true,
+            miscData: true,
+          },
+          props: {
+            isRoot: true,
+          },
+        });
+
+      if (!projectAuth || !projectAuth.authToken) {
+        logger.debug("No project auth found for workspace type");
+        continue;
+      }
+
+      await WorkspaceUtil.getWorkspaceTypeUtil(workspaceType).archiveChannels({
+        authToken: projectAuth.authToken!,
+        channelIds: channelIds,
+        userId: this.getBotUserIdFromprojectAuthToken({
+          projectAuthToken: projectAuth,
+          workspaceType: workspaceType,
+        }),
+        sendMessageBeforeArchiving: data.sendMessageBeforeArchiving,
+      });
+    }
+  }
+
   @CaptureSpan()
   public async sendWorkspaceMarkdownNotification(data: {
     projectId: ObjectID;
@@ -316,43 +397,12 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
           notificationFor: data.notificationFor,
         });
 
-      let monitorChannels: Array<WorkspaceChannel> = [];
-
-      if (data.notificationFor.monitorId) {
-        monitorChannels = await MonitorService.getWorkspaceChannelForMonitor({
-          monitorId: data.notificationFor.monitorId,
-          workspaceType: messageBlocksByWorkspaceType.workspaceType,
-        });
-      }
-
-      // incidents
-      if (data.notificationFor.incidentId) {
-        monitorChannels = await IncidentService.getWorkspaceChannelForIncident({
-          incidentId: data.notificationFor.incidentId,
-          workspaceType: messageBlocksByWorkspaceType.workspaceType,
-        });
-      }
-
-      // alerts
-      if (data.notificationFor.alertId) {
-        monitorChannels = await AlertService.getWorkspaceChannelForAlert({
-          alertId: data.notificationFor.alertId,
+      const monitorChannels: Array<WorkspaceChannel> =
+        await this.getWorkspaceChannelsByNotificationFor({
+          projectId: data.projectId,
+          notificationFor: data.notificationFor,
           workspaceType: messageBlocksByWorkspaceType.workspaceType,
         });
-      }
-
-      // scheduled maintenance
-
-      if (data.notificationFor.scheduledMaintenanceId) {
-        monitorChannels =
-          await ScheduledMaintenanceService.getWorkspaceChannelForScheduledMaintenance(
-            {
-              scheduledMaintenanceId:
-                data.notificationFor.scheduledMaintenanceId,
-              workspaceType: messageBlocksByWorkspaceType.workspaceType,
-            },
-          );
-      }
 
       const workspaceMessagePayload: WorkspaceMessagePayload = {
         _type: "WorkspaceMessagePayload",
@@ -374,6 +424,49 @@ export class Service extends DatabaseService<WorkspaceNotificationRule> {
     });
   }
 
+  private async getWorkspaceChannelsByNotificationFor(data: {
+    projectId: ObjectID;
+    notificationFor: NotificationFor;
+    workspaceType: WorkspaceType;
+  }): Promise<Array<WorkspaceChannel>> {
+    let monitorChannels: Array<WorkspaceChannel> = [];
+
+    if (data.notificationFor.monitorId) {
+      monitorChannels = await MonitorService.getWorkspaceChannelForMonitor({
+        monitorId: data.notificationFor.monitorId,
+        workspaceType: data.workspaceType,
+      });
+    }
+
+    // incidents
+    if (data.notificationFor.incidentId) {
+      monitorChannels = await IncidentService.getWorkspaceChannelForIncident({
+        incidentId: data.notificationFor.incidentId,
+        workspaceType: data.workspaceType,
+      });
+    }
+
+    // alerts
+    if (data.notificationFor.alertId) {
+      monitorChannels = await AlertService.getWorkspaceChannelForAlert({
+        alertId: data.notificationFor.alertId,
+        workspaceType: data.workspaceType,
+      });
+    }
+
+    // scheduled maintenance
+    if (data.notificationFor.scheduledMaintenanceId) {
+      monitorChannels =
+        await ScheduledMaintenanceService.getWorkspaceChannelForScheduledMaintenance(
+          {
+            scheduledMaintenanceId: data.notificationFor.scheduledMaintenanceId,
+            workspaceType: data.workspaceType,
+          },
+        );
+    }
+    return monitorChannels;
+  }
+
   private getNotificationRuleEventType(
     notificationFor: NotificationFor,
   ): NotificationRuleEventType {

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, {