#include "user_interface.hpp"
#include "config.hpp"
#include "input_handler.hpp"

#include <raylib.h>

#define RAYGUI_IMPLEMENTATION
#include <raygui.h>

#ifdef TRACY
#include <tracy/Tracy.hpp>
#endif

auto user_interface::grid::update_bounds(const int _x, const int _y, const int _width, const int _height,
                                         const int _columns, const int _rows) -> void
{
    x = _x;
    y = _y;
    width = _width;
    height = _height;
    columns = _columns;
    rows = _rows;
}

auto user_interface::grid::update_bounds(const int _x, const int _y, const int _width, const int _height) -> void
{
    x = _x;
    y = _y;
    width = _width;
    height = _height;
}

auto user_interface::grid::update_bounds(const int _x, const int _y) -> void
{
    x = _x;
    y = _y;
}

auto user_interface::grid::bounds() const -> Rectangle
{
    Rectangle bounds{0, 0, static_cast<float>(columns), static_cast<float>(rows)};
    bounds.x -= padding;
    bounds.y -= padding;
    bounds.width += 2 * padding;
    bounds.height += 2 * padding;
    return bounds;
}

auto user_interface::grid::bounds(const int _x, const int _y, const int _width, const int _height) const -> Rectangle
{
    if (_x < 0 || _x + _width > columns || _y < 0 || _y + _height > rows) {
        throw std::invalid_argument("Grid bounds out of range");
    }

    const int cell_width = (width - padding) / columns;
    const int cell_height = (height - padding) / rows;

    return Rectangle(x + _x * cell_width + padding, y + _y * cell_height + padding, _width * cell_width - padding,
                     _height * cell_height - padding);
}

auto user_interface::grid::square_bounds() const -> Rectangle
{
    Rectangle bounds = square_bounds(0, 0, columns, rows);
    bounds.x -= padding;
    bounds.y -= padding;
    bounds.width += 2 * padding;
    bounds.height += 2 * padding;
    return bounds;
}

auto user_interface::grid::square_bounds(const int _x, const int _y, const int _width,
                                         const int _height) const -> Rectangle
{
    // Assumes each cell is square, so either width or height are not completely
    // filled

    if (_x < 0 || _x + _width > columns || _y < 0 || _y + _height > rows) {
        throw std::invalid_argument("Grid bounds out of range");
    }

    const int available_width = width - padding * (columns + 1);
    const int available_height = height - padding * (rows + 1);
    const int cell_size = std::min(available_width / columns, available_height / rows);

    const int grid_width = cell_size * columns + padding * (columns + 1);
    const int grid_height = cell_size * rows + padding * (rows + 1);
    const int x_offset = (width - grid_width) / 2;
    const int y_offset = (height - grid_height) / 2;

    return Rectangle(x_offset + _x * (cell_size + padding) + padding, y_offset + _y * (cell_size + padding) + padding,
                     _width * cell_size + padding * (_width - 1), _height * cell_size + padding * (_height - 1));
}

auto user_interface::init() -> void
{
    const Font font = LoadFontEx(FONT, FONT_SIZE, nullptr, 0);
    SetTextureFilter(font.texture, TEXTURE_FILTER_BILINEAR);
    GuiSetFont(font);

    default_style style = get_default_style();
    style.text_size = FONT_SIZE;
    apply_color(style, GRAY);

    set_default_style(style);
}

auto user_interface::apply_color(style& style, const Color color) -> void
{
    style.base_color_normal = ColorToInt(Fade(color, 0.8));
    style.base_color_focused = ColorToInt(Fade(color, 0.3));
    style.base_color_pressed = ColorToInt(Fade(color, 0.8));
    style.base_color_disabled = ColorToInt(Fade(color, 0.5));

    style.border_color_normal = ColorToInt(Fade(color, 1.0));
    style.border_color_focused = ColorToInt(Fade(color, 0.7));
    style.border_color_pressed = ColorToInt(Fade(color, 1.0));
    style.border_color_disabled = ColorToInt(Fade(GRAY, 0.5));

    style.text_color_normal = ColorToInt(Fade(BLACK, 1.0));
    style.text_color_focused = ColorToInt(Fade(BLACK, 1.0));
    style.text_color_pressed = ColorToInt(Fade(BLACK, 1.0));
    style.text_color_disabled = ColorToInt(Fade(BLACK, 0.5));
}

