/**
 * TODO file description
 */

#include <stdbool.h>
#include "app_gui.h"
#include "app_heater.h"
#include "app_buzzer.h"
#include "app_temp.h"
#include "FreeRTOS.h"
#include "task.h"
#include "queue.h"
#include "ufb/framebuffer.h"
#include "ufb/fb_text.h"
#include "ufb/fb_7seg.h"

#define MAX_TEMP 400

/**
 * Screen callback type. The event is either INIT, PAINT, or one of the input events.
 */
typedef void (*screen_t)(GuiEvent event);

/**
 * Choice callback for the generic menu screen
 */
typedef void (*menu_callback_t)(int choice);

/**
 * Generic menu screen (must be called from a screen function with the standard signature)
 *
 * @param event - currently handled event
 * @param options - options for the menu (items to show)
 * @param cb - choice callback
 */
static void screen_menu(GuiEvent event, const char **options, menu_callback_t cb);

static struct State {
  float oven_temp;
  //float soc_temp;

  // manual mode controls
  int set_temp;
  int set_temp_wheel;
  bool heater_enabled;

  bool pushed;
  bool paint_needed;
  uint32_t last_tick_event;
  uint32_t push_tick;
  uint32_t last_read_temp_tick;
  // true if the button is still held since init (i.e. the push action should not work as "enter")
  bool initial_pushed;

  screen_t screen;
  int menu_pos;
  int menu_len;
} s_app = {};

/** Get push time (while held) */
static uint32_t push_time() {
    return s_app.pushed ? (xTaskGetTickCount() - s_app.push_tick) : 0;
}

/** Schedule paint (the screen func will be called with the PAINT event argument */
static void request_paint() {
    s_app.paint_needed = true;
}

/** Home screen */
static void screen_home(GuiEvent event);

/** Manual temperature setting screen */
static void screen_manual(GuiEvent event);

/** Menu in the manual temperature setting screen */
static void screen_manual_menu(GuiEvent event);

/** Draw the common overlay / HUD (with temperatures and heater status) */
static void draw_common_overlay();

/** Input beep (push or knob turn) */
static void input_sound_effect();

/** Switch to a different screen. Handles initial push state handling (so release
 * does not cause a "click" event).
 *
 * @param pScreen - screen to switch to
 * @param init - call the INIT event immediately after
 */
static void switch_screen(screen_t pScreen, bool init);

static char tmp[100];

/** Main loop */
void app_task_gui(void *argument)
{
    // Wait until inited
    ulTaskNotifyTake(pdTRUE, portMAX_DELAY);
    PUTS("GUI task starts\r\n");

    s_app.last_tick_event = xTaskGetTickCount();

    switch_screen(screen_home, true);

    while (1) {
        uint32_t message = GUI_EVENT_NONE;
        int32_t ticks_remain = (int32_t) pdMS_TO_TICKS(10) - (int32_t)(xTaskGetTickCount() - s_app.last_tick_event);
        if (ticks_remain < 0) {
            ticks_remain = 0;
        }
        osMessageQueueGet(guiEventQueHandle, &message, NULL, ticks_remain);

        uint32_t tickNow = xTaskGetTickCount();

        // 10ms tick event
        if (tickNow - s_app.last_tick_event > pdMS_TO_TICKS(10)) {
            s_app.screen(GUI_EVENT_TICK);
            s_app.last_tick_event = tickNow;
        }

        if (tickNow - s_app.last_read_temp_tick > pdMS_TO_TICKS(250)) {
            s_app.oven_temp = app_temp_read_oven();
            //s_app.soc_temp = app_temp_read_soc();
            request_paint();
            s_app.last_read_temp_tick = tickNow;
        }

        switch (message) {
            case GUI_EVENT_KNOB_PRESS:
                s_app.pushed = true;
                s_app.push_tick = tickNow;
                break;

            case GUI_EVENT_KNOB_RELEASE:
                s_app.pushed = false;

                if (s_app.initial_pushed) {
                    s_app.initial_pushed = false;
                    message = GUI_EVENT_NONE;
                }
                break;
        }

        if (message != GUI_EVENT_NONE) {
            s_app.screen(message);
        }

        if (s_app.paint_needed) {
            s_app.paint_needed = false;
            fb_clear();
            draw_common_overlay();
            s_app.screen(GUI_EVENT_PAINT);
            fb_blit();
        }
    }
}

/** Switch to a different screen handler.
 * If "init" is true, immediately call it with the init event. */
static void switch_screen(screen_t pScreen, bool init) {
    s_app.initial_pushed = s_app.pushed;
    s_app.screen = pScreen;
    request_paint();

    if (init) {
        pScreen(GUI_EVENT_SCREEN_INIT);
    }
}

