#include <string.h>
#include <errno.h>
#include <ctype.h>

#include "fh_error.h"
#include "fh_runtime.h"
#include "fh_builtins.h"
#include "fh_stack.h"
#include "fh_mem.h"
#include "fh_globals.h"
#include "fh_print.h"

struct fh_global_s fh_globals = {};

/** State names */
static const char *statenames[FH_STATE_MAX] = {
    [FH_STATE_INTERPRET] = "INTERPRET",
    [FH_STATE_COMPILE] = "COMPILE",
    [FH_STATE_QUIT] = "QUIT",
    [FH_STATE_SHUTDOWN] = "SHUTDOWN",
};

/** State names short */
static const char *stateshort[FH_STATE_MAX] = {
    [FH_STATE_INTERPRET] = "INT",
    [FH_STATE_COMPILE] = "COM",
    [FH_STATE_QUIT] = "QIT",
    [FH_STATE_SHUTDOWN] = "DIE",
};

/** Sub-state names */
static const char *substatenames[FH_SUBSTATE_MAX] = {
    [FH_SUBSTATE_NONE] = "NONE",
    [FH_SUBSTATE_PAREN_COMMENT] = "PAREN_COMMENT",
    [FH_SUBSTATE_LINE_COMMENT] = "LINE_COMMENT",
    [FH_SUBSTATE_EXIT] = "EXIT",
};

/** Sub-state names */
static const char *instrnames[FH_INSTR_MAX] = {
    [FH_INSTR_WORD] = "WORD",
    [FH_INSTR_NUMBER] = "NUMBER",
    [FH_INSTR_ENDWORD] = "ENDWORD",
    [FH_INSTR_ALLOCSTR] = "ALLOCSTR",
    [FH_INSTR_TYPESTR] = "TYPESTR",
    [FH_INSTR_JUMP] = "JUMP",
    [FH_INSTR_JUMPZERO] = "JUMPZERO",
    [FH_INSTR_LEAVE] = "LEAVE",
    [FH_INSTR_DO] = "DO",
    [FH_INSTR_TO] = "TO",
    [FH_INSTR_DO_QUESTION] = "DO_QUESTION",
    [FH_INSTR_LOOP] = "LOOP",
    [FH_INSTR_LOOP_PLUS] = "LOOP_PLUS",
    [FH_INSTR_POSTPONED_WORD] = "POSTPONED_WORD",
};


const char *instr_name(enum fh_instruction_kind kind) {
  if (kind >= FH_INSTR_MAX) {
    return "Unknown";
  } else {
    return instrnames[kind];
  }
}

/** Add a word to the dictionary. */
enum fh_error fh_add_word(const struct fh_word_s *w, struct fh_thread_s *fh)
{
  enum fh_error rv;

  fh_align(fh);
  uint32_t ptr = fh->here;

  TRY(fh_heap_put(fh, w, DICTWORD_SIZE));

  //LOG("Added word \"%s\" at 0x%08x", w->name, ptr);

  // thread it onto the linked list
  fh_word_at(fh, ptr)->previous = fh->dict_last;
  fh->dict_last = ptr;

  return FH_OK;
}

/** Log current runtime state */
static void showstate(const struct fh_thread_s *fh)
{
  if (fh->substate == 0) {
    LOG("state -> %s", statenames[fh->state]);
  } else {
    LOG("state -> %s.%s", statenames[fh->state], substatenames[fh->substate]);
  }
}

/** Set runtime state and sub-state */
void fh_setstate(struct fh_thread_s *fh, enum fh_state state, enum fh_substate substate)
{
  fh->state = state;
  fh->substate = substate;
  showstate(fh);
}

/** Set runtime sub-state (state is unchanged) */
void fh_setsubstate(struct fh_thread_s *fh, enum fh_substate substate)
{
  fh->substate = substate;
  showstate(fh);
}

/** Execute a user word */
enum fh_error w_user_word(struct fh_thread_s *fh, const struct fh_word_s *w0)
{
  enum fh_error rv;
  const struct fh_word_s *w;
  struct fh_word_s *w2;

  w = w0;
  call:
  if (!w) { return FH_ERR_INTERNAL; }

  LOG("Run user word: %s", w->name);

  TRY(rs_push(fh, fh->execptr));
  fh->execptr = w->param;

