1 files changed, 76 insertions, 0 deletions
diff --git a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/notify/Notifier.java b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/notify/Notifier.java
new file mode 100644
index 00000000000..46e1fd904ed
--- /dev/null
+++ b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/notify/Notifier.java
@@ -0,0 +1,76 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.hosted.controller.notify;
+
+import com.yahoo.text.Text;
+import com.yahoo.vespa.hosted.controller.api.integration.organization.Mail;
+import com.yahoo.vespa.hosted.controller.api.integration.organization.Mailer;
+import com.yahoo.vespa.hosted.controller.api.integration.organization.MailerException;
+import com.yahoo.vespa.hosted.controller.notification.Notification;
+import com.yahoo.vespa.hosted.controller.persistence.CuratorDb;
+import com.yahoo.vespa.hosted.controller.tenant.CloudTenant;
+import com.yahoo.vespa.hosted.controller.tenant.TenantContacts;
+
+import java.util.Collection;
+import java.util.Objects;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+import java.util.stream.Collectors;
+
+/**
+ * Notifier is responsible for dispatching user notifications to their chosen Contact points.
+ *
+ * @author enygaard
+ */
+public class Notifier {
+    private final CuratorDb curatorDb;
+    private final Mailer mailer;
+
+    private static final Logger log = Logger.getLogger(Notifier.class.getName());
+
+    public Notifier(CuratorDb curatorDb, Mailer mailer) {
+        this.curatorDb = Objects.requireNonNull(curatorDb);
+        this.mailer = Objects.requireNonNull(mailer);
+    }
+
+    public void dispatch(Notification notification) {
+        var tenant = curatorDb.readTenant(notification.source().tenant());
+        tenant.stream().forEach(t -> {
+            if (t instanceof CloudTenant) {
+                var ct = (CloudTenant) t;
+                ct.info().contacts().all().stream()
+                        .filter(c -> c.audiences().contains(TenantContacts.Audience.NOTIFICATIONS))
+                        .collect(Collectors.groupingBy(TenantContacts.Contact::type, Collectors.toList()))
+                        .entrySet()
+                        .forEach(e -> dispatch(notification, e.getKey(), e.getValue()));
+            }
+        });
+    }
+
+    private void dispatch(Notification notification, TenantContacts.Type type, Collection<? extends TenantContacts.Contact> contacts) {
+        switch (type) {
+            case EMAIL:
+                dispatch(notification, contacts.stream().map(c -> (TenantContacts.EmailContact) c).collect(Collectors.toList()));
+                break;
+            default:
+                throw new IllegalArgumentException("Unknown TenantContacts type " + type.name());
+        }
+    }
+
+    private void dispatch(Notification notification, Collection<TenantContacts.EmailContact> contacts) {
+        try {
+            mailer.send(mailOf(notification, contacts.stream().map(c -> c.email()).collect(Collectors.toList())));
+        } catch (MailerException e) {
+            log.log(Level.SEVERE, "Failed sending email", e);
+        }
+    }
+
+    private Mail mailOf(Notification n, Collection<String> recipients) {
+        var subject = Text.format("[%s] Vespa Notification for %s", n.level().toString().toUpperCase(), n.type().name());
+        var body = new StringBuilder();
+        body.append("Source: ").append(n.source().toString()).append("\n")
+                .append("\n")
+                .append(String.join("\n", n.messages()));
+        return new Mail(recipients, subject.toString(), body.toString());
+    }
+
+}