src/xml/xml.cpp

   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::npos)
 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::npos ? 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::npos ? 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 "&lt;", '>' to "&gt;" and '&' to "&amp;"
 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("&lt;"));
 752                     break;
 753                 case wxT('>'):
 754                     OutputString(stream, wxT("&gt;"));
 755                     break;
 756                 case wxT('&'):
 757                     OutputString(stream, wxT("&amp;"));
 758                     break;
 759                 case wxT('"'):
 760                     OutputString(stream, wxT("&quot;"));
 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