auto user_interface::apply_block_color(style& style, const Color color) -> void
{
    style.base_color_normal = ColorToInt(Fade(color, 0.5));
    style.base_color_focused = ColorToInt(Fade(color, 0.3));
    style.base_color_pressed = ColorToInt(Fade(color, 0.8));
    style.base_color_disabled = ColorToInt(Fade(color, 0.5));

    style.border_color_normal = ColorToInt(Fade(color, 1.0));
    style.border_color_focused = ColorToInt(Fade(color, 0.7));
    style.border_color_pressed = ColorToInt(Fade(color, 1.0));
    style.border_color_disabled = ColorToInt(Fade(GRAY, 0.5));
}

auto user_interface::apply_text_color(style& style, const Color color) -> void
{
    style.text_color_normal = ColorToInt(Fade(color, 1.0));
    style.text_color_focused = ColorToInt(Fade(color, 1.0));
    style.text_color_pressed = ColorToInt(Fade(color, 1.0));
    style.text_color_disabled = ColorToInt(Fade(BLACK, 0.5));
}

auto user_interface::get_default_style() -> default_style
{
    // Could've iterated over the values, but then it wouldn't be as nice to
    // access...
    return {
        {
            GuiGetStyle(DEFAULT, BORDER_COLOR_NORMAL), GuiGetStyle(DEFAULT, BASE_COLOR_NORMAL),
            GuiGetStyle(DEFAULT, TEXT_COLOR_NORMAL), GuiGetStyle(DEFAULT, BORDER_COLOR_FOCUSED),
            GuiGetStyle(DEFAULT, BASE_COLOR_FOCUSED), GuiGetStyle(DEFAULT, TEXT_COLOR_FOCUSED),
            GuiGetStyle(DEFAULT, BORDER_COLOR_PRESSED), GuiGetStyle(DEFAULT, BASE_COLOR_PRESSED),
            GuiGetStyle(DEFAULT, TEXT_COLOR_PRESSED), GuiGetStyle(DEFAULT, BORDER_COLOR_DISABLED),
            GuiGetStyle(DEFAULT, BASE_COLOR_DISABLED), GuiGetStyle(DEFAULT, TEXT_COLOR_DISABLED)
        },
        GuiGetStyle(DEFAULT, BACKGROUND_COLOR), GuiGetStyle(DEFAULT, LINE_COLOR), GuiGetStyle(DEFAULT, TEXT_SIZE),
        GuiGetStyle(DEFAULT, TEXT_SPACING), GuiGetStyle(DEFAULT, TEXT_LINE_SPACING),
        GuiGetStyle(DEFAULT, TEXT_ALIGNMENT_VERTICAL), GuiGetStyle(DEFAULT, TEXT_WRAP_MODE)
    };
}

auto user_interface::set_default_style(const default_style& style) -> void
{
    GuiSetStyle(DEFAULT, BORDER_COLOR_NORMAL, style.border_color_normal);
    GuiSetStyle(DEFAULT, BASE_COLOR_NORMAL, style.base_color_normal);
    GuiSetStyle(DEFAULT, TEXT_COLOR_NORMAL, style.text_color_normal);

    GuiSetStyle(DEFAULT, BORDER_COLOR_FOCUSED, style.border_color_focused);
    GuiSetStyle(DEFAULT, BASE_COLOR_FOCUSED, style.base_color_focused);
    GuiSetStyle(DEFAULT, TEXT_COLOR_FOCUSED, style.text_color_focused);

    GuiSetStyle(DEFAULT, BORDER_COLOR_PRESSED, style.border_color_pressed);
    GuiSetStyle(DEFAULT, BASE_COLOR_PRESSED, style.base_color_pressed);
    GuiSetStyle(DEFAULT, TEXT_COLOR_PRESSED, style.text_color_pressed);

    GuiSetStyle(DEFAULT, BORDER_COLOR_DISABLED, style.border_color_disabled);
    GuiSetStyle(DEFAULT, BASE_COLOR_DISABLED, style.base_color_disabled);
    GuiSetStyle(DEFAULT, TEXT_COLOR_DISABLED, style.text_color_disabled);

    GuiSetStyle(DEFAULT, BACKGROUND_COLOR, style.background_color);
    GuiSetStyle(DEFAULT, LINE_COLOR, style.line_color);

    GuiSetStyle(DEFAULT, TEXT_SIZE, style.text_size);
    GuiSetStyle(DEFAULT, TEXT_SPACING, style.text_spacing);
    GuiSetStyle(DEFAULT, TEXT_LINE_SPACING, style.text_line_spacing);
    GuiSetStyle(DEFAULT, TEXT_ALIGNMENT_VERTICAL, style.text_alignment_vertical);
    GuiSetStyle(DEFAULT, TEXT_WRAP_MODE, style.text_wrap_mode);
}

