#include "wraplibxml.h"
#include <sstream>
#include <stdexcept>
#include <wx/wx.h>

WrapLibxml::WrapLibxml(bool netAccessParameter, const std::string& catalogPathParameter)
  : netAccess(netAccessParameter), catalogPath(catalogPathParameter)
{
  LIBXML_TEST_VERSION
  xmlInitializeCatalog();
}

WrapLibxml::~WrapLibxml()
{
  xsltCleanupGlobals();
  xmlCleanupParser();
  xmlCatalogCleanup();
}

bool WrapLibxml::validate(const std::string& fileName)
{
  output = "";

  xmlCleanupParser();
  xmlParserCtxtPtr ctxt;
  xmlDocPtr docPtr;

  ctxt = xmlNewParserCtxt();
  if (ctxt == NULL)
  {
    return false;
  }

  bool returnValue = false;

  loadCatalog();
  docPtr = xmlCtxtReadFile(
    ctxt,
    fileName.c_str(),
    NULL,
    (netAccess) ? XML_PARSE_DTDVALID : XML_PARSE_DTDVALID | XML_PARSE_NONET);
  if (docPtr == NULL)
    ;
  else if (ctxt->valid == 0)
    ;
  else
    returnValue = true;

  xmlFreeDoc(docPtr);
  xmlFreeParserCtxt(ctxt);

  return returnValue;
}

bool WrapLibxml::validateRelaxNG(
  const std::string& schemaFileName,
  const std::string& fileName)
{
  output = "";

  xmlCleanupParser();

  xmlRelaxNGValidCtxtPtr ctxtPtr;
  xmlDocPtr docPtr;
  xmlRelaxNGParserCtxtPtr schemaParserCtxtPtr;
  xmlRelaxNGPtr schemaPtr;

  schemaParserCtxtPtr = xmlRelaxNGNewParserCtxt(schemaFileName.c_str());
  if (schemaParserCtxtPtr == NULL)
    return false;

  schemaPtr = xmlRelaxNGParse(schemaParserCtxtPtr);
  if (schemaPtr == NULL)
  {
    xmlRelaxNGFreeParserCtxt(schemaParserCtxtPtr);
    return false;
  }

  ctxtPtr = xmlRelaxNGNewValidCtxt(schemaPtr);
  if (ctxtPtr == NULL)
  {
    xmlRelaxNGFree(schemaPtr);
    xmlRelaxNGFreeParserCtxt(schemaParserCtxtPtr);
    return false;
  }

  docPtr = xmlReadFile(fileName.c_str(), NULL,
    (netAccess) ? XML_PARSE_DTDLOAD : XML_PARSE_DTDLOAD | XML_PARSE_NONET);
  if (docPtr == NULL)
  {
    xmlRelaxNGFree(schemaPtr);
    xmlRelaxNGFreeValidCtxt(ctxtPtr);
    return false;
  }

  int res = xmlRelaxNGValidateDoc(ctxtPtr, docPtr);

  bool returnValue = (res) ? false : true;

  xmlFreeDoc(docPtr);
  xmlRelaxNGFree(schemaPtr);
  xmlRelaxNGFreeValidCtxt(ctxtPtr);
  return returnValue;
}

bool WrapLibxml::validateW3CSchema(
  const std::string& schemaFileName,
  const std::string& fileName)
{
  output = "";

  xmlCleanupParser();
  xmlSchemaValidCtxtPtr ctxtPtr;
  xmlDocPtr docPtr;
  xmlSchemaParserCtxtPtr schemaParserCtxtPtr;
  xmlSchemaPtr schemaPtr;

  schemaParserCtxtPtr = xmlSchemaNewParserCtxt(schemaFileName.c_str());
  if (schemaParserCtxtPtr == NULL)
    return false;

  schemaPtr = xmlSchemaParse(schemaParserCtxtPtr);
  if (schemaPtr == NULL)
  {
    xmlSchemaFreeParserCtxt(schemaParserCtxtPtr);
    return false;
  }

  ctxtPtr = xmlSchemaNewValidCtxt(schemaPtr);
  if (ctxtPtr == NULL)
  {
    xmlSchemaFree(schemaPtr);
    xmlSchemaFreeParserCtxt(schemaParserCtxtPtr);
    return false;
  }

  docPtr = xmlReadFile(fileName.c_str(), NULL,
    (netAccess) ? XML_PARSE_DTDLOAD : XML_PARSE_DTDLOAD | XML_PARSE_NONET);
  if (docPtr == NULL)
  {
    xmlSchemaFree(schemaPtr);
    xmlSchemaFreeValidCtxt(ctxtPtr);
    return false;
  }

  int res = xmlSchemaValidateDoc(ctxtPtr, docPtr);

  bool returnValue = (res) ? false : true;

  xmlFreeDoc(docPtr);
  xmlSchemaFree(schemaPtr);
  xmlSchemaFreeValidCtxt(ctxtPtr);
  return returnValue;
}

