#include "ui/service_menu.h"

#include <algorithm>  // Para max

#include "audio.h"          // Para Audio
#include "lang.h"           // Para getText, getCodeFromName, getNameFromCode
#include "menu_option.h"    // Para MenuOption, BoolOption, ActionOption, IntOption, FolderOption, ListOption
#include "menu_renderer.h"  // Para MenuRenderer
#include "options.h"        // Para PendingChanges, VideoOptions, pending_changes, video, AudioOptions, SettingsOptions, audio, checkPendingChanges, settings, WindowOptions, getDifficultyCodeFromName, getDifficultyNameFromCode, window, MusicOptions, SoundOptions
#include "param.h"          // Para Param, param, ParamGame, ParamServiceMenu
#include "resource.h"       // Para Resource
#include "screen.h"         // Para Screen
#include "section.hpp"      // Para Name, name, Options, options
#include "ui/ui_message.h"  // Para UIMessage
#include "utils.h"          // Para Zone

// Singleton
ServiceMenu *ServiceMenu::instance = nullptr;
void ServiceMenu::init() { ServiceMenu::instance = new ServiceMenu(); }
void ServiceMenu::destroy() { delete ServiceMenu::instance; }
auto ServiceMenu::get() -> ServiceMenu * { return ServiceMenu::instance; }

// Constructor
ServiceMenu::ServiceMenu()
    : current_settings_group_(SettingsGroup::MAIN),
      previous_settings_group_(current_settings_group_) {
    auto element_text = Resource::get()->getText("04b_25_flat");
    auto title_text = Resource::get()->getText("04b_25_flat_2x");

    renderer_ = std::make_unique<MenuRenderer>(this, element_text, title_text);
    restart_message_ui_ = std::make_unique<UIMessage>(element_text, Lang::getText("[SERVICE_MENU] NEED_RESTART_MESSAGE"), param.service_menu.title_color);

    reset();
}

void ServiceMenu::toggle() {
    playBackSound();
    enabled_ = !enabled_;
    if (!enabled_) {
        reset();
    }
}

void ServiceMenu::render() {
    if (!enabled_) {
        return;
    }
    renderer_->render(this);

    // El mensaje de reinicio se dibuja por encima, así que lo gestionamos aquí
    const float MSG_X = param.game.game_area.center_x;
    const float MSG_Y = renderer_->getRect().y + 39.0F;
    restart_message_ui_->setPosition(MSG_X, MSG_Y);
    restart_message_ui_->render();
}

void ServiceMenu::update() {
    if (!enabled_) {
        return;
    }

    renderer_->update(this);

    bool now_pending = Options::pending_changes.has_pending_changes;
    if (now_pending != last_pending_changes_) {
        now_pending ? restart_message_ui_->show() : restart_message_ui_->hide();
        last_pending_changes_ = now_pending;
    }
    restart_message_ui_->update();
}

void ServiceMenu::reset() {
    selected_ = 0;
    main_menu_selected_ = 0;
    current_settings_group_ = SettingsGroup::MAIN;
    previous_settings_group_ = SettingsGroup::MAIN;
    initializeOptions();
    updateMenu();
    renderer_->setLayout(this);  // Notifica al renderer para que calcule el layout inicial
}

// --- Lógica de Navegación ---
void ServiceMenu::setSelectorUp() {
    if (display_options_.empty()) {
        return;
    }
    selected_ = (selected_ > 0) ? selected_ - 1 : display_options_.size() - 1;
    playMoveSound();
}

void ServiceMenu::setSelectorDown() {
    if (display_options_.empty()) {
        return;
    }
    selected_ = (selected_ + 1) % display_options_.size();
    playMoveSound();
}

void ServiceMenu::adjustOption(bool adjust_up) {
    if (display_options_.empty()) {
        return;
    }
    auto &selected_option = display_options_.at(selected_);
    if (selected_option->getBehavior() == MenuOption::Behavior::ADJUST) {
        selected_option->adjustValue(adjust_up);
        applySettings();
        playAdjustSound();
    }
}

void ServiceMenu::selectOption() {
    if (display_options_.empty()) {
        return;
    }
    if (current_settings_group_ == SettingsGroup::MAIN) {
        main_menu_selected_ = selected_;
    }

    auto *selected_option = display_options_.at(selected_);
    if (selected_option == nullptr) {
        // This shouldn't happen in normal operation, but protects against null pointer
        return;
    }

    if (auto *folder = dynamic_cast<FolderOption *>(selected_option)) {
        previous_settings_group_ = current_settings_group_;
        current_settings_group_ = folder->getTargetGroup();
        selected_ = 0;
        updateMenu();
    } else if (selected_option->getBehavior() == MenuOption::Behavior::SELECT) {
        selected_option->executeAction();
    }
    playSelectSound();
}

void ServiceMenu::moveBack() {
    playBackSound();
    if (current_settings_group_ == SettingsGroup::MAIN) {
        enabled_ = false;
        return;
    }
    current_settings_group_ = previous_settings_group_;
    selected_ = (current_settings_group_ == SettingsGroup::MAIN) ? main_menu_selected_ : 0;
    updateMenu();
}