auto user_interface::get_component_style(const int component) -> component_style
{
    return {
        {
            GuiGetStyle(component, BORDER_COLOR_NORMAL), GuiGetStyle(component, BASE_COLOR_NORMAL),
            GuiGetStyle(component, TEXT_COLOR_NORMAL), GuiGetStyle(component, BORDER_COLOR_FOCUSED),
            GuiGetStyle(component, BASE_COLOR_FOCUSED), GuiGetStyle(component, TEXT_COLOR_FOCUSED),
            GuiGetStyle(component, BORDER_COLOR_PRESSED), GuiGetStyle(component, BASE_COLOR_PRESSED),
            GuiGetStyle(component, TEXT_COLOR_PRESSED), GuiGetStyle(component, BORDER_COLOR_DISABLED),
            GuiGetStyle(component, BASE_COLOR_DISABLED), GuiGetStyle(component, TEXT_COLOR_DISABLED)
        },
        GuiGetStyle(component, BORDER_WIDTH), GuiGetStyle(component, TEXT_PADDING),
        GuiGetStyle(component, TEXT_ALIGNMENT)
    };
}

auto user_interface::set_component_style(const int component, const component_style& style) -> void
{
    GuiSetStyle(component, BORDER_COLOR_NORMAL, style.border_color_normal);
    GuiSetStyle(component, BASE_COLOR_NORMAL, style.base_color_normal);
    GuiSetStyle(component, TEXT_COLOR_NORMAL, style.text_color_normal);

    GuiSetStyle(component, BORDER_COLOR_FOCUSED, style.border_color_focused);
    GuiSetStyle(component, BASE_COLOR_FOCUSED, style.base_color_focused);
    GuiSetStyle(component, TEXT_COLOR_FOCUSED, style.text_color_focused);

    GuiSetStyle(component, BORDER_COLOR_PRESSED, style.border_color_pressed);
    GuiSetStyle(component, BASE_COLOR_PRESSED, style.base_color_pressed);
    GuiSetStyle(component, TEXT_COLOR_PRESSED, style.text_color_pressed);

    GuiSetStyle(component, BORDER_COLOR_DISABLED, style.border_color_disabled);
    GuiSetStyle(component, BASE_COLOR_DISABLED, style.base_color_disabled);
    GuiSetStyle(component, TEXT_COLOR_DISABLED, style.text_color_disabled);

    GuiSetStyle(component, BORDER_WIDTH, style.border_width);
    GuiSetStyle(component, TEXT_PADDING, style.text_padding);
    GuiSetStyle(component, TEXT_ALIGNMENT, style.text_alignment);
}

auto user_interface::popup_bounds() -> Rectangle
{
    return Rectangle(static_cast<float>(GetScreenWidth()) / 2.0f - POPUP_WIDTH / 2.0f,
                     static_cast<float>(GetScreenHeight()) / 2.0f - POPUP_HEIGHT / 2.0f, POPUP_WIDTH, POPUP_HEIGHT);
}

auto user_interface::draw_button(const Rectangle bounds, const std::string& label, const Color color,
                                 const bool enabled, const int font_size) const -> int
{
    // Save original styling
    const default_style original_default = get_default_style();
    const component_style original_button = get_component_style(BUTTON);

    // Change styling
    default_style style_default = original_default;
    component_style style_button = original_button;
    style_default.text_size = font_size;
    apply_color(style_button, color);
    set_default_style(style_default);
    set_component_style(BUTTON, style_button);

    const int _state = GuiGetState();
    if (!enabled || window_open()) {
        GuiSetState(STATE_DISABLED);
    }
    const int pressed = GuiButton(bounds, label.data());
    if (!enabled || window_open()) {
        GuiSetState(_state);
    }

    // Restore original styling
    set_default_style(original_default);
    set_component_style(BUTTON, original_button);

    return pressed;
}

auto user_interface::draw_menu_button(const int x, const int y, const int width, const int height,
                                      const std::string& label, const Color color, const bool enabled,
                                      const int font_size) const -> int
{
    const Rectangle bounds = menu_grid.bounds(x, y, width, height);
    return draw_button(bounds, label, color, enabled, font_size);
}

