nl.nn.adapterframework.senders.MailSender.java Source code

Java tutorial

Introduction

Here is the source code for nl.nn.adapterframework.senders.MailSender.java

Source

/*
   Copyright 2013 Nationale-Nederlanden
    
   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at
    
   http://www.apache.org/licenses/LICENSE-2.0
    
   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/
package nl.nn.adapterframework.senders;

import java.io.ByteArrayOutputStream;
import java.net.URL;
import java.util.Collection;
import java.util.Date;
import java.util.Iterator;
import java.util.Properties;

import javax.activation.DataHandler;
import javax.activation.DataSource;
import javax.activation.URLDataSource;
import javax.mail.BodyPart;
import javax.mail.Message;
import javax.mail.MessagingException;
import javax.mail.Multipart;
import javax.mail.Session;
import javax.mail.Transport;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeBodyPart;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;

import nl.nn.adapterframework.configuration.ConfigurationException;
import nl.nn.adapterframework.core.ParameterException;
import nl.nn.adapterframework.core.SenderException;
import nl.nn.adapterframework.core.SenderWithParametersBase;
import nl.nn.adapterframework.core.TimeOutException;
import nl.nn.adapterframework.parameters.ParameterResolutionContext;
import nl.nn.adapterframework.parameters.ParameterValue;
import nl.nn.adapterframework.parameters.ParameterValueList;
import nl.nn.adapterframework.util.CredentialFactory;
import nl.nn.adapterframework.util.DomBuilderException;
import nl.nn.adapterframework.util.Misc;
import nl.nn.adapterframework.util.XmlUtils;

import org.apache.commons.lang.StringUtils;
import org.apache.soap.util.mime.ByteArrayDataSource;
import org.apache.xerces.impl.dv.util.Base64;
import org.w3c.dom.Element;

/**
 * {@link nl.nn.adapterframework.core.ISender sender} that sends a mail specified by an XML message. <br/>
 *
 * Sample email.xml:<br/><code><pre>
 *   &lt;email&gt;
 *       &lt;recipients&gt;
 *          &lt;recipient type="to"&gt;***@nn.nl&lt;/recipient&gt;
 *           &lt;recipient type="cc"&gt;***@nn.nl&lt;/recipient&gt;
 *       &lt;/recipients&gt;
 *       &lt;from&gt;***@nn.nl&lt;/from&gt;
 *       &lt;subject&gt;this is the subject&lt;/subject&gt;
 *       &lt;threadTopic&gt;subject&lt;/threadTopic&gt;
 *       &lt;message&gt;dit is de message&lt;/message&gt;
 *       &lt;messageType&gt;text/plain&lt;/messageType&gt;&lt;!-- Optional --&gt;
 *       &lt;messageBase64&gt;false&lt;/messageBase64&gt;&lt;!-- Optional --&gt;
 *       &lt;charset&gt;UTF-8&lt;/charset&gt;&lt;!-- Optional --&gt;
 *       &lt;attachments&gt;
 *           &lt;attachment name="filename1.txt" type="text"&gt;<i>contents of first attachment</i>&lt;/attachment&gt;
 *           &lt;attachment name="filename2.txt" type="text" url="url-to-resource" base64="false"&gt;<i>this is an attachment with a resource</i>&lt;/attachment&gt;
 *       &lt;/attachments&gt;&lt;!-- Optional --&gt;
 *   &lt;/email&gt;
 * </pre></code> <br/>
 * Notice: it must be valid XML. Therefore, especially the message element
 * must be plain text or be wrapped as CDATA.<br/><br/>
 * example:<br/><code><pre>
 * &lt;message&gt;&lt;![CDATA[&lt;h1&gt;This is a HtmlMessage&lt;/h1&gt;]]&gt;&lt;/message&gt;
 * </pre></code><br/>
 *
 * <p><b>Configuration:</b>
 * <table border="1">
 * <tr><th>attributes</th><th>description</th><th>default</th></tr>
 * <tr><td>{@link #setSmtpHost(String) smtpHost}</td><td>name of the host by which the messages are to be send</td><td>&nbsp;</td></tr>
 * <tr><td>{@link #setSmtpAuthAlias(String) smtpAuthAlias}</td><td>alias used to obtain credentials for authentication to smtpHost</td><td>&nbsp;</td></tr>
 * <tr><td>{@link #setSmtpUserid(String) smtpUserid}</td><td>userid on the smtpHost</td><td>&nbsp;</td></tr>
 * <tr><td>{@link #setSmtpPassword(String) smtpPassword}</td><td>password of userid on the smtpHost</td><td>&nbsp;</td></tr>
 * <tr><td>{@link #setDefaultFrom(String) defaultFrom}</td><td>value of the From: header if not specified in message itself</td><td>&nbsp;</td></tr>
 * <tr><td>{@link #setDefaultSubject(String) defaultSubject}</td><td>value of the Subject: header if not specified in message itself</td><td>&nbsp;</td></tr>
 * <tr><td>{@link #setDefaultAttachmentType(String) defaultAttachmentType}</td><td>&nbsp;</td><td>text</td></tr>
 * <tr><td>{@link #setDefaultAttachmentName(String) defaultAttachmentName}</td><td>&nbsp;</td><td>attachment</td></tr>
 * <tr><td>{@link #setTimeout(int) timeout}</td><td>timeout (in milliseconds). Used for socket connection timeout and socket I/O timeout</td><td>20000</td></tr>
 * </table>
 * <table border="1">
 * <p><b>Parameters:</b>
 * <tr><th>name</th><th>type</th><th>remarks</th></tr>
 * <tr><td>from</td><td>string</td><td>email address of the sender</td></tr>
 * <tr><td>subject</td><td>string</td><td>subject field of the message</td></tr>
 * <tr><td>threadTopic</td><td>string</td><td>(optional) conversation field of the message, used to correlate mails in mail viewer (header field "Thread-Topic"). Note: subject must end with value of threadTopic, but cann't be exactly the same</td></tr>
 * <tr><td>message</td><td>string</td><td>message itself. If absent, the complete input message is assumed to be the message</td></tr>
 * <tr><td>messageType</td><td>string</td><td>message MIME type (at this moment only available are text/plain and text/html - default: text/plain)</td></tr>
 * <tr><td>messageBase64</td><td>boolean</td><td>indicates whether the message content is base64 encoded (default: false)</td></tr>
 * <tr><td>charset</td><td>string</td><td>the character encoding (e.g. ISO-8859-1 or UTF-8) used to send the email (default: value of system property mail.mime.charset, when not present the value of system property file.encoding)</td></tr>
 * <tr><td>recipients</td><td>xml</td><td>recipients of the message. must result in a structure like: <code><pre>
 *           &lt;recipient type="to"&gt;***@natned&lt;/recipient&gt;
 *           &lt;recipient type="cc"&gt;***@nn.nl&lt;/recipient&gt;
* </pre></code></td></tr>
 * <tr><td>attachments</td><td>xml</td><td>attachments to the message. must result in a structure like: <code><pre>
 *           &lt;attachment name="filename1.txt" type="text"&gt;<i>contents of first attachment</i>&lt;/attachment&gt;
 *           &lt;attachment name="filename2.txt" type="text" url="url-to-resource" base64="false"&gt;<i>this is an attachment with a resource</i>&lt;/attachment&gt;
 * </pre></code></td></tr>
 * </table>
 * </p>
 * NB Compilation and Deployment Note: mail.jar (v1.2) and activation.jar must appear BEFORE j2ee.jar
 * Otherwise errors like the following might occur:
 *   NoClassDefFoundException: com/sun/mail/util/MailDateFormat 
 * 
 * @author Johan Verrips/Gerrit van Brakel
 */