// --- Lógica Interna ---

void ServiceMenu::updateDisplayOptions() {
    display_options_.clear();
    for (auto &option : options_) {
        if (option->getGroup() == current_settings_group_ && !option->isHidden()) {
            display_options_.push_back(option.get());
        }
    }
    updateOptionPairs();
}

void ServiceMenu::updateOptionPairs() {
    option_pairs_.clear();
    for (const auto &option : display_options_) {
        option_pairs_.emplace_back(option->getCaption(), option->getValueAsString());
    }
}

void ServiceMenu::updateMenu() {
    title_ = settingsGroupToString(current_settings_group_);
    adjustListValues();

    // Actualiza las opciones visibles
    updateDisplayOptions();

    // Notifica al renderer del cambio de layout
    renderer_->onLayoutChanged(this);
}

void ServiceMenu::applySettings() {
    if (current_settings_group_ == SettingsGroup::VIDEO) {
        applyVideoSettings();
    }
    if (current_settings_group_ == SettingsGroup::AUDIO) {
        applyAudioSettings();
    }
    if (current_settings_group_ == SettingsGroup::SETTINGS) {
        applySettingsSettings();
    }

    // Actualiza los valores de las opciones
    updateOptionPairs();
}

void ServiceMenu::applyVideoSettings() {
    Screen::get()->applySettings();
    setHiddenOptions();
}

void ServiceMenu::applyAudioSettings() {
    Audio::get()->applySettings();
}

void ServiceMenu::applySettingsSettings() {
    setHiddenOptions();
}

auto ServiceMenu::getOptionByCaption(const std::string &caption) const -> MenuOption * {
    for (const auto &option : options_) {
        if (option->getCaption() == caption) {
            return option.get();
        }
    }
    return nullptr;
}

// --- Getters y otros ---

auto ServiceMenu::getCurrentGroupAlignment() const -> ServiceMenu::GroupAlignment {
    switch (current_settings_group_) {
        case SettingsGroup::VIDEO:
        case SettingsGroup::AUDIO:
        case SettingsGroup::SETTINGS:
            return GroupAlignment::LEFT;
        default:
            return GroupAlignment::CENTERED;
    }
}

auto ServiceMenu::countOptionsInGroup(SettingsGroup group) const -> size_t {
    size_t count = 0;
    for (const auto &option : options_) {
        if (option->getGroup() == group && !option->isHidden()) {
            count++;
        }
    }
    return count;
}

