net.oneandone.stool.util.Mailer Maven / Gradle / Ivy

Go to download
/*
 * Copyright 1&1 Internet AG, https://github.com/1and1/
 *
 * 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 net.oneandone.stool.util;

import javax.activation.DataHandler;
import javax.activation.DataSource;
import javax.activation.FileDataSource;
import javax.mail.Address;
import javax.mail.BodyPart;
import javax.mail.Message.RecipientType;
import javax.mail.MessagingException;
import javax.mail.Multipart;
import javax.mail.Session;
import javax.mail.Transport;
import javax.mail.internet.AddressException;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeBodyPart;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;
import java.io.File;
import java.util.Date;
import java.util.Properties;

public class Mailer {
    private final String smtphost;
    private final String username;
    private final String password;

    public Mailer(String smtphost, String username, String password) {
        this.smtphost = smtphost;
        this.username = username;
        this.password = password;
    }

    public void send(String from, String[] to, String subject, String text, File ... attachments) throws MessagingException {
        Properties props;
        MimeMessage msg;
        Session session;
        Multipart multipart;
        int i;
        Transport transport;

        props = new Properties();
        props.put("mail.smtp.host", smtphost);
        props.put("mail.smtp.timeout", 30 * 1000);
        props.put("mail.smtp.connectiontimeout", 30 * 1000);
        session = Session.getDefaultInstance(props, null);

        msg = new MimeMessage(session);
        msg.setFrom(new InternetAddress(from));
        msg.setRecipients(RecipientType.TO, addresses(to));
        msg.setSubject(subject);
        msg.setSentDate(new Date());

        if (attachments.length == 0) {
            // avoid evolution attachment marker with no attachments
            msg.setText(text);
        } else {
            multipart = new MimeMultipart();
            addText(multipart, text);
            for (i = 0; i < attachments.length; i++) {
                addAttachment(multipart, attachments[i]);
            }
            msg.setContent(multipart);
        }
        transport = session.getTransport("smtp");
        transport.connect(smtphost, username, password);
        transport.sendMessage(msg, addresses(to));
        transport.close();
    }

    private static Address[] addresses(String[] tos) throws AddressException {
        Address[] result;

        result = new Address[tos.length];
        for (int i = 0; i < tos.length; i++) {
            result[i] = new InternetAddress(tos[i]);
        }
        return result;
    }

    private static void addText(Multipart dest, String text) throws MessagingException {
        BodyPart messageBodyPart;

        messageBodyPart = new MimeBodyPart();
        messageBodyPart.setText(text);
        dest.addBodyPart(messageBodyPart);
    }

    private void addAttachment(Multipart dest, File file) throws MessagingException {
        MimeBodyPart part;
        String filename;
        DataSource source;

        filename = file.getPath();
        part = new MimeBodyPart();
        source = new FileDataSource(filename);
        part.setDataHandler(new DataHandler(source));
        part.setFileName(file.getName());
        dest.addBodyPart(part);
    }

    @Override
    public String toString() {
        return "Mailer(smtphost=" + smtphost + ", username=" + username + ", password=" + password + ")";
    }
}