#include <pland/syscall.h>
#include <knob/block.h>
#include <knob/heap.h>
#include <stdint.h>
#include <stdbool.h>

static const uint8_t caps_and_shift[] = {
  0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
  0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
  0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
  0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x22,
  0x28, 0x29, 0x2a, 0x2b, 0x3c, 0x5f, 0x3e, 0x3f,
  0x29, 0x21, 0x40, 0x23, 0x24, 0x25, 0x5e, 0x26,
  0x2a, 0x28, 0x3a, 0x3a, 0x3c, 0x2d, 0x3e, 0x3f,

  0x40, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67,
  0x68, 0x69, 0x6a, 0x6b, 0x6c, 0x6d, 0x6e, 0x6f,
  0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77,
  0x78, 0x79, 0x7a, 0x57, 0x7c, 0x7d, 0x5e, 0x5f,
  0x7e, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67,
  0x68, 0x69, 0x6a, 0x6b, 0x6c, 0x6d, 0x6e, 0x6f,
  0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77,
  0x78, 0x79, 0x7a, 0x7b, 0x7c, 0x7d, 0x7e, 0x00,

  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,

  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
};

static const uint8_t caps_no_shift[] = {
  0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
  0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
  0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
  0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27,
  0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
  0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37,
  0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f,

  0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47,
  0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
  0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57,
  0x58, 0x59, 0x5a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f,
  0x60, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47,
  0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
  0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57,
  0x58, 0x59, 0x5a, 0x7b, 0x7c, 0x7d, 0x7e, 0x00,

  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,

  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
};

static const uint8_t shifted[] = {
  0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
  0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
  0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
  0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x22,
  0x28, 0x29, 0x2a, 0x2b, 0x3c, 0x5f, 0x3e, 0x3f,
  0x29, 0x21, 0x40, 0x23, 0x24, 0x25, 0x5e, 0x26,
  0x2a, 0x28, 0x3a, 0x3a, 0x3c, 0x2d, 0x3e, 0x3f,

  0x40, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67,
  0x68, 0x69, 0x6a, 0x6b, 0x6c, 0x6d, 0x6e, 0x6f,
  0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77,
  0x78, 0x79, 0x7a, 0x57, 0x7c, 0x7d, 0x5e, 0x5f,
  0x7e, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47,
  0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
  0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57,
  0x58, 0x59, 0x5a, 0x7b, 0x7c, 0x7d, 0x7e, 0x00,

  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,

  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
};

__attribute__ ((const))
char key_to_char(_key_code_t key) {
  return
    key & _KEY_CAPS
    ? key & _KEY_SHIFT
      ? caps_and_shift[key & 0xff]
      : caps_no_shift[key & 0xff]
    : key & _KEY_SHIFT
      ? shifted[key & 0xff]
      : key & 0xff;
}

static char get_key_char() {
  _key_code_t key;
  while (!(key = _get_key()))
    _yield_task();
  return key_to_char(key);
}

void tell_user_sz(const char *sz) {
  _log_string(sz);
}

//return value and max_length don't include null terminator
uint32_t ask_user_line_sz(char *sz, uint32_t max_length) {
  char log_buf[2];
  log_buf[1] = '\0';

  uint32_t i;
  for (i = 0; i != max_length; ++i) {
    char key;
  replace:
    key = get_key_char();
    if (!key)
      goto replace;
    if (key & 0x80)
      goto replace;//TODO
    if (key == '\b') {
      if (i) {
        --i;
        _log_string("\b");
      }
      goto replace;
    }

    log_buf[0] = key;
    _log_string(log_buf);

    if (key == '\n')
      break;
    sz[i] = key;
  }

  sz[i] = '\0';
  return i;
}

struct history_entry {
  struct history_entry *prev;
  struct history_entry *next;
  const char *contents;
};

struct history {
  struct history_entry *earliest_history_item;
  struct history_entry *latest_history_item;
  uint32_t max_entries;
  uint32_t cur_entries;
};

struct history *new_history(uint32_t max_entries) {
  struct history *new = get_block(sizeof(struct history));
  new->latest_history_item = 0;
  new->earliest_history_item = 0;
  new->max_entries = max_entries;
  new->cur_entries = 0;
  return new;
}

void del_history(struct history *hs) {
  free_block(hs);
  for (struct history_entry *he = hs->latest_history_item; he; he = he->prev) {
    free_block(he);
    free_block(he->contents);
  }
}

//return value and max_length don't include null terminator
uint32_t ask_user_line_sz_with_history(char *sz, uint32_t max_length, struct history *hs) {
  char log_buf[2];
  log_buf[1] = '\0';
  struct history_entry *cur_he = 0;

  uint32_t i;
  for (i = 0; i != max_length; ++i) {
    char key;
  replace:
    key = get_key_char();
    if (!key)
      goto replace;
    if (key == (char)_KEY_UP) {
      struct history_entry *new_cur_he = cur_he ? cur_he->prev : hs->latest_history_item;
      if (new_cur_he) {
        cur_he = new_cur_he;
        for (uint32_t j = 0; j < i; ++j)
          _log_string("\b");
        i = strcpy(sz, cur_he->contents);
        _log_string(sz);
      }
      goto replace;
    }
    if (key == (char)_KEY_DOWN) {
      cur_he = cur_he ? cur_he->next : 0;
      for (uint32_t j = 0; j < i; ++j)
        _log_string("\b");
      if (cur_he) {
        i = strcpy(sz, cur_he->contents);
        _log_string(sz);
      }
      else
        i = 0;
      goto replace;
    }
    if (key & 0x80)
      goto replace;//TODO
    if (key == '\b') {
      if (i) {
        --i;
        _log_string("\b");
      }
      goto replace;
    }

    log_buf[0] = key;
    _log_string(log_buf);

    if (key == '\n')
      break;
    sz[i] = key;
  }

  sz[i] = '\0';

  struct history_entry *new_he = get_block(sizeof(struct history_entry));
  new_he->contents = strdup(sz);

  new_he->next = 0;
  if (hs->latest_history_item) {
    new_he->prev = hs->latest_history_item;
    hs->latest_history_item->next = new_he;
  }
  else
    new_he->prev = 0;
  hs->latest_history_item = new_he;
  if (!hs->earliest_history_item)
    hs->earliest_history_item = new_he;
  if (hs->max_entries == hs->cur_entries) {
    free_block(hs->earliest_history_item);
    free_block(hs->earliest_history_item->contents);
    hs->earliest_history_item->next->prev = 0;
    hs->earliest_history_item = hs->earliest_history_item->next;
  }
  else
    ++(hs->cur_entries);

  for (struct history_entry *check_he = hs->latest_history_item->prev; check_he; check_he = check_he->prev) {
    if (blockequ(check_he->contents, sz, i + 1)) {
      check_he->next->prev = check_he->prev;
      if (check_he->prev)
        check_he->prev->next = check_he->next;
      else
        hs->earliest_history_item = check_he->next;
      free_block(check_he);
      free_block(check_he->contents);
    }
  }

  return i;
}