/** Draw GUI common to all screens */
static void draw_common_overlay()
{
    SPRINTF(tmp, "%3.1f°C →%3d°C", s_app.oven_temp, s_app.set_temp);
    fb_text(3, 3, tmp, FONT_3X5, 1);

    if (s_app.heater_enabled) {
        fb_frame(0, 0, FBW, 11, 2, 1);
    }
}

/** Play input sound effect if this is an input event */
static void input_sound_effect()
{
    app_buzzer_beep();
}

// ------------- home screen ----------------

static const char* main_menu_opts[] = {
    "Manual mode",
    "Calibration",
    NULL
};

static void main_menu_cb(int opt) {
    switch (opt) {
        case 0:
            switch_screen(screen_manual, true);
            break;
        case 1:
            // TODO
            break;
    }
}

static void screen_home(GuiEvent event)
{
    if (event == GUI_EVENT_SCREEN_INIT) {
        app_heater_enable(false);
    }

    screen_menu(event, main_menu_opts, main_menu_cb);
}

// --------- manual mode screen ---------------

/** Calc set temperature from the current knob position */
static void manual_calc_set_temp()
{
    int clamped = s_app.set_temp_wheel;
    if (clamped < 0) {
        clamped = 0;
    }
    s_app.set_temp = (clamped / 2) * 5;
    if (s_app.set_temp > MAX_TEMP) {
        s_app.set_temp = MAX_TEMP;
    }

    app_heater_set_target((float) s_app.set_temp);
}

static void screen_manual(GuiEvent event)
{
    if (event == GUI_EVENT_SCREEN_INIT) {
        return;
    }

    // menu is activated by long push
    if (push_time() >= pdMS_TO_TICKS(500)) {
        switch_screen(screen_manual_menu, true);
        return;
    }

    switch (event) {
        case GUI_EVENT_PAINT:
            fb_7seg_number(4, 40, 12, 20, 2, 2,
                           1,
                           s_app.set_temp, 3, 0
            );
            break;

        case GUI_EVENT_KNOB_RELEASE:
            input_sound_effect();
            s_app.heater_enabled ^= 1;
            app_heater_enable(s_app.heater_enabled);
            request_paint();
            break;

        case GUI_EVENT_KNOB_PLUS:
            if (s_app.set_temp < MAX_TEMP) {
                input_sound_effect();
                s_app.set_temp_wheel++;
                manual_calc_set_temp();
                request_paint();
            }
            break;

        case GUI_EVENT_KNOB_MINUS:
            if (s_app.set_temp > 0) {
                input_sound_effect();
                s_app.set_temp_wheel--;
                manual_calc_set_temp();
                request_paint();
            }
            break;
    }
}

// ---------------------------

static const char* manual_menu_opts[] = {
        "Close menu",
        "Exit manual",
        NULL
};

static void manual_menu_cb(int opt) {
    switch (opt) {
        case 0:
            // Close menu
            switch_screen(screen_manual, false);
            break;

        case 1:
            s_app.heater_enabled = false;
            app_heater_enable(false);
            switch_screen(screen_home, true);
            break;
    }
}

static void screen_manual_menu(GuiEvent event)
{
    screen_menu(event, manual_menu_opts, manual_menu_cb);
}

// ------------------------

static void screen_menu(GuiEvent event, const char **options, menu_callback_t cb) {
    switch (event) {
        case GUI_EVENT_SCREEN_INIT:
            s_app.menu_pos = 0;
            s_app.menu_len = 0;
            const char **opt = options;
            while (*opt) {
                s_app.menu_len++;
                opt++;
            }
            break;

        case GUI_EVENT_PAINT:
            for (int i = 0; i < s_app.menu_len; i++) {
                fbcolor_t color = s_app.menu_pos != i;
                fbpos_t y = 27 + i * 10;
                fb_rect(0, y, 64, 10, !color);
                fb_text(1, y + 1, options[i], FONT_5X7, color);
                // ensure the text doesnt escape the screen
                fb_vline(63, y, 10, !color);
            }
            break;

        // the button is held! release is what activates the button
        case GUI_EVENT_KNOB_RELEASE:
            input_sound_effect();
            cb(s_app.menu_pos);
            break;

        case GUI_EVENT_KNOB_PLUS:
            if (s_app.menu_pos < s_app.menu_len - 1) {
                s_app.menu_pos++;
                input_sound_effect();
                request_paint();
            }
            break;

        case GUI_EVENT_KNOB_MINUS:
            if (s_app.menu_pos > 0) {
                s_app.menu_pos--;
                input_sound_effect();
                request_paint();
            }
            break;
    }
}