public class MailSender extends SenderWithParametersBase {

    private String smtpHost;
    private String smtpAuthAlias;
    private String smtpUserid;
    private String smtpPassword;
    private String defaultAttachmentType = "text";
    private String defaultAttachmentName = "attachment";
    private String defaultMessageType = "text/plain";
    private String defaultMessageBase64 = "false";

    private int timeout = 20000;

    // defaults
    private String defaultSubject;
    private String defaultFrom;

    private Session session;
    private Properties properties;

    public void configure() throws ConfigurationException {
        if (StringUtils.isEmpty(getSmtpHost())) {
            throw new ConfigurationException("MailSender [" + getName() + "] has no smtpHost configured");
        }
        properties = System.getProperties();
        try {
            properties.put("mail.smtp.host", getSmtpHost());
        } catch (Throwable t) {
            throw new ConfigurationException(
                    "MailSender [" + getName() + "] cannot set smtpHost [" + getSmtpHost() + "] in properties");
        }
        properties.put("mail.smtp.connectiontimeout", getTimeout() + "");
        properties.put("mail.smtp.timeout", getTimeout() + "");
        if (paramList != null) {
            paramList.configure();
        }
    }

    /**
     * Create a <code>Session</code> and <code>Transport</code> to the
     * smtp host.
      * @throws SenderException
     */
    public void open() throws SenderException {
        try {
            getSession();

        } catch (Exception e) {
            throw new SenderException("Error opening MailSender", e);
        }
    }

