#include "stplugin.h"
#include "duktape.h"

#define DOIT(cmd) duk_eval_string_noresult(ctx, cmd);

#define FAIL_DUK(cmd) { \
  const char *errmsg = duk_safe_to_string(ctx, -1); \
  duk_destroy_heap(ctx); \
  snprintf(failbuf, 255, "error in %s: %s", cmd, errmsg); \
  SF_error(failbuf); \
  return 1; \
}

#define FAIL_LOAD { \
  duk_push_undefined(ctx); \
  SF_error("Error in load_file"); \
  return 1; \
}

static char *read_file(const char *filename, size_t *sz) {
  FILE *f = fopen(filename, "rb");
  if(!f) return NULL;
  long fsize; { fseek(f, 0, SEEK_END); fsize = ftell(f); fseek(f, 0, SEEK_SET); }
  char *buf = (char *)malloc(fsize * sizeof(char));
  *sz = fread((void *) buf, 1, fsize, f);
  fclose(f);
  return buf;
}

static duk_int_t eval_file(duk_context *ctx, const char *filename) {
  size_t len; char *buf = read_file(filename, &len);
  if(!buf) FAIL_LOAD

  duk_push_lstring(ctx, (const char *)buf, (duk_size_t)len);
  duk_int_t retval = duk_peval(ctx);
  duk_pop(ctx);
  return retval;
}

static duk_int_t load_file(duk_context *ctx, const char *filename, const char *var) {
  size_t len; char *buf = read_file(filename, &len);
  if(!buf) FAIL_LOAD

  duk_push_external_buffer(ctx);
  duk_config_buffer(ctx, -1, buf, len);
  duk_put_global_string(ctx, var);
  return 0;
}

static duk_int_t save_file(duk_context *ctx, const char *filename, const char *var) {
  duk_get_global_string(ctx, var);
  duk_size_t sz;
  char *buf = (char *)duk_get_buffer_data(ctx, -1, &sz);

  if(!buf) return 1;
  FILE *f = fopen(filename, "wb"); fwrite(buf, 1, sz, f); fclose(f);
  return 0;
}

STDLL stata_call(int argc, char *argv[])
{
  duk_int_t res = 0;
  char failbuf[255];

  /* initialize duktape */
  duk_context *ctx = duk_create_heap_default();
  /* duktape does not expose a standard "global" by default */
  DOIT("var global = (function(){ return this; }).call(null);")

  /* load SheetJS library */
  res = eval_file(ctx, "shim.min.js");
  if(res != 0) FAIL_DUK("shim load")
  res = eval_file(ctx, "xlsx.full.min.js");
  if(res != 0) FAIL_DUK("library load")

  /* print SheetJS version number */
  //duk_eval_string(ctx, "XLSX.version");
  //char verbuf[255];
  //snprintf(verbuf, 255, "SheetJS library version: %s\n", duk_get_string(ctx, -1));
  //SF_display(verbuf);

  /* read file */
  res = load_file(ctx, argv[0], "buf");
  if(res != 0) FAIL_DUK("file load")

  /* parse workbook */
  DOIT("wb = XLSX.read(buf.slice(0, buf.length), {type:'buffer'});");

  /* print CSV */
  duk_eval_string(ctx, "wb.SheetNames.length");
  duk_uint_t wscnt = duk_get_uint(ctx, -1);
  duk_pop(ctx);

  /* if argument 2 is "verbose", print CSV contents */
  if(argc>1 && !strncmp(argv[1], "verbose", 7)) for(uint32_t wsidx = 0; wsidx < wscnt; ++wsidx) {
    /* select n-th worksheet */
    char wsbuf[80];
    snprintf(wsbuf, 80, "ws = wb.Sheets[wsname = wb.SheetNames[%d]]", wsidx); \
    DOIT(wsbuf);

    duk_eval_string(ctx, "wsname");
    char namebuf[60];
    snprintf(namebuf, 60, "Worksheet %d Name: %s\n", wsidx, duk_get_string(ctx, -1));
    duk_pop(ctx);
    SF_display(namebuf);

    /* convert to CSV */
    duk_eval_string(ctx, "XLSX.utils.sheet_to_csv(ws)");
    const char *csv = duk_get_string(ctx, -1);

    /* print each row in a separate line */
    char *tok = strtok(csv, "\n");
    while(tok != NULL) {
      SF_display(tok);
      SF_display("\n");
      tok = strtok(NULL, "\n");
    }
    duk_pop(ctx);
  }

  /* write to sheetjs.tmp.xlsx */
  DOIT("newbuf = (XLSX.write(wb, {type:'array', bookType:'xlsx'}));");\
  res = save_file(ctx, "sheetjs.tmp.xlsx", "newbuf");\
  if(res != 0) FAIL_DUK("save sheetjsw.xlsx")

  SF_display("\n");
  SF_display("Saved to `sheetjs.tmp.xlsx`\n");
  SF_display("{stata import excel \"sheetjs.tmp.xlsx\", firstrow} will read the first sheet and use headers\n");
  SF_display("for more help, see {help import excel}\n");
  return(0) ;
}