src/main/java/net/pterodactylus/rhynodge/actions/EmailAction.java

   1 /*
   2  * Rhynodge - EmailAction.java - Copyright © 2013 David Roden
   3  *
   4  * This program is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 package net.pterodactylus.rhynodge.actions;
  19
  20 import static java.lang.System.getProperties;
  21 import static javax.mail.Session.getInstance;
  22
  23 import java.util.Properties;
  24
  25 import javax.mail.Message.RecipientType;
  26 import javax.mail.MessagingException;
  27 import javax.mail.Session;
  28 import javax.mail.Transport;
  29 import javax.mail.URLName;
  30 import javax.mail.internet.InternetAddress;
  31 import javax.mail.internet.MimeBodyPart;
  32 import javax.mail.internet.MimeMessage;
  33 import javax.mail.internet.MimeMultipart;
  34
  35 import net.pterodactylus.rhynodge.Action;
  36 import net.pterodactylus.rhynodge.output.Output;
  37
  38 import com.google.common.annotations.VisibleForTesting;
  39 import com.sun.mail.smtp.SMTPTransport;
  40
  41 /**
  42  * {@link Action} implementation that sends an email containing the triggering
  43  * object to an email address.
  44  *
  45  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  46  */
  47 public class EmailAction implements Action {
  48
  49         /** The email address of the sender. */
  50         private final String sender;
  51
  52         /** The email address of the recipient. */
  53         private final String recipient;
  54
  55         private final Transport transport;
  56         private final Session session;
  57
  58         /**
  59          * Creates a new email action.
  60          *
  61          * @param hostname
  62          *            The hostname of the SMTP server
  63          * @param sender
  64          *            The email address of the sender
  65          * @param recipient
  66          *            The email address of the recipient
  67          */
  68         public EmailAction(String hostname, String sender, String recipient) {
  69                 this.sender = sender;
  70                 this.recipient = recipient;
  71                 Properties properties = getProperties();
  72                 properties.put("mail.smtp.host", hostname);
  73                 session = getInstance(properties);
  74                 transport = new SMTPTransport(session, new URLName("smtp", hostname, 25, null, "", ""));
  75         }
  76
  77         @VisibleForTesting
  78         EmailAction(Transport transport, String sender, String recipient) {
  79                 this.transport = transport;
  80                 this.sender = sender;
  81                 this.recipient = recipient;
  82                 this.session = getInstance(getProperties());
  83         }
  84
  85         //
  86         // ACTION METHODS
  87         //
  88
  89         /**
  90          * {@inheritDoc}
  91          */
  92         @Override
  93         public void execute(Output output) {
  94                 MimeMessage message = new MimeMessage(session);
  95                 try {
  96                         /* create message. */
  97                         message.setFrom(new InternetAddress(sender));
  98                         message.setRecipient(RecipientType.TO, new InternetAddress(recipient));
  99                         message.setSubject(output.summary(), "UTF-8");
 100
 101                         /* create text and html parts. */
 102                         MimeMultipart multipart = new MimeMultipart();
 103                         multipart.setSubType("alternative");
 104                         addPlainTextPart(output, multipart);
 105                         addHtmlPart(output, multipart);
 106                         message.setContent(multipart);
 107
 108                         if (!transport.isConnected()) {
 109                                 transport.connect();
 110                         }
 111                         transport.sendMessage(message, message.getAllRecipients());
 112                 } catch (MessagingException me1) {
 113                         /* swallow. */
 114                 }
 115         }
 116
 117         private void addPlainTextPart(Output output, MimeMultipart multipart) throws MessagingException {
 118                 if (output.text("text/plain", -1) == null) {
 119                         return;
 120                 }
 121                 MimeBodyPart textPart = new MimeBodyPart();
 122                 textPart.setContent(output.text("text/plain", -1), "text/plain;charset=utf-8");
 123                 multipart.addBodyPart(textPart);
 124         }
 125
 126         private void addHtmlPart(Output output, MimeMultipart multipart) throws MessagingException {
 127                 if (output.text("text/html", -1) == null) {
 128                         return;
 129                 }
 130                 MimeBodyPart htmlPart = new MimeBodyPart();
 131                 htmlPart.setContent(output.text("text/html", -1), "text/html;charset=utf-8");
 132                 multipart.addBodyPart(htmlPart);
 133         }
 134
 135 }