#include "PointerWarp.hpp"
#include "core/Compositor.hpp"
#include "core/Seat.hpp"
#include "../desktop/view/WLSurface.hpp"
#include "../managers/SeatManager.hpp"
#include "../pointer/PointerManager.hpp"
#include "../desktop/view/Window.hpp"
#include "desktop/view/LayerSurface.hpp"
#include <hyprutils/math/Box.hpp>
#include <hyprutils/math/Vector2D.hpp>

CPointerWarpProtocol::CPointerWarpProtocol(const wl_interface* iface, const int& ver, const std::string& name) : IWaylandProtocol(iface, ver, name) {
    ;
}

void CPointerWarpProtocol::bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id) {
    const auto& RESOURCE = m_managers.emplace_back(makeUnique<CWpPointerWarpV1>(client, ver, id));

    if UNLIKELY (!RESOURCE->resource()) {
        wl_client_post_no_memory(client);
        m_managers.pop_back();
        return;
    }

    RESOURCE->setOnDestroy([this](CWpPointerWarpV1* pMgr) { destroyManager(pMgr); });
    RESOURCE->setDestroy([this](CWpPointerWarpV1* pMgr) { destroyManager(pMgr); });

    RESOURCE->setWarpPointer([](CWpPointerWarpV1* pMgr, wl_resource* surface, wl_resource* pointer, wl_fixed_t x, wl_fixed_t y, uint32_t serial) {
        const auto PSURFACE = CWLSurfaceResource::fromResource(surface);
        if (g_pSeatManager->m_state.pointerFocus != PSURFACE)
            return;

        CBox surfbox;

        auto HLSURF = Desktop::View::CWLSurface::fromResource(PSURFACE);

        if (!HLSURF)
            return;

        auto VIEW   = HLSURF->view();
        auto WINDOW = Desktop::View::CWindow::fromView(VIEW);
        if (WINDOW)
            surfbox = WINDOW->getWindowMainSurfaceBox();
        else {
            auto LAYERSURFACE = Desktop::View::CLayerSurface::fromView(VIEW);
            if (!LAYERSURFACE)
                return;

            auto BOX = LAYERSURFACE->logicalBox();
            if (!BOX.has_value())
                return;

            surfbox = BOX.value();
        }

        const auto LOCALPOS = Vector2D{wl_fixed_to_double(x), wl_fixed_to_double(y)};

        // Allow a margin of 1px on all sides
        surfbox.expand(1);
        const auto GLOBALPOS = LOCALPOS + surfbox.pos() + Vector2D{1., 1.};
        if (!surfbox.containsPoint(GLOBALPOS))
            return;

        const auto POINTER = CWLPointerResource::fromResource(pointer);
        if UNLIKELY (!POINTER) {
            LOGM(Log::ERR, "pointer_warp received an invalid pointer resource");
            return;
        }

        const auto PSEAT = POINTER->m_owner.lock();
        if (!g_pSeatManager->serialValid(PSEAT, serial, false))
            return;

        LOGM(Log::DEBUG, "warped pointer to {}", GLOBALPOS);

        Pointer::mgr()->warpTo(GLOBALPOS);
        g_pSeatManager->sendPointerMotion(Time::millis(Time::steadyNow()), LOCALPOS);
        g_pSeatManager->sendPointerFrame();
    });
}

void CPointerWarpProtocol::destroyManager(CWpPointerWarpV1* manager) {
    std::erase_if(m_managers, [&](const UP<CWpPointerWarpV1>& resource) { return resource.get() == manager; });
}
