#include "ui.h"

// #include "lv_font.h"
#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "esp_log.h"

#include "motor.h"

#define BTN_INCREMENT 100

#define MAX_RPM_SETTING 200
#define MAX_COAT_TIME_SETTING 30

#define MIN_RPM_SETTING 50
#define MIN_COAT_TIME_SETTING 5

#define RPM_STEPSIZE 25
#define COAT_TIME_STEPSIZE 5

static const char *TAG = "spincoat-plater-firmware/ui";

static lv_obj_t * rpm_label = NULL;
static lv_obj_t * coat_time_label = NULL;
static lv_obj_t * run_stop_label = NULL;

static uint16_t rpm_setting = 100; // rpm
static uint16_t coat_time_setting = 5; // seconds

static coat_state motor_state = STOPPED;

/**
 * Callback for pressing the rpm increase button
 */
void rpm_increase_cb(lv_event_t * e) {
    ESP_LOGI(TAG, "RPM increase callback called.");
    if(rpm_setting < MAX_RPM_SETTING) {
      rpm_setting += RPM_STEPSIZE;
      lv_lock();
      lv_label_set_text_fmt(rpm_label, "%d", rpm_setting);
      lv_unlock();
    }
}

/**
 * Callback for pressing the rpm decrease button
 */
void rpm_decrease_cb(lv_event_t * e) {
    ESP_LOGI(TAG, "RPM decrease callback called.");
    if(rpm_setting > MIN_RPM_SETTING) {
      rpm_setting -= RPM_STEPSIZE;
      lv_lock();
      lv_label_set_text_fmt(rpm_label, "%d", rpm_setting);
      lv_unlock();
    }

}

/**
 * Callback for pressing the coat time increase button
 */

void coat_increase_cb(lv_event_t * e) {
    ESP_LOGI(TAG, "Coat time increase callback called.");
    if(coat_time_setting < MAX_COAT_TIME_SETTING) {
        coat_time_setting += COAT_TIME_STEPSIZE;
        lv_lock();
        lv_label_set_text_fmt(coat_time_label, "%d", coat_time_setting);
        lv_unlock();
    }
}

/**
 * Callback for pressing the coat time decrease button
 */
void coat_decrease_cb(lv_event_t * e) {
    ESP_LOGI(TAG, "Coat time decrease callback called.");
    if(coat_time_setting > MIN_COAT_TIME_SETTING) {
        coat_time_setting -= COAT_TIME_STEPSIZE;
        lv_lock();
        lv_label_set_text_fmt(coat_time_label, "%d", coat_time_setting);
        lv_unlock();
    }
}

void run_stop_cb(lv_event_t * e) {
    ESP_LOGI(TAG, "Run/stop callback called.");
    if(run_stop_label == NULL) return;
    lv_label_set_text(run_stop_label, "Stop");
}


/**
 * Factory function that creates a "numberstack" widget and returns the label for it
 */