auto user_interface::draw_toggle_slider(const Rectangle bounds, const std::string& off_label,
                                        const std::string& on_label, int* active, Color color, bool enabled,
                                        int font_size) const -> int
{
    // Save original styling
    const default_style original_default = get_default_style();
    const component_style original_slider = get_component_style(SLIDER);
    const component_style original_toggle = get_component_style(TOGGLE);

    // Change styling
    default_style style_default = original_default;
    component_style style_slider = original_slider;
    component_style style_toggle = original_toggle;
    style_default.text_size = font_size;
    apply_color(style_slider, color);
    apply_color(style_toggle, color);
    set_default_style(style_default);
    set_component_style(SLIDER, style_slider);
    set_component_style(TOGGLE, style_toggle);

    const int _state = GuiGetState();
    if (!enabled || window_open()) {
        GuiSetState(STATE_DISABLED);
    }
    int pressed = GuiToggleSlider(bounds, std::format("{};{}", off_label, on_label).data(), active);
    if (!enabled || window_open()) {
        GuiSetState(_state);
    }

    // Restore original styling
    set_default_style(original_default);
    set_component_style(SLIDER, original_slider);
    set_component_style(TOGGLE, original_toggle);

    return pressed;
}

auto user_interface::draw_menu_toggle_slider(const int x, const int y, const int width, const int height,
                                             const std::string& off_label, const std::string& on_label, int* active,
                                             const Color color, const bool enabled, const int font_size) const -> int
{
    const Rectangle bounds = menu_grid.bounds(x, y, width, height);
    return draw_toggle_slider(bounds, off_label, on_label, active, color, enabled, font_size);
}

auto user_interface::draw_spinner(Rectangle bounds, const std::string& label, int* value, int min, int max, Color color,
                                  bool enabled, int font_size) const -> int
{
    // Save original styling
    const default_style original_default = get_default_style();
    const component_style original_valuebox = get_component_style(VALUEBOX);
    const component_style original_button = get_component_style(BUTTON);

    // Change styling
    default_style style_default = original_default;
    component_style style_valuebox = original_valuebox;
    component_style style_button = original_button;
    style_default.text_size = font_size;
    apply_color(style_valuebox, color);
    apply_color(style_button, color);
    set_default_style(style_default);
    set_component_style(VALUEBOX, style_valuebox);
    set_component_style(BUTTON, style_button);

    const int _state = GuiGetState();
    if (!enabled || window_open()) {
        GuiSetState(STATE_DISABLED);
    }
    int pressed = GuiSpinner(bounds, "", label.data(), value, min, max, false);
    if (!enabled || window_open()) {
        GuiSetState(_state);
    }

    // Restore original styling
    set_default_style(original_default);
    set_component_style(VALUEBOX, original_valuebox);
    set_component_style(BUTTON, style_button);

    return pressed;
}

auto user_interface::draw_menu_spinner(const int x, const int y, const int width, const int height,
                                       const std::string& label, int* value, const int min, const int max,
                                       const Color color, const bool enabled, const int font_size) const -> int
{
    const Rectangle bounds = menu_grid.bounds(x, y, width, height);
    return draw_spinner(bounds, label, value, min, max, color, enabled, font_size);
}

auto user_interface::draw_label(const Rectangle bounds, const std::string& text, const Color color, const bool enabled,
                                const int font_size) const -> int
{
    // Save original styling
    const default_style original_default = get_default_style();
    const component_style original_label = get_component_style(LABEL);

    // Change styling
    default_style style_default = original_default;
    component_style style_label = original_label;
    style_default.text_size = font_size;
    apply_text_color(style_label, color);
    set_default_style(style_default);
    set_component_style(LABEL, style_label);

    const int _state = GuiGetState();
    if (!enabled || window_open()) {
        GuiSetState(STATE_DISABLED);
    }
    const int pressed = GuiLabel(bounds, text.data());
    if (!enabled || window_open()) {
        GuiSetState(_state);
    }

    // Restore original styling
    set_default_style(original_default);
    set_component_style(LABEL, original_label);

    return pressed;
}

