org.tinygroup.message.email.EmailMessageReceiveService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.message Show documentation
The newest version!
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.gnu.org/licenses/gpl.html
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.tinygroup.message.email;

import org.tinygroup.message.MessageException;
import org.tinygroup.message.MessageReceiveService;

import javax.mail.*;
import javax.mail.internet.AddressException;
import javax.mail.internet.InternetAddress;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

/**
 * 邮件接收服务
 * Created by luoguo on 2014/4/18.
 */
public class EmailMessageReceiveService implements MessageReceiveService {
    private EmailMessageFlagMarker emailMessageFlagMarker;

    public EmailMessageFlagMarker getEmailMessageFlagMarker() {
        return emailMessageFlagMarker;
    }

    public void setEmailMessageFlagMarker(EmailMessageFlagMarker emailMessageFlagMarker) {
        this.emailMessageFlagMarker = emailMessageFlagMarker;
    }


    public Collection getMessages(EmailMessageAccount messageAccount) throws MessageException {

        try {
            Session session = EmailMessageUtil.getSession(messageAccount);
            Store store = session.getStore("pop3");
            store.connect(messageAccount.getHost(), messageAccount.getUsername(), messageAccount.getPassword());
            Folder folder = store.getFolder("inbox");
            folder.open(Folder.READ_WRITE);
            Message[] messages = folder.getMessages();
            List receiveMessages = new ArrayList();
            for (int i = 0; i < messages.length; i++) {
                EmailReceiveMessage message = new EmailReceiveMessage();
                receiveMessages.add(message);
                message.setMessage(getEmailMessage(messages[i]));
                message.setMessageSenders(getMessageSender(messages[i]));
                message.setReceiveDate(messages[i].getReceivedDate());
                message.setSendDate(messages[i].getSentDate());
                message.setMessageReceivers(getMessageReceivers(messages[i]));
                if (emailMessageFlagMarker != null) {
                    messages[i].setFlag(emailMessageFlagMarker.getFlag(message, messages[i]), true);
                }
            }
            folder.close(true);
            store.close();
            return receiveMessages;
        } catch (Exception e) {
            throw new MessageException(e);
        }

    }


    private Collection getMessageReceivers(Message message) throws MessagingException {
        List receivers = new ArrayList();

        addReceivers(receivers, Message.RecipientType.TO, message.getReplyTo());
        addReceivers(receivers, Message.RecipientType.TO, message.getRecipients(Message.RecipientType.TO));
        addReceivers(receivers, Message.RecipientType.CC, message.getRecipients(Message.RecipientType.CC));
        addReceivers(receivers, Message.RecipientType.BCC, message.getRecipients(Message.RecipientType.BCC));
        return receivers;
    }

    private void addReceivers(List receivers, Message.RecipientType type, Address[] replyTo) throws AddressException {
        if (replyTo != null) {
            for (Address address : replyTo) {
                InternetAddress internetAddress = new InternetAddress(address.toString());
                EmailMessageReceiver messageReceiver = new EmailMessageReceiver();
                receivers.add(messageReceiver);
                messageReceiver.setEmail(internetAddress.getAddress());
                messageReceiver.setDisplayName(internetAddress.getPersonal());
                messageReceiver.setType(type);
            }
        }
    }

    private Collection getMessageSender(Message message) throws MessagingException, IOException {
        Address[] addresses = message.getFrom();
        List messageSenders = new ArrayList();
        for (Address address : addresses) {
            InternetAddress internetAddress = new InternetAddress(address.toString());
            EmailMessageSender messageSender = new EmailMessageSender();
            messageSenders.add(messageSender);
            messageSender.setEmail(internetAddress.getAddress());
            messageSender.setDisplayName(internetAddress.getPersonal());
        }
        return messageSenders;
    }

    private EmailMessage getEmailMessage(Message message) throws MessagingException, IOException, MessageException {
        EmailMessage emailMessage = new EmailMessage();
        emailMessage.setSubject(message.getSubject());
        getContent(emailMessage, message);
        return emailMessage;
    }

    private void getContent(EmailMessage emailMessage, Part part) throws MessagingException, IOException, MessageException {
        String contentType = part.getContentType();
        int nameIndex = contentType.indexOf("name");
        if (nameIndex != -1) {
            processBodyPart(emailMessage, part);
        } else if (part.isMimeType("text/plain") || part.isMimeType("text/html")) {
            emailMessage.setContent(part.getContent().toString());
        } else if (part.isMimeType("multipart/*")) {
            processMultipart(emailMessage, part);
        } else if (part.isMimeType("message/rfc822")) {
            getContent(emailMessage, (Part) part.getContent());
        } else {
            throw new MessageException("不支持的ContentType:" + contentType);
        }
    }

    private void processMultipart(EmailMessage emailMessage, Part part) throws IOException, MessagingException, MessageException {
        Multipart multipart = (Multipart) part.getContent();
        int counts = multipart.getCount();
        for (int i = 0; i < counts; i++) {
            getContent(emailMessage, multipart.getBodyPart(i));
        }
    }

    private void processBodyPart(EmailMessage emailMessage, Part part) throws MessagingException, IOException {
        BodyPart bodyPart = (BodyPart) part;
        String disposition = bodyPart.getDisposition();
        if ((disposition != null)
                && ((disposition.equals(Part.ATTACHMENT)) || (disposition
                .equals(Part.INLINE)))) {
            EmailAccessory accessory = new EmailAccessory();
            accessory.setFileName(part.getFileName());
            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
            part.writeTo(outputStream);
            accessory.setContent(outputStream.toByteArray());
            emailMessage.getAccessories().add(accessory);
        }
    }
}