#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 "widgets/lv_label.h"

#include "display.hpp"
#include "driver_cache.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 {
  drivers::DriverCache* drivers;
  std::atomic<bool>* quit;
};

void LvglMain(void* voidArgs) {
  LvglArgs* args = reinterpret_cast<LvglArgs*>(voidArgs);
  drivers::DriverCache* drivers = args->drivers;
  std::atomic<bool>* quit = args->quit;
  delete args;

  {
    ESP_LOGI(kTag, "init lvgl");
    lv_init();

    // LVGL has been initialised, so we can now start reporting ticks to it.
    TimerHandle_t tick_timer =
        xTimerCreate("lv_tick", pdMS_TO_TICKS(1), pdTRUE, NULL, &tick_hook);

    ESP_LOGI(kTag, "init display");
    std::shared_ptr<drivers::Display> display = drivers->AcquireDisplay();

    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();
      vTaskDelay(pdMS_TO_TICKS(10));
    }

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

    xTimerDelete(tick_timer, portMAX_DELAY);

    lv_deinit();
  }

  vTaskDelete(NULL);
}

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

auto StartLvgl(drivers::DriverCache* drivers,
               std::atomic<bool>* quit,
               TaskHandle_t* handle) -> bool {
  LvglArgs* args = new LvglArgs();
  args->drivers = drivers;
  args->quit = quit;

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

}  // namespace ui