# SPDX-FileCopyrightText: ☭ Emery Hemingway
# SPDX-License-Identifier: Unlicense

import std/[os, strutils]
import preserves, preserves/sugar, syndicate
import ./schema/[assertions, config]

{.passC: staticExec("pkg-config --cflags libxslt").}
{.passL: staticExec("pkg-config --libs libxslt").}

{.pragma: libxslt, header: "libxslt/xslt.h", importc.}

type
  xmlElementType {.libxslt.} = enum
    XML_ELEMENT_NODE = 1,
    XML_ATTRIBUTE_NODE = 2,
    XML_TEXT_NODE = 3,
    XML_CDATA_SECTION_NODE = 4,
    XML_ENTITY_REF_NODE = 5,
    XML_ENTITY_NODE = 6,
    XML_PI_NODE = 7,
    XML_COMMENT_NODE = 8,
    XML_DOCUMENT_NODE = 9,
    XML_DOCUMENT_TYPE_NODE = 10,
    XML_DOCUMENT_FRAG_NODE = 11,
    XML_NOTATION_NODE = 12,
    XML_HTML_DOCUMENT_NODE = 13,
    XML_DTD_NODE = 14,
    XML_ELEMENT_DECL = 15,
    XML_ATTRIBUTE_DECL = 16,
    XML_ENTITY_DECL = 17,
    XML_NAMESPACE_DECL = 18,
    XML_XINCLUDE_START = 19,
    XML_XINCLUDE_END = 20

  xmlNsPtr = ptr xmlNs
  xmlNs {.libxslt.} = object
    next: xmlNsPtr
    href, prefix: cstring

  xmlAttrPtr = ptr xmlAttr
  xmlAttr {.libxslt.} = object
    name: cstring
    next: xmlAttrPtr
    children: xmlNodePtr

  xmlElementContentPtr = ptr xmlElementContent
  xmlElementContent {.libxslt.} = object
    encoding: cstring

  xmlNodePtr = ptr xmlNode
  xmlNode {.libxslt.} = object
    `type`: xmlElementType
    name: cstring
    children, next: xmlNodePtr
    content: cstring
    properties: xmlAttrPtr
    nsDef: xmlNsPtr

  xmlDocPtr {.libxslt.} = distinct pointer
  xsltStylesheetPtr {.libxslt.} = distinct pointer

proc isNil(x: xmlDocPtr): bool {.borrow.}
proc isNil(x: xsltStylesheetPtr): bool {.borrow.}

proc xmlReadMemory(buf: pointer; len: cint; url, enc: cstring; opts: cint): xmlDocPtr {.libxslt.}

proc xmlReadMemory(buf: string; uri = "noname.xml"): xmlDocPtr =
  xmlReadMemory(buf[0].addr, buf.len.cint, uri, "UTF-8", 0)

proc xmlParseFile(filename: cstring): xmlDocPtr {.libxslt.}

proc xmlFreeDoc(p: xmlDocPtr) {.libxslt.}

proc xmlDocGetRootElement(doc: xmlDocPtr): xmlNodePtr {.libxslt.}

proc loadXmlDoc(text: string): xmlDocPtr =
  if text.startsWith("/") and fileExists(text):
    xmlParseFile(text)
  else:
    xmlReadMemory(text, "noname.xml")

proc xsltParseStylesheetFile(filename: cstring): xsltStylesheetPtr {.libxslt.}

proc xsltParseStylesheetDoc(doc: xmlDocPtr): xsltStylesheetPtr {.libxslt.}

proc xsltParseStylesheetDoc(text: string; uri = "noname.xml"): xsltStylesheetPtr =
  var doc = xmlReadMemory(text, uri)
  result = xsltParseStylesheetDoc(doc)
    # implicit free of doc

proc loadStylesheet(text: string): xsltStylesheetPtr =
  if text.startsWith("/") and fileExists(text):
    xsltParseStylesheetFile(text)
  else:
    xsltParseStylesheetDoc(text, "noname.xsl")

proc xsltApplyStylesheet(
    style: xsltStylesheetPtr, doc: xmlDocPtr,  params: cstringArray): xmlDocPtr {.libxslt.}

