05-27 18:56

Author: stardust95

MicroMailClient.pro.user

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE QtCreatorProject>
-<!-- Written by QtCreator 3.6.1, 2016-05-26T14:23:41. -->
+<!-- Written by QtCreator 3.6.1, 2016-05-27T18:55:56. -->
 <qtcreator>
  <data>
   <variable>EnvironmentId</variable>

Model/IMAPClient.h

@@ -70,6 +70,13 @@ class IMAPClient : public ReceiveMailClient
 
         qDebug() << "In select Folder, folder = "<< folder << ", mailList.size = " << _mailList.size () << "uids.size = " << uids.size ()<< "\n";
 
+        for(int i=0; i<_mailList.size (); i++){
+            for(int j=0; j<i; j++){
+                if( _mailList.at (i).uid == _mailList.at(j).uid )
+                    qDebug() << QString::fromUtf8( _mailList.at (i).subject.c_str()) << " and " << QString::fromUtf8 (_mailList.at (j).subject.c_str()) << " has the same uid, subject = " << QString::fromUtf8 (_mailList.at (i).subject.c_str()) << "\n";
+            }
+        }
+
         _curListIndex = 0;
 
         return uids.size ();
@@ -78,27 +85,40 @@ class IMAPClient : public ReceiveMailClient
 
     int getMailBodies(QList<MAILBODY_PTR> & result, int count ) override{
 
-        std::string msg;
+        std::string partsPaths;
 
         int counter = 0;
 
-        qDebug() << "listIndex = " <<_curListIndex << ", mailList.size=" << _mailList.size () << "\n";
-
         for( ; _curListIndex < _mailList.size (); _curListIndex++, counter++) {
 
             if( counter >= count )
                 break;
 
-            auto info = _mailList.at (_curListIndex);
+            auto info = _mailList.at ( _mailList.size () - _curListIndex - 1);
 
-            _session->loadMessage (_selectedFolder.toStdString (), info, msg);
+//            if ( info.uid != "1577" )
+//                continue;
+
+            _session->loadMessage (_selectedFolder.toStdString (), info);
 
             MAILBODY_PTR newmail = MAILBODY_PTR::create(QString::fromUtf8 (info.subject.c_str ()));
 
-            newmail->setContent (QString::fromUtf8 (msg.c_str ()));
-            newmail->setSender (QString::fromUtf8 (info.from.c_str()));
+            newmail->setContent (QString::fromUtf8 (info.text.c_str ()));
+
+            newmail->setHTMLContent (QString::fromUtf8 (info.htmlText.c_str ()));
+
+//            qDebug() << QString::fromUtf8 ( info.uid.c_str() ) << " content: " << newmail->getHTMLContent ()<< "\n";
+
+            std::string sender = info.from.find_last_of('<') != std::string::npos ?
+                                info.from.substr(info.from.find_last_of('<')+1, info.from.find_last_of('>') - info.from.find_last_of('<')-1) :
+                                info.from ;
+
+            newmail->setSender (QString::fromUtf8 (sender.c_str()));
+
             newmail->addRecipient (QString::fromUtf8 (info.to.c_str()));
+
             newmail->setDateTime (QString::fromUtf8 (info.date.c_str()));
+
             newmail->setIsread (false);
 
             result.push_back (newmail);
@@ -130,6 +150,8 @@ class IMAPClient : public ReceiveMailClient
 
     Poco::Net::IMAPClientSession::MessageInfoVec _mailList;
 
+//    QMap<QString, Poco::Net::IMAPClientSession::MessageInfoVec> _mailList;
+
      SESSION_PTR _session;
 
 };

Model/IMAPClientSession.cpp

@@ -10,12 +10,16 @@
 #include "Poco/TextConverter.h"
 #include "Poco/Net/MessageHeader.h"
 
+#include <Windows.h>
+
 #include <iostream>
+#include <codecvt>
 #include <sstream>
 #include <algorithm>
 #include <iterator>
 #include <string>
 #include <vector>
+#include <fstream>
 
 using Poco::NumberFormatter;
 using Poco::DateTimeFormatter;
@@ -32,6 +36,43 @@ namespace Poco {
 
         //POCO_IMPLEMENT_EXCEPTION(IMAPException, NetException, "IMAP Exception")
 
+        // GB2312转UTF8
+        std::string GB2312ToUTF8(const char* lpszGb32Text)
+        {
+            int nUnicodeBufLen = MultiByteToWideChar(CP_ACP, 0, lpszGb32Text, -1, 0, 0);
+                if (nUnicodeBufLen == 0)
+                    return "";
+
+                WCHAR* pUnicodeBuf = new WCHAR[nUnicodeBufLen];
+                if (pUnicodeBuf == 0)
+                    return "";
+
+                MultiByteToWideChar(CP_ACP, 0, lpszGb32Text, -1, pUnicodeBuf, nUnicodeBufLen);
+
+                int nUtf8BufLen = WideCharToMultiByte(CP_UTF8, 0, pUnicodeBuf, -1, 0, 0, NULL, NULL);
+                if (nUtf8BufLen == 0)
+                {
+                    delete[] pUnicodeBuf;
+                    return "";
+                }
+
+                char* pUft8Buf = new char[nUtf8BufLen];
+                if (pUft8Buf == 0)
+                {
+                    delete[] pUnicodeBuf;
+                    return "";
+                }
+
+                WideCharToMultiByte(CP_UTF8, 0, pUnicodeBuf, -1, pUft8Buf, nUtf8BufLen, NULL, NULL);
+
+                std::string strUtf8 = pUft8Buf;
+
+                delete[] pUnicodeBuf;
+                delete[] pUft8Buf;
+
+                return strUtf8;
+        }
+
         template <class S=std::string>
         S trimchar(const S& str, const char ch)
         {
@@ -317,35 +358,41 @@ namespace Poco {
             if (!sendCommand("DELETE \"" + folder + "\" *", response, data)) throw NetException("Can't delete folder", response);
         }
 
-        void IMAPClientSession::loadMessage (const std::string & folder, const MessageInfo & info, std::string & message) {
+        void IMAPClientSession::loadMessage (const std::string & folder, MessageInfo & info) {
             std::string response, tmpdata;
             std::vector<std::string> data, data1;
-            message.clear ( );
 
             if ( !sendCommand ("SELECT \"" + folder + "\"", response, data) ) throw NetException ("Can't select from folder", response);
 
-            loadText (info.uid, info.parts, "", message);
+            loadText (info.uid, info.parts, "", "HTML" , info.htmlText);
+            loadText (info.uid, info.parts, "", "PLAIN" , info.text);
 
 //            sendCommand ("CLOSE", response, data1);
 
-            if ( data.size ( ) <= 2 ) return;
+//            if ( data.size ( ) <= 2 ) return;
+
+//            for ( int i = 1; i < data.size ( ) - 2; i++ ) {
+//                if( data[i].length() > 0 && data[i][0] != '*' ){
+//                    message += data[i];
+//                    message += "\r\n";
+//                }
+//            }
 
-            for ( int i = 1; i < data.size ( ) - 2; i++ ) {
-                if( data[i].length() > 0 && data[i][0] != '*' ){
-                    message += data[i];
-                    message += "\r\n";
-                }
-            }
         }
 
-        void IMAPClientSession::loadText (const std::string & uid, const PartInfo & info, const std::string & index, std::string & text) {
+        void IMAPClientSession::loadText (const std::string & uid, const PartInfo & info, const std::string & index, const std::string & _type, std::string & text) {
+
             auto & attrs = info.attributes;
 
+            std::string type = _type;
+
+            std::transform(type.begin (), type.end (), type.begin (), ::toupper );
+
             std::string response;
             std::vector<std::string> result;
             std::stringstream ss;
 
-            if ( std::find (attrs.begin ( ), attrs.end ( ), "TEXT") != info.attributes.end ( ) ) {	// 如果是正文类型
+            if ( std::find (attrs.begin ( ), attrs.end ( ), type) != info.attributes.end ( ) ) {      // 如果是正文类型
 
                 if ( !sendCommand ("UID FETCH " + uid + " BODY.PEEK[" + ( index.length() == 0 ? "1" : index ) + "]", response, result) )
                     throw NetException ("Cannot Load Mail Text", response);
@@ -371,29 +418,54 @@ namespace Poco {
                     }
 
                 } else {
-                    std::cout << "UNKNONW CONTENT TRANSFER TYPE: " ;
+                    std::cout << "Unknown Content Transfer Type: " ;
                     for(auto s : attrs){
                         std::cout << s << ";";
                     }
-                    std::cout << std::endl;
+//                    std::cout << "Content: " << ss.str ()<< std::endl;
                     text += ss.str ( );
                     //throw NetException ("UNKNOWN CONTENT TRANSFER TYPE");
                 }
 
-            } else if( std::find (attrs.begin ( ), attrs.end ( ), "APPLICATION") != attrs.end()) {		// 附件类型, 只记录index和文件名(需要用decoder解码), 不需要读内容
-
             }
 
             for ( int i = 0; i < info.childs.size ( ); i++ ) {
-                loadText (uid, info.childs.at(i), index + "." + std::to_string(i+1), text);
+                std::string nextindex = index.length () == 0 ? std::to_string(i+1) : index + "." + std::to_string(i+1);
+                loadText (uid, info.childs.at(i), nextindex, type,  text);
             }
 
-            text += "\n";
+//            text += "\n";
 
             return;
 
         }
 
+        void IMAPClientSession::loadParts (const std::string & uid, const PartInfo & info, const std::string & index, const std::string & _type, std::vector<std::string> & paths){
+
+            auto & attrs = info.attributes;
+
+            std::string type = _type;
+
+            std::transform(type.begin (), type.end (), type.begin (), ::toupper );
+
+            std::string response;
+            std::vector<std::string> result;
+
+            if ( std::find (attrs.begin ( ), attrs.end ( ), type) != info.attributes.end ( ) ) {
+
+                paths.push_back (index);
+
+            }
+
+            for ( int i = 0; i < info.childs.size ( ); i++ ) {
+                std::string nextindex = index.length () == 0 ? std::to_string(i+1) : index + "." + std::to_string(i+1);
+                loadParts(uid, info.childs.at(i), nextindex, type,  paths);
+            }
+
+
+        }
+
+
         void IMAPClientSession::getMessages(const std::string& folder, std::vector<std::string>& uids, MessageInfoVec& messages)
         {
             std::string response, response1;
@@ -481,7 +553,7 @@ namespace Poco {
                         m.from = IMAPClientSession::decoder(trim (tokens4[1]));
                     //m.from = MessageHeader::decodeWord(trim(tokens4[1]));
                     else if ( cmd == "TO" )
-                        m.from = IMAPClientSession::decoder (trim (tokens4[1]));
+                        m.to = IMAPClientSession::decoder (trim (tokens4[1]));
                         //m.to = MessageHeader::decodeWord(trim(tokens4[1]));
                 }
                 messages.push_back(m);
@@ -541,30 +613,41 @@ namespace Poco {
 
             if ( encoding[0] == 'B' ) {	// Base64编码
                 Base64Decoder decode (iss);
-                while ( !decode.eof ( ) && (c = decode.get()) ) {
+                while ( ( c = decode.get ( ) ) != -1 ) {
                     tmp += c;
                 }
             } else if ( encoding[0] == 'Q' ) {		// Quote-Printable编码
                 QuotedPrintableDecoder qpd (iss);
-                while ( !qpd.eof ( ) && ( c = qpd.get ( ) ) ) {
-                    text += c;
+                while ( ( c = qpd.get ( ) ) != -1  ) {
+                    tmp += c;
                 }
 
             } else {				// 编码未知, 直接返回原字符串
                 outs = ins;
                 return;
             }
 
-            // 转换字符集
+            // 标题转换字符集
             if ( charset != charset_to ) {
-                try {
-                    TextEncoding & enc = TextEncoding::byName (charset);
-                    TextEncoding & dec = TextEncoding::byName (charset_to);
-                    TextConverter converter (enc, dec);
-                    converter.convert (tmp, outs);
-                } catch ( ... ) {			// 无法转换的未知字符集(包括GB2312)
-                    std::cout << "Unknown charset " << charset << std::endl;
-                    outs = tmp;
+                std::transform(charset.begin (), charset.end (), charset.begin (), ::toupper);
+                if( charset == "GB2312" ){
+                    std::cout << "charset == gb2312" << std::endl;
+                    try{
+                        outs = GB2312ToUTF8 (tmp.c_str ());
+                    }catch( std::exception & e ){
+                        std::cout << e.what () << std::endl;
+                    }
+                }else{
+                    try {
+                        TextEncoding & enc = TextEncoding::byName (charset);
+                        TextEncoding & dec = TextEncoding::byName (charset_to);
+                        TextConverter converter (enc, dec);
+                        converter.convert (tmp, outs);
+                    } catch ( ... ) {			// 无法转换的未知字符集(包括GB2312)
+                        std::cout << "Unknown charset " << charset << std::endl;
+                        std::cout << "Content: " << tmp << std::endl;
+                        outs = tmp;
+                    }
                 }
             } else {			// 不需要转换字符集
                 outs = tmp;
@@ -581,7 +664,7 @@ namespace Poco {
                     break;
                 }
                 if ( start > 0 ) {
-                    outs = text.substr (0, start);
+                    outs += text.substr (0, start);
                 }
 
                 text = text.substr (start+2);		// 从=?之后开始查找其余信息

Model/IMAPClientSession.h

@@ -57,7 +57,7 @@ namespace Poco {
                 std::string flags;
                 PartInfo    parts;
 
-
+                std::string htmlText;
                 std::string text;
 
                 bool        seen;
@@ -94,8 +94,9 @@ namespace Poco {
             virtual void copyMessage (const std::string& uid, const std::string& from_folder, const std::string& to_folder);
             virtual void moveMessage (const std::string& uid, const std::string& from_folder, const std::string& to_folder);
             virtual void deleteMessage (const std::string& uid, const std::string& folder, bool expunge = true);
-            virtual void loadMessage (const std::string & folder, const MessageInfo & info, std::string & message);
-            virtual void loadText (const std::string & uid, const PartInfo & info, const std::string & index, std::string & text);
+            virtual void loadMessage (const std::string & folder, MessageInfo & info);
+            virtual void loadText (const std::string & uid, const PartInfo & info, const std::string & index, const std::string & type, std::string & text);
+            virtual void loadParts (const std::string & uid, const PartInfo & info, const std::string & index, const std::string & type, std::vector<std::string> & paths);
 
             virtual void createFolder (const std::string& folder);
             virtual void deleteFolder (const std::string& folder);

Model/MailBody.h

@@ -52,6 +52,14 @@ class MailBody : public QObject {
         return _content;
     }
 
+    QString getHTMLContent(){
+        return _htmlContent;
+    }
+
+    void setHTMLContent (QString c){
+        _htmlContent = c;
+    }
+
     void setContent(QString c){
         _content = c;
     }
@@ -99,6 +107,8 @@ class MailBody : public QObject {
 
     QString _content;
 
+    QString _htmlContent;
+
     QString _sender;
 
     QString _datetime;

Model/MailClient.h

@@ -22,7 +22,7 @@ class MailClient{
 
 		virtual int getTimeout() = 0;
 
-		virtual void setTimeout(int) = 0;
+        virtual void setTimeout(int)     = 0;
 
 //       virtual ~MailClient() = 0;