MailMessagePool.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.util.Date;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.function.Consumer;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.mail.internet.InternetAddress;
import org.apache.commons.mail.EmailException;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.impl.abstracts.AbstractImplementation;
import org.turro.marker.ElephantMarker;
 
public class MailMessagePool extends AbstractImplementation {
 
  private final Date date = new Date();
  private final Set<MailMessage> pool = new HashSet<>();
  private String encoding;
 
  public MailMessagePool() {
  }
 
  public void setEncoding(String encoding) {
    this.encoding = encoding;
  }
 
  public MailMessage addToPoolTemplate(String from, String to, String cc, String subject, 
          String message, String template, String name, String login) {
    return addToPoolTemplate(from, to, cc, subject, message, template, name, login, null);
  }
  
  public MailMessage addToPoolTemplate(String from, String to, String cc, String subject, 
          String message, String template, String name, String login, Map args) {
    try {
      if (to == null) {
        return null;
      }
      MailMessageTemplate mmt = new MailMessageTemplate();
      mmt.setCharset(encoding);
      mmt.setSubject(subject);
      mmt.setFrom(from == null ? (String) getAttributes().get("from") : from);
      mmt.addTo(to);
      if (cc != null) {
        mmt.addCc(cc);
      }
      ElephantMarker em = new ElephantMarker(constructor, true);
      em.put("email", to);
      em.put("body", message);
      if(args != null) {
        em.putAll(args);
      }
      mmt.setMessage(em, template, to, name);
      return addToPool(mmt);
    } catch (EmailException ex) {
      Logger.getLogger(MailMessagePool.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
    return null;
  }
 
  public MailMessage addToPool(String from, String to, String cc, String subject, String message) {
    try {
      if (to == null) {
        return null;
      }
      MailMessage mail = new MailMessage();
      mail.setCharset(encoding);
      mail.setFrom(from == null ? (String) getAttributes().get("from") : from);
      mail.addTo(to);
      if (cc != null) {
        mail.addCc(cc);
      }
      mail.setSubject(subject);
      mail.setMessage(message);
      return addToPool(mail);
    } catch (EmailException ex) {
      Logger.getLogger(MailMessagePool.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
    return null;
  }
 
  public MailMessage addToPool(MailMessage mail) throws EmailException {
    try {
      mail.addHeader("X-Mailer", "Elephant Mail System");
      mail.setSentDate(date);
      mail.setHostName((String) getAttributes().get("mailhost"));
      mail.setPort((String) getAttributes().get("port"));
      mail.setTls((String) getAttributes().get("tls"));
      mail.setSsl((String) getAttributes().get("ssl"));
      mail.setUser((String) getAttributes().get("user"));
      String password = (String) getAttributes().get("password");
      if(password == null) {
        password = ElephantContext.decrypt((String) getAttributes().get("cryptpass"));
      }
      mail.setPassword(password);
      mail.setFromIfNull((String) getAttributes().get("from"));
      mail.setSubject(cleanSubject(mail.getSubject()));
 
      pool.add(mail);
 
      Logger.getLogger(MailMessagePool.class.getName()).log(Level.INFO, "Added mail from {0} to {1}. Subject: {2}",
          new Object[]{mail.getEmail().getFromAddress().getAddress(),
          InternetAddress.toString((InternetAddress[]) mail.getEmail().getToAddresses().toArray(new InternetAddress[0])), mail.getSubject()});
 
      return mail;
    } catch(Exception ex) {
      Logger.getLogger(MailMessagePool.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
    return null;
  }
 
  public void sendPool() {
    sendPool(null);
  }
  
  public void sendPool(Consumer onFinish) {
    MailQueueConsumer.getInstance().addToQueue(pool, onFinish);
    //new MailMessagePoolSender(pool).send(onFinish);
    pool.clear();
  }
 
  private String cleanSubject(String subject) {
    subject = subject.replaceAll("\\<\\/?[\\?a-zA-Z\\-\\:\\_0-9]+\\ ?.*?\\>", "");
    subject = subject.replaceAll("\\n|\\r", " ");
    return subject;
  }
  
}