auto user_interface::draw_board_block(const int x, const int y, const int width, const int height, const Color color,
                                      const bool enabled) const -> bool
{
    component_style s = get_component_style(BUTTON);
    apply_block_color(s, color);

    const Rectangle bounds = board_grid.square_bounds(x, y, width, height);

    const bool focused = CheckCollisionPointRec(input.mouse - Vector2(0, MENU_HEIGHT), bounds);
    const bool pressed = puzzle::block(x, y, width, height, false).covers(input.sel_x, input.sel_y);

    // Background to make faded colors work
    DrawRectangleRec(bounds, RAYWHITE);

    Color base = GetColor(s.base_color_normal);
    Color border = GetColor(s.base_color_normal);
    if (pressed) {
        base = GetColor(s.base_color_pressed);
        border = GetColor(s.base_color_pressed);
    }
    if (focused) {
        base = GetColor(s.base_color_focused);
        border = GetColor(s.base_color_focused);
    }
    if (focused && IsMouseButtonDown(MOUSE_BUTTON_LEFT)) {
        base = GetColor(s.base_color_pressed);
        border = GetColor(s.base_color_pressed);
    }
    if (!enabled) {
        base = BOARD_COLOR_RESTRICTED;
    }
    DrawRectangleRec(bounds, base);
    if (enabled) {
        DrawRectangleLinesEx(bounds, 2.0, border);
    }

    return focused && enabled;
}

auto user_interface::window_open() const -> bool
{
    return save_window || help_window || ok_message || yes_no_message;
}

auto user_interface::draw_menu_header(const Color color) const -> void
{
    int preset = static_cast<int>(state.get_current_preset());
    draw_menu_spinner(0, 0, 1, 1, "Preset: ", &preset, -1, static_cast<int>(state.get_preset_count()), color,
                      !input.editing);
    if (preset > static_cast<int>(state.get_current_preset())) {
        input.load_next_preset();
    } else if (preset < static_cast<int>(state.get_current_preset())) {
        input.load_previous_preset();
    }

    draw_menu_button(1, 0, 1, 1, std::format("{}: {}/{} Blocks",
                                             state.was_edited()
                                                 ? "Modified"
                                                 : std::format("\"{}\"", state.get_current_preset_comment().substr(2)),
                                             state.get_current_state().block_count(), puzzle::MAX_BLOCKS), color);

    int editing = input.editing;
    draw_menu_toggle_slider(2, 0, 1, 1, "Puzzle Mode (Tab)", "Edit Mode (Tab)", &editing, color);
    if (editing != input.editing) {
        input.toggle_editing();
    }
}

auto user_interface::draw_graph_info(const Color color) const -> void
{
    draw_menu_button(0, 1, 1, 1, std::format("Found {} States ({} Winning)", state.get_state_count(),
                                             state.get_target_count()), color);

    draw_menu_button(1, 1, 1, 1, std::format("Found {} Transitions", state.get_link_count()), color);

    draw_menu_button(2, 1, 1, 1, std::format("{} Moves to Nearest Solution",
                                             state.get_path_length() > 0 ? state.get_path_length() - 1 : 0), color);
}

auto user_interface::draw_graph_controls(const Color color) const -> void
{
    if (draw_menu_button(0, 2, 1, 1, "Populate Graph (G)", color)) {
        input.populate_graph();
    }

    // int mark_path = input.mark_path;
    // DrawMenuToggleSlider(2, 2, 1, 1, "Path Hidden (U)", "Path Shown (U)",
    //                      &mark_path, color);
    // if (mark_path != input.mark_path) {
    //   input.ToggleMarkPath();
    // }

    if (draw_menu_button(1, 2, 1, 1, "Clear Graph (C)", color)) {
        input.clear_graph();
    }

    int mark_solutions = input.mark_solutions;
    draw_menu_toggle_slider(2, 2, 1, 1, "Solution Hidden (I)", "Solution Shown (I)", &mark_solutions, color);
    if (mark_solutions != input.mark_solutions) {
        input.toggle_mark_solutions();
    }
    input.mark_path = input.mark_solutions;
}

auto user_interface::draw_camera_controls(const Color color) const -> void
{
    int lock_camera = input.camera_lock;
    draw_menu_toggle_slider(0, 3, 1, 1, "Free Camera (L)", "Locked Camera (L)", &lock_camera, color);
    if (lock_camera != input.camera_lock) {
        input.toggle_camera_lock();
    }

    int lock_camera_mass_center = input.camera_mass_center_lock;
    draw_menu_toggle_slider(1, 3, 1, 1, "Current Block (U)", "Graph Center (U)", &lock_camera_mass_center, color,
                            input.camera_lock);
    if (lock_camera_mass_center != input.camera_mass_center_lock) {
        input.toggle_camera_mass_center_lock();
    }

    int projection = camera.projection == CAMERA_ORTHOGRAPHIC;
    draw_menu_toggle_slider(2, 3, 1, 1, "Perspective (Alt)", "Orthographic (Alt)", &projection, color);
    if (projection != (camera.projection == CAMERA_ORTHOGRAPHIC)) {
        input.toggle_camera_projection();
    }
}

