/*
 * This product includes software developed by the
 * Apache Software Foundation (http://www.apache.org/).
 */
package ch05.module;

import java.sql.*;
import java.util.*;
import java.text.*;

import javax.mail.*;
import javax.mail.internet.*; 

import javax.servlet.http.*;

import ch05.*;

/**
 * 針對注冊頁面的后臺處理類
 * @author ShenYK
 * @version 1.0
 */
public class MInbox
{
    public boolean getNewestMail ( HttpSession mySession )
    {
        //設置用戶信息用
        Hashtable myValues = (Hashtable)mySession.getAttribute(CommonConst.VIEWID_INBOXLIST);
        
        String sHost = "";
        String sUser = "";
        String sPass = "";
        
        //嘗試查找用戶郵箱設置
        try
        {
            //載入MySQL的JDBC驅動類
            Class.forName(CommonConst.DB_DRIVER_CLASSNAME);
            //獲得數據庫連接
            Connection conn = DriverManager.getConnection( CommonConst.DB_CONN_STRING );

            Statement stmt = null;
            ResultSet rs = null;
            
            String sUsername = (String)mySession.getAttribute("username");
            
            try
            {
                //設置數據庫的rollback點
                conn.setAutoCommit(false);
                
                //檢查數據庫中是否已經有該用戶的郵箱設置
                stmt = conn.createStatement();
                
                //執行SQL語句
                String sQuery = "select * from mail_setting where username='" + sUsername + "'";
                rs = stmt.executeQuery( sQuery );
                if (rs.next() )
                {
                    sHost  = rs.getString(2);
                    sUser = rs.getString(3);
                    sPass = rs.getString(4);
                }
                else
                {
                    mySession.setAttribute("errMsg","請首先進行郵箱設置！");
                    return false;
                }
                
                //嘗試連接Pop3服務器
                Properties props = System.getProperties();
                Session session = Session.getDefaultInstance(props, null);
                //獲得收件箱
                Store store = session.getStore("pop3");
                store.connect( sHost, sUser, sPass);
                //獲得Folder，并按照只讀打開
                Folder folder = store.getFolder( "INBOX" );
                folder.open( Folder.READ_WRITE );
                //獲得所有郵件，并依次獲取內容
                Message messages[] = folder.getMessages();
                
                //往數據庫中登錄
                for ( int i=0; i<messages.length; i++ )
                {
                    //只要不是html格式，大概沒有亂碼問題。。。
                    String sFrom = MailContent.getFrom( messages[i].getFrom()[0].toString() );
                    String sTo   = MailContent.getFrom( messages[i].getRecipients( Message.RecipientType.TO )[0].toString() );
                    String sDate = (new SimpleDateFormat("yyyy/MM/dd HH:mm:ss")).format(messages[i].getSentDate());
                    String sSubject = messages[i].getSubject();
                    String sContent = "";
                    //如果純文本郵件內容
                    if ( messages[i].getContent() instanceof String )
                    {
                        sContent = (String) messages[i].getContent();
                    }
                    //如果HTML格式郵件內容
                    else
                    {
                        sContent = ((MimeMultipart)messages[i].getContent()).getBodyPart(0).getContent().toString();
                    }
                    
                    //插入數據庫
                    String sUpdateQuery = "insert into inbox values('" + sUsername + "', '"
                                                                       + sFrom + "', '"
                                                                       + sTo + "', '"
                                                                       + sDate + "', '"
                                                                       + sSubject + "', '"
                                                                       + sContent + "', '0') ";
                    
                    stmt.executeUpdate( sUpdateQuery );
                }
                
                conn.commit();
                conn.setAutoCommit(true);
                
                //刪除服務器上的郵件
                for ( int i=0; i<messages.length; i++ )
                {
                    messages[i].setFlag( Flags.Flag.DELETED, true );
                }
                folder.close(true);
                store.close();
                
                //重新獲取數據庫中的所有郵件，并表示在頁面上
                Vector vMails = new Vector();
                sQuery = "select * from inbox where username='" + sUsername + "' order by send_time desc";
                rs = stmt.executeQuery( sQuery );
                while ( rs.next() )
                {
                    String sFrom    = rs.getString(2);
                    String sTo      = rs.getString(3);
                    String sDate    = rs.getString(4);
                    String sSubject = rs.getString(5);
                    String sContent = rs.getString(6);
                    String sReadFlg = rs.getString(7);
                    
                    MailContent mc = new MailContent();
                    mc.setSender(sFrom);
                    mc.setReceiver(sTo);
                    mc.setSendTime(sDate);
                    mc.setSubject(sSubject);
                    mc.setContent(sContent);
                    mc.setReadFlg(sReadFlg);
                    
                    vMails.add( mc );
                }
                
                myValues.put( "mails", vMails );
            }
            catch(Exception e)
            {
                conn.rollback();
                e.printStackTrace();
                mySession.setAttribute("errMsg","查找郵件時出現錯誤！");
                return false;
            }
            finally
            {
                try
                {
                    rs.close();
                    stmt.close();
                }catch(Exception ex)
                {
                }
            }
        }catch(Exception ex)
        {
            ex.printStackTrace();
            mySession.setAttribute("errMsg","查找郵件時出現錯誤！");
            return false;
        }
        
        return true;
    }
    