proc xsltFreeStylesheet(style: xsltStylesheetPtr) {.libxslt.}

proc xsltSaveResultToString(txt: ptr pointer; len: ptr cint; res: xmlDocPtr; style: xsltStylesheetPtr): cint {.libxslt.}

proc c_free*(p: pointer) {.importc: "free", header: "<stdlib.h>".}

proc xsltSaveResultToString(res: xmlDocPtr; style: xsltStylesheetPtr): string =
  var
    txt: pointer
    len: cint
  if xsltSaveResultToString(addr txt, addr len, res, style) < 0:
    raise newException(CatchableError, "xsltSaveResultToString failed")
  if len > 0:
    result = newString(int len)
    copyMem(result[0].addr, txt, len)
    c_free(txt)

proc initLibXml =
  discard

proc XML_GET_CONTENT(xn: xmlNodePtr): xmlElementContentPtr {.libxslt.}

proc textContent(xn: xmlNodePtr): string =
  if xn.content != nil: result = $xn.content

proc content(attr: xmlAttrPtr): string =
  var child = attr.children
  while not child.isNil:
    result.add child.content
    child = child.next

proc preserveSiblings(result: var seq[Value]; first: xmlNodePtr) =
  var xn = first
  while not xn.isNil:
    case xn.type
    of XML_ELEMENT_NODE:
      var child = Value(kind: pkRecord)
      if not xn.nsDef.isNil:
        child.record.add initDictionary()
        var ns = xn.nsDef
        while not ns.isNil:
          if not ns.href.isNil:
            var key = Value(kind: pkString)
            if ns.prefix.isNil:
              key.string = "xmlns"
            else:
              key.string = "xmlns:" & $ns.prefix
            child.record[0][key] = toPreserves($ns.href)
          ns = ns.next

      if not xn.properties.isNil:
        if child.record.len < 1:
          child.record.add initDictionary()
        var attr = xn.properties
        while not attr.isNil:
          var
            key = toPreserves($attr.name)
            val = toPreserves(attr.content)
          child.record[0][key] = val
          attr = attr.next
      if not xn.children.isNil:
        preserveSiblings(child.record, xn.children)
      child.record.add tosymbol($xn.name)
      result.add child
    of XML_TEXT_NODE:
      result.add textContent(xn).toPreserves
    else:
      stderr.writeLine "not an XML_ELEMENT_NODE - ", $xn.type
    xn = xn.next

proc toPreservesHook*(xn: xmlNodePtr): Value =
  var items = newSeqofCap[Value](1)
  preserveSiblings(items, xn)
  items[0]

proc spawnXsltActor*(turn: Turn; root: Cap): Actor {.discardable.} =
  spawnActor(turn, "xslt") do (turn: Turn):
    initLibXml()
    during(turn, root, ?:XsltArguments) do (ds: Cap):
      let sheetsPat = observePattern(!XsltTransform, {@[%0]: grab(), @[%1]: grab()})
      during(turn, ds, sheetsPat) do (stylesheet: Literal[string], input: Literal[string]):
        let cur = loadStylesheet(stylesheet.value)
        if cur.isNil:
          stderr.writeLine "failed to parse stylesheet"
        else:
          let doc = loadXmlDoc(input.value)
          if doc.isNil:
            stderr.writeLine "failed to parse input document"
          else:
            let
              params = allocCStringArray([])
              res = xsltApplyStylesheet(cur, doc, params)
            if res.isNil:
              stderr.writeLine "failed to apply stylesheet transformation"
            else:
              let output = xsltSaveResultToString(res, cur)
              deallocCStringArray(params)
              publish(turn, ds, XsltTransform(
                  stylesheet: stylesheet.value,
                  input: input.value,
                  output: xmlDocGetRootElement(res).toPreservesHook,
                ))
              xmlFreeDoc(res)
            xmlFreeDoc(doc)
          xsltFreeStylesheet(cur)

when isMainModule:
  import syndicate/relays
  runActor("main") do (turn: Turn):
    resolveEnvironment(turn) do (turn: Turn; ds: Cap):
      spawnXsltActor(turn, ds)