  instr:;
  if (fh->state == FH_STATE_QUIT) {
    /* abort or quit was called, return to interactive mode */
    fh_setstate(fh, FH_STATE_INTERPRET, FH_SUBSTATE_NONE);
    return FH_OK;
  }
  // make sure it's aligned
  fh->execptr = WORDALIGNED(fh->execptr);
  const struct fh_instruction_s *instr = fh_instr_at(fh, fh->execptr);
  fh->execptr += INSTR_SIZE;

  uint32_t strl;
  uint32_t val;
  uint32_t addr = 0;
  uint32_t limit, index, index0;

  switch (instr->kind) {
    case FH_INSTR_NUMBER:
      TRY(ds_push(fh, instr->data));
      goto instr;

    case FH_INSTR_POSTPONED_WORD:
      if (fh->state == FH_STATE_COMPILE) {
        w2 = fh_word_at(fh, instr->data);
        if (w2->flags & WORDFLAG_IMMEDIATE) {
          LOG("Call immediate postponed word: %s", w2->name);
          TRY(w2->handler(fh, w2));
        } else {
          LOG("Add postponed word: %s", w2->name);
          TRY(fh_put_instr(fh, FH_INSTR_WORD, instr->data));
        }
      } else {
        LOGE("Postpone in interpret mode!");
        goto end;
      }
      goto instr;

    case FH_INSTR_WORD:
      w2 = fh_word_at(fh, instr->data);
      if (w2->flags & WORDFLAG_BUILTIN) {
        LOG("Exec: native-word \"%s\"", w2->name);
        w2->handler(fh, w2);
        if (fh->substate == FH_SUBSTATE_EXIT) {
          fh_setsubstate(fh, 0);
          LOG("Exec: early return");
          TRY(rs_pop(fh, &fh->execptr));
          if (fh->execptr == MAGICADDR_EXEC_INTERACTIVE) {
            goto end;
          }
        }
        goto instr;
      } else {
        LOG("Exec: user-word %s (CALL)", w2->name);
        w = fh_word_at(fh, instr->data);
        goto call;
      }

    case FH_INSTR_JUMPZERO:
      if (instr->data == MAGICADDR_UNRESOLVED) {
        LOGE("Encountered unresolved jump!");
        goto end;
      }

      TRY(ds_pop(fh, &val));
      if (0 == val) {
        fh->execptr = instr->data;
      }
      goto instr;

    case FH_INSTR_JUMP:
      if (instr->data == MAGICADDR_UNRESOLVED) {
        LOGE("Encountered unresolved jump!");
        goto end;
      }
      fh->execptr = instr->data;
      goto instr;

    case FH_INSTR_TO:
      TRY(ds_pop(fh, &val));
      w2 = fh_word_at(fh, instr->data);
      LOG("Exec: %d->%s", val, w2->name);

      if (w2->flags & WORDFLAG_CONSTANT) {
        LOGE("Cannot assign to constant!");
        return FH_ERR_ILLEGAL_STORE;
      }

      w2->param = val;
      goto instr;

    case FH_INSTR_DO:
      TRY(ds_pop(fh, &index));
      TRY(ds_pop(fh, &limit)); // just make sure it exists
      TRY(fh_loop_nest(fh, index));
      TRY(rs_push(fh, limit));
      goto instr;

    case FH_INSTR_DO_QUESTION:
      if (instr->data == MAGICADDR_UNRESOLVED) {
        LOGE("Encountered unresolved jump!");
        goto end;
      }

      TRY(ds_pop(fh, &index));
      TRY(ds_pop(fh, &limit));
      if (index == limit) {
        // jump to end
        fh->execptr = instr->data;
      } else {
        TRY(fh_loop_nest(fh, index));
        TRY(rs_push(fh, limit));
      }
      goto instr;

    case FH_INSTR_LOOP_PLUS:
      TRY(ds_pop(fh, &val));
      // fall-through
    case FH_INSTR_LOOP:
      if (instr->kind == FH_INSTR_LOOP) {
        val = 1;
      }

      // R: index,limit
      TRY(rs_peek(fh, &limit));

      LOG("+LOOP, i=%d, step %d, limit %d", fh->loop_i, val, limit);

      index0 = fh->loop_i;
      fh->loop_i += val;

      LOG("after add: %d", fh->loop_i);

      if (((int32_t)index0 < (int32_t)limit) == ((int32_t)fh->loop_i < (int32_t)limit)) { // boundary not crossed, continue
        fh->execptr = instr->data; // go to beginning
      } else {
        // end of loop
        TRY(rs_pop(fh, &limit));
        TRY(fh_loop_unnest(fh));
      }
      goto instr;

    case FH_INSTR_LEAVE:
      if (instr->data == MAGICADDR_UNRESOLVED) {
        LOGE("Encountered unresolved jump!");
        goto end;
      }
      TRY(rs_pop(fh, &limit));
      TRY(fh_loop_unnest(fh));
      fh->execptr = instr->data;
      goto instr;

      /* special case for strings stored in compile memory */
    case FH_INSTR_ALLOCSTR:
    case FH_INSTR_TYPESTR:
      strl = instr->data;
      if (instr->kind == FH_INSTR_ALLOCSTR) {
        LOG("Exec: alloc-str \"%.*s\"", strl, fh_str_at(fh, fh->execptr));
        TRY(ds_push(fh, fh->execptr)); // give pointer directly into the definition
        TRY(ds_push(fh, strl));
      } else {
        LOG("Exec: type-str \"%.*s\"", strl, fh_str_at(fh, fh->execptr));
        FHPRINT("%.*s", (int) strl, fh_str_at(fh, fh->execptr));
      }
      fh->execptr += strl;
      goto instr;

    case FH_INSTR_ENDWORD:
      LOG("Exec: word-end (RETURN)");
      TRY(rs_pop(fh, &fh->execptr));
      if (fh->execptr == MAGICADDR_EXEC_INTERACTIVE) {
        goto end;
      }
      goto instr;
  }