bool WrapLibxml::parse(
  const std::string& fileName,
  bool indent,
  bool resolveEntities)
{
  output = "";

  xmlCleanupParser();
  xmlParserCtxtPtr ctxt;
  xmlDocPtr docPtr;

  ctxt = xmlNewParserCtxt();
  if (ctxt == NULL)
  {
    return false;
  }

  loadCatalog();

  int flags = XML_PARSE_DTDLOAD;
  if (resolveEntities)
    flags |= XML_PARSE_NOENT;
  if (!netAccess)
    flags |= XML_PARSE_NONET;

  docPtr = xmlCtxtReadFile(
    ctxt,
    fileName.c_str(),
    NULL,
    flags);
  
  if (docPtr == NULL)
  {
    xmlFreeParserCtxt(ctxt);
    return false;
  }

  if (indent)
    xmlKeepBlanksDefault(0);
  
  xmlChar *buf = NULL;
  int size;

  // tbd: link output encoding to input encoding?
  xmlDocDumpFormatMemoryEnc(
    docPtr, 
    &buf, 
    &size, 
    "UTF-8", 
    indent);

  if (buf)
  {
    output.append((const char *)buf);
    free(buf);
  }

  bool returnValue = (!ctxt->errNo) ? true : false;

  xmlFreeDoc(docPtr);
  xmlFreeParserCtxt(ctxt);

  return returnValue;
}

bool WrapLibxml::xpath(const std::string& path, const std::string& fileName)
{
  output = "";

  xmlCleanupParser();

  xmlParserCtxtPtr ctxt;
  xmlDocPtr docPtr;

  ctxt = xmlNewParserCtxt();
  if (ctxt == NULL)
  {
    return false;
  }

  loadCatalog();
  docPtr = xmlCtxtReadFile(
    ctxt,
    fileName.c_str(),
    NULL,
    //(netAccess) ? XML_PARSE_DTDLOAD | XML_PARSE_NOENT : XML_PARSE_DTDLOAD | XML_PARSE_NONET | XML_PARSE_NOENT
    XML_PARSE_NOENT | XML_PARSE_NONET | XML_PARSE_NSCLEAN
  );
  
  if (docPtr == NULL)
  {
    xmlFreeParserCtxt(ctxt);
    return false;
  }

  xmlXPathContextPtr context;
  xmlXPathObjectPtr result;
	xmlNodeSetPtr nodeset;

	context = xmlXPathNewContext(docPtr);
	if (!context)
	{
    xmlFreeDoc(docPtr);
    xmlFreeParserCtxt(ctxt);
    return false;
  }
  
  // enable namespace prefixes
  xmlXPathRegisterNs(context, (xmlChar *)"xhtml", (xmlChar *)"http://www.w3.org/1999/xhtml");
  // add others as necessary!
  
	result = xmlXPathEvalExpression((const xmlChar *)path.c_str(), context);
	
  xmlKeepBlanksDefault(0);

	bool xpathIsValid = (result) ? true : false;
  
  while (result != NULL)
  {
	  if(xmlXPathNodeSetIsEmpty(result->nodesetval))
		  break;
		nodeset = result->nodesetval;
		for (int i = 0; i < nodeset->nodeNr; i++)
    {
      xmlNodePtr node = nodeset->nodeTab[i];
      xmlBufferPtr bufferPtr = xmlBufferCreate();
      if (!node || !bufferPtr)
        break;
      xmlNodeDump(bufferPtr, NULL, node, 0, 1);
      std::string nodeBuffer = (const char *)xmlBufferContent(bufferPtr);

      output += nodeBuffer;
      output += '\n';
    }
    break;
	}
  if (result)
    xmlXPathFreeObject(result);
	xmlXPathFreeContext(context);
  xmlFreeDoc(docPtr);
  xmlFreeParserCtxt(ctxt);

  xmlCleanupParser();

  return xpathIsValid;
}

bool WrapLibxml::xslt(
  const std::string& styleFileName,
  const std::string& fileName
)
{
  xmlCleanupParser();
  output = "";

  xsltStylesheetPtr cur;
	xmlDocPtr doc, res;
  xmlSubstituteEntitiesDefault(1);
	xmlLoadExtDtdDefaultValue = 1;
	cur = xsltParseStylesheetFile((const xmlChar *)styleFileName.c_str());
	if (!cur)
  {
    return false;
  }

  loadCatalog();
	doc = xmlParseFile(fileName.c_str());
	if (!doc)
	{
    xsltFreeStylesheet(cur);
    return false;
  }

  // ensure entity warnings are treated as errors
  if (!getLastError().empty())
  {
    xmlFreeDoc(doc);
    xsltFreeStylesheet(cur);
    return false;
  }

	res = xsltApplyStylesheet(cur, doc, NULL);
	if (!res)
	{
    xmlFreeDoc(doc);
    xsltFreeStylesheet(cur);
    return false;
  }
	
  xmlChar *buf = NULL;
  int size;
  xmlDocDumpFormatMemoryEnc(res, &buf, &size, "UTF-8", 1);
  if (buf)
  {
    output.append((char *)buf, size);
    free(buf); // as above?
  }

	xsltFreeStylesheet(cur);
	xmlFreeDoc(doc);
	xmlFreeDoc(res);
	
  return true;
}