auto user_interface::draw_puzzle_controls(const Color color) const -> void
{
    auto nth = [&](const int n)
    {
        if (n == 11 || n == 12 || n == 13) {
            return "th";
        }
        if (n % 10 == 1) {
            return "st";
        }
        if (n % 10 == 2) {
            return "nd";
        }
        if (n % 10 == 3) {
            return "rd";
        }
        return "th";
    };

    const int visits = state.get_current_visits();
    draw_menu_button(0, 4, 1, 1, std::format("{} Moves ({}{} Time at this State)", state.get_total_moves(), visits,
                                             nth(visits)), color);

    if (draw_menu_button(1, 4, 1, 1, "Make Optimal Move (Space)", color, state.has_distances())) {
        input.goto_optimal_next_state();
    }

    if (draw_menu_button(2, 4, 1, 1, "Undo Last Move (Backspace)", color, state.has_history())) {
        input.goto_previous_state();
    }

    if (draw_menu_button(0, 5, 1, 1, "Go to Nearest Solution (B)", color, state.has_distances())) {
        input.goto_closest_target_state();
    }

    if (draw_menu_button(1, 5, 1, 1, "Go to Worst State (V)", color, state.has_distances())) {
        input.goto_most_distant_state();
    }

    if (draw_menu_button(2, 5, 1, 1, "Go to Starting State (R)", color,
                         state.get_current_index() != state.get_starting_index())) {
        input.goto_starting_state();
    }
}

auto user_interface::draw_edit_controls(const Color color) const -> void
{
    const puzzle& current = state.get_current_state();

    // Toggle Target Block
    if (draw_menu_button(0, 4, 1, 1, "Toggle Target Block (T)", color)) {
        input.toggle_target_block();
    }

    // Toggle Wall Block
    if (draw_menu_button(0, 5, 1, 1, "Toggle Wall Block (Y)", color)) {
        input.toggle_wall_block();
    }

    // Toggle Restricted/Free Block Movement
    int free = !current.get_restricted();
    draw_menu_toggle_slider(1, 4, 1, 1, "Restricted (F)", "Free (F)", &free, color);
    if (free != !current.get_restricted()) {
        input.toggle_restricted_movement();
    }

    // Clear Goal
    if (draw_menu_button(1, 5, 1, 1, "Clear Goal (X)", color)) {}

    // Column Count Spinner
    int columns = current.get_width();
    draw_menu_spinner(2, 4, 1, 1, "Cols: ", &columns, puzzle::MIN_WIDTH, puzzle::MAX_WIDTH, color);
    if (columns > current.get_width()) {
        input.add_board_column();
    } else if (columns < current.get_width()) {
        input.remove_board_column();
    }

    // Row Count Spinner
    int rows = current.get_height();
    draw_menu_spinner(2, 5, 1, 1, "Rows: ", &rows, puzzle::MIN_WIDTH, puzzle::MAX_WIDTH, color);
    if (rows > current.get_height()) {
        input.add_board_row();
    } else if (rows < current.get_height()) {
        input.remove_board_row();
    }
}

auto user_interface::draw_menu_footer(const Color color) -> void
{
    draw_menu_button(0, 6, 2, 1, state.get_current_state().string_repr().data(), color);

    if (draw_menu_button(2, 6, 1, 1, "Save as Preset", color)) {
        if (const std::optional<std::string>& reason = state.get_current_state().try_get_invalid_reason()) {
            message_title = "Can't Save Preset";
            message_message = std::format("Invalid Board: {}.", *reason);
            ok_message = true;
        } else {
            save_window = true;
        }
    }
}

auto user_interface::get_background_color() -> Color
{
    return GetColor(GuiGetStyle(DEFAULT, BACKGROUND_COLOR));
}

auto user_interface::help_popup() -> void {}