    public boolean getDetailMail( HttpSession mySession, String sMailIndex )
    {
        try
        {
            
            //所有郵件的列表
            Hashtable listValues = (Hashtable)mySession.getAttribute(CommonConst.VIEWID_INBOXLIST);
        
            //本頁面使用的值
            Hashtable myValues = (Hashtable)mySession.getAttribute(CommonConst.VIEWID_INBOXDETAIL);
        
            Vector vMails = (Vector)listValues.get("mails");
        
            int iIndex = Integer.parseInt(sMailIndex);
        
            MailContent mailContent = (MailContent)vMails.get(iIndex);
            
            myValues.put( "mail", mailContent );
            
            //設置郵件已讀標志
            String sReadFlg = mailContent.getReadFlg();
            if ( sReadFlg.equals("0") )
            {
                //載入MySQL的JDBC驅動類
                Class.forName(CommonConst.DB_DRIVER_CLASSNAME);
                //獲得數據庫連接
                Connection conn = DriverManager.getConnection( CommonConst.DB_CONN_STRING );

                Statement stmt = null;
                ResultSet rs = null;
                
                String sUsername = (String)mySession.getAttribute("username");
                String sSender = mailContent.getSender();
                String sSendTime = mailContent.getSendTime();
                try
                {
                    //檢查數據庫中是否已經有該用戶的郵箱設置
                    stmt = conn.createStatement();
                    
                    String sUpdateQuery = "update inbox set read_flg='1' "
                                        + "where username='" + sUsername + "' and "
                                        + "sender = '" + sSender + "' and "
                                        + "send_time = '" + sSendTime + "'";
                    stmt.executeUpdate( sUpdateQuery );
                    
                }
                catch(Exception e)
                {
                    e.printStackTrace();
                    mySession.setAttribute("errMsg","閱讀郵件時出現錯誤！");
                    return false;
                }
                finally
                {
                    try
                    {
                        rs.close();
                        stmt.close();
                    }catch(Exception ex)
                    {
                    }
                }
            }

            
            return true;
        }
        catch(Exception e)
        {
            e.printStackTrace();
            mySession.setAttribute("errMsg","顯示郵件時出現錯誤！");
            return false;
        }
    }
    
    //刪除指定郵件
    public boolean deleteMail( HttpSession mySession, String sSender, String sSendTime )
    {
        //刪除對應的郵件
        try
        {
            //載入MySQL的JDBC驅動類
            Class.forName(CommonConst.DB_DRIVER_CLASSNAME);
            //獲得數據庫連接
            Connection conn = DriverManager.getConnection( CommonConst.DB_CONN_STRING );

            Statement stmt = null;
            ResultSet rs = null;
            
            String sUsername = (String)mySession.getAttribute("username");
            
            try
            {
                //檢查數據庫中是否已經有該用戶的郵箱設置
                stmt = conn.createStatement();
                
                String sDeleteQuery = "delete from inbox where username='" + sUsername + "' and "
                                    + "sender = '" + sSender + "' and "
                                    + "send_time = '" + sSendTime + "'";
                stmt.executeUpdate( sDeleteQuery );
                
            }
            catch(Exception e)
            {
                e.printStackTrace();
                mySession.setAttribute("errMsg","刪除郵件時出現錯誤！");
                return false;
            }
            finally
            {
                try
                {
                    rs.close();
                    stmt.close();
                }catch(Exception ex)
                {
                }
            }
        }catch(Exception ex)
        {
            ex.printStackTrace();
            mySession.setAttribute("errMsg","刪除郵件時出現錯誤！");
            return false;
        }
        
        return true;
        
    }
    
    //回復指定郵件
    public boolean replyMail( HttpSession mySession, String sSender, String sSubject, String sContent )
    {
        try
        {
            //本頁面使用的值
            Hashtable myValues = (Hashtable)mySession.getAttribute(CommonConst.VIEWID_SENDBOXDETAIL);
            
            //如果是回復的話，應該明示郵件標題
            String sNewSubject = "";
            if ( sSubject.toUpperCase().startsWith("RE:") )
            {
                sNewSubject = sSubject;
            }
            else
            {
                sNewSubject = "Re:" + sSubject;
            }
           
            //改善：在原郵件內容每行的開頭加上“> ”的符號
            String[] sContents = sContent.split("\r\n");
            StringBuffer sbContent = new StringBuffer();
            for ( int i=0; i<sContents.length; i++ )
            {
                sbContent.append( "\r\n> "+sContents[i]);
            }
            
            myValues.put( "receiver", sSender );
            myValues.put( "subject", sNewSubject );
            myValues.put( "content", sbContent.toString() );
            
            return true;
        }
        catch(Exception e)
        {
            e.printStackTrace();
            mySession.setAttribute("errMsg","準備回復郵件時出現錯誤！");
            return false;
        }
        
    }

}