#include "content/views/view_tools.hpp"
#include <imgui_internal.h>

#include <hex/api/content_registry.hpp>
#include <hex/api/layout_manager.hpp>

#include <fonts/codicons_font.h>

namespace hex::plugin::builtin {

    ViewTools::ViewTools() : View::Window("hex.builtin.view.tools.name", ICON_VS_TOOLS) {
        m_dragStartIterator = ContentRegistry::Tools::impl::getEntries().end();

        LayoutManager::registerLoadCallback([this](std::string_view line) {
            auto parts = wolv::util::splitString(std::string(line), "=");
            if (parts.size() != 2)
                return;

            m_detachedTools[parts[0]] = parts[1] == "1";
        });

        LayoutManager::registerStoreCallback([this](ImGuiTextBuffer *buffer) {
            for (auto &[unlocalizedName, function] : ContentRegistry::Tools::impl::getEntries()) {
                auto detached = m_detachedTools[unlocalizedName];
                buffer->appendf("%s=%d\n", unlocalizedName.get().c_str(), detached);
            }
        });
    }

    void ViewTools::drawContent() {
        auto &tools = ContentRegistry::Tools::impl::getEntries();

        // Draw all tools
        for (auto iter = tools.begin(); iter != tools.end(); ++iter) {
            auto &[unlocalizedName, function] = *iter;

            // If the tool has been detached from the main window, don't draw it here anymore
            if (m_detachedTools[unlocalizedName]) continue;

            // Draw the tool
            if (ImGui::CollapsingHeader(Lang(unlocalizedName))) {
                function();
                ImGui::NewLine();
            } else {
                // Handle dragging the tool out of the main window

                // If the user clicks on the header, start dragging the tool remember the iterator
                if (ImGui::IsMouseClicked(0) && ImGui::IsItemActivated() && m_dragStartIterator == tools.end())
                    m_dragStartIterator = iter;

                // If the user released the mouse button, stop dragging the tool
                if (!ImGui::IsMouseDown(0))
                    m_dragStartIterator = tools.end();

                // Detach the tool if the user dragged it out of the main window
                if (!ImGui::IsItemHovered() && m_dragStartIterator == iter) {
                    m_detachedTools[unlocalizedName] = true;
                }

            }
        }
    }

    void ViewTools::drawAlwaysVisibleContent() {
        // Make sure the tool windows never get drawn over the welcome screen
        if (!ImHexApi::Provider::isValid())
            return;

        auto &tools = ContentRegistry::Tools::impl::getEntries();

        for (auto iter = tools.begin(); iter != tools.end(); ++iter) {
            auto &[unlocalizedName, function] = *iter;

            // If the tool is still attached to the main window, don't draw it here
            if (!m_detachedTools[unlocalizedName]) continue;

            // Load the window height that is dependent on the tool content
            const auto windowName = View::toWindowName(unlocalizedName);
            const auto height = m_windowHeights[ImGui::FindWindowByName(windowName.c_str())];
            if (height > 0)
                ImGui::SetNextWindowSizeConstraints(ImVec2(400_scaled, height), ImVec2(FLT_MAX, height));

            // Create a new window for the tool
            if (ImGui::Begin(windowName.c_str(), &m_detachedTools[unlocalizedName], ImGuiWindowFlags_NoCollapse)) {
                // Draw the tool content
                function();

                // Handle the first frame after the tool has been detached
                if (ImGui::IsWindowAppearing() && m_dragStartIterator == iter) {
                    m_dragStartIterator = tools.end();

                    // Attach the newly created window to the cursor, so it gets dragged around
                    GImGui->MovingWindow = ImGui::GetCurrentWindowRead();
                    GImGui->ActiveId = GImGui->MovingWindow->MoveId;
                }

                const auto window = ImGui::GetCurrentWindowRead();
                m_windowHeights[ImGui::GetCurrentWindowRead()] = ImGui::CalcWindowNextAutoFitSize(window).y;
            }
            ImGui::End();
        }
    }

}