#include "SDL2/SDL_keycode.h"
#include "editor_scenes.hpp"
#include "../../sdlpp/sdlpp.hpp"
#include "../../sdlpp/sdlpp_mouse.hpp"
#include "../global_vars.hpp"
#include "../objectids.hpp"
#include "../editor_visitor.hpp"

bool __update_scenes_yesno_dialog = false;
bool __quit_scenes_yesno_dialog = false;
bool __started_yesno_dialog = false;
uint64_t __cur_button_index_yesno_dialog = -1;
uint64_t __cur_button_index_yesno_dialog_down = -1;
std::vector<std::shared_ptr<Button>> __buttons_yesno_dialog{};
std::shared_ptr<SDLPP::RectangleRender> __mouse_yesno_dialog{};

std::function<void(bool)> __yesno_dialog_finalizer;

void __quitYesNoDialog(bool result = false) {
    __yesno_dialog_finalizer(result);
    __quit_scenes_yesno_dialog = true;
}

void __updateSelectedButton_YesNoDialog(uint64_t new_index) {
    if (new_index != __cur_button_index_yesno_dialog &&
        new_index != (uint64_t)-1) {
        __buttons_yesno_dialog[new_index]->setHighlight();

        if (__cur_button_index_yesno_dialog != (uint64_t)-1) {
            __buttons_yesno_dialog[__cur_button_index_yesno_dialog]
                ->unsetHighlight();
        }
        __cur_button_index_yesno_dialog = new_index;
    }
}

void __handleKeyUp_YesNoDialog(SDL_Keycode key, SDLPP::Scene & /*UNUSED*/) {
    switch (key) {
    case SDLK_ESCAPE:
        __quitYesNoDialog();
        break;
    case SDLK_LEFT:
    case SDLK_a:
        __updateSelectedButton_YesNoDialog(0);
        break;
    case SDLK_RIGHT:
    case SDLK_d:
        __updateSelectedButton_YesNoDialog(1);
        break;
    case SDLK_RETURN:
        __quitYesNoDialog(__cur_button_index_yesno_dialog == 0);
        break;
    default:
        break;
    }
}

void __returnYesCallback_YesNoDialog(void * /*UNUSED*/, Button * /*UNUSED*/) {
    __quitYesNoDialog(true);
}

void __returnNoCallback_YesNoDialog(void * /*UNUSED*/, Button * /*UNUSED*/) {
    __quitYesNoDialog(false);
}

std::shared_ptr<SDLPP::Scene>
createSceneYesNoDialog(std::shared_ptr<SDLPP::Renderer> &renderer,
                       const std::string &text) {
    auto scene = std::make_shared<SDLPP::Scene>(renderer);
    auto bg = std::make_shared<SDLPP::RectangleRender>(0, 0, 10, 10, renderer,
                                                       "#00000088", true);
    bg->setPermanent();
    bg->setId(1);
    scene->addObject(bg);
    __mouse_yesno_dialog =
        std::make_shared<SDLPP::RectangleRender>(0.01, 0.01, 0, 0, renderer);
    __mouse_yesno_dialog->setMinWidth(1);
    __mouse_yesno_dialog->setMinHeight(1);
    __mouse_yesno_dialog->setAlignment(SDLPP::OBJ_CENTER, SDLPP::OBJ_CENTER);
    __mouse_yesno_dialog->setId(EDITOR_MOUSE_ID);
    __mouse_yesno_dialog->setColiderColor("#00FF00");
    __mouse_yesno_dialog->addCollision(SDLPP::RectColider({ 0, 0 }, { 1, 1 }));
    scene->addObject(__mouse_yesno_dialog);

    ButtonConfig default_button_theme{};
    default_button_theme.bg_color = "#FFFFFF88";
    default_button_theme.bg_color_highlight = "#FFFFFFBB";
    default_button_theme.bg_color_disabled = "#AAAAAA88";
    default_button_theme.font_color = "#000000";
    default_button_theme.font_color_highlight = "#000000";
    default_button_theme.font_color_disabled = "#555555";
    default_button_theme.font_outline_color = "#FFFFFF88";
    default_button_theme.font_outline_color_highlight = "#FFFFFFAA";
    default_button_theme.font_outline_color_disabled = "#787878";
    default_button_theme.outline = 0.1;
    // buttons
    __buttons_yesno_dialog.emplace_back(std::make_shared<Button>(
        0.1, 0.5, 0.4, 0.2, renderer, "YES", default_button_theme,
        __returnYesCallback_YesNoDialog, nullptr));
    __buttons_yesno_dialog.back()->setAlignment(SDLPP::OBJ_CENTER,
                                                SDLPP::OBJ_CENTER);
    __buttons_yesno_dialog.back()->setPermanent();
    __buttons_yesno_dialog.back()->setButtonIndex(
        __buttons_yesno_dialog.size() - 1);
    __buttons_yesno_dialog.emplace_back(std::make_shared<Button>(
        0.55, 0.5, 0.4, 0.2, renderer, "NO", default_button_theme,
        __returnNoCallback_YesNoDialog, nullptr));
    __buttons_yesno_dialog.back()->setAlignment(SDLPP::OBJ_CENTER,
                                                SDLPP::OBJ_CENTER);
    __buttons_yesno_dialog.back()->setPermanent();
    __buttons_yesno_dialog.back()->setButtonIndex(
        __buttons_yesno_dialog.size() - 1);
    for (auto &button : __buttons_yesno_dialog) {
        scene->addObject(button);
    }
    auto font_config = std::make_shared<SDLPP::FontConfiguration>(
        g_text_config->getFont(), "#000000", "#FFFFFFAA", 0.10);
    auto dialog_text = std::make_shared<SDLPP::TextRenderer>(
        0.1, 0.15, 0.8, 0.3, renderer, text, font_config, SDLPP_TEXT_CENTER);
    dialog_text->setAlignment(SDLPP::OBJ_CENTER, SDLPP::OBJ_CENTER);
    dialog_text->setPermanent();
    scene->addObject(dialog_text);
    return scene;
}

