#include <string>
#include <vector>
#include <stdexcept>
#include <expat.h>
#include "xmlassociatexsl.h"

XmlAssociateXsl::XmlAssociateXsl(const std::string& path, size_t size) :
  d(new XslData())
{
  d->buffer.reserve(size);
  d->path = path;
  d->rootElementSeen = false;
  XML_SetUserData(p, d.get());
  XML_SetElementHandler(p, start, end);
  XML_SetProcessingInstructionHandler(p, processinghandler);
  XML_SetDefaultHandlerExpand(p, defaulthandler);
}

XmlAssociateXsl::~XmlAssociateXsl()
{
}

void XMLCALL XmlAssociateXsl::defaulthandler(
  void *data,
  const XML_Char *s,
  int len)
{
  XslData *d;
  d = (XslData *)data;
  d->buffer.append(s, len);
}

void XMLCALL XmlAssociateXsl::start(void *data,
  const XML_Char *el,
  const XML_Char **attr)
{
  XslData *d;
  d = (XslData *)data;
  
  if (!d->rootElementSeen)
  {
    d->buffer += "<?xml-stylesheet type=\"text/xsl\" href=\"";
    d->buffer += d->path;
    d->buffer += "\"?>\n";
    d->rootElementSeen = true;
  }

  d->buffer += "<";
  d->buffer += el;

  while (*attr)
  {
    d->buffer += " ";
    d->buffer += *attr;
    d->buffer += "=\"";
    d->buffer += xmliseAttribute(*(attr + 1));
    d->buffer += "\"";
    attr += 2;
  }
  d->buffer += ">";
}

void XMLCALL XmlAssociateXsl::end(void *data, const XML_Char *el)
{
  XslData *d;
  d = (XslData *)data;
  d->buffer += "</";
  d->buffer += el;
  d->buffer += ">";
}

void XMLCALL XmlAssociateXsl::processinghandler(
  void *data,
  const XML_Char *target,
  const XML_Char *datastring)
{
  XslData *d;
  d = (XslData *)data;

  if (!strcmp(target, "xml-stylesheet"))
    return;
  
  d->buffer += "<?";
  d->buffer += target;
  d->buffer += " ";
  d->buffer += datastring;
  d->buffer += "?>";
}