lv_obj_t * build_numberstack(lv_obj_t * parent,
        const char * label_text,
        lv_obj_t ** label_value,
        ns_btn_cb_t top_btn_cb,
        ns_btn_cb_t bottom_btn_cb,
        uint16_t start_value,
        uint16_t stepsize) {

    lv_obj_t * container = lv_obj_create(parent);
    lv_obj_set_flex_flow(container, LV_FLEX_FLOW_COLUMN);
    lv_obj_set_flex_align(container,
            LV_FLEX_ALIGN_START,
            LV_FLEX_ALIGN_CENTER,
            LV_FLEX_ALIGN_CENTER);

    lv_obj_set_style_pad_all(container, 0, 0);

    /* -------- Top label -------- */
    lv_obj_t * label_cont = lv_obj_create(container);

    lv_obj_remove_flag(label_cont, LV_OBJ_FLAG_SCROLLABLE);

    lv_obj_set_flex_grow(label_cont, 1);
    lv_obj_set_width(label_cont, lv_pct(100));

    lv_obj_t * label = lv_label_create(label_cont);
    lv_label_set_text(label, label_text);

    lv_obj_set_style_bg_opa(label, LV_OPA_COVER, 0);

    lv_obj_set_style_text_font(label, &lv_font_montserrat_12, 0);
    lv_obj_center(label);
    
    /* -------- Top button -------- */
    lv_obj_t * btn_top = lv_button_create(container);
    lv_obj_t * lbl_top = lv_label_create(btn_top);
    lv_label_set_text_fmt(lbl_top, "+%d", stepsize);
    lv_obj_add_event_cb(btn_top, top_btn_cb, LV_EVENT_CLICKED, (void *) 1);
    lv_obj_set_flex_grow(btn_top, 1);
    lv_obj_set_width(btn_top, lv_pct(100));
    lv_obj_center(lbl_top);

    /* -------- Center label -------- */
    lv_obj_t * label2_cont = lv_obj_create(container);
    
    lv_obj_remove_flag(label2_cont, LV_OBJ_FLAG_SCROLLABLE);

    lv_obj_set_flex_grow(label2_cont, 1);
    lv_obj_set_width(label2_cont, lv_pct(100));

    * label_value = lv_label_create(label2_cont);
    lv_label_set_text_fmt(*label_value, "%d", start_value);

    lv_obj_set_style_bg_opa(*label_value, LV_OPA_COVER, 0);

    lv_obj_set_style_text_font(*label_value, &lv_font_montserrat_12, 0);
    lv_obj_center(*label_value);

    /* -------- Bottom button -------- */
    lv_obj_t * btn_bottom = lv_button_create(container);
    lv_obj_t * lbl_bottom = lv_label_create(btn_bottom);
    lv_label_set_text_fmt(lbl_bottom, "-%d", stepsize);
    lv_obj_add_event_cb(btn_bottom, bottom_btn_cb, LV_EVENT_CLICKED, (void *) 2);
    lv_obj_set_flex_grow(btn_bottom, 1);
    lv_obj_set_width(btn_bottom, lv_pct(100));
    lv_obj_center(lbl_bottom);
    
    return container;
}

void update_rpm_readout(uint16_t rpm) {
    if(rpm_label != NULL) {
        lv_label_set_text_fmt(rpm_label, "RPM: %d", rpm);
    }
};

void build_ui(void) {
    lv_obj_t * container = lv_obj_create(lv_screen_active());
    lv_obj_set_size(container, lv_pct(100), lv_pct(100));
    lv_obj_set_flex_flow(container, LV_FLEX_FLOW_COLUMN);
    lv_obj_set_flex_align(container,
            LV_FLEX_ALIGN_START,
            LV_FLEX_ALIGN_CENTER,
            LV_FLEX_ALIGN_CENTER);

    lv_obj_t * numberstacks_container = lv_obj_create(container);
    lv_obj_set_size(numberstacks_container , lv_pct(100), lv_pct(70));
    lv_obj_set_flex_flow(numberstacks_container, LV_FLEX_FLOW_ROW);
    lv_obj_set_flex_align(numberstacks_container,
            LV_FLEX_ALIGN_SPACE_EVENLY,
            LV_FLEX_ALIGN_CENTER,
            LV_FLEX_ALIGN_CENTER);

    lv_obj_t * rpm_container = build_numberstack(numberstacks_container, "RPM", &rpm_label,
        rpm_increase_cb, rpm_decrease_cb, rpm_setting, RPM_STEPSIZE);
    lv_obj_t * coat_time_container = build_numberstack(numberstacks_container, "Coat Time",
        &coat_time_label, coat_increase_cb, coat_decrease_cb, coat_time_setting, COAT_TIME_STEPSIZE);

    lv_obj_set_size(rpm_container, lv_pct(48), lv_pct(96));
    lv_obj_set_size(coat_time_container, lv_pct(48), lv_pct(96));

    lv_obj_set_style_pad_all(container, 0, 0);
    lv_obj_set_style_pad_all(numberstacks_container, 0, 0);
    lv_obj_set_style_pad_all(rpm_container, 0, 0);
    lv_obj_set_style_pad_all(coat_time_container, 0, 0);

    /* -------- Run/Stop button -------- */
    lv_obj_t * btn_run_stop = lv_button_create(container);
    lv_obj_t * lbl_run_stop = lv_label_create(btn_run_stop);

    run_stop_label = lbl_run_stop;

    lv_label_set_text(lbl_run_stop, "Run");

    lv_obj_center(lbl_run_stop);

    lv_obj_add_event_cb(btn_run_stop, run_stop_cb, LV_EVENT_CLICKED, NULL);

    lv_obj_set_size(btn_run_stop, lv_pct(96), lv_pct(18));
    lv_obj_set_style_pad_all(btn_run_stop, 0, 0);
}