com.cubusmail.server.services.ShowMessageSourceServlet.java Source code

Introduction

Here is the source code for com.cubusmail.server.services.ShowMessageSourceServlet.java

Source

/* ShowMessageSourceServlet.java
    
   Copyright (c) 2009 Juergen Schlierf, All Rights Reserved
       
   This file is part of Cubusmail (http://code.google.com/p/cubusmail/).
       
   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 3 of the License, or (at your option) any later version.
       
   This library 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
   Lesser General Public License for more details.
       
   You should have received a copy of the GNU Lesser General Public
   License along with Cubusmail. If not, see <http://www.gnu.org/licenses/>.
       
 */
package com.cubusmail.server.services;

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.UnsupportedEncodingException;
import java.util.Enumeration;

import javax.mail.Header;
import javax.mail.Message;
import javax.mail.MessagingException;
import javax.mail.internet.ContentType;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.cubusmail.common.util.CubusConstants;
import com.cubusmail.server.mail.IMailbox;
import com.cubusmail.server.mail.SessionManager;

/**
 * Servlet for showing the message source.
 *
 * @author Juergen Schlierf
 */
public class ShowMessageSourceServlet extends HttpServlet {

    private Log log = LogFactory.getLog(this.getClass());

    /**
     * 
     */
    private static final long serialVersionUID = 1739028172595259352L;

    /*
     * (non-Javadoc)
     * 
     * @see javax.servlet.http.HttpServlet#service(javax.servlet.ServletRequest,
     * javax.servlet.ServletResponse)
     */
    @Override
    public void service(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        try {
            String messageId = request.getParameter("messageId");
            if (messageId != null) {
                IMailbox mailbox = SessionManager.get().getMailbox();
                Message msg = mailbox.getCurrentFolder().getMessageById(Long.parseLong(messageId));

                ContentType contentType = new ContentType("text/plain");
                response.setContentType(contentType.getBaseType());
                response.setHeader("expires", "0");
                String charset = null;
                if (msg.getContentType() != null) {
                    try {
                        charset = new ContentType(msg.getContentType()).getParameter("charset");
                    } catch (Throwable e) {
                        // should never happen
                    }
                }
                if (null == charset || charset.equalsIgnoreCase(CubusConstants.US_ASCII)) {
                    charset = CubusConstants.DEFAULT_CHARSET;
                }

                OutputStream outputStream = response.getOutputStream();

                // writing the header
                String header = generateHeader(msg);
                outputStream.write(header.getBytes(), 0, header.length());

                BufferedInputStream bufInputStream = new BufferedInputStream(msg.getInputStream());

                InputStreamReader reader = null;

                try {
                    reader = new InputStreamReader(bufInputStream, charset);
                } catch (UnsupportedEncodingException e) {
                    log.error(e.getMessage(), e);
                    reader = new InputStreamReader(bufInputStream);
                }

                OutputStreamWriter writer = new OutputStreamWriter(outputStream);
                char[] inBuf = new char[1024];
                int len = 0;
                try {
                    while ((len = reader.read(inBuf)) > 0) {
                        writer.write(inBuf, 0, len);
                    }
                } catch (Throwable e) {
                    log.warn("Download canceled!");
                }

                writer.flush();
                writer.close();
                outputStream.flush();
                outputStream.close();
            }

        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }
    }

    /**
     * @param msg
     * @return
     */
    @SuppressWarnings("unchecked")
    private String generateHeader(Message msg) {

        StringBuffer headerString = new StringBuffer();

        try {
            Enumeration<Header> e = msg.getAllHeaders();
            if (e != null) {
                for (; e.hasMoreElements();) {
                    Header header = e.nextElement();
                    headerString.append(header.getName());
                    headerString.append(": ");
                    headerString.append(header.getValue());
                    headerString.append('\n');
                }
            }
        } catch (MessagingException e) {
            log.error(e.getMessage(), e);
        }

        return headerString.toString();
    }
}