#include <furi.h>
#include <furi_hal_gpio.h>
#include <gui/gui.h>
#include <gui/modules/button_panel.h>

const GpioPin* const pin_up = &gpio_ext_pa4;
const GpioPin* const pin_down = &gpio_ext_pb2;
const GpioPin* const pin_right = &gpio_ext_pa7;
const GpioPin* const pin_left = &gpio_ext_pa6;
const GpioPin* const pin_fire = &gpio_ext_pc3;
const GpioPin* const pin_mode = &gpio_ext_pc1;

typedef enum
{
  EventTypeTick,
  EventTypeKey,
} EventType;

typedef struct
{
  EventType type;
  InputEvent input;
} PluginEvent;

typedef enum
{
  WorkerEvtStop = (1 << 0),
  WorkerEvtModeChange = (1 << 1),
  WorkerEvtUpChange = (1 << 2),
  WorkerEvtDownChange = (1 << 3),
  WorkerEvtRightChange = (1 << 4),
  WorkerEvtLeftChange = (1 << 5),
  WorkerEvtCodeChange = (1 << 6),
} WorkerEvtFlags;

typedef struct
{
  FuriThread* thread;
  volatile bool mode;

  bool button_up;
  bool button_down;
  bool button_right;
  bool button_left;
  bool button_fire;
  unsigned int button_code;

  ButtonPanel* button_panel;
  bool dpad;
} Coleco;

static void mode_isr(void* context)
{
  Coleco* coleco = (Coleco*)context;
  coleco->mode = furi_hal_gpio_read(&gpio_ext_pc1);
  furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtModeChange);
}

static int32_t coleco_output_worker(void* context)
{
  Coleco* coleco = (Coleco*)context;

  furi_hal_gpio_add_int_callback(pin_mode, mode_isr, coleco);

  while(1)
  {
    uint32_t events = furi_thread_flags_wait(0x7F, FuriFlagWaitAny, FuriWaitForever);
    furi_check((events & FuriFlagError) == 0);

    if (events & WorkerEvtStop)
    {
      break;
    }

    if (events & WorkerEvtModeChange)
    {
      if (coleco->mode)
      {
        furi_hal_gpio_write(pin_up, !coleco->button_up);
        furi_hal_gpio_write(pin_down, !coleco->button_down);
        furi_hal_gpio_write(pin_right, !coleco->button_right);
        furi_hal_gpio_write(pin_left, !coleco->button_left);
      }
      else
      {
        furi_hal_gpio_write(pin_left, (coleco->button_code & 1));
        furi_hal_gpio_write(pin_down, (coleco->button_code & 2));
        furi_hal_gpio_write(pin_right, (coleco->button_code & 4));
        furi_hal_gpio_write(pin_up, (coleco->button_code & 8));
      }
    }

    if (events & WorkerEvtUpChange)
    {
      if (coleco->mode)
      {
        furi_hal_gpio_write(pin_up, !coleco->button_up);
      }
    }

    if (events & WorkerEvtDownChange)
    {
      if (coleco->mode)
      {
        furi_hal_gpio_write(pin_down, !coleco->button_down);
      }
    }

    if (events & WorkerEvtRightChange)
    {
      if (coleco->mode)
      {
        furi_hal_gpio_write(pin_right, !coleco->button_right);
      }
    }

    if (events & WorkerEvtLeftChange)
    {
      if (coleco->mode)
      {
        furi_hal_gpio_write(pin_left, !coleco->button_left);
      }
    }

    if (events & WorkerEvtCodeChange)
    {
      if (!coleco->mode)
      {
        furi_hal_gpio_write(pin_left, !(coleco->button_code & 1));
        furi_hal_gpio_write(pin_down, !(coleco->button_code & 2));
        furi_hal_gpio_write(pin_right, !(coleco->button_code & 4));
        furi_hal_gpio_write(pin_up, !(coleco->button_code & 8));
      }
    }
  }

  return 0;
}

static void render_callback(Canvas* const canvas, void* context)
{
  Coleco* coleco = acquire_mutex((ValueMutex*)context, 25);
  if (coleco == NULL)
  {
    return;
  }

  canvas_set_font(canvas, FontPrimary);
  canvas_draw_str_aligned(canvas, 64, 8, AlignCenter, AlignBottom, "ColecoVision");

  canvas_set_font(canvas, FontSecondary);

  if (coleco->dpad)
  {
    canvas_draw_str_aligned(canvas, 64, 18, AlignCenter, AlignBottom, "d-pad");
  }
  else
  {
    canvas_draw_str_aligned(canvas, 64, 18, AlignCenter, AlignBottom, "numbers");
  }

  release_mutex((ValueMutex*)context, coleco);
}

static void input_callback(InputEvent* input_event, FuriMessageQueue* event_queue)
{
  furi_assert(event_queue);

  PluginEvent event = {.type = EventTypeKey, .input = *input_event};
  furi_message_queue_put(event_queue, &event, FuriWaitForever);
}

