xml.cpp

#include <iostream>
#include <fstream>
#include <map>
#include <list>
#include <string>
#include <stack>
#include <sstream>


#include "gm/gm.h"
#include "gm/xml.h"

namespace GNUMessenger
{

using namespace std;

string XMLParser::escape(string str)
{
    string a[] = {"&", "<", ">", "\"", "'"};
    string b[] = {"&amp;",  "&lt;",  "&gt;",  "&quot;",  "&apos;"};

    for (int i=0; i<5; i++) 
    {
        size_t pos=0;
        while ( ( pos = str.find(a[i], pos ) ) != std::string::npos ) 
        {
           str.replace( pos, a[i].length(), b[i] );
           pos += b[i].length(); // new starting point of search is just after 'b'
        }
    }
    return str;
}

void XMLParser::init(const string &line, Type type)
{
    switch (type)
    {
    case String:
        try {
            parse(line);
        }
        catch (Expat::ParseFailure &e)
        {
            LOG_THROW(e.what(), ParseFailure);
        }
        break;
    case File:
    {
        string str;
        ifstream in(line.c_str());
        if (!in)
        {
            LOG_THROW("XMLParser: Specified file cannot be opened.", DiskError);
            return;
        }

        while (in.good())
        {
           getline(in,str);
           str+="\n";
           parse(str);
        }
        in.close();
    }
    break;
    };

}

XMLParser::XMLParser(const string &line, Type type)
: Expat()
{
    init(line, type);
}

XMLParser::XMLParser(XMLNode &_root)
: Expat()
{
    root = _root;  
}

XMLParser::~XMLParser()
{
    
}

int XMLParser::tree_to_xml(string &line)
{
    printTag(root,line);
    return 0;
}

int XMLParser::tree_to_file(const string &filename)
{
    string line;
    ofstream out(filename.c_str());

    if (!out)
    {
        LOG_THROW("File Error!", DiskError);
        return -1;
    }

    printTag(root,line);
    out << line;

    out.close();
    return 0;
}

vector<XMLNode> XMLParser::get_tags(const string &key)
{
    vector<XMLNode> ret;
    return ret;
}

int XMLParser::printTag(const XMLNode &node, string &ret, int indent)
{
    int s(0);
    
    // print indent
    for (s=0; s<indent; s++) 
        ret += "\t";

    // print tag name
    ret += "<" + XMLParser::escape(node.m_xmlData->m_name);
    
    // print properties
    for (map<string,string>::const_iterator i=node.m_xmlData->m_properties.begin();i!=node.m_xmlData->m_properties.end();i++)
    {
        if (i->second == "")
            ret+= " " + XMLParser::escape(i->first) + "=\"\"";
        else
            ret+= " " + XMLParser::escape(i->first) + "=\"" + XMLParser::escape(i->second) + "\"";
    }

    // if tag has no children, close and return
    if (( node.m_xmlData->m_data == "" || node.m_xmlData->m_data == "\n")  && node.m_xmlData->m_children.empty())
    {
        ret+="/>";
        return 0;
    }
    
    // if has data/children
    ret+=">";
    
    // print children
    bool childprinted = false;
    for (vector<XMLNode>::const_iterator it=node.m_xmlData->m_children.begin();it!=node.m_xmlData->m_children.end();it++)
    {
       ret+="\n";
       childprinted = true;
       string tmp;
       printTag(*it,tmp,indent+1);
       ret+=tmp;
    }
  
    // indent for data
    if (childprinted && (node.m_xmlData->m_data.length()>0))
        for (s=0; s<indent + 1; s++) 
           ret += "\t";

    if (node.m_xmlData->data_is_cdata)
        ret+="<![CDATA[";


    if (node.m_xmlData->m_data.length()>0)
    {
        if (node.m_xmlData->data_is_cdata)
           ret+=node.m_xmlData->m_data;
        else
           ret+=XMLParser::escape(node.m_xmlData->m_data);
    }

    if (node.m_xmlData->data_is_cdata)
        ret+="]]>";

    if (childprinted)
        ret+='\n';

    // print indent
    if (childprinted)
        for (s=0; s<indent; s++) ret += "\t";
    
    ret += "</" + XMLParser::escape(node.m_xmlData->m_name) + ">\n";
    
    return 0;
}


void XMLParser::recievedTag(XMLNode& n)
{
    root.addChild(n);
}

void XMLParser::rootTagRecieved(XMLNode& _root)
{
    root=_root;
}

void XMLParser::parserError(const string& errorMsg, int line)
{
    stringstream error;
    error << errorMsg << ": " << line;
    reset();

    LOG_THROW(error.str(), ParseFailure);
    
}

} 

---