com.spartasystems.holdmail.smtp.SMTPHandler.java Source code

Introduction

Here is the source code for com.spartasystems.holdmail.smtp.SMTPHandler.java

Source

/*******************************************************************************
 * Copyright 2016 - 2017 Sparta Systems, Inc
 *
 * Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
 *
 * 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 com.spartasystems.holdmail.smtp;

import com.spartasystems.holdmail.domain.Message;
import com.spartasystems.holdmail.domain.MessageHeaders;
import com.spartasystems.holdmail.service.MessageService;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.CharEncoding;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.subethamail.smtp.MessageContext;
import org.subethamail.smtp.MessageHandler;

import javax.mail.Header;
import javax.mail.MessagingException;
import javax.mail.Session;
import javax.mail.internet.MimeMessage;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.Date;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;

@Component
public class SMTPHandler implements MessageHandler {

    private Logger logger = LoggerFactory.getLogger(SMTPHandler.class);

    @Autowired
    private MessageService messageService;

    private byte[] data = {};
    private String senderHost;
    private String senderEmail;
    private List<String> recipients = new ArrayList<>();

    public SMTPHandler() {
    }

    public SMTPHandler(MessageContext ctx) {

        InetSocketAddress hostAddr = (InetSocketAddress) ctx.getRemoteAddress();
        this.senderHost = hostAddr.getAddress().getHostAddress();

    }

    @Override
    public void from(String from) {
        this.senderEmail = from;
    }

    @Override
    public void recipient(String recipient) {

        this.recipients.add(recipient);
    }

    @Override
    public void data(InputStream is) throws IOException {

        data = IOUtils.toByteArray(is);
    }

    @Override
    public void done() {

        try {

            Session s = Session.getDefaultInstance(new Properties());
            MimeMessage mimeMsg = new MimeMessage(s, new ByteArrayInputStream(data));

            // set any data from the mimemessage itself
            MessageHeaders headers = getHeaders(mimeMsg);

            Message message = new Message(0, mimeMsg.getMessageID(), headers.get("Subject"), this.senderEmail,
                    new Date(), senderHost, this.data.length, IOUtils.toString(data, CharEncoding.UTF_8),
                    this.recipients, headers);

            messageService.saveMessage(message);

            logger.info("Stored SMTP message from " + message.getSenderEmail() + " to "
                    + StringUtils.join(message.getRecipients(), ","));

        } catch (Exception e) {

            logger.error("Couldn't handle message: " + e.getMessage(), e);

        }

    }

    protected MessageHeaders getHeaders(MimeMessage message) throws MessagingException {

        Map<String, String> headerMap = new HashMap<>();

        // oh wow 2015 and it's untyped and uses Enumeration
        Enumeration allHeaders = message.getAllHeaders();
        while (allHeaders.hasMoreElements()) {
            Header header = (Header) allHeaders.nextElement();
            String headerName = header.getName();
            String headerVal = header.getValue();

            headerMap.put(headerName, headerVal);

        }

        return new MessageHeaders(headerMap);
    }

}