mailrepository.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;

import org.jpxx.mail.Config.SysConfig;
import java.io.*;
import java.util.*;
import org.jpxx.mail.Util.*;
import org.jpxx.mail.Message.*;

/**
 * Managing an e-mail Repository.
 * 
 * @author Jun Li
 * @version $Revision: 0.0.1 $, $Date: 2008/04/26 19:11:00 $
 * 
 */
public class MailRepository {

    private Pop3User user = Pop3UserMonitor.getSingletonInstance();
    private EmailAddress emailAddress = null;
    private SysConfig sc = new SysConfig();
    private ArrayList mailList = null;

    /**
     * 
     * @param emailAddress
     */
    public MailRepository(EmailAddress emailAddress) {
        this.emailAddress = emailAddress;
    }

    /**
     * Mark the mail state.
     * @param num the mail id
     * @param isDelete If isDelete==true, it will be deleted when quit.
     * @throws java.lang.IndexOutOfBoundsException
     */
    public void setDelete(int num, boolean isDelete)
            throws IndexOutOfBoundsException {
        ArrayList list = this.mailList;
        if (num < 1 || num > list.size()) {
            throw new IndexOutOfBoundsException();
        }

        Mail m = (Mail) list.get(num - 1);
        m.setState(isDelete);
        list.set(num - 1, m);
        this.setMailList(list);
    }

    /**
     * Remove all the mails which were marked to delete.
     *  
     * @throws java.io.IOException
     */
    public void remove() throws IOException {
        String userDir = sc.getHomeDir(emailAddress.getUserName(),
                emailAddress.getDomain());
        ArrayList list = list();
        for (int i = 0; i < list.size(); i++) {
            Mail m = (Mail) list.get(i);
            if (m.getState()) {
                String path = userDir + m.getName();
                File f = new File(path);
                if (f.exists()) {
                    f.delete();
                }
            }
        }
    }

    /**
     * The POP3 server sends the entire message.
     * 
     * @param pw The object of PrintWriter
     * @param m This is the mail that will be sent to client.
     * @throws java.io.IOException
     * @throws org.jpxx.mail.Message.MessageException
     */
    public void write(PrintWriter pw, Mail m)
            throws IOException, MessageException {
        ArrayList contents = m.getMimeMessage().getMailMessageContent(
                new FileInputStream(m.getPath()));
        for (int i = 0; i < contents.size(); i++) {
            pw.println(contents.get(i).toString());
            pw.flush();
        }
    }

    /**Write <b>n</b> line message of the mail. 
     * Used by the command - TOP [number] [line].
     * Note that if the number of lines requested by the POP3
     * client is greater than the number of lines in the
     * body, then the POP3 server sends the entire message.
     * 
     * @param pw The object of PrintWriter
     * @param m an Mail object. This is the mail that will be sent to client.
     * @param n line message of the mail
     * @throws java.io.IOException
     * @throws org.jpxx.mail.Message.MessageException
     */
    public void write(PrintWriter pw, Mail m, int n)
            throws IOException, MessageException {
        ArrayList contents = m.getMimeMessage().getMailMessageContent(
                new FileInputStream(m.getPath()), n);
        for (int i = 0; i < contents.size(); i++) {
            pw.println(contents.get(i).toString());
            pw.flush();
        }
    }

    /**
     * Get all mail files of current user
     * @return a list of mail files
     */
    public ArrayList list() {
        if (mailList == null) {
            mailList = this.getMailList();
        }
        return mailList;
    }

    /**
     * Lock this box so no one else can use it. 
     * This way you don't have two people getting and deleting mail from 
     * the same user at the same time.
     */
    public void lockMailRepository() {
        user.lockUser(emailAddress);
    }

    /**
     * Unlock this box so it can be used again.
     */
    public void unlockMailRepository() {
        user.unlockUser(emailAddress);
    }

    /**
     * Is user is lock now. Return current user's state.
     * @return Current user's state. When user is login, lock it. This user
     * can't login again before logout.
     */
    public boolean isLock() {
        return user.isOnline(emailAddress);
    }

    /**
     * Set current user's mail list.
     * @param mailList current user's mails list
     */
    public void setMailList(ArrayList mailList) {
        this.mailList = mailList;
    }

    /**
     * List all mails of current user's.
     * @return An list of current user's. The Objects of 
     * org.jpxx.mail.Pop3.MailRepository.Mail are saving in this list.
     */
    private ArrayList getMailList() {
        String userDir = sc.getHomeDir(emailAddress.getUserName(),
                emailAddress.getDomain());
        File dir = new File(userDir);
        // Read all mail files
        File files[] = dir.listFiles();
        mailList = new ArrayList();
        for (int i = 0; i < files.length; i++) {
            try {
                Mail m = new MailImpl(files[i].getAbsolutePath());
                mailList.add(m);
            } catch (Exception e) {
                // Ignore
            }
        }
        return mailList;
    }
}