elephant-mail/src/main/java/org/turro/mail/pool/MailMessagePool.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.pool;
 
import java.util.Date;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.function.Consumer;
import javax.mail.internet.InternetAddress;
import org.apache.commons.mail.EmailException;
import org.turro.elephant.context.IConstructor;
import org.turro.log.WebLoggers;
import org.turro.mail.message.MailMessage;
import org.turro.mail.message.MailMessageTemplate;
import org.turro.mail.provider.MailProvider;
import org.turro.marker.ElephantMarker;
 
public class MailMessagePool {
  
  private final Date date = new Date();
  private final Set<MailMessage> pool = new HashSet<>();
 
  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 = MailMessageTemplate.of(provider);
      mmt.setSubject(subject);
      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) {
      WebLoggers.severe(this).exception(ex).log();
    }
    return null;
  }
 
  public MailMessage addToPool(String from, String to, String cc, String subject, String message) {
    try {
      if (to == null) {
        return null;
      }
      MailMessage mail = MailMessage.of(provider);
      mail.addTo(to);
      if (cc != null) {
        mail.addCc(cc);
      }
      mail.setSubject(subject);
      mail.setMessage(message);
      return addToPool(mail);
    } catch (EmailException ex) {
      WebLoggers.severe(this).exception(ex).log();
    }
    return null;
  }
 
  public MailMessage addToPool(MailMessage mail) throws EmailException {
    try {
      mail.addHeader("X-Mailer", "Elephant Mail System");
      mail.setSentDate(date);
      mail.setSubject(cleanSubject(mail.getSubject()));
 
      pool.add(mail);
 
      WebLoggers.info(this).message("Added mail from %s to %s. Subject: %s", 
              mail.getProvider().getFrom().getMail(),
              InternetAddress.toString((InternetAddress[]) mail.getEmail().getToAddresses().toArray(new InternetAddress[0])),
              mail.getSubject()
              ).log();
 
      return mail;
    } catch(Exception ex) {
      WebLoggers.severe(this).exception(ex).log();
    }
    return null;
  }
 
  public void sendPool() {
    sendPool(null);
  }
  
  public void sendPool(Consumer onFinish) {
    MailQueueConsumer.getInstance().addToQueue(pool, onFinish);
    pool.clear();
  }
 
  public MailProvider getProvider() {
    return provider;
  }
 
  private String cleanSubject(String subject) {
    subject = subject.replaceAll("\\<\\/?[\\?a-zA-Z\\-\\:\\_0-9]+\\ ?.*?\\>", "");
    subject = subject.replaceAll("\\n|\\r", " ");
    return subject;
  }
  
  /* Factory */
  
  private final IConstructor constructor;
  private final MailProvider provider;
  
  public static MailMessagePool of(IConstructor constructor, MailProvider provider) {
    return new MailMessagePool(constructor, provider);
  }
 
  public MailMessagePool(IConstructor constructor, MailProvider provider) {
    this.constructor = constructor;
    this.provider = provider;
  }
  
}