elephant/src/main/java/org/turro/mail/impl/MailPool.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.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.Date;
import java.util.HashSet;
import java.util.Properties;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.mail.Message;
import javax.mail.MessagingException;
import javax.mail.Session;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeMessage;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import org.turro.string.Strings;
import org.turro.assistant.Assistant;
import org.turro.assistant.AssistantSet;
import org.turro.assistant.Assistants;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.context.HeadlessApplication;
import org.turro.elephant.impl.abstracts.AbstractImplementation;
import org.turro.elephant.impl.context.ContextFactory;
import org.turro.elephant.impl.util.FileUtil;
import org.turro.elephant.impl.util.Parser;
import org.turro.elephant.impl.util.StringParser;
 
@Deprecated
public class MailPool extends AbstractImplementation {
 
  protected Session session = null;
  private final Date date = new Date();
  private final Set<MimeMessage> pool = new HashSet<>();
  private final Set<String> cssFiles = new HashSet<>();
  private String encoding;
 
  public MailPool() {
  }
 
  public void setEncoding(String encoding) {
    this.encoding = encoding;
  }
 
  public String getTemplateString(String attribute) {
    return loadFromFile((String) getAttributes().get(attribute));
  }
  
  public void addToPool(String from, String to, String cc, String subject, String message, String content) {
    String poolFrom = (String) getAttributes().get("from");
    String poolTo = (String) getAttributes().get("to");
    String template = getTemplateString("template");
 
    if(to == null) {
      to = poolTo;
    }
    if(to == null) {
      return;
    }
    try {
 
      if (session == null) {
        session = getMailSession();
      }
 
      MimeMessage msg = new MimeMessage(session);
 
      if (from != null) {
        msg.setFrom(new InternetAddress(from));
      } else if (poolFrom != null) {
        msg.setFrom(new InternetAddress(poolFrom));
      } else {
        msg.setFrom();
      }
      msg.setRecipients(Message.RecipientType.TO, InternetAddress.parse(to, false));
 
      if (cc != null) {
        msg.setRecipients(Message.RecipientType.CC, InternetAddress.parse(cc, false));
      }
      msg.setSubject(cleanSubject(subject), encoding);
 
      message = template == null ? message : template.replaceAll("\\#body", Parser.escapeGrouping(message));
      
      if (content == null) {
        msg.setText(addStyles(message), encoding);
      } else {
        msg.setContent(addStyles(message), content + ";charset=\"" + encoding + "\"");
      }
      msg.setHeader("X-Mailer", "Elephant Mail System");
      msg.setHeader("Content-Transfer-Encoding", "quoted-printable");
      msg.setSentDate(date);
 
      pool.add(msg);
 
      Logger.getLogger(MailPool.class.getName()).log(Level.INFO, "Added mail from {0} to {1}. Subject: {2}",
          new Object[]{InternetAddress.toString(msg.getFrom()),
          InternetAddress.toString(msg.getAllRecipients()), msg.getSubject()});
 
    } catch (MessagingException mex) {
      Logger.getLogger(MailPool.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), mex);
    }
  }
 
  public void sendPool() {
    String mailhost = (String) getAttributes().get("mailhost");
    String user = (String) getAttributes().get("user");
    String password = (String) getAttributes().get("password");
    
    if(password == null) {
      try {
        password = ElephantContext.decrypt((String) getAttributes().get("cryptpass"));
      } catch (Exception ex) {
        Logger.getLogger(MailPool.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      }
    }    
    
    //new MailPoolSender(session, pool, mailhost, user, password).send();
  }
 
  protected Session getMailSession() throws MessagingException {
    String mailhost = (String) getAttributes().get("mailhost");
    String user = (String) getAttributes().get("user");
 
    Session tmpSession = null;
 
    if (mailhost.startsWith("java:")) {
      try {
        InitialContext ic = new InitialContext();
        tmpSession = (Session) ic.lookup(mailhost);
      } catch (NamingException ex) {
        Logger.getLogger(MailPool.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
        throw new MessagingException("Naming problem with: " + mailhost + "\n\n" + ex.getExplanation());
      }
    } else {
      Properties props = System.getProperties();
      tmpSession = Session.getDefaultInstance(props, null);
      if (mailhost != null) {
        props.put("mail.smtp.host", mailhost);
      }
      if (user != null) {
        props.put("mail.smtp.auth", "true");
      } else {
        props.put("mail.smtp.auth", "false");
      }
    }
 
    return tmpSession;
  }
 
  public void addCssFile(String cssFile) {
    if(cssFiles.isEmpty()) {
      String cssS = (String) getAttributes().get("cssS");
      if(!Strings.isBlank(cssS)) {
        String[] cssA = cssS.split(",");
        for(String css1 : cssA) {
          cssFiles.add(ElephantContext.getRealPath(css1));
        }
      }
    }
    cssFiles.add(cssFile);
  }
 
  private String cleanSubject(String subject) {
    subject = subject.replaceAll("\\<\\/?[\\?a-zA-Z\\-\\:\\_0-9]+\\ ?.*?\\>", "");
    subject = subject.replaceAll("\\n|\\r", " ");
    return subject;
  }
 
  private String addStyles(String message) {
    File css;
    String style = "";
 
    for (String fileName : cssFiles) {
      css = new File(fileName);
      if (css.exists()) {
        try {
          BufferedReader br = new BufferedReader(new FileReader(css));
          String l;
          while ((l = br.readLine()) != null) {
            style += l + "\n";
          }
        } catch (Exception ex) {
          Logger.getLogger(MailPool.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
        }
      }
    }
 
    if(!Strings.isBlank(style)) {
      style = "<style type='text/css'>\n" + style + "\n</style>\n";
    }
 
    return style + "<div class='mail-body'>" + message + "</div>";
  }
 
  private String loadFromFile(String file) {
    if(!Strings.isBlank(file)) {
      try {
        return FileUtil.getContent(new File(ElephantContext.getRealPath(file)));
      } catch (IOException ex) {
        return null;
      }
    } else {
      return null;
    }
  }
 
  public static void sendMailToAssistants(String path, String subject, String message, Object data) {
    AssistantSet al = new AssistantSet();
    Assistants.addAssistants(path, true, al, data);
    if(!al.isEmpty()) {
      MailPool mp = (MailPool) ContextFactory.getImplementation(HeadlessApplication.getInstance().getConstructor(), "IMailPool");
      mp.setEncoding(ElephantContext.getEncoding());
      mp.addCssFile(ElephantContext.getRealPath("/_internal/css/mail.css"));
      for(Assistant a : al) {
        mp.addToPool(null, a.email, null, subject + " : " + ElephantContext.getSiteName(), 
                StringParser.toHTML(al.getSubject() + "\n\n" + message), "text/html");
      }
      mp.sendPool();
    }
  }
 
  public static void sendMailToAdmin(String subject, String message) {
    MailPool mp = (MailPool) ContextFactory.getImplementation(HeadlessApplication.getInstance().getConstructor(), "IMailPool");
    mp.setEncoding(ElephantContext.getEncoding());
    mp.addCssFile(ElephantContext.getRealPath("/_internal/css/mail.css"));
    mp.addToPool(null, null, null, subject + " : " + ElephantContext.getSiteName(), 
            StringParser.toHTML(message), "text/html");
    mp.sendPool();
  }
 
  public static void sendMailToRole(String role, String subject, String message, Object data) {
    AssistantSet al = new AssistantSet();
    Assistants.addAssistants(role, al, data);
    if(!al.isEmpty()) {
      MailPool mp = (MailPool) ContextFactory.getImplementation(HeadlessApplication.getInstance().getConstructor(), "IMailPool");
      mp.setEncoding(ElephantContext.getEncoding());
      mp.addCssFile(ElephantContext.getRealPath("/_internal/css/mail.css"));
      for(Assistant a : al) {
        mp.addToPool(null, a.email, null, subject + " : " + ElephantContext.getSiteName(), 
                StringParser.toHTML(al.getSubject() + "\n\n" + message), "text/html");
      }
      mp.sendPool();
    }
  }
 
  public static void sendMailToAssistantsAndAdmin(String path, String subject, String message, Object data) {
    MailPool mp = (MailPool) ContextFactory.getImplementation(HeadlessApplication.getInstance().getConstructor(), "IMailPool");
    mp.setEncoding(ElephantContext.getEncoding());
    mp.addCssFile(ElephantContext.getRealPath("/_internal/css/mail.css"));
    AssistantSet al = new AssistantSet();
    Assistants.addAssistants(path, true, al, data);
    String to = (String) mp.getAttributes().get("to");
    if(to != null) {
      al.add(new Assistant(null, to, null, null));
    }
    if(!al.isEmpty()) {
      for(Assistant a : al) {
        mp.addToPool(null, a.email, null, subject + " : " + ElephantContext.getSiteName(), 
                StringParser.toHTML(al.getSubject() + "\n\n" + message), "text/html");
      }
      mp.sendPool();
    }
  }
 
  public static void sendMailToRoleAndAdmin(String role, String subject, String message, Object data) {
    MailPool mp = (MailPool) ContextFactory.getImplementation(HeadlessApplication.getInstance().getConstructor(), "IMailPool");
    mp.setEncoding(ElephantContext.getEncoding());
    mp.addCssFile(ElephantContext.getRealPath("/_internal/css/mail.css"));
    AssistantSet al = new AssistantSet();
    Assistants.addAssistants(role, al, data);
    String to = (String) mp.getAttributes().get("to");
    if(to != null) {
      al.add(new Assistant(null, to, null, null));
    }
    if(!al.isEmpty()) {
      for(Assistant a : al) {
        mp.addToPool(null, a.email, null, subject + " : " + ElephantContext.getSiteName(), 
                StringParser.toHTML(al.getSubject() + "\n\n" + message), "text/html");
      }
      mp.sendPool();
    }
  }
 
}