#include "lvgl_task.hpp"

#include <dirent.h>
#include <stdint.h>
#include <stdio.h>

#include <cstddef>
#include <cstdint>
#include <memory>

#include "core/lv_disp.h"
#include "core/lv_obj.h"
#include "core/lv_obj_pos.h"
#include "core/lv_obj_tree.h"
#include "esp_log.h"
#include "font/lv_font.h"
#include "freertos/portmacro.h"
#include "freertos/projdefs.h"
#include "freertos/timers.h"
#include "hal/gpio_types.h"
#include "hal/spi_types.h"
#include "lvgl/lvgl.h"
#include "misc/lv_color.h"
#include "misc/lv_style.h"
#include "misc/lv_timer.h"
#include "touchwheel.hpp"
#include "widgets/lv_label.h"

#include "display.hpp"
#include "gpio_expander.hpp"

namespace ui {

static const char* kTag = "lv_task";

auto tick_hook(TimerHandle_t xTimer) -> void {
  lv_tick_inc(1);
}

struct LvglArgs {
  std::weak_ptr<drivers::TouchWheel> touch_wheel;
  std::weak_ptr<drivers::Display> display;
  std::atomic<bool>* quit;
};

void LvglMain(void* voidArgs) {
  LvglArgs* args = reinterpret_cast<LvglArgs*>(voidArgs);
  std::weak_ptr<drivers::TouchWheel> weak_touch_wheel = args->touch_wheel;
  std::weak_ptr<drivers::Display> weak_display = args->display;

  std::atomic<bool>* quit = args->quit;
  delete args;

  {
    TimerHandle_t tick_timer =
        xTimerCreate("lv_tick", pdMS_TO_TICKS(1), pdTRUE, NULL, &tick_hook);

    lv_style_t style;
    lv_style_init(&style);
    lv_style_set_text_color(&style, LV_COLOR_MAKE(0xFF, 0, 0));
    // TODO: find a nice bitmap font for this display size and density.
    // lv_style_set_text_font(&style, &lv_font_montserrat_24);

    auto label = lv_label_create(NULL);
    lv_label_set_text(label, "COLOURS!!");
    lv_obj_add_style(label, &style, 0);

    lv_obj_center(label);
    lv_scr_load(label);

    while (!quit->load()) {
      lv_timer_handler();
      // 30 FPS
      // TODO(jacqueline): make this dynamic
      vTaskDelay(pdMS_TO_TICKS(33));
    }

    // TODO(robin? daniel?): De-init the UI stack here.
    lv_obj_del(label);
    lv_style_reset(&style);

    xTimerDelete(tick_timer, portMAX_DELAY);
  }

  vTaskDelete(NULL);
}

static const size_t kLvglStackSize = 8 * 1024;
static StaticTask_t sLvglTaskBuffer = {};
static StackType_t sLvglStack[kLvglStackSize] = {0};

auto StartLvgl(std::weak_ptr<drivers::TouchWheel> touch_wheel,
               std::weak_ptr<drivers::Display> display,
               std::atomic<bool>* quit) -> bool {
  LvglArgs* args = new LvglArgs();
  args->touch_wheel = touch_wheel;
  args->display = display;
  args->quit = quit;

  return xTaskCreateStaticPinnedToCore(&LvglMain, "LVGL", kLvglStackSize,
                                       reinterpret_cast<void*>(args), 1,
                                       sLvglStack, &sLvglTaskBuffer, 1);
}

}  // namespace ui