/* xml++.cc
 * libxml++ and this file are copyright (C) 2000 by Ari Johnson, and
 * are covered by the GNU Lesser General Public License, which should be
 * included with libxml++ as the file COPYING.
 */

#include "xml++.hpp"

static XMLNode *readnode(xmlNodePtr);
static void writenode(xmlDocPtr, XMLNode *, xmlNodePtr, int);

XMLTree::XMLTree(const XMLTree * from)
{
    _filename = from->filename();
    _root = new XMLNode(*from->root());
    _compression = from->compression();
    _initialized = true;
}

XMLTree::~XMLTree()
{
    if (_initialized && _root)
		delete _root;
}

int XMLTree::set_compression(int c)
{
    if (c > 9)
		c = 9;

    if (c < 0)
		c = 0;

    _compression = c;

    return _compression;
}

bool XMLTree::read(void)
{
    xmlDocPtr doc;

    if (_root) {
		delete _root;
		_root = NULL;
    }

    xmlKeepBlanksDefault(0);

    doc = xmlParseFile(_filename.c_str());
    if (!doc) {
		_initialized = false;
		return false;
    }

    _root = readnode(xmlDocGetRootElement(doc));
    xmlFreeDoc(doc);
    _initialized = true;

    return true;
}

bool XMLTree::read_buffer(const string & buffer)
{
    xmlDocPtr doc;

    _filename = "";

    if (_root) {
		delete _root;
		_root = NULL;
    }

    doc = xmlParseMemory((char *) buffer.c_str(), buffer.length());
    if (!doc) {
		_initialized = false;
		return false;
    }

    _root = readnode(xmlDocGetRootElement(doc));
    xmlFreeDoc(doc);
    _initialized = true;

    return true;
}

bool XMLTree::write(void) const
{
    xmlDocPtr doc;
    XMLNodeList children;
    int result;

    xmlKeepBlanksDefault(0);
    doc = xmlNewDoc((xmlChar *) "1.0");
    xmlSetDocCompressMode(doc, _compression);
    writenode(doc, _root, doc->children, 1);
    result = xmlSaveFormatFile(_filename.c_str(), doc, 1);
    xmlFreeDoc(doc);

    if (result == -1)
		return false;

    return true;
}

const string & XMLTree::write_buffer(void) const
{
    static string retval;
    char *ptr;
    int len;
    xmlDocPtr doc;
    XMLNodeList children;

    xmlKeepBlanksDefault(0);
    doc = xmlNewDoc((xmlChar *) "1.0");
    xmlSetDocCompressMode(doc, _compression);
    writenode(doc, _root, doc->children, 1);
    xmlDocDumpMemory(doc, (xmlChar **) & ptr, &len);
    xmlFreeDoc(doc);

    retval = ptr;

    free(ptr);

    return retval;
}

XMLNode::XMLNode(const string & n)
	:  _name(n), _is_content(false), _content(string())
{

    if (_name.empty())
		_initialized = false;
    else
		_initialized = true;
}

XMLNode::XMLNode(const string & n, const string & c)
	:_name(string()), _is_content(true), _content(c)
{
    _initialized = true;
}

XMLNode::XMLNode(const XMLNode& from)
	: _initialized(false)
{
    XMLPropertyList props;
    XMLPropertyIterator curprop;
    XMLNodeList nodes;
    XMLNodeIterator curnode;

    _name = from.name();
    set_content(from.content());

    props = from.properties();
    for (curprop = props.begin(); curprop != props.end(); curprop++)
		add_property((*curprop)->name(), (*curprop)->value());

    nodes = from.children();
    for (curnode = nodes.begin(); curnode != nodes.end(); curnode++)
		add_child_copy(**curnode);
}

XMLNode::~XMLNode()
{
    XMLNodeIterator curchild;
    XMLPropertyIterator curprop;

    for (curchild = _children.begin(); curchild != _children.end();
			curchild++)
		delete *curchild;

    for (curprop = _proplist.begin(); curprop != _proplist.end();
			curprop++)
		delete *curprop;
}

const string & XMLNode::set_content(const string & c)
{
    if (c.empty())
		_is_content = false;
    else
		_is_content = true;

    _content = c;

    return _content;
}

