#include "flipper_v2.h"
#include <cli/cli.h>
#include <gui/gui.h>
#include "menu/menu.h"
#include "menu/menu_item.h"
#include "applications.h"
#include <assets_icons.h>

typedef struct {
    FuriApp* handler;
    Widget* widget;
    const FlipperStartupApp* current_app;
} AppLoaderState;

typedef struct {
    AppLoaderState* state;
    const FlipperStartupApp* app;
} AppLoaderContext;

// TODO add mutex for contex

static void render_callback(CanvasApi* canvas, void* _ctx) {
    AppLoaderState* ctx = (AppLoaderState*)_ctx;

    canvas->clear(canvas);
    canvas->set_color(canvas, ColorBlack);
    canvas->set_font(canvas, FontPrimary);
    canvas->draw_str(canvas, 2, 32, ctx->current_app->name);

    canvas->set_font(canvas, FontSecondary);
    canvas->draw_str(canvas, 2, 44, "press back to exit");
}

static void input_callback(InputEvent* input_event, void* _ctx) {
    AppLoaderState* ctx = (AppLoaderState*)_ctx;

    if(input_event->state && input_event->input == InputBack) {
        furiac_kill(ctx->handler);
        widget_enabled_set(ctx->widget, false);
    }
}

static void handle_menu(void* _ctx) {
    AppLoaderContext* ctx = (AppLoaderContext*)_ctx;

    if(ctx->app->app == NULL) return;

    widget_enabled_set(ctx->state->widget, true);

    // TODO how to call this?
    // furiac_wait_libs(&FLIPPER_STARTUP[i].libs);

    ctx->state->current_app = ctx->app;
    ctx->state->handler = furiac_start(ctx->app->app, ctx->app->name, NULL);
}

static void handle_cli(string_t args, void* _ctx) {
    AppLoaderContext* ctx = (AppLoaderContext*)_ctx;

    if(ctx->app->app == NULL) return;

    cli_print("Starting furi application\r\n");

    ctx->state->current_app = ctx->app;
    ctx->state->handler = furiac_start(ctx->app->app, ctx->app->name, NULL);

    cli_print("Press any key to kill application");

    char c;
    cli_read(&c, 1);
    furiac_kill(ctx->state->handler);
}

void app_loader(void* p) {
    osThreadId_t self_id = osThreadGetId();
    furi_check(self_id);

    AppLoaderState state;
    state.handler = NULL;

    state.widget = widget_alloc();
    widget_enabled_set(state.widget, false);
    widget_draw_callback_set(state.widget, render_callback, &state);
    widget_input_callback_set(state.widget, input_callback, &state);

    ValueMutex* menu_mutex = furi_open("menu");
    if(menu_mutex == NULL) {
        printf("menu is not available\n");
        furiac_exit(NULL);
    }

    Cli* cli = furi_open("cli");

    // Open GUI and register widget
    GuiApi* gui = furi_open("gui");
    if(gui == NULL) {
        printf("gui is not available\n");
        furiac_exit(NULL);
    }
    gui->add_widget(gui, state.widget, GuiLayerFullscreen);

    // FURI startup
    const size_t flipper_app_count = sizeof(FLIPPER_APPS) / sizeof(FLIPPER_APPS[0]);
    const size_t flipper_plugins_count = sizeof(FLIPPER_PLUGINS) / sizeof(FLIPPER_PLUGINS[0]);

    // Main menu
    with_value_mutex(
        menu_mutex, (Menu * menu) {
            for(size_t i = 0; i < flipper_app_count; i++) {
                AppLoaderContext* ctx = furi_alloc(sizeof(AppLoaderContext));
                ctx->state = &state;
                ctx->app = &FLIPPER_APPS[i];

                menu_item_add(
                    menu,
                    menu_item_alloc_function(
                        FLIPPER_APPS[i].name,
                        assets_icons_get(FLIPPER_APPS[i].icon),
                        handle_menu,
                        ctx));

                // Add cli command
                if(cli) {
                    string_t cli_name;
                    string_init_set_str(cli_name, "app_");
                    string_cat_str(cli_name, FLIPPER_APPS[i].name);
                    cli_add_command(cli, string_get_cstr(cli_name), handle_cli, ctx);
                    string_clear(cli_name);
                }
            }
        });

    with_value_mutex(
        menu_mutex, (Menu * menu) {
            menu_item_add(
                menu, menu_item_alloc_function("U2F", assets_icons_get(A_U2F_14), NULL, NULL));
            menu_item_add(
                menu,
                menu_item_alloc_function(
                    "File Manager", assets_icons_get(A_FileManager_14), NULL, NULL));
            menu_item_add(
                menu, menu_item_alloc_function("Games", assets_icons_get(A_Games_14), NULL, NULL));
            menu_item_add(
                menu,
                menu_item_alloc_function("Passport", assets_icons_get(A_Passport_14), NULL, NULL));
            menu_item_add(
                menu,
                menu_item_alloc_function("Settings", assets_icons_get(A_Settings_14), NULL, NULL));
        });

    // plugins
    with_value_mutex(
        menu_mutex, (Menu * menu) {
            MenuItem* menu_plugins =
                menu_item_alloc_menu("Plugins", assets_icons_get(A_Plugins_14));

            for(size_t i = 0; i < flipper_plugins_count; i++) {
                AppLoaderContext* ctx = furi_alloc(sizeof(AppLoaderContext));
                ctx->state = &state;
                ctx->app = &FLIPPER_PLUGINS[i];

                menu_item_subitem_add(
                    menu_plugins,
                    menu_item_alloc_function(
                        FLIPPER_PLUGINS[i].name,
                        assets_icons_get(FLIPPER_PLUGINS[i].icon),
                        handle_menu,
                        ctx));

                // Add cli command
                if(cli) {
                    string_t cli_name;
                    string_init_set_str(cli_name, "app_");
                    string_cat_str(cli_name, FLIPPER_PLUGINS[i].name);
                    cli_add_command(cli, string_get_cstr(cli_name), handle_cli, ctx);
                    string_clear(cli_name);
                }
            }

            menu_item_add(menu, menu_plugins);
        });

    printf("[app loader] start\n");

    osThreadSuspend(self_id);
}