Class: com.technoetic.xplanner.mail.EmailMessageImpl   ©

package com.technoetic.xplanner.mail;

import com.technoetic.xplanner.XPlannerProperties;
import com.technoetic.xplanner.db.hibernate.ThreadSession;
import com.technoetic.xplanner.domain.Person;
import net.sf.hibernate.HibernateException;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;

import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.mail.Message;
import javax.mail.MessagingException;
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.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Date;
import java.util.Iterator;
import java.util.Properties;

public class EmailMessageImpl implements EmailMessage {
    protected Logger log = Logger.getLogger(getClass());
    private Message msg;
    private StringWriter bodyWriter;
    private ArrayList attachments = new ArrayList();

    /*package scope*/
    EmailMessageImpl() throws MessagingException {
        Properties transportProperties = new Properties();
        transportProperties.put("mail.smtp.host", new XPlannerProperties().get().get("xplanner.mail.smtp.host"));
        Session session = Session.getDefaultInstance(transportProperties, null);
        session.setDebug(log.isDebugEnabled());
        msg = new MimeMessage(session);
        msg.setSentDate(new Date());
    }

    public void setFrom(String from) throws AddressException, MessagingException {
        msg.setFrom(new InternetAddress(from));
    }

    public void setRecipient(int personId) throws MessagingException {
        try {
            Person person = getPerson(personId);
Rate            if (StringUtils.isEmpty(person.getEmail())) {
                throw new MessagingException("no email address for user: uid=" + person.getUserId() + ",id=" + person.getId());
            }
            setRecipients(person.getEmail());
        } catch (MessagingException e) {
            throw e;
        } catch (Exception e) {
            throw new MessagingException("error setting recipient", e);
        }
    }

    private Person getPerson(int personId) throws HibernateException {
        return (Person)ThreadSession.get().load(Person.class, new Integer(personId));
    }

    public void setRecipients(String recipients) throws AddressException, MessagingException {
        if (StringUtils.isNotEmpty(recipients)) {
            setRecipients(Message.RecipientType.TO, recipients.split(","));
        }
    }

    public void setRecipients(String[] recipients) throws AddressException, MessagingException {
        InternetAddress[] addresses = new InternetAddress[recipients.length];
        for (int i = 0; i < recipients.length; i++) {
            addresses[i] = new InternetAddress(recipients[i]);
        }
        msg.setRecipients(Message.RecipientType.TO, addresses);
    }

    public void setCcRecipients(String recipients) throws MessagingException, AddressException {
        if (StringUtils.isNotEmpty(recipients)) {
            setRecipients(Message.RecipientType.CC, recipients.split(","));
        }
    }

    private void setRecipients(Message.RecipientType recipientType, String[] recipients) throws AddressException, MessagingException {
        if (recipients.length > 0) {
            InternetAddress[] addresses = new InternetAddress[recipients.length];
            for (int i = 0; i < recipients.length; i++) {
                addresses[i] = new InternetAddress(recipients[i]);
            }
            msg.setRecipients(recipientType, addresses);
        }
    }

    public void setBody(String body) throws MessagingException {
        bodyWriter = new StringWriter();
        bodyWriter.write(body);
    }

    public PrintWriter getBodyWriter() {
        bodyWriter = new StringWriter();
        return new PrintWriter(bodyWriter);
    }

    public void setSubject(String subject) throws MessagingException {
        msg.setSubject(subject);
    }

    public void setSentDate(Date sentDate) throws MessagingException {
        msg.setSentDate(sentDate);
    }

    public void addAttachment(String filename) throws MessagingException {
        File file = new File(filename);
        addAttachment(file.getName(), file);
    }

    public void addAttachment(String filename, File file) throws MessagingException {
        MimeBodyPart part = new MimeBodyPart();
        FileDataSource fds = new FileDataSource(file);
        part.setDataHandler(new DataHandler(fds));
        part.setFileName(filename);
        attachments.add(part);
    }

    public void send() throws MessagingException {
        MimeMultipart parts = new MimeMultipart();
        if (bodyWriter == null) {
            setBody("");
        }
        MimeBodyPart bodyPart = new MimeBodyPart();
        bodyPart.setText(bodyWriter.toString());
        parts.addBodyPart(bodyPart);
        Iterator iter = attachments.iterator();
        while (iter.hasNext()) {
            parts.addBodyPart((MimeBodyPart)iter.next());
        }
        msg.setContent(parts);
        Transport.send(msg);
    }
}