    /**
     * Close the <code>transport</code> layer.
     */
    public void close() throws SenderException {
        /*
        try {
           if (transport!=null) {
        transport.close();
           }
        } catch (Exception e) {
           throw new SenderException("error closing transport", e);
        }
        */
    }

    public boolean isSynchronous() {
        return false;
    }

    protected Session getSession() {
        if (session == null) {
            session = Session.getInstance(properties, null);
            session.setDebug(log.isDebugEnabled());
            //         log.debug("MailSender [" + getName() + "] got session to [" + properties + "]");
        }
        return session;
    }

    public String sendMessage(String correlationID, String message, ParameterResolutionContext prc)
            throws SenderException, TimeOutException {
        String from = null;
        String subject = null;
        String threadTopic = null;
        String messageType = null;
        String messageBase64 = null;
        String charset = null;
        Collection recipients = null;
        Collection attachments = null;
        ParameterValueList pvl;
        ParameterValue pv;

        String messageInMailSafeForm;
        if (paramList == null) {
            messageInMailSafeForm = sendEmail(message);
        } else {
            try {
                pvl = prc.getValues(paramList);
                pv = pvl.getParameterValue("from");
                if (pv != null) {
                    from = pv.asStringValue(null);
                    log.debug("MailSender [" + getName() + "] retrieved from-parameter [" + from + "]");
                }
                pv = pvl.getParameterValue("subject");
                if (pv != null) {
                    subject = pv.asStringValue(null);
                    log.debug("MailSender [" + getName() + "] retrieved subject-parameter [" + subject + "]");
                }
                pv = pvl.getParameterValue("threadTopic");
                if (pv != null) {
                    threadTopic = pv.asStringValue(null);
                    log.debug(
                            "MailSender [" + getName() + "] retrieved threadTopic-parameter [" + threadTopic + "]");
                }
                pv = pvl.getParameterValue("message");
                if (pv != null) {
                    message = pv.asStringValue(message);
                    log.debug("MailSender [" + getName() + "] retrieved message-parameter [" + message + "]");
                }
                pv = pvl.getParameterValue("messageType");
                if (pv != null) {
                    messageType = pv.asStringValue(null);
                    log.debug(
                            "MailSender [" + getName() + "] retrieved messageType-parameter [" + messageType + "]");
                }
                pv = pvl.getParameterValue("messageBase64");
                if (pv != null) {
                    messageBase64 = pv.asStringValue(null);
                    log.debug("MailSender [" + getName() + "] retrieved messageBase64-parameter [" + messageBase64
                            + "]");
                }
                pv = pvl.getParameterValue("charset");
                if (pv != null) {
                    charset = pv.asStringValue(null);
                    log.debug("MailSender [" + getName() + "] retrieved charset-parameter [" + charset + "]");
                }
                pv = pvl.getParameterValue("recipients");
                if (pv != null) {
                    recipients = pv.asCollection();
                }
                pv = pvl.getParameterValue("attachments");
                if (pv != null) {
                    attachments = pv.asCollection();
                }
            } catch (ParameterException e) {
                throw new SenderException(
                        "MailSender [" + getName() + "] got exception determining parametervalues", e);
            }
            messageInMailSafeForm = sendEmail(from, subject, threadTopic, message, messageType, messageBase64,
                    charset, recipients, attachments);
        }
        prc.getSession().put("messageInMailSafeForm", messageInMailSafeForm);
        return correlationID;
    }

