Improve serialization of messages

Author: dinhvh

src/core/rfc822/MCAttachment.cpp

@@ -2,6 +2,7 @@
 
 #include "MCAttachment.h"
 
+#include "MCDefines.h"
 #include "MCMultipart.h"
 #include "MCMessagePart.h"
 #include "MCMessageHeader.h"
@@ -610,3 +611,13 @@ MessagePart * Attachment::attachmentWithMessageMIME(struct mailmime * mime)
 
     return (MessagePart *) attachment->autorelease();
 }
+
+static void * createObject()
+{
+    return new Attachment();
+}
+
+INITIALIZE(Attachment)
+{
+    Object::registerObjectConstructor("mailcore::Attachment", &createObject);
+}

src/core/rfc822/MCMessageParser.cpp

@@ -5,6 +5,7 @@
 #include <CoreFoundation/CoreFoundation.h>
 #endif
 
+#include "MCDefines.h"
 #include "MCAttachment.h"
 #include "MCMessageHeader.h"
 #include "MCHTMLRenderer.h"
@@ -140,6 +141,15 @@ HashMap * MessageParser::serializable()
     return result;
 }
 
+void MessageParser::importSerializable(HashMap * serializable)
+{
+    AbstractMessage::importSerializable(serializable);
+    MC_SAFE_REPLACE_RETAIN(AbstractPart, mMainPart, Object::objectWithSerializable((HashMap *) serializable->objectForKey(MCSTR("mainPart"))));
+    if (mMainPart != NULL) {
+        mMainPart->applyUniquePartID();
+    }
+}
+
 Object * MessageParser::copy()
 {
     return new MessageParser(this);
@@ -190,3 +200,13 @@ String * MessageParser::plainTextBodyRendering(bool stripWhitespace)
     }
     return plainTextBodyString;
 }
+
+static void * createObject()
+{
+    return new MessageParser();
+}
+
+INITIALIZE(MessageParser)
+{
+    Object::registerObjectConstructor("mailcore::MessageParser", &createObject);
+}

src/core/rfc822/MCMessageParser.h

@@ -42,7 +42,8 @@ namespace mailcore {
         virtual AbstractPart * partForUniqueID(String * uniqueID);
 
         virtual HashMap * serializable();
-        
+        virtual void importSerializable(HashMap * serializable);
+
 #ifdef __APPLE__
     public:
         static MessageParser * messageParserWithData(CFDataRef data);

src/core/rfc822/MCMessagePart.cpp

@@ -1,5 +1,7 @@
 #include "MCMessagePart.h"
 
+#include "MCDefines.h"
+
 using namespace mailcore;
 
 MessagePart::MessagePart()
@@ -18,3 +20,13 @@ Object * MessagePart::copy()
 {
     return new MessagePart(this);
 }
+
+static void * createObject()
+{
+    return new MessagePart();
+}
+
+INITIALIZE(MessagePart)
+{
+    Object::registerObjectConstructor("mailcore::MessagePart", &createObject);
+}

src/core/rfc822/MCMultipart.cpp

@@ -1,5 +1,7 @@
 #include "MCMultipart.h"
 
+#include "MCDefines.h"
+
 using namespace mailcore;
 
 Multipart::Multipart()
@@ -19,3 +21,12 @@ Object * Multipart::copy()
     return new Multipart(this);
 }
 
+static void * createObject()
+{
+    return new Multipart();
+}
+
+INITIALIZE(Multipart)
+{
+    Object::registerObjectConstructor("mailcore::Multipart", &createObject);
+}