#include "../../include/gui/ComboBox.hpp"
#include "../../include/gui/Utils.hpp"
#include "../../include/Theme.hpp"
#include <mglpp/graphics/Rectangle.hpp>
#include <mglpp/graphics/Font.hpp>
#include <mglpp/window/Window.hpp>
#include <mglpp/window/Event.hpp>
#include <assert.h>

namespace gsr {
    static const float padding_top = 10.0f;
    static const float padding_bottom = 10.0f;
    static const float padding_left = 10.0f;
    static const float padding_right = 10.0f;

    ComboBox::ComboBox(mgl::Font *font) : font(font) {
        assert(font);
    }

    bool ComboBox::on_event(mgl::Event &event, mgl::Window&, mgl::vec2f offset) {
        if(event.type == mgl::Event::MouseButtonPressed && event.mouse_button.button == mgl::Mouse::Left) {
            const mgl::vec2f draw_pos = position + offset;
            const mgl::vec2f mouse_pos = { (float)event.mouse_button.x, (float)event.mouse_button.y };
            const mgl::vec2f item_size(max_size.x, font->get_character_size() + padding_top + padding_bottom);

            if(show_dropdown && !items.empty()) {
                mgl::vec2f pos = draw_pos + mgl::vec2f(padding_left, padding_top);
                pos.y += items[selected_item].text.get_bounds().size.y + padding_top + padding_bottom;

                for(size_t i = 0; i < items.size(); ++i) {
                    Item &item = items[i];
                    const mgl::FloatRect text_bounds = item.text.get_bounds();
                    if(mgl::FloatRect(pos - mgl::vec2f(padding_left, padding_top), item_size).contains(mouse_pos)) {
                        selected_item = i;
                        show_dropdown = false;
                        remove_widget_as_selected_in_parent();
                        return false;
                    }
                    pos.y += text_bounds.size.y + padding_top + padding_bottom;
                }
            }

            if(mgl::FloatRect(draw_pos, item_size).contains(mouse_pos)) {
                show_dropdown = !show_dropdown;
                if(show_dropdown)
                    set_widget_as_selected_in_parent();
                else
                    remove_widget_as_selected_in_parent();
                return false;
            } else {
                show_dropdown = false;
                remove_widget_as_selected_in_parent();
            }
        }
        return true;
    }

    void ComboBox::draw(mgl::Window &window, mgl::vec2f offset) {
        update_if_dirty();

        if(items.empty())
            return;

        const mgl::vec2f draw_pos = position + offset;

        const mgl::vec2f item_size(max_size.x, font->get_character_size() + padding_top + padding_bottom);
        const mgl::vec2i mouse_pos = window.get_mouse_position();
        bool inside = false;

        mgl::Rectangle background(draw_pos, mgl::vec2f(max_size.x, item_size.y));
        if(show_dropdown) {
            background.set_size(max_size);
            background.set_color(mgl::Color(0, 0, 0));
        } else {
            background.set_color(mgl::Color(0, 0, 0, 120));
        }
        window.draw(background);

        mgl::vec2f pos = draw_pos + mgl::vec2f(padding_left, padding_top);

        Item &item = items[selected_item];
        item.text.set_position(pos.floor());
        if(show_dropdown) {
            const int border_size = 3;
            const mgl::Color border_color = gsr::get_theme().tint_color;
            draw_rectangle_outline(window, pos - mgl::vec2f(padding_left, padding_top), item_size, border_color, border_size);
        }
        window.draw(item.text);
        pos.y += item.text.get_bounds().size.y + padding_top + padding_bottom;

        for(size_t i = 0; i < items.size(); ++i) {
            Item &item = items[i];
            item.text.set_position(pos.floor());
            const mgl::FloatRect text_bounds = item.text.get_bounds();

            if(show_dropdown) {
                if(!inside) {
                    inside = mgl::FloatRect(text_bounds.position - mgl::vec2f(padding_left, padding_top), item_size).contains({ (float)mouse_pos.x, (float)mouse_pos.y });
                    if(inside) {
                        mgl::Rectangle item_background(text_bounds.position - mgl::vec2f(padding_left, padding_top), item_size);
                        item_background.set_color(gsr::get_theme().tint_color);
                        window.draw(item_background);
                    } else {
                        /*const int border_size = 3;
                        const mgl::Color border_color(150, 150, 150);
                        draw_rectangle_outline(window, text_bounds.position, item_size, border_color, border_size);*/
                    }
                }
                window.draw(item.text);
            }

            pos.y += text_bounds.size.y + padding_top + padding_bottom;
        }
    }

    void ComboBox::add_item(const std::string &text, const std::string &id) {
        items.push_back({mgl::Text(text, *font), id});
        dirty = true;
    }

    void ComboBox::update_if_dirty() {
        if(!dirty)
            return;

        max_size = { 0.0f, font->get_character_size() + padding_top + padding_bottom };
        for(Item &item : items) {
            const mgl::vec2f bounds = item.text.get_bounds().size;
            max_size.x = std::max(max_size.x, bounds.x + padding_left + padding_right);
            max_size.y += bounds.y + padding_top + padding_bottom;
        }
        dirty = false;
    }

    mgl::vec2f ComboBox::get_size() {
        update_if_dirty();
        return { max_size.x, font->get_character_size() + padding_top + padding_bottom };
    }
}