/**
 * @file view_circuit.cpp
 * @brief SQL function @c provsql.view_circuit() – GraphViz DOT circuit visualisation.
 *
 * Implements @c provsql.view_circuit(), which serialises the provenance
 * circuit rooted at a given UUID token to a GraphViz DOT string using the
 * @c DotCircuit class.  The returned string can be rendered to an image
 * with @c dot(1) or @c graph-easy(1).
 *
 * An SPI query retrieves the optional provenance-mapping labels (human-
 * readable identifiers for input gates) and attaches them to the DOT nodes
 * before calling @c DotCircuit::render().
 */
extern "C"
{
#include "postgres.h"
#include "fmgr.h"
#include "catalog/pg_type.h"
#include "utils/uuid.h"
#include "executor/spi.h"
#include "provsql_shmem.h"
#include "provsql_utils.h"

PG_FUNCTION_INFO_V1(view_circuit);
}

#include "DotCircuit.h"
#include "tool_registry_sync.h"
#include <csignal>
#include <utility>
#include <sstream>
#include <algorithm>

using namespace std;

/**
 * @brief Parse a PostgreSQL text representation of an array of integer pairs.
 * @param s  String in the form @c {{a,b},{c,d},...}.
 * @return   Vector of (first, second) integer pairs.
 */
static vector<pair<int, int> > parse_array(string s)
{
  s = s.substr(1, s.size() - 2); // Remove initial '{' and final '}'

  vector<pair<int, int> > result;

  istringstream iss(s);
  string p;

  while(getline(iss, p, '}'))
  {
    if(!p.empty() && p[0]==',')
      p=p.substr(1);

    int k = p.find(",", 1);
    string s1 = p.substr(1, k - 1);
    int i1;
    if (s1 == "NULL" || s1 == "")
      i1 = 0;
    else
      i1 = stoi(p.substr(1, k - 1));
    int i2 = stoi(p.substr(k + 1, p.size() - k));
    result.push_back(make_pair(i1, i2));
  }

  return result;
}

/**
 * @brief Build a GraphViz DOT representation of the provenance circuit at @p token.
 * @param token      Datum containing the root provenance gate UUID.
 * @param token2prob Datum with the OID of the probability mapping relation.
 * @param is_debug   Datum boolean; if true, include debug information.
 * @return           DOT string for the circuit.
 */
static std::string view_circuit_internal(Datum token, Datum token2prob, Datum is_debug)
{
  Datum arguments[2]={token,token2prob};
  constants_t constants=get_constants(true);
  Oid argtypes[2]={constants.OID_TYPE_UUID,REGCLASSOID};
  char nulls[2] = {' ',' '};

  SPI_connect();

  DotCircuit c;

  int proc = 0;

  if (SPI_execute_with_args(
        "SELECT * FROM provsql.sub_circuit_with_desc($1,$2)", 2, argtypes, arguments, nulls, true, 0) == SPI_OK_SELECT)
  {
    proc = SPI_processed;
    TupleDesc tupdesc = SPI_tuptable->tupdesc;
    SPITupleTable *tuptable = SPI_tuptable;

    for (int i = 0; i < proc; i++)
    {
      HeapTuple tuple = tuptable->vals[i];

      string f = SPI_getvalue(tuple, tupdesc, 1);
      string type = SPI_getvalue(tuple, tupdesc, 3);
      if (type == "input")
      {
        c.setGate(f, DotGate::IN, SPI_getvalue(tuple, tupdesc, 4));

      } else {
        auto id=c.getGate(f);

        if (type == "times" || type == "assumed_boolean" || type == "annotation")
        {
          /* assumed_boolean and annotation are single-child structural
           * wrappers.  A single-wire TIMES renders as identity in the DOT
           * output, which is what we want here; rendering a distinct node
           * would clutter the graph for no semantic gain.  PROV-XML export
           * keeps the wrapper explicit. */
          c.setGate(f, DotGate::OTIMES);
        }
        else if (type == "plus")
        {
          c.setGate(f, DotGate::OPLUS);
        }
        else if (type == "monus")
        {
          c.setGate(f, DotGate::OMINUS);
        }
        else if (type == "monusr")
        {
          c.setGate(f, DotGate::OMINUSR);
        }
        else if (type == "monusl")
        {
          c.setGate(f, DotGate::OMINUSL);
        }
        else if (type == "delta")
        {
          c.setGate(f, DotGate::DELTA);
        }
        else if (type == "eq")
        {
          vector<pair<int, int> > v = parse_array(std::string("{")+SPI_getvalue(tuple, tupdesc, 5)+"}");
          if (v.size() != 1)
            provsql_error("Incorrect extra information on eq gate");
          std::string cond = std::to_string(v[0].first) + std::string("=") + std::to_string(v[0].second);
          c.setGate(f, DotGate::EQ, cond);
        }
        else if (type == "project")
        {
          vector<pair<int, int> > v = parse_array(SPI_getvalue(tuple, tupdesc, 6));
          sort(v.begin(), v.end(), [](auto &left, auto &right) {
            return left.second < right.second;
          });
          std::string cond("(");
          for (auto p : v)
          {
            cond += std::to_string(p.first) + ",";
          }
          std::string cond_final(cond.substr(0, cond.size() - 1) + ")");
          //elog(WARNING, "PROJECT cond: %s", cond_final.c_str());
          c.setGate(f, DotGate::PROJECT, cond_final);
        }
        else
        {
          provsql_error("Unknown type of gate: %s", type.c_str());
        }
        //elog(WARNING, "%d -- %d", id, c.getGate(SPI_getvalue(tuple, tupdesc, 2)));
        c.addWire(id, c.getGate(SPI_getvalue(tuple, tupdesc, 2)));
      }
    }
  }

  SPI_finish();

  // Display the circuit for debugging:
  int display = DatumGetInt64(is_debug);
  if(display)
    provsql_warning("%s", c.toString(gate_t{0}).c_str());

  //Calling the dot renderer
  return c.render();
}

/** @brief PostgreSQL-callable wrapper for view_circuit(). */
Datum view_circuit(PG_FUNCTION_ARGS)
{
  provsql_sync_tool_registry();  // honour persisted tool-registry overrides
  try
  {
    Datum token = PG_GETARG_DATUM(0);
    Datum token2prob = PG_GETARG_DATUM(1);
    Datum is_debug = PG_GETARG_DATUM(2);

    if (PG_ARGISNULL(1))
      PG_RETURN_NULL();

    std::string s = view_circuit_internal(token, token2prob, is_debug);

    text *result = (text *) palloc(VARHDRSZ + s.size() + 1);
    SET_VARSIZE(result, VARHDRSZ + s.size());

    memcpy((void *) VARDATA(result),
           s.c_str(),
           s.size());
    PG_RETURN_TEXT_P(result);
  } catch(const std::exception &e) {
    provsql_error("view_circuit: %s", e.what());
  }
  catch (...)
  {
    provsql_error("view_circuit: Unknown exception");
  }

  PG_RETURN_NULL();
}