  end:
  return FH_OK;
}


/** Initialize a runtime */
enum fh_error fh_init(struct fh_thread_s *fh)
{
  enum fh_error rv;

  /* Make sure we have a clean state */
  memset(fh, 0, sizeof(struct fh_thread_s));

  fh->dict_last = MAGICADDR_DICTFIRST;
  TRY(register_builtin_words(fh));

  fh->execptr = MAGICADDR_EXEC_INTERACTIVE;
  fh->base = 10;
  return FH_OK;
}

enum fh_error fh_handle_word(struct fh_thread_s *fh, uint32_t addr)
{
  enum fh_error rv;
  struct fh_word_s *w = fh_word_at(fh, addr);
  if (fh->state == FH_STATE_COMPILE && 0 == (w->flags & WORDFLAG_IMMEDIATE)) {
    LOG("\x1b[34m[%s] Compile word:\x1b[m %s", stateshort[fh->state], w->name);
    TRY(fh_put_instr(fh, FH_INSTR_WORD, addr));
  } else {
    /* interpret or immediate in compiled code */
//    enum fh_state oldstate = fh->state;
//    fh_setstate(fh, FH_STATE_INTERPRET, 0);
    LOG("\x1b[35m[%s] Run word:\x1b[m %s%s", stateshort[fh->state], w->name, (w->flags&WORDFLAG_IMMEDIATE)?" 'imm":"");
    TRY(w->handler(fh, w));
//    if (fh->state == FH_STATE_INTERPRET && fh->substate == FH_SUBSTATE_NONE) {
//      fh_setstate(fh, oldstate, 0);
//    }
  }
  return FH_OK;
}

enum fh_error fh_find_word(struct fh_thread_s *fh, const char *name, size_t wordlen, uint32_t *addr_out)
{
  if (wordlen == 0) {
    wordlen = strlen(name);
  }
  uint32_t addr = fh->dict_last;
  while (addr != MAGICADDR_DICTFIRST) {
    struct fh_word_s *w = fh_word_at(fh, addr);
    if (0 == strncasecmp(name, w->name, wordlen) && w->name[wordlen] == 0) {
      if (addr_out) {
        *addr_out = addr;
      }
      return FH_OK;
    }
    addr = w->previous;
  }
  // no log message, this can be OK - e.g. parsing a number
  //LOG("fail to find word %.*s", wordlen, name);
  return FH_ERR_UNKNOWN_WORD;
}

enum fh_error fh_loop_nest(struct fh_thread_s *fh, uint32_t indexvalue)
{
  enum fh_error rv;
  TRY(rs_push(fh, fh->loop_j));
  fh->loop_j = fh->loop_i;
  fh->loop_i = indexvalue;
  return FH_OK;
}

enum fh_error fh_loop_unnest(struct fh_thread_s *fh)
{
  enum fh_error rv;
  fh->loop_i = fh->loop_j;
  TRY(rs_pop(fh, &fh->loop_j));
  return FH_OK;
}

