package getmail;
import java.io.*;
import java.text.*;
import java.util.*;
import javax.mail.*;
import javax.mail.internet.*;
public class GetmailBean {
    private MimeMessage mimeMessage = null;
    private Message message[] =null;
    private String saveAttachPath = "";          //附件下載后的存放目錄
    private String dateformat = "yy-MM-dd HH:mm";       //默認的日前顯示格式

    public void setMessage(String host,String username,String password)throws Exception{
        Properties props = new Properties();
        Session sessions = Session.getDefaultInstance(props, null);
        Store store = sessions.getStore("pop3");
        store.connect(host, username, password);

        Folder folder = store.getFolder("INBOX");
        folder.open(Folder.READ_ONLY);
        message= folder.getMessages();
    }
    public void setMimeMessage(MimeMessage mimeMessage){
        this.mimeMessage = mimeMessage;
    }
    public Message[] getMessage(){
        return this.message;
    }
    /**
     * 獲得發(fā)件人的地址和姓名
     */
    public String getFrom()throws Exception{
        InternetAddress address[] = (InternetAddress[])mimeMessage.getFrom();
        String from = address[0].getAddress();
        if(from == null) from="";
        String personal = address[0].getPersonal();
        if(personal == null) personal="";
        String fromaddr = personal+"<"+from+">";
        return fromaddr;
    }

    /**
     * 獲得郵件的收件人，抄送，和密送的地址和姓名，根據(jù)所傳遞的參數(shù)的不同
     * "to"----收件人 "cc"---抄送人地址 "bcc"---密送人地址
     */
    public String getMailAddress(String type)throws Exception{
        String mailaddr = "";
        String addtype = type.toUpperCase();
        InternetAddress []address = null;
        if(addtype.equals("TO") || addtype.equals("CC") ||addtype.equals("BCC")){
            if(addtype.equals("TO")){
                address = (InternetAddress[])mimeMessage.getRecipients(Message.RecipientType.TO);
            }else if(addtype.equals("CC")){
                address = (InternetAddress[])mimeMessage.getRecipients(Message.RecipientType.CC);
            }else{
                address = (InternetAddress[])mimeMessage.getRecipients(Message.RecipientType.BCC);
            }
            if(address != null){
                for(int i=0;i<address.length;i++){
                    String email=address[i].getAddress();
                    if(email==null) email="";
                    else{
                        email=MimeUtility.decodeText(email);
                    }
                    String personal=address[i].getPersonal();
                    if(personal==null) personal="";
                    else{
                        personal=MimeUtility.decodeText(personal);
                    }
                    String compositeto=personal+"<"+email+">";
                    mailaddr+=","+compositeto;
                }
                mailaddr=mailaddr.substring(1);
            }
        }else{
            throw new Exception("Error emailaddr type!");
        }
        return mailaddr;
    }

    /**
     * 獲得郵件主題
     */
    public String getSubject()throws MessagingException{
        String subject = "";
        try{
            subject = MimeUtility.decodeText(mimeMessage.getSubject());
            if(subject == null) subject="";
        }catch(Exception exce){
        }
        return subject;
    }

    /**
     * 獲得郵件發(fā)送日期
     */
    public String getSentDate()throws Exception{
        Date sentdate = mimeMessage.getSentDate();
        SimpleDateFormat format = new SimpleDateFormat(dateformat);
        return format.format(sentdate);
    }

    /**
     * 獲得郵件正文內(nèi)容
     */
    public Part[] getBody(int i)throws Exception{
        if (message[i].getContent() instanceof MimeMultipart) {
            MimeMultipart mp = (MimeMultipart) message[i].getContent();
            int count = mp.getCount();
            Part[] p = new Part[count];
            for (int j = 0; j < count; j++) {
                p[j] = mp.getBodyPart(j);
            }
            return p;
        }
        else
            return null;
    }
    /**
     * 解析郵件，把得到的郵件內(nèi)容保存到一個StringBuffer對象中，解析郵件
     * 主要是根據(jù)MimeType類型的不同執(zhí)行不同的操作，一步一步的解析
     */
    public String getMailContent(Part part)throws Exception{
        StringBuffer bodytexts = new StringBuffer();
        String contenttype = part.getContentType();
        int nameindex = contenttype.indexOf("name");
        boolean conname =false;
        if(nameindex != -1) conname=true;
        System.out.println("CONTENTTYPE: "+contenttype);
        if(part.isMimeType("text/plain") && !conname){
            bodytexts.append((String)part.getContent());
        }else if(part.isMimeType("text/html") && !conname){
            bodytexts.append((String)part.getContent());
        }else if(part.isMimeType("multipart/*")){
            Multipart multipart = (Multipart)part.getContent();
            int counts = multipart.getCount();
            for(int i=0;i<counts;i++){
                getMailContent(multipart.getBodyPart(i));
            }
        }else if(part.isMimeType("message/rfc822")){
            getMailContent((Part)part.getContent());
        }else{}
         return bodytexts.toString();
    }

