ec5946e72facab20b3e1ef19965dc425501be3f0
[fms.git] / src / freenet / messagexml.cpp
1 #include "../../include/freenet/messagexml.h"\r
2 \r
3 #ifdef XMEM\r
4         #include <xmem.h>\r
5 #endif\r
6 \r
7 MessageXML::MessageXML()\r
8 {\r
9         Initialize();\r
10 }\r
11 \r
12 std::string MessageXML::GetXML()\r
13 {\r
14         TiXmlDocument td;\r
15         TiXmlDeclaration *tdec=new TiXmlDeclaration("1.0","UTF-8","");\r
16         TiXmlElement *tid;\r
17         TiXmlPrinter tp;\r
18 \r
19         td.LinkEndChild(tdec);\r
20         tid=new TiXmlElement("Message");\r
21         td.LinkEndChild(tid);\r
22 \r
23         tid->LinkEndChild(XMLCreateTextElement("Date",m_date));\r
24         tid->LinkEndChild(XMLCreateTextElement("Time",m_time));\r
25         tid->LinkEndChild(XMLCreateCDATAElement("Subject",m_subject));\r
26         tid->LinkEndChild(XMLCreateCDATAElement("MessageID",m_messageid));\r
27         tid->LinkEndChild(XMLCreateCDATAElement("ReplyBoard",m_replyboard));\r
28         // a little hack because TinyXML doesn't handle ]]> inside a CDATA section - manually separate into multiple CDATA sections\r
29         // TinyXML still won't be able to parse past the 1st CDATA section, but other implementations might\r
30         tid->LinkEndChild(XMLCreateCDATAElement("Body",StringFunctions::Replace(m_body,"]]>","]]]]><![CDATA[>")));\r
31 \r
32         TiXmlElement *brds=new TiXmlElement("Boards");\r
33         tid->LinkEndChild(brds);\r
34         for(std::vector<std::string>::iterator i=m_boards.begin(); i!=m_boards.end(); i++)\r
35         {\r
36                 std::string boardname=(*i);\r
37                 StringFunctions::Convert(boardname,boardname);\r
38                 brds->LinkEndChild(XMLCreateCDATAElement("Board",boardname));\r
39         }\r
40 \r
41         if(m_inreplyto.size()>0)\r
42         {\r
43                 TiXmlElement *rply=new TiXmlElement("InReplyTo");\r
44                 tid->LinkEndChild(rply);\r
45                 for(std::map<long,std::string>::iterator j=m_inreplyto.begin(); j!=m_inreplyto.end(); j++)\r
46                 {\r
47                         TiXmlElement *mess=new TiXmlElement("Message");\r
48                         rply->LinkEndChild(mess);\r
49                         mess->LinkEndChild(XMLCreateTextElement("Order",(*j).first));\r
50                         mess->LinkEndChild(XMLCreateCDATAElement("MessageID",(*j).second));\r
51                 }\r
52         }\r
53 \r
54         td.Accept(&tp);\r
55         return std::string(tp.CStr());\r
56 }\r
57 \r
58 void MessageXML::Initialize()\r
59 {\r
60         m_date="";\r
61         m_time="";\r
62         m_subject="";\r
63         m_boards.clear();\r
64         m_replyboard="";\r
65         m_inreplyto.clear();\r
66         m_body="";\r
67 }\r
68 \r
69 const bool MessageXML::ParseXML(const std::string &xml)\r
70 {\r
71         TiXmlDocument td;\r
72         td.Parse(xml.c_str());\r
73 \r
74         if(!td.Error())\r
75         {\r
76                 TiXmlHandle hnd(&td);\r
77                 TiXmlNode *node2;\r
78                 TiXmlText *txt;\r
79 \r
80                 Initialize();\r
81 \r
82                 txt=hnd.FirstChild("Message").FirstChild("Date").FirstChild().ToText();\r
83                 if(txt)\r
84                 {\r
85                         m_date=txt->ValueStr();\r
86                 }\r
87                 txt=hnd.FirstChild("Message").FirstChild("Time").FirstChild().ToText();\r
88                 if(txt)\r
89                 {\r
90                         m_time=txt->ValueStr();\r
91                 }\r
92                 txt=hnd.FirstChild("Message").FirstChild("Subject").FirstChild().ToText();\r
93                 if(txt)\r
94                 {\r
95                         m_subject=txt->ValueStr();\r
96                 }\r
97                 txt=hnd.FirstChild("Message").FirstChild("MessageID").FirstChild().ToText();\r
98                 if(txt)\r
99                 {\r
100                         m_messageid=txt->ValueStr();\r
101                 }\r
102                 txt=hnd.FirstChild("Message").FirstChild("ReplyBoard").FirstChild().ToText();\r
103                 if(txt)\r
104                 {\r
105                         m_replyboard=txt->ValueStr();\r
106                         StringFunctions::LowerCase(m_replyboard,m_replyboard);\r
107                         if(m_replyboard.size()>40)\r
108                         {\r
109                                 m_replyboard.erase(40);\r
110                         }\r
111                 }\r
112                 txt=hnd.FirstChild("Message").FirstChild("Body").FirstChild().ToText();\r
113                 if(txt)\r
114                 {\r
115                         m_body=txt->ValueStr();\r
116                 }\r
117 \r
118                 node2=hnd.FirstChild("Message").FirstChild("Boards").FirstChild("Board").ToNode();\r
119                 while(node2)\r
120                 {\r
121                         if(node2->FirstChild())\r
122                         {\r
123                                 std::string boardname=node2->FirstChild()->ValueStr();\r
124                                 StringFunctions::LowerCase(boardname,boardname);\r
125                                 if(boardname.size()>40)\r
126                                 {\r
127                                         boardname.erase(40);\r
128                                 }\r
129                                 m_boards.push_back(boardname);\r
130                         }\r
131                         node2=node2->NextSibling("Board");\r
132                 }\r
133 \r
134                 node2=hnd.FirstChild("Message").FirstChild("InReplyTo").FirstChild("Message").ToNode();\r
135                 while(node2)\r
136                 {\r
137                         std::string orderstr;\r
138                         long order=-1;\r
139                         std::string messageid="";\r
140                         TiXmlHandle hnd2(node2);\r
141                         txt=hnd2.FirstChild("Order").FirstChild().ToText();\r
142                         if(txt)\r
143                         {\r
144                                 orderstr=txt->ValueStr();\r
145                                 StringFunctions::Convert(orderstr,order);\r
146                         }\r
147                         txt=hnd2.FirstChild("MessageID").FirstChild().ToText();\r
148                         if(txt)\r
149                         {\r
150                                 messageid=txt->ValueStr();\r
151                         }\r
152 \r
153                         if(order!=-1 && messageid!="")\r
154                         {\r
155                                 m_inreplyto[order]=messageid;\r
156                         }\r
157 \r
158                         node2=node2->NextSibling("Message");\r
159                 }\r
160 \r
161                 return true;\r
162         }\r
163         else\r
164         {\r
165                 return false;\r
166         }\r
167 }\r