auto user_interface::draw_save_preset_popup() -> void
{
    if (!save_window) {
        return;
    }

    // Returns the pressed button index
    const int button = GuiTextInputBox(popup_bounds(), "Save as Preset", "Enter Preset Name", "Ok;Cancel",
                                       preset_name.data(), 255, nullptr);
    if (button == 1) {
        state.append_preset_file(preset_name.data());
    }
    if (button == 0 || button == 1 || button == 2) {
        save_window = false;
        TextCopy(preset_name.data(), "\0");
    }
}

auto user_interface::draw_ok_message_box() -> void
{
    if (!ok_message) {
        return;
    }

    const int button = GuiMessageBox(popup_bounds(), message_title.data(), message_message.data(), "Ok");
    if (button == 0 || button == 1) {
        message_title = "";
        message_message = "";
        ok_message = false;
    }
}

auto user_interface::draw_yes_no_message_box() -> void
{
    if (!yes_no_message) {
        return;
    }

    const int button = GuiMessageBox(popup_bounds(), message_title.data(), message_message.data(), "Yes;No");
    if (button == 1) {
        yes_no_handler();
    }
    if (button == 0 || button == 1 || button == 2) {
        message_title = "";
        message_message = "";
        yes_no_message = false;
    }
}

auto user_interface::draw_main_menu() -> void
{
    menu_grid.update_bounds(0, 0, GetScreenWidth(), MENU_HEIGHT);

    draw_menu_header(GRAY);
    draw_graph_info(ORANGE);
    draw_graph_controls(RED);
    draw_camera_controls(DARKGREEN);

    if (input.editing) {
        draw_edit_controls(PURPLE);
    } else {
        draw_puzzle_controls(BLUE);
    }

    draw_menu_footer(GRAY);
}

auto user_interface::draw_puzzle_board() -> void
{
    const puzzle& current = state.get_current_state();

    board_grid.update_bounds(0, MENU_HEIGHT, GetScreenWidth() / 2, GetScreenHeight() - MENU_HEIGHT, current.get_width(),
                             current.get_height());

    // Draw outer border
    const Rectangle bounds = board_grid.square_bounds();
    DrawRectangleRec(bounds, current.goal_reached() ? BOARD_COLOR_WON : BOARD_COLOR_RESTRICTED);

    // Draw inner borders
    DrawRectangle(bounds.x + BOARD_PADDING, bounds.y + BOARD_PADDING, bounds.width - 2 * BOARD_PADDING,
                  bounds.height - 2 * BOARD_PADDING,
                  current.get_restricted() ? BOARD_COLOR_RESTRICTED : BOARD_COLOR_FREE);

    // Draw target opening
    // TODO: Only draw single direction (in corner) if restricted (use target block principal
    // direction)
    const std::optional<puzzle::block> target_block = current.try_get_target_block();
    const int target_x = current.get_goal_x();
    const int target_y = current.get_goal_y();
    if (current.get_goal() && target_block) {
        auto [x, y, width, height] = board_grid.square_bounds(target_x, target_y, target_block->get_width(),
                                                              target_block->get_height());

        const Color opening_color = Fade(current.goal_reached() ? BOARD_COLOR_WON : BOARD_COLOR_RESTRICTED, 0.3);

        if (target_x == 0) {
            // Left opening
            DrawRectangle(x - BOARD_PADDING, y, BOARD_PADDING, height, RAYWHITE);
            DrawRectangle(x - BOARD_PADDING, y, BOARD_PADDING, height, opening_color);
        }
        if (target_x + target_block->get_width() == current.get_width()) {
            // Right opening
            DrawRectangle(x + width, y, BOARD_PADDING, height, RAYWHITE);
            DrawRectangle(x + width, y, BOARD_PADDING, height, opening_color);
        }
        if (target_y == 0) {
            // Top opening
            DrawRectangle(x, y - BOARD_PADDING, width, BOARD_PADDING, RAYWHITE);
            DrawRectangle(x, y - BOARD_PADDING, width, BOARD_PADDING, opening_color);
        }
        if (target_y + target_block->get_height() == current.get_height()) {
            // Bottom opening
            DrawRectangle(x, y + height, width, BOARD_PADDING, RAYWHITE);
            DrawRectangle(x, y + height, width, BOARD_PADDING, opening_color);
        }
    }

    // Draw empty cells. Also set hovered blocks
    input.hov_x = -1;
    input.hov_y = -1;
    for (int x = 0; x < board_grid.columns; ++x) {
        for (int y = 0; y < board_grid.rows; ++y) {
            DrawRectangleRec(board_grid.square_bounds(x, y, 1, 1), RAYWHITE);

            Rectangle hov_bounds = board_grid.square_bounds(x, y, 1, 1);
            hov_bounds.x -= BOARD_PADDING;
            hov_bounds.y -= BOARD_PADDING;
            hov_bounds.width += BOARD_PADDING;
            hov_bounds.height += BOARD_PADDING;
            if (CheckCollisionPointRec(GetMousePosition() - Vector2(0, MENU_HEIGHT), hov_bounds)) {
                input.hov_x = x;
                input.hov_y = y;
            }
        }
    }

    // Draw blocks
    for (const puzzle::block b : current.block_view()) {
        Color c = BLOCK_COLOR;
        if (b.get_target()) {
            c = TARGET_BLOCK_COLOR;
        } else if (b.get_immovable()) {
            c = WALL_COLOR;
        }

        const auto [x, y, w, h, t, i] = b.unpack_repr();

        draw_board_block(x, y, w, h, c, !i);
    }

    // Draw block placing
    if (input.editing && input.has_block_add_xy) {
        if (current.covers(input.block_add_x, input.block_add_y) && input.hov_x >= input.block_add_x && input.hov_y >=
            input.block_add_y) {
            bool collides = false;
            for (const puzzle::block b : current.block_view()) {
                if (b.collides(puzzle::block(input.block_add_x, input.block_add_y, input.hov_x - input.block_add_x + 1,
                                             input.hov_y - input.block_add_y + 1, false))) {
                    collides = true;
                    break;
                }
            }
            if (!collides) {
                draw_board_block(input.block_add_x, input.block_add_y, input.hov_x - input.block_add_x + 1,
                                 input.hov_y - input.block_add_y + 1, PURPLE);
            }
        }
    }

    // TODO: In edit mode
    //       - Clear Goal button doesn't work
    //       - Toggle Target Block button throws "Grid bounds out of range"
    //       - Clicking the goal to remove it throws "Grid bounds out of range"
    // Draw goal boundaries when editing
    if (input.editing && current.get_goal() && target_block) {
        DrawRectangleLinesEx(
            board_grid.square_bounds(target_x, target_y, target_block->get_width(), target_block->get_height()), 2.0,
            TARGET_BLOCK_COLOR);
    }
}

