/***************************************************************************
 *             __________               __   ___.
 *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
 *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
 *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
 *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
 *                     \/            \/     \/    \/            \/
 *
 * Copyright (C) 2021-2022 Aidan MacDonald
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
 * KIND, either express or implied.
 *
 ****************************************************************************/

#include "x1000bootloader.h"
#include "system.h"
#include "kernel.h"
#include "lcd.h"
#include "backlight.h"
#include "backlight-target.h"
#include "font.h"
#include "button.h"
#include "version.h"
#include <string.h>
#include <stdarg.h>
#include <stdio.h>

static bool lcd_inited = false;
extern bool is_usb_connected;
extern intptr_t usb_connection_seqnum;

void clearscreen(void)
{
    init_lcd();
    lcd_clear_display();
    putversion();
}

void putversion(void)
{
    int x = (LCD_WIDTH - SYSFONT_WIDTH*strlen(rbversion)) / 2;
    int y = LCD_HEIGHT - SYSFONT_HEIGHT;
    lcd_putsxy(x, y, rbversion);
}

void putcenter_y(int y, const char* msg)
{
    int x = (LCD_WIDTH - SYSFONT_WIDTH*strlen(msg)) / 2;
    lcd_putsxy(x, y, msg);
}

static void get_splash_size(const char* str, int* width, int* height)
{
    *width = 0;
    *height = 0;

    while(str) {
        const char* np = strchr(str, '\n');
        int len;
        if(np) {
            len = np - str;
            np++;
        } else {
            len = strlen(str);
        }

        *width = MAX(len, *width);
        *height += 1;
        str = np;
    }
}

void splashf(long delay, const char* msg, ...)
{
    static char buf[512];

    va_list ap;
    va_start(ap, msg);
    int len = vsnprintf(buf, sizeof(buf), msg, ap);
    va_end(ap);

    if(len < 0)
        return;

    int xpos, ypos;
    int width, height;
    get_splash_size(buf, &width, &height);

    width *= SYSFONT_WIDTH;
    height *= SYSFONT_HEIGHT;
    xpos = (LCD_WIDTH - width) / 2;
    ypos = (LCD_HEIGHT - height) / 2;

    const int padding = 10;
    clearscreen();
    lcd_drawrect(xpos-padding, ypos-padding,
                 width+2*padding, height+2*padding);

    char* str = buf;
    do {
        char* np = strchr(str, '\n');
        if(np) {
            *np = '\0';
            np++;
        }

        lcd_putsxyf(xpos, ypos, "%s", str);
        ypos += SYSFONT_HEIGHT;
        str = np;
    } while(str);

    lcd_update();

    if(delay == TIMEOUT_BLOCK) {
        while(get_button(TIMEOUT_BLOCK) != BL_QUIT);
    } else if(delay > 0) {
        long end_tick = current_tick + delay;
        do {
            get_button(end_tick - current_tick);
        } while(current_tick < end_tick);
    }
}

int get_button(int timeout)
{
    int btn = button_get_w_tmo(timeout);
    switch(btn) {
    case SYS_USB_CONNECTED:
    case SYS_USB_DISCONNECTED:
        is_usb_connected = (btn == SYS_USB_CONNECTED);
        usb_connection_seqnum = button_get_data();
        break;
#ifdef HAVE_SCREENDUMP
    case BL_SCREENSHOT:
        screenshot();
        break;
#endif
    }

    return btn;
}

void init_lcd(void)
{
    if(lcd_inited)
        return;

    lcd_init();
    font_init();
    lcd_setfont(FONT_SYSFIXED);

    /* Clear screen before turning backlight on, otherwise we might
     * display random garbage on the screen */
    lcd_clear_display();
    lcd_update();

    backlight_init();

    lcd_inited = true;
}

void gui_shutdown(void)
{
    if(!lcd_inited)
        return;

    backlight_hw_off();
}

void gui_list_init(struct bl_list* list, struct viewport* vp)
{
    list->vp = vp;
    list->num_items = 0;
    list->selected_item = 0;
    list->top_item = 0;
    list->item_height = SYSFONT_HEIGHT;
    list->draw_item = NULL;
}

void gui_list_draw(struct bl_list* list)
{
    struct bl_listitem item = {
        .list = list,
        .x = 0, .y = 0,
        .width = list->vp->width,
        .height = list->item_height,
    };

    struct viewport* old_vp = lcd_set_viewport(list->vp);
    lcd_clear_viewport();

    int items_on_screen = list->vp->height / list->item_height;
    for(int i = 0; i < items_on_screen; ++i) {
        item.index = list->top_item + i;
        if(item.index >= list->num_items)
            break;

        list->draw_item(&item);

        item.y += item.height;
    }

    lcd_set_viewport(old_vp);
}

void gui_list_select(struct bl_list* list, int item_index)
{
    /* clamp the selection */
    list->selected_item = item_index;

    if(list->selected_item < 0)
        list->selected_item = 0;
    else if(list->selected_item >= list->num_items)
        list->selected_item = list->num_items - 1;

    /* handle scrolling the list view */
    int items_on_screen = list->vp->height / list->item_height;
    int bottom_item = list->top_item + items_on_screen;

    if(list->selected_item < list->top_item) {
        list->top_item = list->selected_item;
    } else if(list->selected_item >= bottom_item) {
        list->top_item = list->selected_item - items_on_screen + 1;
        if(list->top_item < 0)
            list->top_item = 0;
    }
}

void gui_list_scroll(struct bl_list* list, int delta)
{
    gui_list_select(list, list->selected_item + delta);
}