/** Process a word read from input */
static enum fh_error fh_handle_ascii_word(
    struct fh_thread_s *fh,
    const char *name,
    const size_t wordlen
)
{
  enum fh_error rv;
  if (wordlen >= MAX_NAME_LEN) {
    return FH_ERR_NAME_TOO_LONG;
  }

  /* First, try if it's a known word */

  uint32_t wadr = 0;
  if (FH_OK == fh_find_word(fh, name, wordlen, &wadr)) {
    TRY(fh_handle_word(fh, wadr));
    return FH_OK;
  }

  /* word not found, try parsing as number */
  errno = 0;
  char *endptr;
  int base = (int) fh->base;

  // prefix can override BASE - this is a syntax extension
  if (name[0] == '0') {
    if (name[1] == 'x') {
      base = 16;
    } else if (name[1] == 'b') {
      base = 2;
    } else if (name[1] == 'o') {
      base = 8;
    }
  }

  long v = strtol(name, &endptr, base); // XXX if base is 0, this will use auto-detection
  if (errno != 0 || (endptr - name) != wordlen) {
    LOGE("Unknown word and fail to parse as number: \"%.*s\"", (int) wordlen, name);
    return FH_ERR_UNKNOWN_WORD;
  }

  struct fh_instruction_s instr;
  if (fh->state == FH_STATE_COMPILE) {
    LOG("Compile number: %ld", v);
    TRY(fh_put_instr(fh, FH_INSTR_NUMBER, (uint32_t) v));
  } else {
    /* interpret */
    LOG("Interpret number: %ld", v);
    TRY(ds_push(fh, (uint32_t) v));
  }

  return FH_OK;
}


/** Postpone a word */
enum fh_error fh_postpone_word(
    struct fh_thread_s *fh,
    const char *name,
    const size_t wordlen
)
{
  enum fh_error rv;
  uint32_t wadr;
  TRY(fh_find_word(fh, name, wordlen, &wadr));

  LOG("Postpone word");
  TRY(fh_put_instr(fh, FH_INSTR_POSTPONED_WORD, wadr));

  return FH_OK;
}

/** True if the character is CR or LF */
static inline bool isnl(char c)
{
  return c == '\n' || c == '\r';
}

/** Process a line read from input */
enum fh_error fh_process_line(struct fh_thread_s *fh, const char *linebuf, size_t len)
{
  enum fh_error rv;

#define ReadPtr ((char*)(&fh->heap[INPUTBUF_ADDR + fh->inputptr]))
#define ReadPos (fh->inputptr)
#define ReadLen (fh->inputlen)

  fh_fill_input_buffer(fh, linebuf, len);

  char c;

  if (fh_globals.echo && !fh_globals.interactive) {
    LOGI("%s", linebuf);
  }

  while (ReadPos < ReadLen && fh->state != FH_STATE_SHUTDOWN) {
    c = *ReadPtr;
    /* end on newline */
    if (isnl(c)) {
      goto done;
    }
    /* skip whitespace */
    if (isspace(c)) {
      ReadPos++;
      continue;
    }

    const char * const rp = ReadPtr;

    char *end;
    size_t length;
    switch (fh->substate) {
      case FH_SUBSTATE_NONE:
        /* try to read a word */
        end = strchr(rp, ' ');
        if (end) {
          length = end - rp; /* exclude the space */
        } else {
          length = strlen(rp);
        }

        ReadPos += length + 1;

        /* eval a word */
        //LOG("Handle \"%.*s\"", (int) length, rp);
        TRY(fh_handle_ascii_word(fh, rp, length));

        if (!end) {
          goto done;
        }
        break;

      case FH_SUBSTATE_PAREN_COMMENT:
        end = strchr(rp, ')');
        if (end) {
          length = end - rp;
          LOG("Discard inline comment");
          fh_setsubstate(fh, FH_SUBSTATE_NONE);
          ReadPos += length + 1;
        } else {
          /* no end, discard all */
          LOGE("Unterminated parenthesis comment");
          goto done;
        }
        break;

      case FH_SUBSTATE_LINE_COMMENT:
        LOG("Discard line comment");
        fh_setsubstate(fh, 0);
        goto done; // just discard the rest

      default:
        LOGE("Bad substate %s", substatenames[fh->substate]);
    }
  }
  done:
  LOG("Line done.");
  return FH_OK;
}