    public String sendMessage(String correlationID, String input) throws SenderException {
        sendEmail(input);
        return correlationID;
    }

    /**
     * Send a mail conforming to the XML input
     */
    protected String sendEmail(String input) throws SenderException {
        // initialize this request
        String from;
        String subject;
        String threadTopic;
        String message;
        String messageType;
        String messageBase64;
        String charset;
        Collection recipients;
        Collection attachments;

        Element emailElement;
        try {
            emailElement = XmlUtils.buildElement(input);

            from = XmlUtils.getChildTagAsString(emailElement, "from");
            subject = XmlUtils.getChildTagAsString(emailElement, "subject");
            threadTopic = XmlUtils.getChildTagAsString(emailElement, "threadTopic");
            message = XmlUtils.getChildTagAsString(emailElement, "message");
            messageType = XmlUtils.getChildTagAsString(emailElement, "messageType");
            messageBase64 = XmlUtils.getChildTagAsString(emailElement, "messageBase64");
            charset = XmlUtils.getChildTagAsString(emailElement, "charset");

            Element recipientsElement = XmlUtils.getFirstChildTag(emailElement, "recipients");
            recipients = XmlUtils.getChildTags(recipientsElement, "recipient");

            Element attachmentsElement = XmlUtils.getFirstChildTag(emailElement, "attachments");
            attachments = attachmentsElement == null ? null
                    : XmlUtils.getChildTags(attachmentsElement, "attachment");

        } catch (DomBuilderException e) {
            throw new SenderException("exception parsing [" + input + "]", e);
        }

        return sendEmail(from, subject, threadTopic, message, messageType, messageBase64, charset, recipients,
                attachments);
    }

