package ch08.section05;

import java.io.*;
import java.util.*;
import javax.mail.*;
import javax.servlet.*;
import javax.servlet.http.*;

public class GetMail
    extends HttpServlet {
  private static final String CONTENT_TYPE = "text/html; charset=GBK";
  MailUser mailUser;
  MailInfo mf;
  public void init() throws ServletException {
  }

  public void doGet(HttpServletRequest request, HttpServletResponse response) throws
      ServletException, IOException {
    response.setContentType(CONTENT_TYPE);
    PrintWriter out = response.getWriter();

    String transport = request.getParameter("transport");
    String transportHost = request.getParameter("storeHost");
    String store = request.getParameter("store");
    String user = request.getParameter("user");
    String password = request.getParameter("password");
    mailUser = new MailUser();
    try {

      mailUser.login(store, transportHost,
                     transport, null,
                     user, password);
    }
    catch (MessagingException ex) {
      out.println("登錄錯(cuò)誤：" + ex.getMessage());
      return;
    }
    javax.servlet.http.HttpSession session = request.getSession();
    session.setAttribute("_mailUser", mailUser);
    Vector v_mf = getMail();
    request.setAttribute("_mails", v_mf);
    ServletContext sc = getServletContext();
    RequestDispatcher rd = sc.getRequestDispatcher(
        "/ch08/section05/ListMail.jsp");
    rd.forward(request, response);

  }

  private Vector getMail() {
    Vector v_mf = new Vector();
    try {
      Folder inbox = mailUser.getInbox();
      int n = inbox.getMessageCount();
      Message[] msgs = inbox.getMessages();
      for (int i = 0; i < n; i++) {
        Message m = msgs[i];
        if (m.isSet(Flags.Flag.DELETED)) {
          continue;
        }
        mf = new MailInfo();
        Address from[] = m.getFrom();
        Address addr = null;
        if ( (from != null) && (from.length > 0)) {
          addr = from[0];
        }

        mf.setM_From(PubPro.getAddress(addr));
        mf.setM_Sent(m.getSentDate());
        mf.setM_Subject(m.getSubject());
        mf.setM_Number(i);
        v_mf.add(mf);
      }
    }
    catch (MessagingException ex) {
      System.out.println("錯(cuò)誤: " + ex.getMessage());
    }
    finally {
      return v_mf;
    }
  }

  public void doPost(HttpServletRequest request, HttpServletResponse response) throws
      ServletException, IOException {
    doGet(request, response);
  }

  public void destroy() {
  }
}