elephant-mail/src/main/java/org/turro/mail/message/MailMessage.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2022 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.message;
 
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.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.mail.provider.MailProvider;
import org.turro.string.Strings;
 
public class MailMessage  {
 
  private final HtmlEmail email = new HtmlEmail();
 
  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 subject, message;
  
  public HtmlEmail getEmail() {
    return email;
  }
 
  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 MailProvider getProvider() {
    return provider;
  }
 
  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 {
    provider.setupMail(email);
 
    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 MailInline().inline(message);
    
    email.setHtmlMsg(message);
    email.setTextMsg("NO HTML SUPPORTED");
 
    email.send();
  }
 
  /* 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, ElephantContext.getEncoding());
  }
 
  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], ElephantContext.getEncoding());
      }
      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, ElephantContext.getEncoding());
  }
 
  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, ElephantContext.getEncoding());
  }
 
  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, ElephantContext.getEncoding());
  }
 
  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, ElephantContext.getEncoding());
  }
 
  public Email addBcc(String email) throws EmailException {
    return this.email.addBcc(email);
  }
 
  /* Factory */
  
  public static MailMessage of(MailProvider provider) {
    return new MailMessage(provider);
  }
  
  protected final MailProvider provider;
 
  protected MailMessage(MailProvider provider) {
    this.provider = provider;
    email.setCharset(ElephantContext.getEncoding());
  }
  
}