    protected String sendEmail(String from, String subject, String threadTopic, String message, String messageType,
            String messageBase64, String charset, Collection recipients, Collection attachments)
            throws SenderException {

        StringBuffer sb = new StringBuffer();

        if (recipients == null || recipients.size() == 0) {
            throw new SenderException("MailSender [" + getName() + "] has no recipients for message");
        }
        if (StringUtils.isEmpty(from)) {
            from = defaultFrom;
        }
        if (StringUtils.isEmpty(subject)) {
            subject = defaultSubject;
        }
        log.debug("MailSender [" + getName() + "] requested to send message from [" + from + "] subject [" + subject
                + "] to #recipients [" + recipients.size() + "]");

        if (StringUtils.isEmpty(messageType)) {
            messageType = defaultMessageType;
        }

        if (StringUtils.isEmpty(messageBase64)) {
            messageBase64 = defaultMessageBase64;
        }

        try {
            if (log.isDebugEnabled()) {
                sb.append("MailSender [" + getName() + "] sending message ");
                sb.append("[smtpHost=" + smtpHost);
                sb.append("[from=" + from + "]");
                sb.append("[subject=" + subject + "]");
                sb.append("[threadTopic=" + threadTopic + "]");
                sb.append("[text=" + message + "]");
                sb.append("[type=" + messageType + "]");
                sb.append("[base64=" + messageBase64 + "]");
            }

            if ("true".equalsIgnoreCase(messageBase64) && StringUtils.isNotEmpty(message)) {
                message = decodeBase64ToString(message);
            }

            // construct a message  
            MimeMessage msg = new MimeMessage(session);
            msg.setFrom(new InternetAddress(from));
            msg.setSubject(subject, charset);
            if (StringUtils.isNotEmpty(threadTopic)) {
                msg.setHeader("Thread-Topic", threadTopic);
            }
            Iterator iter = recipients.iterator();
            boolean recipientsFound = false;
            while (iter.hasNext()) {
                Element recipientElement = (Element) iter.next();
                String recipient = XmlUtils.getStringValue(recipientElement);
                if (StringUtils.isNotEmpty(recipient)) {
                    String typeAttr = recipientElement.getAttribute("type");
                    Message.RecipientType recipientType = Message.RecipientType.TO;
                    if ("cc".equalsIgnoreCase(typeAttr)) {
                        recipientType = Message.RecipientType.CC;
                    }
                    if ("bcc".equalsIgnoreCase(typeAttr)) {
                        recipientType = Message.RecipientType.BCC;
                    }
                    msg.addRecipient(recipientType, new InternetAddress(recipient));
                    recipientsFound = true;
                    if (log.isDebugEnabled()) {
                        sb.append("[recipient(" + typeAttr + ")=" + recipient + "]");
                    }
                } else {
                    log.debug("empty recipient found, ignoring");
                }
            }
            if (!recipientsFound) {
                throw new SenderException("MailSender [" + getName() + "] did not find any valid recipients");
            }

            String messageTypeWithCharset;
            if (charset == null) {
                charset = System.getProperty("mail.mime.charset");
                if (charset == null) {
                    charset = System.getProperty("file.encoding");
                }
            }
            if (charset != null) {
                messageTypeWithCharset = messageType + ";charset=" + charset;
            } else {
                messageTypeWithCharset = messageType;
            }
            log.debug("MailSender [" + getName() + "] uses encoding [" + messageTypeWithCharset + "]");

            if (attachments == null || attachments.size() == 0) {
                //msg.setContent(message, messageType);
                msg.setContent(message, messageTypeWithCharset);
            } else {
                Multipart multipart = new MimeMultipart();
                BodyPart messageBodyPart = new MimeBodyPart();
                //messageBodyPart.setContent(message, messageType);
                messageBodyPart.setContent(message, messageTypeWithCharset);
                multipart.addBodyPart(messageBodyPart);

                iter = attachments.iterator();
                while (iter.hasNext()) {
                    Element attachmentElement = (Element) iter.next();
                    String attachmentText = XmlUtils.getStringValue(attachmentElement);
                    String attachmentName = attachmentElement.getAttribute("name");
                    String attachmentUrl = attachmentElement.getAttribute("url");
                    String attachmentType = attachmentElement.getAttribute("type");
                    String attachmentBase64 = attachmentElement.getAttribute("base64");
                    if (StringUtils.isEmpty(attachmentType)) {
                        attachmentType = getDefaultAttachmentType();
                    }
                    if (StringUtils.isEmpty(attachmentName)) {
                        attachmentName = getDefaultAttachmentName();
                    }
                    log.debug("found attachment [" + attachmentName + "] type [" + attachmentType + "] url ["
                            + attachmentUrl + "]contents [" + attachmentText + "]");

                    messageBodyPart = new MimeBodyPart();

                    DataSource attachmentDataSource;
                    if (!StringUtils.isEmpty(attachmentUrl)) {
                        attachmentDataSource = new URLDataSource(new URL(attachmentUrl));
                        messageBodyPart.setDataHandler(new DataHandler(attachmentDataSource));
                    }

                    messageBodyPart.setFileName(attachmentName);

                    if ("true".equalsIgnoreCase(attachmentBase64)) {
                        messageBodyPart.setDataHandler(decodeBase64(attachmentText));
                    } else {
                        messageBodyPart.setText(attachmentText);
                    }

                    multipart.addBodyPart(messageBodyPart);
                }
                msg.setContent(multipart);
            }

            log.debug(sb.toString());
            msg.setSentDate(new Date());
            msg.saveChanges();
            // send the message
            putOnTransport(msg);
            // return the mail in mail-safe from
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            msg.writeTo(out);
            byte[] byteArray = out.toByteArray();
            return Misc.byteArrayToString(byteArray, "\n", false);
        } catch (Exception e) {
            throw new SenderException("MailSender got error", e);
        }
    }