const XMLNodeList & XMLNode::children(const string & n) const
{
    static XMLNodeList retval;
    XMLNodeConstIterator cur;

    if (n.length() == 0)
		return _children;

    retval.erase(retval.begin(), retval.end());

    for (cur = _children.begin(); cur != _children.end(); cur++)
		if ((*cur)->name() == n)
		    retval.insert(retval.end(), *cur);

    return retval;
}

XMLNode *XMLNode::add_child(const string & n)
{
	return add_child_copy(XMLNode (n));
}

void
XMLNode::add_child_nocopy (XMLNode& n)
{
	_children.insert(_children.end(), &n);
}

XMLNode *
XMLNode::add_child_copy(const XMLNode& n)
{
	XMLNode *copy = new XMLNode (n);
	_children.insert(_children.end(), copy);
	return copy;
}

XMLNode *XMLNode::add_content(const string & c)
{
    return add_child_copy(XMLNode (string(), c));
}

XMLProperty *XMLNode::property(const string & n)
{
    if (_propmap.find(n) == _propmap.end())
		return NULL;
    return _propmap[n];
}

XMLProperty *XMLNode::add_property(const string & n, const string & v)
{
	if(_propmap.find(n) != _propmap.end()){
		remove_property(n);
	}

    XMLProperty *tmp = new XMLProperty(n, v);

    if (!tmp)
		return NULL;

    _propmap[tmp->name()] = tmp;
    _proplist.insert(_proplist.end(), tmp);

    return tmp;
}

void XMLNode::remove_property(const string & n)
{
    if (_propmap.find(n) != _propmap.end()) {
		_proplist.remove(_propmap[n]);
		_propmap.erase(n);
    }
}

void XMLNode::remove_nodes(const string & n)
{
    XMLNodeIterator i = _children.begin();
    XMLNodeIterator tmp;

    while (i != _children.end()) {
	    tmp = i;
	    ++tmp;
	    if ((*i)->name() == n) {
		    _children.remove(*i);
	    }
	    i = tmp;
    }
}

XMLNode *
XMLNode::find_named_node (const string & name)
{
        XMLNodeList nlist;
        XMLNodeConstIterator niter;
        XMLNode* child;
                                                                                                      
        nlist = this->children();
                                                                                                      
        for (niter = nlist.begin(); niter != nlist.end(); ++niter) {
                                                                                                      
                child = *niter;
                                                                                                      
                if (child->name() == name) {
                        return child;
                }
        }
                                                                                                      
        return 0;
}


static XMLNode *readnode(xmlNodePtr node)
{
    string name, content;
    xmlNodePtr child;
    XMLNode *tmp;
    xmlAttrPtr attr;

    if (node->name)
		name = (char *) node->name;
    else
		name = string();

    tmp = new XMLNode(name);

    for (attr = node->properties; attr; attr = attr->next) {
		name = (char *) attr->name;
		content = "";
		if (attr->children)
		    content = (char *) attr->children->content;
		tmp->add_property(name, content);
    }

    if (node->content)
		tmp->set_content((char *) node->content);
    else
		tmp->set_content(string());

    for (child = node->children; child; child = child->next)
	    tmp->add_child_nocopy (*readnode(child));

    return tmp;
}

static void writenode(xmlDocPtr doc, XMLNode * n, xmlNodePtr p, int root =
		      0)
{
    XMLPropertyList props;
    XMLPropertyIterator curprop;
    XMLNodeList children;
    XMLNodeIterator curchild;
    xmlNodePtr node;

    if (root)
		node = doc->children =
			    xmlNewDocNode(doc, NULL, (xmlChar *) n->name().c_str(), NULL);

    else
		node = xmlNewChild(p, NULL, (xmlChar *) n->name().c_str(), NULL);

    if (n->is_content()) {
		node->type = XML_TEXT_NODE;
		xmlNodeSetContentLen(node, (const xmlChar *) n->content().c_str(),
			     n->content().length());
    }

    props = n->properties();
    for (curprop = props.begin(); curprop != props.end(); curprop++)
		xmlSetProp(node, (xmlChar *) (*curprop)->name().c_str(),
			   (xmlChar *) (*curprop)->value().c_str());

    children = n->children();
    for (curchild = children.begin(); curchild != children.end();
			 curchild++)
		writenode(doc, *curchild, node);
}