static Coleco* coleco_alloc()
{
  Coleco* coleco = malloc(sizeof(Coleco));

  coleco->button_panel = button_panel_alloc();

  coleco->mode = true;
  coleco->button_up = false;
  coleco->button_down = false;
  coleco->button_right = false;
  coleco->button_left = false;
  coleco->button_fire = false;
  coleco->button_code = 0x0F;
  coleco->dpad = false;

  // configure output pins
  furi_hal_gpio_init(pin_up, GpioModeOutputPushPull, GpioPullNo, GpioSpeedVeryHigh);
  furi_hal_gpio_init(pin_down, GpioModeOutputPushPull, GpioPullNo, GpioSpeedVeryHigh);
  furi_hal_gpio_init(pin_right, GpioModeOutputPushPull, GpioPullNo, GpioSpeedVeryHigh);
  furi_hal_gpio_init(pin_left, GpioModeOutputPushPull, GpioPullNo, GpioSpeedVeryHigh);
  furi_hal_gpio_init(pin_fire, GpioModeOutputPushPull, GpioPullNo, GpioSpeedVeryHigh);

  furi_hal_gpio_write(pin_up, true);
  furi_hal_gpio_write(pin_down, true);
  furi_hal_gpio_write(pin_right, true);
  furi_hal_gpio_write(pin_left, true);
  furi_hal_gpio_write(pin_fire, true);

  // configure mode pin
  furi_hal_gpio_init(pin_mode, GpioModeInterruptRiseFall, GpioPullNo, GpioSpeedVeryHigh);

  // set up output worker thread
  coleco->thread = furi_thread_alloc();
  furi_thread_set_name(coleco->thread, "ColecoOutputWorker");
  furi_thread_set_stack_size(coleco->thread, 1024);
  furi_thread_set_context(coleco->thread, coleco);
  furi_thread_set_callback(coleco->thread, coleco_output_worker);
  furi_thread_start(coleco->thread);

  return coleco;
}

static void coleco_free(Coleco* coleco)
{
  furi_assert(coleco);

  furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtStop);
  furi_thread_join(coleco->thread);
  furi_thread_free(coleco->thread);

  button_panel_free(coleco->button_panel);

  free(coleco);
}

int32_t coleco_app(void* p)
{
  UNUSED(p);

  FuriMessageQueue* event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));

  Coleco* coleco = coleco_alloc();

  ValueMutex coleco_mutex;
  if (!init_mutex(&coleco_mutex, coleco, sizeof(Coleco)))
  {
    FURI_LOG_E("Coleco", "cannot create mutex\r\n");
    coleco_free(coleco);
    return 255;
  }

  // set system callbacks
  ViewPort* view_port = view_port_alloc();
  view_port_draw_callback_set(view_port, render_callback, &coleco_mutex);
  view_port_input_callback_set(view_port, input_callback, event_queue);

  // open GUI and register view_port
  Gui* gui = furi_record_open("gui");
  gui_add_view_port(gui, view_port, GuiLayerFullscreen);

  PluginEvent event;
  for (bool processing = true; processing;)
  {
    FuriStatus event_status = furi_message_queue_get(event_queue, &event, 100);

    Coleco* coleco = (Coleco*)acquire_mutex_block(&coleco_mutex);

    if (event_status == FuriStatusOk)
    {
      // press events
      if (event.type == EventTypeKey)
      {
        switch (event.input.key)
        {
          case InputKeyUp:
            if (coleco->dpad)
            {
              if (event.input.type == InputTypePress)
              {
                coleco->button_up = true;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtUpChange);
              }
              else if (event.input.type == InputTypeRelease)
              {
                coleco->button_up = false;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtUpChange);
              }
            }
            else  // FIXME: hack to allow the 1 to be pressed
            {
              if (event.input.type == InputTypePress)
              {
                coleco->button_code = 0x0D;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtCodeChange);
              }
              else if (event.input.type == InputTypeRelease)
              {
                coleco->button_code = 0x0F;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtCodeChange);
              }
            }
            break;
          case InputKeyDown:
            if (coleco->dpad)
            {
              if (event.input.type == InputTypePress)
              {
                coleco->button_down = true;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtDownChange);
              }
              else if (event.input.type == InputTypeRelease)
              {
                coleco->button_down = false;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtDownChange);
              }
            }
            break;
          case InputKeyRight:
            if (coleco->dpad)
            {
              if (event.input.type == InputTypePress)
              {
                coleco->button_right = true;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtRightChange);
              }
              else if (event.input.type == InputTypeRelease)
              {
                coleco->button_right = false;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtRightChange);
              }
            }
            break;
          case InputKeyLeft:
            if (coleco->dpad)
            {
              if (event.input.type == InputTypePress)
              {
                coleco->button_left = true;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtLeftChange);
              }
              else if (event.input.type == InputTypeRelease)
              {
                coleco->button_left = false;
                furi_thread_flags_set(furi_thread_get_id(coleco->thread), WorkerEvtLeftChange);
              }
            }
            break;
          case InputKeyOk:
            if (coleco->dpad)
            {
              if (event.input.type == InputTypePress)
              {
                coleco->button_fire = true;
                furi_hal_gpio_write(pin_fire, !coleco->button_fire);
              }
              else if (event.input.type == InputTypeRelease)
              {
                coleco->button_fire = false;
                furi_hal_gpio_write(pin_fire, !coleco->button_fire);
              }
            }
            else
            {
              coleco->dpad = true;
            }
            break;
          case InputKeyBack:
            if (event.input.type == InputTypePress)
            {
              if (coleco->dpad)
              {
                coleco->dpad = false;
              }
              else
              {
                processing = false;
              }
            }
            break;
          default:
            break;
        }

        view_port_update(view_port);
      }
    }
    else
    {
      FURI_LOG_D("Coleco", "FuriMessageQueue: event timeout");
    }

    release_mutex(&coleco_mutex, coleco);
  }

  furi_hal_gpio_remove_int_callback(pin_mode);
  view_port_enabled_set(view_port, false);
  gui_remove_view_port(gui, view_port);
  furi_record_close("gui");
  view_port_free(view_port);
  furi_message_queue_free(event_queue);
  delete_mutex(&coleco_mutex);
  coleco_free(coleco);
  return 0;
}