protocols: implement pointer-warp-v1 (#11469)

4 files changed, 89 insertions, 0 deletions

diff --git a/src/protocols/PointerWarp.cpp b/src/protocols/PointerWarp.cpp
new file mode 100644
index 00000000..bde0b913
--- /dev/null
+++ b/src/protocols/PointerWarp.cpp
@@ -0,0 +1,53 @@
+#include "PointerWarp.hpp"
+#include "core/Compositor.hpp"
+#include "core/Seat.hpp"
+#include "../desktop/WLSurface.hpp"
+#include "../managers/SeatManager.hpp"
+#include "../managers/PointerManager.hpp"
+#include "../desktop/Window.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;
+
+        auto SURFBOXV = CWLSurface::fromResource(PSURFACE)->getSurfaceBoxGlobal();
+        if (!SURFBOXV.has_value())
+            return;
+
+        const auto SURFBOX   = SURFBOXV->expand(1);
+        const auto LOCALPOS  = Vector2D{wl_fixed_to_double(x), wl_fixed_to_double(y)};
+        const auto GLOBALPOS = LOCALPOS + SURFBOX.pos();
+        if (!SURFBOX.containsPoint(GLOBALPOS))
+            return;
+
+        const auto PSEAT = CWLPointerResource::fromResource(pointer)->m_owner.lock();
+        if (!g_pSeatManager->serialValid(PSEAT, serial, false))
+            return;
+
+        LOGM(LOG, "warped pointer to {}", GLOBALPOS);
+
+        g_pPointerManager->warpTo(GLOBALPOS);
+        g_pSeatManager->sendPointerMotion(Time::millis(Time::steadyNow()), LOCALPOS);
+    });
+}
+
+void CPointerWarpProtocol::destroyManager(CWpPointerWarpV1* manager) {
+    std::erase_if(m_managers, [&](const UP<CWpPointerWarpV1>& resource) { return resource.get() == manager; });
+}
diff --git a/src/protocols/PointerWarp.hpp b/src/protocols/PointerWarp.hpp
new file mode 100644
index 00000000..d1ce0062
--- /dev/null
+++ b/src/protocols/PointerWarp.hpp
@@ -0,0 +1,21 @@
+#pragma once
+
+#include "WaylandProtocol.hpp"
+#include "pointer-warp-v1.hpp"
+
+class CPointerWarpProtocol : public IWaylandProtocol {
+  public:
+    CPointerWarpProtocol(const wl_interface* iface, const int& ver, const std::string& name);
+
+    virtual void bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id);
+
+  private:
+    void destroyManager(CWpPointerWarpV1* manager);
+
+    //
+    std::vector<UP<CWpPointerWarpV1>> m_managers;
+};
+
+namespace PROTO {
+    inline UP<CPointerWarpProtocol> pointerWarp;
+};
diff --git a/src/protocols/core/Seat.cpp b/src/protocols/core/Seat.cpp
index e89c49ae..0b356c6f 100644
--- a/src/protocols/core/Seat.cpp
+++ b/src/protocols/core/Seat.cpp
@@ -109,6 +109,8 @@ CWLPointerResource::CWLPointerResource(SP<CWlPointer> resource_, SP<CWLSeatResou
     if UNLIKELY (!good())
         return;
 
+    m_resource->setData(this);
+
     m_resource->setRelease([this](CWlPointer* r) { PROTO::seat->destroyResource(this); });
     m_resource->setOnDestroy([this](CWlPointer* r) { PROTO::seat->destroyResource(this); });
 
@@ -145,6 +147,11 @@ bool CWLPointerResource::good() {
     return m_resource->resource();
 }
 
+SP<CWLPointerResource> CWLPointerResource::fromResource(wl_resource* res) {
+    auto data = sc<CWLPointerResource*>(sc<CWlPointer*>(wl_resource_get_user_data(res))->data());
+    return data ? data->m_self.lock() : nullptr;
+}
+
 void CWLPointerResource::sendEnter(SP<CWLSurfaceResource> surface, const Vector2D& local) {
     if (!m_owner || m_currentSurface == surface || !surface->getResource()->resource())
         return;
@@ -439,6 +446,8 @@ CWLSeatResource::CWLSeatResource(SP<CWlSeat> resource_) : m_resource(resource_)
             return;
         }
 
+        RESOURCE->m_self = RESOURCE;
+
         m_pointers.emplace_back(RESOURCE);
     });
 
diff --git a/src/protocols/core/Seat.hpp b/src/protocols/core/Seat.hpp
index 8b8f6004..29399a27 100644
--- a/src/protocols/core/Seat.hpp
+++ b/src/protocols/core/Seat.hpp
@@ -88,15 +88,21 @@ class CWLPointerResource {
 
     WP<CWLSeatResource> m_owner;
 
+    //
+    static SP<CWLPointerResource> fromResource(wl_resource* res);
+
   private:
     SP<CWlPointer>         m_resource;
     WP<CWLSurfaceResource> m_currentSurface;
+    WP<CWLPointerResource> m_self;
 
     std::vector<uint32_t>  m_pressedButtons;
 
     struct {
         CHyprSignalListener destroySurface;
     } m_listeners;
+
+    friend class CWLSeatResource;
 };
 
 class CWLKeyboardResource {