bool WrapLibxml::bufferWellFormed(const std::string& buffer)
{
  xmlCleanupParser();

  xmlParserCtxtPtr ctxt = xmlNewParserCtxt();
  if (!ctxt)
    return false;

  loadCatalog();
	xmlDocPtr docPtr = xmlCtxtReadMemory(
    ctxt,
    buffer.c_str(),
    buffer.size(),
    "",
    "UTF-8",
    0);
  bool returnValue = (docPtr) ? true : false;
  
  xmlFreeDoc(docPtr);
  xmlFreeParserCtxt(ctxt);

  return returnValue;
}

int WrapLibxml::saveEncoding(
  const std::string& buffer,
  const std::string& fileName,
  const std::string& encoding)
{
  xmlCleanupParser();

  xmlParserCtxtPtr ctxt = xmlNewParserCtxt();
  if (!ctxt)
    return -1;

  xmlSubstituteEntitiesDefault(0);

  loadCatalog();

  xmlKeepBlanksDefault(1); // prevents single-line output

  xmlDocPtr docPtr = xmlCtxtReadMemory(
    ctxt,
    buffer.c_str(),
    buffer.size(),
    "",
    NULL,
    XML_PARSE_DTDLOAD | XML_PARSE_DTDVALID | XML_PARSE_PEDANTIC//XML_PARSE_NONET//XML_PARSE_DTDLOAD//0//(netAccess) ? XML_PARSE_DTDLOAD | XML_PARSE_NOENT : XML_PARSE_DTDLOAD | XML_PARSE_NONET | XML_PARSE_NOENT//0
    );
  if (!docPtr)
  {
    xmlFreeParserCtxt(ctxt);
    return -1;
  }

  int result = xmlSaveFileEnc(
    fileName.c_str(),
    docPtr,
    encoding.c_str());
  xmlFreeDoc(docPtr);
  xmlFreeParserCtxt(ctxt);
  return result;
}

int WrapLibxml::saveEncodingFromFile(
  const std::string& fileNameSource,
  const std::string& fileNameDestination,
  const std::string& encoding)
{
  xmlCleanupParser();

  xmlParserCtxtPtr ctxt = xmlNewParserCtxt();
  if (!ctxt)
    return -1;

  xmlSubstituteEntitiesDefault(0);

  loadCatalog();

  xmlKeepBlanksDefault(1); // prevents single-line output

  xmlDocPtr docPtr = xmlCtxtReadFile(
    ctxt,
    fileNameSource.c_str(),
    "UTF-8",
    (netAccess) ? XML_PARSE_DTDLOAD : XML_PARSE_DTDLOAD | XML_PARSE_NONET);//XML_PARSE_NONET//XML_PARSE_DTDLOAD//0//(netAccess) ? XML_PARSE_DTDLOAD | XML_PARSE_NOENT : XML_PARSE_DTDLOAD | XML_PARSE_NONET | XML_PARSE_NOENT//0
  if (!docPtr)
  {
    xmlFreeParserCtxt(ctxt);
    return -1;
  }

  int result = xmlSaveFileEnc(
    fileNameDestination.c_str(),
    docPtr,
    encoding.c_str());
  
  // ensure entity warnings are treated as errors!
  if (!getLastError().empty())
    return -1;
  
  xmlFreeDoc(docPtr);
  xmlFreeParserCtxt(ctxt);
  return result;
}

std::string WrapLibxml::getLastError()
{
  xmlErrorPtr err = xmlGetLastError();
  
  if (!err)
    return "";

  std::stringstream ss;
  ss << "Error at line ";
  ss << err->line;
  if (err->int2)
  {
    ss << ", column ";
    ss << err->int2;
  }
  ss << ": ";
	ss << err->message;  
	return ss.str();
}

std::pair<int, int> WrapLibxml::getErrorPosition()
{
  xmlErrorPtr err = xmlGetLastError();
  if (!err)
    return std::make_pair(1, 1);

  return std::make_pair(
    err->line,
    err->int2);
}

std::string WrapLibxml::getOutput() { return output; }

void WrapLibxml::loadCatalog()
{ 
#ifdef __WXMSW__
  xmlLoadCatalog(catalogPath.c_str());
#endif
}

std::string WrapLibxml::lookupPublicId(const std::string& id)
{
  loadCatalog();
  std::string ret;
  char *s = (char *)xmlCatalogResolvePublic((const xmlChar *)id.c_str());
  if (!s)
    return ret;
  ret = s;
  free(s);
  return ret;
}