retrcommand.java

Java Mail Server JAVA编写的邮件服务器

/****************************************************************
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.     *
 *                                                              *
 * Copyright 2008 Jun Li(SiChuan University, the School of      *
 * Software Engineering). All rights reserved.                  *
 *                                                              *
 * Licensed to the JMS under one  or more contributor license   *
 * agreements.  See the LICENCE file  distributed with this     *
 * work for additional information regarding copyright          *
 * ownership.  The JMS licenses this file  you may not use this *
 * file except in compliance  with the License.                 *
 *                                                              *
 * 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.jpxx.mail.Pop3.Command;

import org.jpxx.mail.Pop3.Pop3States;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import org.jpxx.mail.Exception.ArgumentException;
import org.jpxx.mail.Message.Mail;
import org.jpxx.mail.Pop3.MailRepository;
import org.jpxx.mail.Service.CommandHandler;
import org.jpxx.mail.Service.Session;

/**
 * If the POP3 server issues a positive response, then the
 * response given is multi-line. After the initial +OK, the
 * POP3 server sends the message corresponding to the given
 * message-number, being careful to byte-stuff the termination
 * character (as with all multi-line responses).
 * 
 * @author Jun Li
 * @version $Revision: 0.0.1 $, $Date: 2008/04/27 19:33:00 $
 * 
 */
public class RetrCommand implements CommandHandler, Pop3States {

    /**
     * @see org.jpxx.mail.Service.CommandHandler#onCommand(Session session)
     * @param session  The session of client and server
     */
    public void onCommand(Session session) {
        String responseString = null;
        int state = session.getLastAction();
        if (state == TRANSACTION) {
            String argument = null;
            try {
                argument = session.getCommandLine().getArgument();
            } catch (ArgumentException ae) {
                argument = null;
            }
            
            int num = 0;
            try {
                num = Integer.parseInt(argument.trim());
            } catch (Exception e) {
                responseString = ERR + " Usage: RETR [mail number]";
                session.writeResponse(responseString);
                return;
            }

            try {
                ArrayList userMailbox = (ArrayList) session.getOperation().get(
                        "USER_MAIL_BOX");
                Mail m = (Mail) userMailbox.get(num - 1);
                MailRepository mr = (MailRepository) session.getOperation().get(
                        "MAIL_REPOSITORY");

                if (!m.getState()) {
                    StringBuffer buffer = new StringBuffer(64);
                    buffer.append(OK);
                    buffer.append(" ");
                    buffer.append("Message follows");
                    session.writeResponse(buffer.toString());
                    /**
                     * Write mail content to client.
                     */
                    try {
                        mr.write(new PrintWriter(session.getClinetSocket().
                                getOutputStream(), true), m);
                    } finally {
                        session.writeResponse(".");
                    }
                } else {
                    StringBuffer buffer = new StringBuffer(64);
                    buffer.append(ERR);
                    buffer.append(" Message (");
                    buffer.append(num);
                    buffer.append(") already deleted.");
                    session.writeResponse(buffer.toString());
                }
            } catch (IndexOutOfBoundsException iob) {
                StringBuffer buffer = new StringBuffer(64);
                buffer.append(ERR);
                buffer.append(" ");
                buffer.append("Message (");
                buffer.append(num);
                buffer.append(") does not exist.");
                session.writeResponse(buffer.toString());
            } catch (IOException ioe) {
                StringBuffer buffer = new StringBuffer(64);
                buffer.append(ERR);
                buffer.append(" ");
                buffer.append("Error while retrieving message.");
                session.writeResponse(buffer.toString());
            } catch (Exception e) {
                StringBuffer buffer = new StringBuffer(64);
                buffer.append(ERR);
                buffer.append(" ");
                buffer.append("Error while retrieving message.");
                session.writeResponse(buffer.toString());
            }
        } else {
            StringBuffer buffer = new StringBuffer(64);
            buffer.append(ERR);
            buffer.append(" ");
            buffer.append("RETR Not allowed here.");
            session.writeResponse(buffer.toString());
        }
    }
}