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

XmlAssociateDtd::XmlAssociateDtd(
  const std::string& path,
  const std::string& publicIdentifier,
  size_t size) :
  d(new DtdData())
{
  d->buffer.reserve(size);
  d->path = path;
  d->publicIdentifier = publicIdentifier;
  d->rootElementSeen = false;
  d->insideDtd = false;
  XML_SetUserData(p, d.get());
  XML_SetElementHandler(p, start, end);
  XML_SetDoctypeDeclHandler(p, startdoctypehandler, enddoctypehandler);
  XML_SetDefaultHandlerExpand(p, defaulthandler);
}

XmlAssociateDtd::~XmlAssociateDtd()
{
}

void XMLCALL XmlAssociateDtd::defaulthandler(
  void *data,
  const XML_Char *s,
  int len)
{
  DtdData *d;
  d = (DtdData *)data;
  if (!d->insideDtd)
    d->buffer.append(s, len);
}

void XMLCALL XmlAssociateDtd::start(void *data,
  const XML_Char *el,
  const XML_Char **attr)
{
  DtdData *d;
  d = (DtdData *)data;
  
  if (!d->rootElementSeen)
  {
    if (d->publicIdentifier.empty())
    {
      d->buffer += "<!DOCTYPE ";
      d->buffer += el;
      d->buffer += " SYSTEM \"";
      d->buffer += d->path;
      d->buffer += "\">\n";
    }
    else
    {
      d->buffer += "<!DOCTYPE ";
      d->buffer += el;
      d->buffer += " PUBLIC \"";
      d->buffer += d->publicIdentifier;
      d->buffer += "\" \"";
      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 XmlAssociateDtd::end(void *data, const XML_Char *el)
{
  DtdData *d;
  d = (DtdData *)data;
  d->buffer += "</";
  d->buffer += el;
  d->buffer += ">";
}

void XMLCALL XmlAssociateDtd::startdoctypehandler(
  void *data,
  const XML_Char *doctypeName,
  const XML_Char *sysid,
  const XML_Char *pubid,
  int has_internal_subset)
{
  DtdData *d;
  d = (DtdData *)data;
  d->insideDtd = true;    
}

void XMLCALL XmlAssociateDtd::enddoctypehandler(void *data)
{
  DtdData *d;
  d = (DtdData *)data;
  d->insideDtd = false;
}