1 files changed, 51 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/frontend/src/app/shared/services/prometheus-notification.service.ts b/src/pybind/mgr/dashboard/frontend/src/app/shared/services/prometheus-notification.service.ts
new file mode 100644
index 000000000..ab94c686e
--- /dev/null
+++ b/src/pybind/mgr/dashboard/frontend/src/app/shared/services/prometheus-notification.service.ts
@@ -0,0 +1,51 @@
+import { Injectable } from '@angular/core';
+
+import _ from 'lodash';
+
+import { PrometheusService } from '../api/prometheus.service';
+import { CdNotificationConfig } from '../models/cd-notification';
+import { AlertmanagerNotification } from '../models/prometheus-alerts';
+import { PrometheusAlertFormatter } from './prometheus-alert-formatter';
+
+@Injectable({
+  providedIn: 'root'
+})
+export class PrometheusNotificationService {
+  private notifications: AlertmanagerNotification[];
+  private backendFailure = false;
+
+  constructor(
+    private alertFormatter: PrometheusAlertFormatter,
+    private prometheusService: PrometheusService
+  ) {
+    this.notifications = [];
+  }
+
+  refresh() {
+    if (this.backendFailure) {
+      return;
+    }
+    this.prometheusService.getNotifications(_.last(this.notifications)).subscribe(
+      (notifications) => this.handleNotifications(notifications),
+      () => (this.backendFailure = true)
+    );
+  }
+
+  private handleNotifications(notifications: AlertmanagerNotification[]) {
+    if (notifications.length === 0) {
+      return;
+    }
+    if (this.notifications.length > 0) {
+      this.alertFormatter.sendNotifications(
+        _.flatten(notifications.map((notification) => this.formatNotification(notification)))
+      );
+    }
+    this.notifications = this.notifications.concat(notifications);
+  }
+
+  private formatNotification(notification: AlertmanagerNotification): CdNotificationConfig[] {
+    return this.alertFormatter
+      .convertToCustomAlerts(notification.alerts)
+      .map((alert) => this.alertFormatter.convertAlertToNotification(alert));
+  }
+}