MailReader.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2018 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.reader;
 
import java.util.Properties;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.mail.Flags;
import javax.mail.Folder;
import javax.mail.Message;
import javax.mail.MessagingException;
import javax.mail.Session;
import javax.mail.Store;
import org.turro.collections.KeyValueMap;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.impl.abstracts.AbstractImplementation;
 
public class MailReader extends AbstractImplementation {
 
  private Store store;
  private Folder emailFolder;
 
  public MailReader() {
  }
 
  public boolean isReadOnly() {
    return getAttributes().get("type").startsWith("pop3");
  }
  
  public boolean isSeen(Message message) {
    try {
      return message.isSet(Flags.Flag.SEEN);
    } catch (MessagingException ex) {
      Logger.getLogger(MailReader.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
    return false;
  }
  
  public void seen(Message message) {
    try {
      message.setFlag(Flags.Flag.SEEN, true);
    } catch (MessagingException ex) {
      Logger.getLogger(MailReader.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
  }
  
  public void delete(Message message) {
    try {
      message.setFlag(Flags.Flag.DELETED, true);
    } catch (MessagingException ex) {
      Logger.getLogger(MailReader.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
  }
  
  public void close() {
    try {
      if (emailFolder != null) {
        emailFolder.close(false);
      }
      if (store != null) {
        store.close();
      }
    } catch (MessagingException ex) {
      Logger.getLogger(MailReader.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
  }
 
  public Message[] getMessages() {
    try {
      KeyValueMap kvm = getAttributes();
      Properties properties = new Properties();
 
      if(kvm.get("type").startsWith("pop3")) {
        properties.put("mail.pop3.host", kvm.get("mailhost"));
        properties.put("mail.pop3.port", kvm.get("port"));
        properties.put("mail.pop3.starttls.enable", kvm.get("tls"));
        properties.put("mail.pop3.ssl.enable", kvm.get("ssl"));
      } else if(kvm.get("type").startsWith("imap")) {
        properties.put("mail.imap.host", kvm.get("mailhost"));
        properties.put("mail.imap.port", kvm.get("port"));
        properties.put("mail.imap.starttls.enable", kvm.get("tls"));
        properties.put("mail.imap.ssl.enable", kvm.get("ssl"));
      }
      Session emailSession = Session.getDefaultInstance(properties);
 
      store = emailSession.getStore(kvm.get("type"));
 
      String password = kvm.get("password");
      if (password == null) {
        try {
          password = ElephantContext.decrypt(kvm.get("cryptpass"));
        } catch (Exception ex) {
          Logger.getLogger(MailReader.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
        }
      }
      store.connect(
              kvm.get("mailhost"),
              kvm.get(Integer.class, "port"),
              kvm.get("user"), 
              password);
 
      emailFolder = store.getFolder(kvm.get(String.class, "folder", "INBOX"));
      emailFolder.open(Folder.READ_WRITE);
      
      int count = kvm.get(Integer.class, "count", 50) - 1;
              
      int messageCount = emailFolder.getMessageCount();
 
      return emailFolder.getMessages(messageCount - count, messageCount);      
    } catch (MessagingException ex) {
      Logger.getLogger(MailReader.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
    return null;
  }
 
}