MailMessage.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.turro.mail.impl;
 
import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import javax.naming.NamingException;
import org.amic.util.string.Strings;
import org.apache.commons.mail.Email;
import org.apache.commons.mail.EmailAttachment;
import org.apache.commons.mail.EmailException;
import org.apache.commons.mail.HtmlEmail;
import org.turro.elephant.context.ElephantContext;
import org.turro.math.StringNumber;
 
public class MailMessage {
 
  private final Map<String, File> imageMap = new HashMap<>();
  private final Map<String, URL> imageURLMap = new HashMap<>();
  private final Set<EmailAttachment> attachments = new HashSet<>();
 
  private String hostName, user, password, subject, message, port, tls, ssl;
 
  private HtmlEmail email = new HtmlEmail();
 
  public MailMessage() {
  }
 
  public MailMessage(String hostName, String user, String password) {
    this.hostName = hostName;
    this.user = user;
    this.password = password;
  }
 
  public HtmlEmail getEmail() {
    return email;
  }
 
  public void setHostName(String hostName) {
    this.hostName = hostName;
  }
 
  public void setPassword(String password) {
    this.password = password;
  }
 
  public void setUser(String user) {
    this.user = user;
  }
 
  public String getMessage() {
    return message;
  }
 
  public void setMessage(String message) {
    this.message = message;
  }
 
  public String getSubject() {
    return subject;
  }
 
  public void setSubject(String subject) {
    this.subject = subject;
  }
 
  public String getPort() {
    return port;
  }
 
  public void setPort(String port) {
    this.port = port;
  }
 
  public String getTls() {
    return tls;
  }
 
  public void setTls(String tls) {
    this.tls = tls;
  }
 
  public String getSsl() {
    return ssl;
  }
 
  public void setSsl(String ssl) {
    this.ssl = ssl;
  }
 
  public void attachRelative(String path, String description, String name) {
    EmailAttachment attachment = new EmailAttachment();
    attachment.setPath(ElephantContext.getRealPath(path));
    attachment.setDisposition(EmailAttachment.ATTACHMENT);
    attachment.setDescription(description == null ? "" : description);
    attachment.setName(name == null ? "" : name);
    attachments.add(attachment);
  }
 
  public void attach(String path, String description, String name) {
    EmailAttachment attachment = new EmailAttachment();
    attachment.setPath(path);
    attachment.setDisposition(EmailAttachment.ATTACHMENT);
    attachment.setDescription(description == null ? "" : description);
    attachment.setName(name == null ? "" : name);
    attachments.add(attachment);
  }
 
  public void attachExternal(URL url, String description, String name) {
    EmailAttachment attachment = new EmailAttachment();
    attachment.setURL(url);
    attachment.setDisposition(EmailAttachment.ATTACHMENT);
    attachment.setDescription(description == null ? "" : description);
    attachment.setName(name == null ? "" : name);
    attachments.add(attachment);
  }
 
  public void embedRelative(String name, String path) {
    embed(name, new File(ElephantContext.getRealPath(path)));
  }
 
  public void embed(String name, File file) {
    imageMap.put(name, file);
  }
 
  public void embedExternal(String name, String url) throws MalformedURLException {
    imageURLMap.put(name, new URL(url));
  }
 
  public void send() throws EmailException, MalformedURLException, NamingException {
    email.setCharset(ElephantContext.getEncoding());
    if (hostName.startsWith("java:")) {
      email.setMailSessionFromJNDI(hostName);
    } else {
      email.setHostName(hostName);
    }
    if(StringNumber.toInteger(port) != null) {
      email.setSmtpPort(StringNumber.toInteger(port));
    }
    if(StringNumber.toBoolean(ssl)) {
      email.setSSLOnConnect(true);
      if(StringNumber.toInteger(port) != null) {
        email.setSslSmtpPort(port);
      }
    }
    if(StringNumber.toBoolean(tls)) {
      email.setStartTLSEnabled(true);
    }
    if(user != null && password != null) {
      email.setAuthentication(user, password);
    }
 
    for(String n : imageMap.keySet()) {
      String cid = email.embed(imageMap.get(n), n);
      message = message.replaceAll("image\\(" + n + "\\)", "<img src=\"cid:" + cid + "\"/>");
    }
 
    for(String n : imageURLMap.keySet()) {
      String cid = email.embed(imageURLMap.get(n), n);
      message = message.replaceAll("imageURL\\(" + n + "\\)", "<img src=\"cid:" + cid + "\"/>");
    }
 
    email.setSubject(subject);
    
    for(EmailAttachment ea : attachments) {
      email.attach(ea);
    }
    
    String serverBase = ElephantContext.getServerBase("http");
    message = message.replaceAll("href=\\'(?![a-z]+:)(\\/?)\\/?([^\\']*)\\'", "href=\\'" + serverBase + "$1$2\\'")
                   .replaceAll("src=\\'(?![a-z]+:)(\\/?)\\/?([^\\']*)\\'", "src=\\'" + serverBase + "$1$2\\'")
                   .replaceAll("href=\\\"(?![a-z]+:)(\\/?)\\/?([^\\\"]*)\\\"", "href=\\\"" + serverBase + "$1$2\\\"")
                   .replaceAll("src=\\\"(?![a-z]+:)(\\/?)\\/?([^\\\"]*)\\\"", "src=\\\"" + serverBase + "$1$2\\\"");
    
    message = new MailFormat().inline(message);
    
    email.setHtmlMsg(message);
    email.setTextMsg("NO HTML SUPPORTED");
 
    email.send();
  }
 
  public Email setFromIfNull(String email) throws EmailException {
    if(this.email.getFromAddress() == null && !Strings.isBlank(email)) {
      return setFrom(email);
    }
    return this.email;
  }
 
  /* Delegates */
 
  public Date getSentDate() {
    return email.getSentDate();
  }
 
  public void setSentDate(Date date) {
    email.setSentDate(date);
  }
 
  public Email setFrom(String email, String name) throws EmailException {
    return this.email.setFrom(email, name);
  }
 
  public Email setFrom(String email) throws EmailException {
    if(email != null) {
      if(email.contains("|")) {
        String[] parts = email.split("\\|");
        return this.email.setFrom(parts[0], parts[1]);
      }
      return this.email.setFrom(email);
    }
    return this.email;
  }
 
  public Email addTo(String email, String name) throws EmailException {
    if(Strings.isBlank(email)) return null;
    return this.email.addTo(email, name);
  }
 
  public Email addTo(String email) throws EmailException {
    if(Strings.isBlank(email)) return null;
    return this.email.addTo(email);
  }
 
  public Email addReplyTo(String email, String name) throws EmailException {
    return this.email.addReplyTo(email, name);
  }
 
  public Email addReplyTo(String email) throws EmailException {
    return this.email.addReplyTo(email);
  }
 
  public void addHeader(String name, String value) {
    email.addHeader(name, value);
  }
 
  public Email addCc(String email, String name) throws EmailException {
    return this.email.addCc(email, name);
  }
 
  public Email addCc(String email) throws EmailException {
    return this.email.addCc(email);
  }
 
  public Email addBcc(String email, String name) throws EmailException {
    return this.email.addBcc(email, name);
  }
 
  public Email addBcc(String email) throws EmailException {
    return this.email.addBcc(email);
  }
 
  public void setCharset(String newCharset) {
    email.setCharset(newCharset);
  }
 
}