auto user_interface::draw_graph_overlay(int fps, int ups, size_t mass_count, size_t spring_count) -> void
{
    graph_overlay_grid.update_bounds(GetScreenWidth() / 2, MENU_HEIGHT);
    debug_overlay_grid.update_bounds(GetScreenWidth() / 2, GetScreenHeight() - 75);

    draw_label(graph_overlay_grid.bounds(0, 0, 1, 1), std::format("Dist: {:0>7.2f}", camera.distance), BLACK);
    draw_label(graph_overlay_grid.bounds(0, 1, 1, 1), std::format("FoV:  {:0>6.2f}", camera.fov), BLACK);
    draw_label(graph_overlay_grid.bounds(0, 2, 1, 1), std::format("FPS:  {:0>3}", fps), LIME);
    draw_label(graph_overlay_grid.bounds(0, 3, 1, 1), std::format("UPS:  {:0>3}", ups), ORANGE);

    // Debug
    draw_label(debug_overlay_grid.bounds(0, 0, 1, 1), std::format("Physics Debug:"), BLACK);
    draw_label(debug_overlay_grid.bounds(0, 1, 1, 1), std::format("Masses:  {}", mass_count), BLACK);
    draw_label(debug_overlay_grid.bounds(0, 2, 1, 1), std::format("Springs: {}", spring_count), BLACK);
}

auto user_interface::draw(const int fps, const int ups, const size_t mass_count, const size_t spring_count) -> void
{
    const auto visitor = overloads{
        [&](const show_ok_message& msg)
        {
            message_title = msg.title;
            message_message = msg.message;
            ok_message = true;
        },
        [&](const show_yes_no_message& msg)
        {
            message_title = msg.title;
            message_message = msg.message;
            yes_no_handler = msg.on_yes;
            yes_no_message = true;
        },
        [&](const show_save_preset_window& msg)
        {
            save_window = true;
        }
    };

    while (!input.ui_commands.empty()) {
        const ui_command& cmd = input.ui_commands.front();

        cmd.visit(visitor);

        input.ui_commands.pop();
    }

    input.disable = window_open();

    draw_graph_overlay(fps, ups, mass_count, spring_count);
    draw_save_preset_popup();
    draw_ok_message_box();
    draw_yes_no_message_box();
}