    private DataHandler decodeBase64(String str) {
        byte[] bytesDecoded = Base64.decode(str);
        String encodingType = "application/octet-stream";
        DataSource ads = new ByteArrayDataSource(bytesDecoded, encodingType);
        return new DataHandler(ads);
    }

    private String decodeBase64ToString(String str) {
        byte[] bytesDecoded = Base64.decode(str);
        return new String(bytesDecoded);
    }

    protected void putOnTransport(Message msg) throws SenderException {
        // connect to the transport 
        Transport transport = null;
        try {
            CredentialFactory cf = new CredentialFactory(getSmtpAuthAlias(), getSmtpUserid(), getSmtpPassword());
            transport = session.getTransport("smtp");
            transport.connect(getSmtpHost(), cf.getUsername(), cf.getPassword());
            if (log.isDebugEnabled()) {
                log.debug("MailSender [" + getName() + "] connected transport to URL [" + transport.getURLName()
                        + "]");
            }
            transport.sendMessage(msg, msg.getAllRecipients());
            transport.close();
        } catch (Exception e) {
            throw new SenderException("MailSender [" + getName() + "] cannot connect send message to smtpHost ["
                    + getSmtpHost() + "]", e);
        } finally {
            if (transport != null) {
                try {
                    transport.close();
                } catch (MessagingException e1) {
                    log.warn("MailSender [" + getName() + "] got exception closing connection", e1);
                }
            }
        }
    }

    /**
     * Set the default for From
     */
    public void setDefaultFrom(String newFrom) {
        defaultFrom = newFrom;
    }

    /**
     * Set the default for Subject>
     */
    public void setDefaultSubject(String newSubject) {
        defaultSubject = newSubject;
    }

    /**
     * Name of the SMTP Host.
     */
    public void setSmtpHost(String newSmtpHost) {
        smtpHost = newSmtpHost;
    }

    public String getSmtpHost() {
        return smtpHost;
    }

    public void setSmtpAuthAlias(String string) {
        smtpAuthAlias = string;
    }

    public String getSmtpAuthAlias() {
        return smtpAuthAlias;
    }

    public void setSmtpUserid(java.lang.String newSmtpUserid) {
        smtpUserid = newSmtpUserid;
    }

    public String getSmtpUserid() {
        return smtpUserid;
    }

    public void setSmtpPassword(String newSmtpPassword) {
        smtpPassword = newSmtpPassword;
    }

    public String getSmtpPassword() {
        return smtpPassword;
    }

    public void setDefaultAttachmentName(String string) {
        defaultAttachmentName = string;
    }

    public String getDefaultAttachmentName() {
        return defaultAttachmentName;
    }

    public void setDefaultAttachmentType(String string) {
        defaultAttachmentType = string;
    }

    public String getDefaultAttachmentType() {
        return defaultAttachmentType;
    }

    public int getTimeout() {
        return timeout;
    }

    public void setTimeout(int timeout) {
        this.timeout = timeout;
    }

}