    /**
     * 判斷此郵件是否需要回執(zhí)，如果需要回執(zhí)返回"true",否則返回"false"
     */
    public boolean getReplySign()throws MessagingException{
        boolean replysign = false;
        String needreply[] = mimeMessage.getHeader("Disposition-Notification-To");
        if(needreply != null){
            replysign = true;
        }
        return replysign;
    }

    /**
     * 獲得此郵件的Message-ID
     */
    public String getMessageId()throws MessagingException{
        return mimeMessage.getMessageID();
    }

    /**
     * 【判斷此郵件是否已讀，如果未讀返回返回false,反之返回true】
     */
    public boolean isNew()throws MessagingException{
        boolean isnew = false;
        Flags flags = ((Message)mimeMessage).getFlags();
        Flags.Flag []flag = flags.getSystemFlags();
        System.out.println("flags's length: "+flag.length);
        for(int i=0;i<flag.length;i++){
            if(flag[i] == Flags.Flag.SEEN){
                isnew=true;
                System.out.println("seen Message.......");
                break;
            }
        }
        return isnew;
    }

    /**
     * 判斷此郵件是否包含附件
     */
    public boolean isContainAttach(Part part)throws Exception{
        boolean attachflag = false;
        String contentType = part.getContentType();
        if(part.isMimeType("multipart/*")){
            Multipart mp = (Multipart)part.getContent();
            for(int i=0;i<mp.getCount();i++){
                BodyPart mpart = mp.getBodyPart(i);
                String disposition = mpart.getDisposition();
                if((disposition != null) &&((disposition.equals(Part.ATTACHMENT)) ||(disposition.equals(Part.INLINE))))
                    attachflag = true;
                else if(mpart.isMimeType("multipart/*")){
                    attachflag = isContainAttach((Part)mpart);
                }else{
                    String contype = mpart.getContentType();
                    if(contype.toLowerCase().indexOf("application") != -1) attachflag=true;
                    if(contype.toLowerCase().indexOf("name") != -1) attachflag=true;
                }
            }
        }else if(part.isMimeType("message/rfc822")){
            attachflag = isContainAttach((Part)part.getContent());
        }
        return attachflag;
    }

    /**
     * 【保存附件】
     */
    public void saveAttachMent(Part part)throws Exception{
        String fileName = "";
        if(part.isMimeType("multipart/*")){
            Multipart mp = (Multipart)part.getContent();
            for(int i=0;i<mp.getCount();i++){
                BodyPart mpart = mp.getBodyPart(i);
                String disposition = mpart.getDisposition();
                if((disposition != null) &&((disposition.equals(Part.ATTACHMENT)) ||(disposition.equals(Part.INLINE)))){
                    fileName = mpart.getFileName();
                    if(fileName.toLowerCase().indexOf("gb2312") != -1){
                        fileName = MimeUtility.decodeText(fileName);
                    }
                                        saveFile(fileName,mpart.getInputStream());
                }else if(mpart.isMimeType("multipart/*")){
                    saveAttachMent(mpart);
                }else{
                    fileName = mpart.getFileName();
                    if((fileName != null) && (fileName.toLowerCase().indexOf("GB2312") != -1)){
                        fileName=MimeUtility.decodeText(fileName);
                        saveFile(fileName,mpart.getInputStream());
                    }
                }
            }
        }else if(part.isMimeType("message/rfc822")){
            saveAttachMent((Part)part.getContent());
        }
    }

    /**
     * 【設(shè)置附件存放路徑】
     */
    public void setAttachPath(String attachpath){
        this.saveAttachPath = attachpath;
    }

    /**
     * 【設(shè)置日期顯示格式】
     */
    public void setDateFormat(String format)throws Exception{
        this.dateformat = format;
    }

    /**
     * 【獲得附件存放路徑】
     */
    public String getAttachPath(){
        return saveAttachPath;
    }

    /**
     * 【真正的保存附件到指定目錄里】
     */
    private void saveFile(String fileName,InputStream in)throws Exception{
        String osName = System.getProperty("os.name");
        String storedir = getAttachPath();
        String separator = "";
        if(osName == null) osName="";
        if(osName.toLowerCase().indexOf("win") != -1){
            separator = "\\";
            if(storedir == null || storedir.equals("")) storedir="c:\\tmp";
        }else{
            separator = "/";
            storedir = "/tmp";
        }
        File storefile = new File(storedir+separator+fileName);
        System.out.println("storefile's path: "+storefile.toString());
        //for(int i=0;storefile.exists();i++){
            //storefile = new File(storedir+separator+fileName+i);
        //}
        BufferedOutputStream bos = null;
        BufferedInputStream  bis = null;
        try{
            bos = new BufferedOutputStream(new FileOutputStream(storefile));
            bis = new BufferedInputStream(in);
            int c;
            while((c=bis.read()) != -1){
                bos.write(c);
                bos.flush();
            }
        }catch(Exception exception){
            exception.printStackTrace();
            throw new Exception("文件保存失敗!");
        }finally{
            bos.close();
            bis.close();
        }
    }
}