]>
Commit | Line | Data |
---|---|---|
1 | ///////////////////////////////////////////////////////////////////////////// | |
2 | // Name: src/xml/xml.cpp | |
3 | // Purpose: wxXmlDocument - XML parser & data holder class | |
4 | // Author: Vaclav Slavik | |
5 | // Created: 2000/03/05 | |
6 | // RCS-ID: $Id$ | |
7 | // Copyright: (c) 2000 Vaclav Slavik | |
8 | // Licence: wxWindows licence | |
9 | ///////////////////////////////////////////////////////////////////////////// | |
10 | ||
11 | // For compilers that support precompilation, includes "wx.h". | |
12 | #include "wx/wxprec.h" | |
13 | ||
14 | #ifdef __BORLANDC__ | |
15 | #pragma hdrstop | |
16 | #endif | |
17 | ||
18 | #if wxUSE_XML | |
19 | ||
20 | #include "wx/xml/xml.h" | |
21 | ||
22 | #ifndef WX_PRECOMP | |
23 | #include "wx/intl.h" | |
24 | #include "wx/log.h" | |
25 | #include "wx/app.h" | |
26 | #endif | |
27 | ||
28 | #include "wx/wfstream.h" | |
29 | #include "wx/datstrm.h" | |
30 | #include "wx/zstream.h" | |
31 | #include "wx/strconv.h" | |
32 | ||
33 | #include "expat.h" // from Expat | |
34 | ||
35 | // DLL options compatibility check: | |
36 | WX_CHECK_BUILD_OPTIONS("wxXML") | |
37 | ||
38 | ||
39 | IMPLEMENT_CLASS(wxXmlDocument, wxObject) | |
40 | ||
41 | ||
42 | // a private utility used by wxXML | |
43 | static bool wxIsWhiteOnly(const wxChar *buf); | |
44 | ||
45 | ||
46 | //----------------------------------------------------------------------------- | |
47 | // wxXmlNode | |
48 | //----------------------------------------------------------------------------- | |
49 | ||
50 | wxXmlNode::wxXmlNode(wxXmlNode *parent,wxXmlNodeType type, | |
51 | const wxString& name, const wxString& content, | |
52 | wxXmlProperty *props, wxXmlNode *next) | |
53 | : m_type(type), m_name(name), m_content(content), | |
54 | m_properties(props), m_parent(parent), | |
55 | m_children(NULL), m_next(next) | |
56 | { | |
57 | if (m_parent) | |
58 | { | |
59 | if (m_parent->m_children) | |
60 | { | |
61 | m_next = m_parent->m_children; | |
62 | m_parent->m_children = this; | |
63 | } | |
64 | else | |
65 | m_parent->m_children = this; | |
66 | } | |
67 | } | |
68 | ||
69 | wxXmlNode::wxXmlNode(wxXmlNodeType type, const wxString& name, | |
70 | const wxString& content) | |
71 | : m_type(type), m_name(name), m_content(content), | |
72 | m_properties(NULL), m_parent(NULL), | |
73 | m_children(NULL), m_next(NULL) | |
74 | {} | |
75 | ||
76 | wxXmlNode::wxXmlNode(const wxXmlNode& node) | |
77 | { | |
78 | m_next = NULL; | |
79 | m_parent = NULL; | |
80 | DoCopy(node); | |
81 | } | |
82 | ||
83 | wxXmlNode::~wxXmlNode() | |
84 | { | |
85 | wxXmlNode *c, *c2; | |
86 | for (c = m_children; c; c = c2) | |
87 | { | |
88 | c2 = c->m_next; | |
89 | delete c; | |
90 | } | |
91 | ||
92 | wxXmlProperty *p, *p2; | |
93 | for (p = m_properties; p; p = p2) | |
94 | { | |
95 | p2 = p->GetNext(); | |
96 | delete p; | |
97 | } | |
98 | } | |
99 | ||
100 | wxXmlNode& wxXmlNode::operator=(const wxXmlNode& node) | |
101 | { | |
102 | wxDELETE(m_properties); | |
103 | wxDELETE(m_children); | |
104 | DoCopy(node); | |
105 | return *this; | |
106 | } | |
107 | ||
108 | void wxXmlNode::DoCopy(const wxXmlNode& node) | |
109 | { | |
110 | m_type = node.m_type; | |
111 | m_name = node.m_name; | |
112 | m_content = node.m_content; | |
113 | m_children = NULL; | |
114 | ||
115 | wxXmlNode *n = node.m_children; | |
116 | while (n) | |
117 | { | |
118 | AddChild(new wxXmlNode(*n)); | |
119 | n = n->GetNext(); | |
120 | } | |
121 | ||
122 | m_properties = NULL; | |
123 | wxXmlProperty *p = node.m_properties; | |
124 | while (p) | |
125 | { | |
126 | AddProperty(p->GetName(), p->GetValue()); | |
127 | p = p->GetNext(); | |
128 | } | |
129 | } | |
130 | ||
131 | bool wxXmlNode::HasProp(const wxString& propName) const | |
132 | { | |
133 | wxXmlProperty *prop = GetProperties(); | |
134 | ||
135 | while (prop) | |
136 | { | |
137 | if (prop->GetName() == propName) return true; | |
138 | prop = prop->GetNext(); | |
139 | } | |
140 | ||
141 | return false; | |
142 | } | |
143 | ||
144 | bool wxXmlNode::GetPropVal(const wxString& propName, wxString *value) const | |
145 | { | |
146 | wxXmlProperty *prop = GetProperties(); | |
147 | ||
148 | while (prop) | |
149 | { | |
150 | if (prop->GetName() == propName) | |
151 | { | |
152 | *value = prop->GetValue(); | |
153 | return true; | |
154 | } | |
155 | prop = prop->GetNext(); | |
156 | } | |
157 | ||
158 | return false; | |
159 | } | |
160 | ||
161 | wxString wxXmlNode::GetPropVal(const wxString& propName, const wxString& defaultVal) const | |
162 | { | |
163 | wxString tmp; | |
164 | if (GetPropVal(propName, &tmp)) | |
165 | return tmp; | |
166 | ||
167 | return defaultVal; | |
168 | } | |
169 | ||
170 | void wxXmlNode::AddChild(wxXmlNode *child) | |
171 | { | |
172 | if (m_children == NULL) | |
173 | m_children = child; | |
174 | else | |
175 | { | |
176 | wxXmlNode *ch = m_children; | |
177 | while (ch->m_next) ch = ch->m_next; | |
178 | ch->m_next = child; | |
179 | } | |
180 | child->m_next = NULL; | |
181 | child->m_parent = this; | |
182 | } | |
183 | ||
184 | bool wxXmlNode::InsertChild(wxXmlNode *child, wxXmlNode *before_node) | |
185 | { | |
186 | wxCHECK_MSG(before_node == NULL || before_node->GetParent() == this, false, | |
187 | wxT("wxXmlNode::InsertChild - the node has incorrect parent")); | |
188 | wxCHECK_MSG(child, false, wxT("Cannot insert a NULL pointer!")); | |
189 | ||
190 | if (m_children == before_node) | |
191 | m_children = child; | |
192 | else if (m_children == NULL) | |
193 | { | |
194 | if (before_node != NULL) | |
195 | return false; // we have no children so we don't need to search | |
196 | m_children = child; | |
197 | } | |
198 | else if (before_node == NULL) | |
199 | { | |
200 | // prepend child | |
201 | child->m_parent = this; | |
202 | child->m_next = m_children; | |
203 | m_children = child; | |
204 | return true; | |
205 | } | |
206 | else | |
207 | { | |
208 | wxXmlNode *ch = m_children; | |
209 | while (ch && ch->m_next != before_node) ch = ch->m_next; | |
210 | if (!ch) | |
211 | return false; // before_node not found | |
212 | ch->m_next = child; | |
213 | } | |
214 | ||
215 | child->m_parent = this; | |
216 | child->m_next = before_node; | |
217 | return true; | |
218 | } | |
219 | ||
220 | bool wxXmlNode::RemoveChild(wxXmlNode *child) | |
221 | { | |
222 | if (m_children == NULL) | |
223 | return false; | |
224 | else if (m_children == child) | |
225 | { | |
226 | m_children = child->m_next; | |
227 | child->m_parent = NULL; | |
228 | child->m_next = NULL; | |
229 | return true; | |
230 | } | |
231 | else | |
232 | { | |
233 | wxXmlNode *ch = m_children; | |
234 | while (ch->m_next) | |
235 | { | |
236 | if (ch->m_next == child) | |
237 | { | |
238 | ch->m_next = child->m_next; | |
239 | child->m_parent = NULL; | |
240 | child->m_next = NULL; | |
241 | return true; | |
242 | } | |
243 | ch = ch->m_next; | |
244 | } | |
245 | return false; | |
246 | } | |
247 | } | |
248 | ||
249 | void wxXmlNode::AddProperty(const wxString& name, const wxString& value) | |
250 | { | |
251 | AddProperty(new wxXmlProperty(name, value, NULL)); | |
252 | } | |
253 | ||
254 | void wxXmlNode::AddProperty(wxXmlProperty *prop) | |
255 | { | |
256 | if (m_properties == NULL) | |
257 | m_properties = prop; | |
258 | else | |
259 | { | |
260 | wxXmlProperty *p = m_properties; | |
261 | while (p->GetNext()) p = p->GetNext(); | |
262 | p->SetNext(prop); | |
263 | } | |
264 | } | |
265 | ||
266 | bool wxXmlNode::DeleteProperty(const wxString& name) | |
267 | { | |
268 | wxXmlProperty *prop; | |
269 | ||
270 | if (m_properties == NULL) | |
271 | return false; | |
272 | ||
273 | else if (m_properties->GetName() == name) | |
274 | { | |
275 | prop = m_properties; | |
276 | m_properties = prop->GetNext(); | |
277 | prop->SetNext(NULL); | |
278 | delete prop; | |
279 | return true; | |
280 | } | |
281 | ||
282 | else | |
283 | { | |
284 | wxXmlProperty *p = m_properties; | |
285 | while (p->GetNext()) | |
286 | { | |
287 | if (p->GetNext()->GetName() == name) | |
288 | { | |
289 | prop = p->GetNext(); | |
290 | p->SetNext(prop->GetNext()); | |
291 | prop->SetNext(NULL); | |
292 | delete prop; | |
293 | return true; | |
294 | } | |
295 | p = p->GetNext(); | |
296 | } | |
297 | return false; | |
298 | } | |
299 | } | |
300 | ||
301 | wxString wxXmlNode::GetNodeContent() const | |
302 | { | |
303 | wxXmlNode *n = GetChildren(); | |
304 | ||
305 | while (n) | |
306 | { | |
307 | if (n->GetType() == wxXML_TEXT_NODE || | |
308 | n->GetType() == wxXML_CDATA_SECTION_NODE) | |
309 | return n->GetContent(); | |
310 | n = n->GetNext(); | |
311 | } | |
312 | return wxEmptyString; | |
313 | } | |
314 | ||
315 | int wxXmlNode::GetDepth(wxXmlNode *grandparent) const | |
316 | { | |
317 | const wxXmlNode *n = this; | |
318 | int ret = -1; | |
319 | ||
320 | do | |
321 | { | |
322 | ret++; | |
323 | n = n->GetParent(); | |
324 | if (n == grandparent) | |
325 | return ret; | |
326 | ||
327 | } while (n); | |
328 | ||
329 | return wxNOT_FOUND; | |
330 | } | |
331 | ||
332 | bool wxXmlNode::IsWhitespaceOnly() const | |
333 | { | |
334 | return wxIsWhiteOnly(m_content); | |
335 | } | |
336 | ||
337 | ||
338 | ||
339 | //----------------------------------------------------------------------------- | |
340 | // wxXmlDocument | |
341 | //----------------------------------------------------------------------------- | |
342 | ||
343 | wxXmlDocument::wxXmlDocument() | |
344 | : m_version(wxT("1.0")), m_fileEncoding(wxT("utf-8")), m_root(NULL) | |
345 | { | |
346 | #if !wxUSE_UNICODE | |
347 | m_encoding = wxT("UTF-8"); | |
348 | #endif | |
349 | } | |
350 | ||
351 | wxXmlDocument::wxXmlDocument(const wxString& filename, const wxString& encoding) | |
352 | :wxObject(), m_root(NULL) | |
353 | { | |
354 | if ( !Load(filename, encoding) ) | |
355 | { | |
356 | wxDELETE(m_root); | |
357 | } | |
358 | } | |
359 | ||
360 | wxXmlDocument::wxXmlDocument(wxInputStream& stream, const wxString& encoding) | |
361 | :wxObject(), m_root(NULL) | |
362 | { | |
363 | if ( !Load(stream, encoding) ) | |
364 | { | |
365 | wxDELETE(m_root); | |
366 | } | |
367 | } | |
368 | ||
369 | wxXmlDocument::wxXmlDocument(const wxXmlDocument& doc) | |
370 | :wxObject() | |
371 | { | |
372 | DoCopy(doc); | |
373 | } | |
374 | ||
375 | wxXmlDocument& wxXmlDocument::operator=(const wxXmlDocument& doc) | |
376 | { | |
377 | wxDELETE(m_root); | |
378 | DoCopy(doc); | |
379 | return *this; | |
380 | } | |
381 | ||
382 | void wxXmlDocument::DoCopy(const wxXmlDocument& doc) | |
383 | { | |
384 | m_version = doc.m_version; | |
385 | #if !wxUSE_UNICODE | |
386 | m_encoding = doc.m_encoding; | |
387 | #endif | |
388 | m_fileEncoding = doc.m_fileEncoding; | |
389 | m_root = new wxXmlNode(*doc.m_root); | |
390 | } | |
391 | ||
392 | bool wxXmlDocument::Load(const wxString& filename, const wxString& encoding, int flags) | |
393 | { | |
394 | wxFileInputStream stream(filename); | |
395 | if (!stream.Ok()) | |
396 | return false; | |
397 | return Load(stream, encoding, flags); | |
398 | } | |
399 | ||
400 | bool wxXmlDocument::Save(const wxString& filename, int indentstep) const | |
401 | { | |
402 | wxFileOutputStream stream(filename); | |
403 | if (!stream.Ok()) | |
404 | return false; | |
405 | return Save(stream, indentstep); | |
406 | } | |
407 | ||
408 | ||
409 | ||
410 | //----------------------------------------------------------------------------- | |
411 | // wxXmlDocument loading routines | |
412 | //----------------------------------------------------------------------------- | |
413 | ||
414 | // converts Expat-produced string in UTF-8 into wxString using the specified | |
415 | // conv or keep in UTF-8 if conv is NULL | |
416 | static wxString CharToString(wxMBConv *conv, | |
417 | const char *s, size_t len = wxSTRING_MAXLEN) | |
418 | { | |
419 | #if wxUSE_UNICODE | |
420 | wxUnusedVar(conv); | |
421 | ||
422 | return wxString(s, wxConvUTF8, len); | |
423 | #else // !wxUSE_UNICODE | |
424 | if ( conv ) | |
425 | { | |
426 | // there can be no embedded NULs in this string so we don't need the | |
427 | // output length, it will be NUL-terminated | |
428 | const wxWCharBuffer wbuf( | |
429 | wxConvUTF8.cMB2WC(s, len == wxSTRING_MAXLEN ? wxNO_LEN : len, NULL)); | |
430 | ||
431 | return wxString(wbuf, *conv); | |
432 | } | |
433 | else // already in UTF-8, no conversion needed | |
434 | { | |
435 | return wxString(s, len != wxSTRING_MAXLEN ? len : strlen(s)); | |
436 | } | |
437 | #endif // wxUSE_UNICODE/!wxUSE_UNICODE | |
438 | } | |
439 | ||
440 | // returns true if the given string contains only whitespaces | |
441 | bool wxIsWhiteOnly(const wxChar *buf) | |
442 | { | |
443 | for (const wxChar *c = buf; *c != wxT('\0'); c++) | |
444 | if (*c != wxT(' ') && *c != wxT('\t') && *c != wxT('\n') && *c != wxT('\r')) | |
445 | return false; | |
446 | return true; | |
447 | } | |
448 | ||
449 | ||
450 | struct wxXmlParsingContext | |
451 | { | |
452 | wxMBConv *conv; | |
453 | wxXmlNode *root; | |
454 | wxXmlNode *node; | |
455 | wxXmlNode *lastAsText; | |
456 | wxString encoding; | |
457 | wxString version; | |
458 | bool bLastCdata; | |
459 | bool removeWhiteOnlyNodes; | |
460 | }; | |
461 | ||
462 | extern "C" { | |
463 | static void StartElementHnd(void *userData, const char *name, const char **atts) | |
464 | { | |
465 | wxXmlParsingContext *ctx = (wxXmlParsingContext*)userData; | |
466 | wxXmlNode *node = new wxXmlNode(wxXML_ELEMENT_NODE, CharToString(ctx->conv, name)); | |
467 | const char **a = atts; | |
468 | while (*a) | |
469 | { | |
470 | node->AddProperty(CharToString(ctx->conv, a[0]), CharToString(ctx->conv, a[1])); | |
471 | a += 2; | |
472 | } | |
473 | if (ctx->root == NULL) | |
474 | ctx->root = node; | |
475 | else | |
476 | ctx->node->AddChild(node); | |
477 | ctx->node = node; | |
478 | ctx->lastAsText = NULL; | |
479 | } | |
480 | } | |
481 | ||
482 | extern "C" { | |
483 | static void EndElementHnd(void *userData, const char* WXUNUSED(name)) | |
484 | { | |
485 | wxXmlParsingContext *ctx = (wxXmlParsingContext*)userData; | |
486 | ||
487 | ctx->node = ctx->node->GetParent(); | |
488 | ctx->lastAsText = NULL; | |
489 | } | |
490 | } | |
491 | ||
492 | extern "C" { | |
493 | static void TextHnd(void *userData, const char *s, int len) | |
494 | { | |
495 | wxXmlParsingContext *ctx = (wxXmlParsingContext*)userData; | |
496 | wxString str = CharToString(ctx->conv, s, len); | |
497 | ||
498 | if (ctx->lastAsText) | |
499 | { | |
500 | if ( ctx->bLastCdata ) | |
501 | { | |
502 | ctx->lastAsText->SetContent(ctx->lastAsText->GetContent() + | |
503 | CharToString(NULL, s, len)); | |
504 | } | |
505 | else | |
506 | { | |
507 | ctx->lastAsText->SetContent(ctx->lastAsText->GetContent() + str); | |
508 | } | |
509 | } | |
510 | else | |
511 | { | |
512 | bool whiteOnly = false; | |
513 | if (ctx->removeWhiteOnlyNodes) | |
514 | whiteOnly = wxIsWhiteOnly(str); | |
515 | ||
516 | if (!whiteOnly) | |
517 | { | |
518 | ctx->lastAsText = new wxXmlNode(wxXML_TEXT_NODE, wxT("text"), str); | |
519 | ctx->node->AddChild(ctx->lastAsText); | |
520 | } | |
521 | } | |
522 | } | |
523 | } | |
524 | ||
525 | extern "C" { | |
526 | static void StartCdataHnd(void *userData) | |
527 | { | |
528 | wxXmlParsingContext *ctx = (wxXmlParsingContext*)userData; | |
529 | ||
530 | ctx->bLastCdata = true; | |
531 | ||
532 | ctx->lastAsText = new wxXmlNode(wxXML_CDATA_SECTION_NODE, wxT("cdata"),wxT("")); | |
533 | ctx->node->AddChild(ctx->lastAsText); | |
534 | } | |
535 | } | |
536 | ||
537 | extern "C" { | |
538 | static void EndCdataHnd(void *userData) | |
539 | { | |
540 | wxXmlParsingContext *ctx = (wxXmlParsingContext*)userData; | |
541 | ||
542 | ctx->bLastCdata = false; | |
543 | } | |
544 | } | |
545 | ||
546 | extern "C" { | |
547 | static void CommentHnd(void *userData, const char *data) | |
548 | { | |
549 | wxXmlParsingContext *ctx = (wxXmlParsingContext*)userData; | |
550 | ||
551 | if (ctx->node) | |
552 | { | |
553 | // VS: ctx->node == NULL happens if there is a comment before | |
554 | // the root element (e.g. wxDesigner's output). We ignore such | |
555 | // comments, no big deal... | |
556 | ctx->node->AddChild(new wxXmlNode(wxXML_COMMENT_NODE, | |
557 | wxT("comment"), CharToString(ctx->conv, data))); | |
558 | } | |
559 | ctx->lastAsText = NULL; | |
560 | } | |
561 | } | |
562 | ||
563 | extern "C" { | |
564 | static void DefaultHnd(void *userData, const char *s, int len) | |
565 | { | |
566 | // XML header: | |
567 | if (len > 6 && memcmp(s, "<?xml ", 6) == 0) | |
568 | { | |
569 | wxXmlParsingContext *ctx = (wxXmlParsingContext*)userData; | |
570 | ||
571 | wxString buf = CharToString(ctx->conv, s, (size_t)len); | |
572 | int pos; | |
573 | pos = buf.Find(wxT("encoding=")); | |
574 | if (pos != wxNOT_FOUND) | |
575 | ctx->encoding = buf.Mid(pos + 10).BeforeFirst(buf[(size_t)pos+9]); | |
576 | pos = buf.Find(wxT("version=")); | |
577 | if (pos != wxNOT_FOUND) | |
578 | ctx->version = buf.Mid(pos + 9).BeforeFirst(buf[(size_t)pos+8]); | |
579 | } | |
580 | } | |
581 | } | |
582 | ||
583 | extern "C" { | |
584 | static int UnknownEncodingHnd(void * WXUNUSED(encodingHandlerData), | |
585 | const XML_Char *name, XML_Encoding *info) | |
586 | { | |
587 | // We must build conversion table for expat. The easiest way to do so | |
588 | // is to let wxCSConv convert as string containing all characters to | |
589 | // wide character representation: | |
590 | wxString str(name, wxConvLibc); | |
591 | wxCSConv conv(str); | |
592 | char mbBuf[2]; | |
593 | wchar_t wcBuf[10]; | |
594 | size_t i; | |
595 | ||
596 | mbBuf[1] = 0; | |
597 | info->map[0] = 0; | |
598 | for (i = 0; i < 255; i++) | |
599 | { | |
600 | mbBuf[0] = (char)(i+1); | |
601 | if (conv.MB2WC(wcBuf, mbBuf, 2) == (size_t)-1) | |
602 | { | |
603 | // invalid/undefined byte in the encoding: | |
604 | info->map[i+1] = -1; | |
605 | } | |
606 | info->map[i+1] = (int)wcBuf[0]; | |
607 | } | |
608 | ||
609 | info->data = NULL; | |
610 | info->convert = NULL; | |
611 | info->release = NULL; | |
612 | ||
613 | return 1; | |
614 | } | |
615 | } | |
616 | ||
617 | bool wxXmlDocument::Load(wxInputStream& stream, const wxString& encoding, int flags) | |
618 | { | |
619 | #if wxUSE_UNICODE | |
620 | (void)encoding; | |
621 | #else | |
622 | m_encoding = encoding; | |
623 | #endif | |
624 | ||
625 | const size_t BUFSIZE = 1024; | |
626 | char buf[BUFSIZE]; | |
627 | wxXmlParsingContext ctx; | |
628 | bool done; | |
629 | XML_Parser parser = XML_ParserCreate(NULL); | |
630 | ||
631 | ctx.root = ctx.node = NULL; | |
632 | ctx.encoding = wxT("UTF-8"); // default in absence of encoding="" | |
633 | ctx.conv = NULL; | |
634 | #if !wxUSE_UNICODE | |
635 | if ( encoding != wxT("UTF-8") && encoding != wxT("utf-8") ) | |
636 | ctx.conv = new wxCSConv(encoding); | |
637 | #endif | |
638 | ctx.removeWhiteOnlyNodes = (flags & wxXMLDOC_KEEP_WHITESPACE_NODES) == 0; | |
639 | ctx.bLastCdata = false; | |
640 | ||
641 | XML_SetUserData(parser, (void*)&ctx); | |
642 | XML_SetElementHandler(parser, StartElementHnd, EndElementHnd); | |
643 | XML_SetCharacterDataHandler(parser, TextHnd); | |
644 | XML_SetCdataSectionHandler(parser, StartCdataHnd, EndCdataHnd ); | |
645 | XML_SetCommentHandler(parser, CommentHnd); | |
646 | XML_SetDefaultHandler(parser, DefaultHnd); | |
647 | XML_SetUnknownEncodingHandler(parser, UnknownEncodingHnd, NULL); | |
648 | ||
649 | bool ok = true; | |
650 | do | |
651 | { | |
652 | size_t len = stream.Read(buf, BUFSIZE).LastRead(); | |
653 | done = (len < BUFSIZE); | |
654 | if (!XML_Parse(parser, buf, len, done)) | |
655 | { | |
656 | wxString error(XML_ErrorString(XML_GetErrorCode(parser)), | |
657 | *wxConvCurrent); | |
658 | wxLogError(_("XML parsing error: '%s' at line %d"), | |
659 | error.c_str(), | |
660 | XML_GetCurrentLineNumber(parser)); | |
661 | ok = false; | |
662 | break; | |
663 | } | |
664 | } while (!done); | |
665 | ||
666 | if (ok) | |
667 | { | |
668 | if (!ctx.version.empty()) | |
669 | SetVersion(ctx.version); | |
670 | if (!ctx.encoding.empty()) | |
671 | SetFileEncoding(ctx.encoding); | |
672 | SetRoot(ctx.root); | |
673 | } | |
674 | else | |
675 | { | |
676 | delete ctx.root; | |
677 | } | |
678 | ||
679 | XML_ParserFree(parser); | |
680 | #if !wxUSE_UNICODE | |
681 | if ( ctx.conv ) | |
682 | delete ctx.conv; | |
683 | #endif | |
684 | ||
685 | return ok; | |
686 | ||
687 | } | |
688 | ||
689 | ||
690 | ||
691 | //----------------------------------------------------------------------------- | |
692 | // wxXmlDocument saving routines | |
693 | //----------------------------------------------------------------------------- | |
694 | ||
695 | // write string to output: | |
696 | inline static void OutputString(wxOutputStream& stream, const wxString& str, | |
697 | wxMBConv *convMem = NULL, | |
698 | wxMBConv *convFile = NULL) | |
699 | { | |
700 | if (str.empty()) | |
701 | return; | |
702 | ||
703 | #if wxUSE_UNICODE | |
704 | wxUnusedVar(convMem); | |
705 | ||
706 | const wxWX2MBbuf buf(str.mb_str(*(convFile ? convFile : &wxConvUTF8))); | |
707 | stream.Write((const char*)buf, strlen((const char*)buf)); | |
708 | #else // !wxUSE_UNICODE | |
709 | if ( convFile && convMem ) | |
710 | { | |
711 | wxString str2(str.wc_str(*convMem), *convFile); | |
712 | stream.Write(str2.mb_str(), str2.Len()); | |
713 | } | |
714 | else // no conversions to do | |
715 | { | |
716 | stream.Write(str.mb_str(), str.Len()); | |
717 | } | |
718 | #endif // wxUSE_UNICODE/!wxUSE_UNICODE | |
719 | } | |
720 | ||
721 | // flags for OutputStringEnt() | |
722 | enum | |
723 | { | |
724 | XML_ESCAPE_QUOTES = 1 | |
725 | }; | |
726 | ||
727 | // Same as above, but create entities first. | |
728 | // Translates '<' to "<", '>' to ">" and '&' to "&" | |
729 | static void OutputStringEnt(wxOutputStream& stream, const wxString& str, | |
730 | wxMBConv *convMem = NULL, | |
731 | wxMBConv *convFile = NULL, | |
732 | int flags = 0) | |
733 | { | |
734 | wxString buf; | |
735 | size_t i, last, len; | |
736 | wxChar c; | |
737 | ||
738 | len = str.Len(); | |
739 | last = 0; | |
740 | for (i = 0; i < len; i++) | |
741 | { | |
742 | c = str.GetChar(i); | |
743 | if (c == wxT('<') || c == wxT('>') || | |
744 | (c == wxT('&') && str.Mid(i+1, 4) != wxT("amp;")) || | |
745 | ((flags & XML_ESCAPE_QUOTES) && c == wxT('"'))) | |
746 | { | |
747 | OutputString(stream, str.Mid(last, i - last), convMem, convFile); | |
748 | switch (c) | |
749 | { | |
750 | case wxT('<'): | |
751 | OutputString(stream, wxT("<")); | |
752 | break; | |
753 | case wxT('>'): | |
754 | OutputString(stream, wxT(">")); | |
755 | break; | |
756 | case wxT('&'): | |
757 | OutputString(stream, wxT("&")); | |
758 | break; | |
759 | case wxT('"'): | |
760 | OutputString(stream, wxT(""")); | |
761 | break; | |
762 | default: | |
763 | break; | |
764 | } | |
765 | last = i + 1; | |
766 | } | |
767 | } | |
768 | OutputString(stream, str.Mid(last, i - last), convMem, convFile); | |
769 | } | |
770 | ||
771 | inline static void OutputIndentation(wxOutputStream& stream, int indent) | |
772 | { | |
773 | wxString str = wxT("\n"); | |
774 | for (int i = 0; i < indent; i++) | |
775 | str << wxT(' ') << wxT(' '); | |
776 | OutputString(stream, str); | |
777 | } | |
778 | ||
779 | static void OutputNode(wxOutputStream& stream, wxXmlNode *node, int indent, | |
780 | wxMBConv *convMem, wxMBConv *convFile, int indentstep) | |
781 | { | |
782 | wxXmlNode *n, *prev; | |
783 | wxXmlProperty *prop; | |
784 | ||
785 | switch (node->GetType()) | |
786 | { | |
787 | case wxXML_CDATA_SECTION_NODE: | |
788 | OutputString( stream, wxT("<![CDATA[")); | |
789 | OutputString( stream, node->GetContent() ); | |
790 | OutputString( stream, wxT("]]>") ); | |
791 | break; | |
792 | ||
793 | case wxXML_TEXT_NODE: | |
794 | OutputStringEnt(stream, node->GetContent(), convMem, convFile); | |
795 | break; | |
796 | ||
797 | case wxXML_ELEMENT_NODE: | |
798 | OutputString(stream, wxT("<")); | |
799 | OutputString(stream, node->GetName()); | |
800 | ||
801 | prop = node->GetProperties(); | |
802 | while (prop) | |
803 | { | |
804 | OutputString(stream, wxT(" ") + prop->GetName() + wxT("=\"")); | |
805 | OutputStringEnt(stream, prop->GetValue(), convMem, convFile, | |
806 | XML_ESCAPE_QUOTES); | |
807 | OutputString(stream, wxT("\"")); | |
808 | prop = prop->GetNext(); | |
809 | } | |
810 | ||
811 | if (node->GetChildren()) | |
812 | { | |
813 | OutputString(stream, wxT(">")); | |
814 | prev = NULL; | |
815 | n = node->GetChildren(); | |
816 | while (n) | |
817 | { | |
818 | if (indentstep >= 0 && n && n->GetType() != wxXML_TEXT_NODE) | |
819 | OutputIndentation(stream, indent + indentstep); | |
820 | OutputNode(stream, n, indent + indentstep, convMem, convFile, indentstep); | |
821 | prev = n; | |
822 | n = n->GetNext(); | |
823 | } | |
824 | if (indentstep >= 0 && prev && prev->GetType() != wxXML_TEXT_NODE) | |
825 | OutputIndentation(stream, indent); | |
826 | OutputString(stream, wxT("</")); | |
827 | OutputString(stream, node->GetName()); | |
828 | OutputString(stream, wxT(">")); | |
829 | } | |
830 | else | |
831 | OutputString(stream, wxT("/>")); | |
832 | break; | |
833 | ||
834 | case wxXML_COMMENT_NODE: | |
835 | OutputString(stream, wxT("<!--")); | |
836 | OutputString(stream, node->GetContent(), convMem, convFile); | |
837 | OutputString(stream, wxT("-->")); | |
838 | break; | |
839 | ||
840 | default: | |
841 | wxFAIL_MSG(wxT("unsupported node type")); | |
842 | } | |
843 | } | |
844 | ||
845 | bool wxXmlDocument::Save(wxOutputStream& stream, int indentstep) const | |
846 | { | |
847 | if ( !IsOk() ) | |
848 | return false; | |
849 | ||
850 | wxString s; | |
851 | ||
852 | wxMBConv *convMem = NULL; | |
853 | ||
854 | #if wxUSE_UNICODE | |
855 | wxMBConv *convFile = new wxCSConv(GetFileEncoding()); | |
856 | #else | |
857 | wxMBConv *convFile = NULL; | |
858 | if ( GetFileEncoding() != GetEncoding() ) | |
859 | { | |
860 | convFile = new wxCSConv(GetFileEncoding()); | |
861 | convMem = new wxCSConv(GetEncoding()); | |
862 | } | |
863 | #endif | |
864 | ||
865 | s.Printf(wxT("<?xml version=\"%s\" encoding=\"%s\"?>\n"), | |
866 | GetVersion().c_str(), GetFileEncoding().c_str()); | |
867 | OutputString(stream, s); | |
868 | ||
869 | OutputNode(stream, GetRoot(), 0, convMem, convFile, indentstep); | |
870 | OutputString(stream, wxT("\n")); | |
871 | ||
872 | if ( convFile ) | |
873 | delete convFile; | |
874 | if ( convMem ) | |
875 | delete convMem; | |
876 | ||
877 | return true; | |
878 | } | |
879 | ||
880 | #endif // wxUSE_XML |