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

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

#include <fonts/vscode_icons.hpp>

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, icon, 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, icon, function] = *iter;

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

            if (!m_collapsedTools.contains(unlocalizedName))
                m_collapsedTools[unlocalizedName] = true;

            // Draw the tool
            auto &collapsed = m_collapsedTools[unlocalizedName];
            if (ImGuiExt::BeginSubWindow(fmt::format("{} {}", icon, Lang(unlocalizedName)).c_str(), &collapsed, ImVec2(0, collapsed ? 1 : 0))) {
                function();
            } 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::IsWindowHovered() && 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::IsWindowHovered() && m_dragStartIterator == iter) {
                    m_detachedTools[unlocalizedName] = true;
                }
            }
            ImGuiExt::EndSubWindow();
        }
    }

    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, icon, 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 = fmt::format("{} {}", icon, 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
            ImGui::SetNextWindowPos(ImGui::GetMousePos() - ImVec2(0, ImGui::GetTextLineHeightWithSpacing()), ImGuiCond_Appearing);
            if (ImGui::Begin(windowName.c_str(), &m_detachedTools[unlocalizedName], ImGuiWindowFlags_NoCollapse)) {
                // Draw the tool content
                function();

                const auto window = ImGui::GetCurrentWindowRead();

                // 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
                    ImGui::StartMouseMovingWindowOrNode(window, nullptr, true);
                }

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

    void ViewTools::drawHelpText() {
        ImGuiExt::TextFormattedWrapped("This view contains various standalone tools that didn't fit anywhere else but are useful nonetheless.");
        ImGui::NewLine();
        ImGuiExt::TextFormattedWrapped("Click on the arrow icon in the title bar of each tool to open or collapse it. When collapsed, you can drag the tool out of this window to create a separate floating window which can also be docked anywhere you like.");

    }

}