Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright (C) 2020 Graylog, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the Server Side Public License, version 1,
* as published by MongoDB, Inc.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* Server Side Public License for more details.
*
* You should have received a copy of the Server Side Public License
* along with this program. If not, see
* .
*/
package org.graylog.events.notifications.types;
import com.floreysoft.jmte.Engine;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.ImmutableList;
import org.apache.commons.mail.Email;
import org.apache.commons.mail.EmailException;
import org.apache.commons.mail.HtmlEmail;
import org.apache.commons.mail.SimpleEmail;
import org.graylog.events.notifications.EventNotificationContext;
import org.graylog.events.notifications.EventNotificationModelData;
import org.graylog2.alerts.EmailRecipients;
import org.graylog2.jackson.TypeReferences;
import org.graylog2.notifications.Notification;
import org.graylog2.notifications.NotificationService;
import org.graylog2.plugin.MessageSummary;
import org.graylog2.plugin.alarms.transports.TransportConfigurationException;
import org.graylog2.plugin.system.NodeId;
import org.graylog2.shared.bindings.providers.ObjectMapperProvider;
import org.graylog2.shared.email.EmailFactory;
import org.joda.time.DateTimeZone;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.inject.Inject;
import javax.inject.Named;
import java.util.ArrayList;
import java.util.Map;
import java.util.Set;
import static com.google.common.base.Strings.isNullOrEmpty;
import static java.util.Objects.requireNonNull;
public class EmailSender {
private static final Logger LOG = LoggerFactory.getLogger(EmailSender.class);
private final EmailRecipients.Factory emailRecipientsFactory;
private final NotificationService notificationService;
private final NodeId nodeId;
private final ObjectMapperProvider objectMapperProvider;
private final Engine templateEngine;
private final Engine htmlTemplateEngine;
private final EmailFactory emailFactory;
@Inject
public EmailSender(EmailRecipients.Factory emailRecipientsFactory,
NotificationService notificationService,
NodeId nodeId,
ObjectMapperProvider objectMapperProvider,
Engine templateEngine,
@Named("HtmlSafe") Engine htmlTemplateEngine,
EmailFactory emailFactory) {
this.emailRecipientsFactory = requireNonNull(emailRecipientsFactory, "emailRecipientsFactory");
this.notificationService = requireNonNull(notificationService, "notificationService");
this.nodeId = requireNonNull(nodeId, "nodeId");
this.objectMapperProvider = requireNonNull(objectMapperProvider, "objectMapperProvider)");
this.templateEngine = requireNonNull(templateEngine, "templateEngine");
this.htmlTemplateEngine = requireNonNull(htmlTemplateEngine, "htmlTemplateEngine");
this.emailFactory = requireNonNull(emailFactory, "emailFactory");
}
@VisibleForTesting
private String buildSubject(EmailEventNotificationConfig config, Map model) {
final String template;
if (isNullOrEmpty(config.subject())) {
template = EmailEventNotificationConfig.DEFAULT_SUBJECT;
} else {
template = config.subject();
}
return templateEngine.transform(template, model);
}
@VisibleForTesting
private String buildBody(EmailEventNotificationConfig config, Map model) {
final String template;
if (isNullOrEmpty(config.bodyTemplate())) {
template = EmailEventNotificationConfig.DEFAULT_BODY_TEMPLATE;
} else {
template = config.bodyTemplate();
}
return templateEngine.transform(template, model);
}
@VisibleForTesting
private String buildHtmlBody(EmailEventNotificationConfig config, Map model) {
return htmlTemplateEngine.transform(config.htmlBodyTemplate(), model);
}
private Map getModel(EventNotificationContext ctx, ImmutableList backlog, DateTimeZone timeZone) {
final EventNotificationModelData modelData = EventNotificationModelData.of(ctx, backlog);
return objectMapperProvider.getForTimeZone(timeZone).convertValue(modelData, TypeReferences.MAP_STRING_OBJECT);
}
private void sendEmail(EmailEventNotificationConfig config, String emailAddress, Map model) throws TransportConfigurationException, EmailException {
LOG.debug("Sending mail to " + emailAddress);
if (!emailFactory.isEmailTransportEnabled()) {
throw new TransportConfigurationException("Email transport is not enabled in server configuration file!");
}
final Email email = createEmailWithBody(config, model);
if (!isNullOrEmpty(config.sender())) {
email.setFrom(config.sender());
}
if (email.getFromAddress() == null) {
throw new TransportConfigurationException("No from address specified for email transport.");
}
email.setSubject(buildSubject(config, model));
email.addTo(emailAddress);
email.send();
}
Email createEmailWithBody(EmailEventNotificationConfig config, Map model) throws EmailException {
if (!isNullOrEmpty(config.htmlBodyTemplate())) {
HtmlEmail email = emailFactory.htmlEmail();
email.setTextMsg(buildBody(config, model));
email.setHtmlMsg(buildHtmlBody(config, model));
return email;
} else {
SimpleEmail email = emailFactory.simpleEmail();
email.setMsg(buildBody(config, model));
return email;
}
}
// TODO: move EmailRecipients class to events code
void sendEmails(EmailEventNotificationConfig notificationConfig, EventNotificationContext ctx, ImmutableList backlog) throws TransportConfigurationException, EmailException, ConfigurationError {
if (!emailFactory.isEmailTransportEnabled()) {
throw new TransportConfigurationException("Email transport is not enabled in server configuration file!");
}
final EmailRecipients emailRecipients = emailRecipientsFactory.create(
new ArrayList<>(notificationConfig.userRecipients()),
new ArrayList<>(notificationConfig.emailRecipients())
);
if (emailRecipients.isEmpty()) {
LOG.debug("Cannot send emails: empty recipient list.");
return;
}
final Set recipientsSet = emailRecipients.getEmailRecipients();
if (recipientsSet.size() == 0) {
final Notification notification = notificationService.buildNow()
.addNode(nodeId.toString())
.addType(Notification.Type.GENERIC)
.addSeverity(Notification.Severity.NORMAL)
.addDetail("title", "No recipients have been defined!")
.addDetail("description", "To fix this, go to the notification configuration and add at least one alert recipient.");
notificationService.publishIfFirst(notification);
}
final Map model = getModel(ctx, backlog, notificationConfig.timeZone());
for (String email : recipientsSet) {
sendEmail(notificationConfig, email, model);
}
}
static class ConfigurationError extends Exception {
ConfigurationError(String message) {
super(message);
}
}
}