// Inicializa todas las opciones del menú
void ServiceMenu::initializeOptions() {
    options_.clear();

    // --- Video ---
    options_.push_back(std::make_unique<BoolOption>(Lang::getText("[SERVICE_MENU] FULLSCREEN"), SettingsGroup::VIDEO, &Options::video.fullscreen));
    options_.push_back(std::make_unique<IntOption>(Lang::getText("[SERVICE_MENU] WINDOW_SIZE"), SettingsGroup::VIDEO, &Options::window.size, 1, Options::window.max_size, 1));
    options_.push_back(std::make_unique<BoolOption>(Lang::getText("[SERVICE_MENU] SHADERS"), SettingsGroup::VIDEO, &Options::video.shaders));
    options_.push_back(std::make_unique<BoolOption>(Lang::getText("[SERVICE_MENU] VSYNC"), SettingsGroup::VIDEO, &Options::video.v_sync));
    options_.push_back(std::make_unique<BoolOption>(Lang::getText("[SERVICE_MENU] INTEGER_SCALE"), SettingsGroup::VIDEO, &Options::video.integer_scale));

    // --- Audio ---
    options_.push_back(std::make_unique<BoolOption>(Lang::getText("[SERVICE_MENU] AUDIO"), SettingsGroup::AUDIO, &Options::audio.enabled));
    options_.push_back(std::make_unique<IntOption>(Lang::getText("[SERVICE_MENU] MAIN_VOLUME"), SettingsGroup::AUDIO, &Options::audio.volume, 0, 100, 5));
    options_.push_back(std::make_unique<IntOption>(Lang::getText("[SERVICE_MENU] MUSIC_VOLUME"), SettingsGroup::AUDIO, &Options::audio.music.volume, 0, 100, 5));
    options_.push_back(std::make_unique<IntOption>(Lang::getText("[SERVICE_MENU] SFX_VOLUME"), SettingsGroup::AUDIO, &Options::audio.sound.volume, 0, 100, 5));

    // --- Settings ---
    options_.push_back(std::make_unique<BoolOption>(Lang::getText("[SERVICE_MENU] AUTOFIRE"), SettingsGroup::SETTINGS, &Options::settings.autofire));
    options_.push_back(std::make_unique<ListOption>(Lang::getText("[SERVICE_MENU] LANGUAGE"), SettingsGroup::SETTINGS, std::vector<std::string>{Lang::getText("[SERVICE_MENU] LANG_ES"), Lang::getText("[SERVICE_MENU] LANG_BA"), Lang::getText("[SERVICE_MENU] LANG_EN")}, []() { return Lang::getNameFromCode(Options::pending_changes.new_language); }, [](const std::string &val) { Options::pending_changes.new_language = Lang::getCodeFromName(val); Options::checkPendingChanges(); }));
    options_.push_back(std::make_unique<ListOption>(Lang::getText("[SERVICE_MENU] DIFFICULTY"), SettingsGroup::SETTINGS, std::vector<std::string>{Lang::getText("[SERVICE_MENU] EASY"), Lang::getText("[SERVICE_MENU] NORMAL"), Lang::getText("[SERVICE_MENU] HARD")}, []() { return Options::getDifficultyNameFromCode(Options::pending_changes.new_difficulty); }, [](const std::string &val) { Options::pending_changes.new_difficulty = Options::getDifficultyCodeFromName(val); Options::checkPendingChanges(); }));
    options_.push_back(std::make_unique<BoolOption>(Lang::getText("[SERVICE_MENU] ENABLE_SHUTDOWN"), SettingsGroup::SETTINGS, &Options::settings.shutdown_enabled));

    // --- System ---
    options_.push_back(std::make_unique<ActionOption>(Lang::getText("[SERVICE_MENU] RESET"), SettingsGroup::SYSTEM, [this]() { Section::name = Section::Name::RESET; toggle(); }));
    options_.push_back(std::make_unique<ActionOption>(Lang::getText("[SERVICE_MENU] QUIT"), SettingsGroup::SYSTEM, []() { Section::name = Section::Name::QUIT; Section::options = Section::Options::NONE; }));
    options_.push_back(std::make_unique<ActionOption>(Lang::getText("[SERVICE_MENU] SHUTDOWN"), SettingsGroup::SYSTEM, []() { Section::name = Section::Name::QUIT; Section::options = Section::Options::SHUTDOWN; }, !Options::settings.shutdown_enabled));

    // --- Main Menu ---
    options_.push_back(std::make_unique<FolderOption>(Lang::getText("[SERVICE_MENU] VIDEO"), SettingsGroup::MAIN, SettingsGroup::VIDEO));
    options_.push_back(std::make_unique<FolderOption>(Lang::getText("[SERVICE_MENU] AUDIO"), SettingsGroup::MAIN, SettingsGroup::AUDIO));
    options_.push_back(std::make_unique<FolderOption>(Lang::getText("[SERVICE_MENU] SETTINGS"), SettingsGroup::MAIN, SettingsGroup::SETTINGS));
    options_.push_back(std::make_unique<FolderOption>(Lang::getText("[SERVICE_MENU] SYSTEM"), SettingsGroup::MAIN, SettingsGroup::SYSTEM));

    setHiddenOptions();
}

// Sincroniza los valores de las opciones tipo lista
void ServiceMenu::adjustListValues() {
    for (auto &option : options_) {
        if (auto *list_option = dynamic_cast<ListOption *>(option.get())) {
            list_option->sync();
        }
    }
}

// Reproduce el sonido de navegación del menú
void ServiceMenu::playAdjustSound() { Audio::get()->playSound("service_menu_adjust.wav", Audio::Group::INTERFACE); }
void ServiceMenu::playMoveSound() { Audio::get()->playSound("service_menu_move.wav", Audio::Group::INTERFACE); }
void ServiceMenu::playSelectSound() { Audio::get()->playSound("service_menu_select.wav", Audio::Group::INTERFACE); }
void ServiceMenu::playBackSound() { Audio::get()->playSound("service_menu_select.wav", Audio::Group::INTERFACE); }

// Devuelve el nombre del grupo como string para el título
auto ServiceMenu::settingsGroupToString(SettingsGroup group) -> std::string {
    switch (group) {
        case SettingsGroup::MAIN:
            return Lang::getText("[SERVICE_MENU] TITLE");
        case SettingsGroup::VIDEO:
            return Lang::getText("[SERVICE_MENU] VIDEO");
        case SettingsGroup::AUDIO:
            return Lang::getText("[SERVICE_MENU] AUDIO");
        case SettingsGroup::SETTINGS:
            return Lang::getText("[SERVICE_MENU] SETTINGS");
        case SettingsGroup::SYSTEM:
            return Lang::getText("[SERVICE_MENU] SYSTEM");
        default:
            return Lang::getText("[SERVICE_MENU] TITLE");
    }
}

// Establece el estado de oculto de ciertas opciones
void ServiceMenu::setHiddenOptions() {
    {
        auto *option = getOptionByCaption(Lang::getText("[SERVICE_MENU] WINDOW_SIZE"));
        if (option != nullptr) {
            option->setHidden(Options::video.fullscreen);
        }
    }

    {
        auto *option = getOptionByCaption(Lang::getText("[SERVICE_MENU] SHUTDOWN"));
        if (option != nullptr) {
            option->setHidden(!Options::settings.shutdown_enabled);
        }
    }

    updateMenu();  // El menú debe refrescarse si algo se oculta
}