void __resetGlobals_YesNoDialog() {
    __update_scenes_yesno_dialog = false;
    __quit_scenes_yesno_dialog = false;
    __started_yesno_dialog = false;
    __cur_button_index_yesno_dialog = -1;
    __cur_button_index_yesno_dialog_down = -1;
    __buttons_yesno_dialog.clear();
    __mouse_yesno_dialog.reset();
}

void __additionalRender_YesNoDialog(
    std::shared_ptr<SDLPP::Scene> & /*UNUSED*/) {
    if (__update_scenes_yesno_dialog) {
        for (auto &_scene : game_scenes) {
            _scene.scene->updateSizeAndPosition();
        }
        if (__started_yesno_dialog) {
            __update_scenes_yesno_dialog = false;
        } else {
            __started_yesno_dialog = true;
        }
    }
    if (__quit_scenes_yesno_dialog) {
        game_scenes.pop_back();
        __resetGlobals_YesNoDialog();
    }
}

void __getMousePositionFlags_YesNoDialog(SDLPP::Scene &scene) {
    auto mouse = scene.getObjects({ EDITOR_MOUSE_ID })[0];
    // move mouse colider to mouse position
    mouse->setPos(SDLPP::Mouse::getMousePositionDouble(
        scene.getRenderer(), SDLPP::OBJ_CENTER, SDLPP::OBJ_CENTER));

    MouseVisitor visitor;
    scene.visitCollisions(*mouse, visitor);
    __updateSelectedButton_YesNoDialog(visitor.getCurButton());
}

void __pollEvents_YesNoDialog(std::shared_ptr<SDLPP::Scene> &scene) {
    SDL_Event event;
    while (SDLPP::getSDLEvent(event)) {
        switch (event.type) {
        case SDL_QUIT:
            __quitYesNoDialog();
            break;
        case SDL_KEYUP:
            __handleKeyUp_YesNoDialog(event.key.keysym.sym, *scene);
            break;
        case SDL_WINDOWEVENT:
            if (event.window.event == SDL_WINDOWEVENT_RESIZED) {
                __update_scenes_yesno_dialog = true;
            }
            break;
        case SDL_MOUSEMOTION:
            __getMousePositionFlags_YesNoDialog(*scene);
            break;
        case SDL_MOUSEBUTTONUP:
            if (__cur_button_index_yesno_dialog_down ==
                    __cur_button_index_yesno_dialog &&
                __cur_button_index_yesno_dialog != (uint64_t)-1) {
                __buttons_yesno_dialog[__cur_button_index_yesno_dialog]
                    ->performFunction();
            }
            break;
        case SDL_MOUSEBUTTONDOWN:
            // store current mouse flags in previous mouse flags
            __cur_button_index_yesno_dialog_down =
                __cur_button_index_yesno_dialog;
            break;
        default:
            break;
        }
    }
}

SceneStruct createYesNoScene(std::shared_ptr<SDLPP::Renderer> renderer,
                             const std::string &text,
                             std::function<void(bool)> finalizer) {
    __yesno_dialog_finalizer = finalizer;
    SceneStruct ret{};
    ret.scene = createSceneYesNoDialog(renderer, text);
    ret.additionalRender = __additionalRender_YesNoDialog;
    ret.doInput = __pollEvents_YesNoDialog;
    __update_scenes_yesno_dialog = true;
    __updateSelectedButton_YesNoDialog(1);
    return ret;
}