1 files changed, 260 insertions, 0 deletions
diff --git a/src/protocols/PointerConstraints.cpp b/src/protocols/PointerConstraints.cpp
new file mode 100644
index 00000000..ab228549
--- /dev/null
+++ b/src/protocols/PointerConstraints.cpp
@@ -0,0 +1,260 @@
+#include "PointerConstraints.hpp"
+#include "../desktop/WLSurface.hpp"
+#include "../Compositor.hpp"
+#include "../config/ConfigValue.hpp"
+
+#define LOGM PROTO::constraints->protoLog
+
+CPointerConstraint::CPointerConstraint(SP<CZwpLockedPointerV1> resource_, wlr_surface* surf, wl_resource* region_, zwpPointerConstraintsV1Lifetime lifetime) :
+    resourceL(resource_), locked(true) {
+    if (!resource_->resource())
+        return;
+
+    resource_->setOnDestroy([this](CZwpLockedPointerV1* p) { PROTO::constraints->destroyPointerConstraint(this); });
+    resource_->setDestroy([this](CZwpLockedPointerV1* p) { PROTO::constraints->destroyPointerConstraint(this); });
+
+    pHLSurface = CWLSurface::surfaceFromWlr(surf);
+
+    if (!pHLSurface)
+        return;
+
+    if (region_)
+        region.set(wlr_region_from_resource(region_));
+
+    resource_->setSetRegion([this](CZwpLockedPointerV1* p, wl_resource* region) { onSetRegion(region); });
+    resource_->setSetCursorPositionHint([this](CZwpLockedPointerV1* p, wl_fixed_t x, wl_fixed_t y) {
+        static auto PXWLFORCESCALEZERO = CConfigValue<Hyprlang::INT>("xwayland:force_zero_scaling");
+
+        if (!pHLSurface)
+            return;
+
+        hintSet = true;
+
+        float      scale   = 1.f;
+        const auto PWINDOW = pHLSurface->getWindow();
+        if (PWINDOW) {
+            const auto ISXWL = PWINDOW->m_bIsX11;
+            scale            = ISXWL && *PXWLFORCESCALEZERO ? PWINDOW->m_fX11SurfaceScaledBy : 1.f;
+        }
+
+        positionHint = {wl_fixed_to_double(x) / scale, wl_fixed_to_double(y) / scale};
+        g_pInputManager->simulateMouseMovement();
+    });
+
+    sharedConstructions();
+}
+
+CPointerConstraint::CPointerConstraint(SP<CZwpConfinedPointerV1> resource_, wlr_surface* surf, wl_resource* region_, zwpPointerConstraintsV1Lifetime lifetime) :
+    resourceC(resource_), locked(false) {
+    if (!resource_->resource())
+        return;
+
+    resource_->setOnDestroy([this](CZwpConfinedPointerV1* p) { PROTO::constraints->destroyPointerConstraint(this); });
+    resource_->setDestroy([this](CZwpConfinedPointerV1* p) { PROTO::constraints->destroyPointerConstraint(this); });
+
+    pHLSurface = CWLSurface::surfaceFromWlr(surf);
+
+    if (!pHLSurface)
+        return;
+
+    if (region_)
+        region.set(wlr_region_from_resource(region_));
+
+    resource_->setSetRegion([this](CZwpConfinedPointerV1* p, wl_resource* region) { onSetRegion(region); });
+
+    sharedConstructions();
+}
+
+CPointerConstraint::~CPointerConstraint() {
+    std::erase_if(g_pInputManager->m_vConstraints, [this](const auto& c) {
+        const auto SHP = c.lock();
+        return !SHP || SHP.get() == this;
+    });
+
+    if (pHLSurface)
+        pHLSurface->m_pConstraint.reset();
+}
+
+void CPointerConstraint::sharedConstructions() {
+    if (pHLSurface) {
+        listeners.destroySurface = pHLSurface->events.destroy.registerListener([this](std::any d) {
+            pHLSurface = nullptr;
+            if (active)
+                deactivate();
+
+            std::erase_if(g_pInputManager->m_vConstraints, [this](const auto& c) {
+                const auto SHP = c.lock();
+                return !SHP || SHP.get() == this;
+            });
+        });
+    }
+
+    cursorPosOnActivate = g_pInputManager->getMouseCoordsInternal();
+
+    if (g_pCompositor->m_pLastFocus == pHLSurface->wlr())
+        activate();
+}
+
+bool CPointerConstraint::good() {
+    return locked ? resourceL->resource() : resourceC->resource();
+}
+
+void CPointerConstraint::deactivate() {
+    if (!active)
+        return;
+
+    if (locked)
+        resourceL->sendUnlocked();
+    else
+        resourceC->sendUnconfined();
+
+    active = false;
+
+    if (locked)
+        g_pCompositor->warpCursorTo(logicPositionHint(), true);
+
+    if (lifetime == ZWP_POINTER_CONSTRAINTS_V1_LIFETIME_ONESHOT) {
+        dead = true;
+        // remove from inputmgr
+        std::erase_if(g_pInputManager->m_vConstraints, [this](const auto& c) {
+            const auto SHP = c.lock();
+            return !SHP || SHP.get() == this;
+        });
+    }
+}
+
+void CPointerConstraint::activate() {
+    if (dead || active)
+        return;
+
+    // TODO: hack, probably not a super duper great idea
+    if (g_pCompositor->m_sSeat.seat->pointer_state.focused_surface != pHLSurface->wlr()) {
+        const auto SURFBOX = pHLSurface->getSurfaceBoxGlobal();
+        const auto LOCAL   = SURFBOX.has_value() ? logicPositionHint() - SURFBOX->pos() : Vector2D{};
+        wlr_seat_pointer_enter(g_pCompositor->m_sSeat.seat, pHLSurface->wlr(), LOCAL.x, LOCAL.y);
+    }
+
+    g_pCompositor->warpCursorTo(logicPositionHint(), true);
+
+    if (locked)
+        resourceL->sendLocked();
+    else
+        resourceC->sendConfined();
+
+    active = true;
+
+    g_pInputManager->simulateMouseMovement();
+}
+
+bool CPointerConstraint::isActive() {
+    return active;
+}
+
+void CPointerConstraint::onSetRegion(wl_resource* wlRegion) {
+    if (!wlRegion) {
+        region.clear();
+        return;
+    }
+
+    const auto REGION = wlr_region_from_resource(wlRegion);
+
+    region.set(REGION);
+    positionHint = region.closestPoint(positionHint);
+    g_pInputManager->simulateMouseMovement(); // to warp the cursor if anything's amiss
+}
+
+CWLSurface* CPointerConstraint::owner() {
+    return pHLSurface;
+}
+
+CRegion CPointerConstraint::logicConstraintRegion() {
+    CRegion    rg            = region;
+    const auto SURFBOX       = pHLSurface->getSurfaceBoxGlobal();
+    const auto CONSTRAINTPOS = SURFBOX.has_value() ? SURFBOX->pos() : Vector2D{};
+    rg.translate(CONSTRAINTPOS);
+    return rg;
+}
+
+bool CPointerConstraint::isLocked() {
+    return locked;
+}
+
+Vector2D CPointerConstraint::logicPositionHint() {
+    if (!pHLSurface)
+        return {};
+
+    const auto SURFBOX       = pHLSurface->getSurfaceBoxGlobal();
+    const auto CONSTRAINTPOS = SURFBOX.has_value() ? SURFBOX->pos() : Vector2D{};
+
+    return hintSet ? CONSTRAINTPOS + positionHint : (locked ? CONSTRAINTPOS + SURFBOX->size() / 2.f : cursorPosOnActivate);
+}
+
+CPointerConstraintsProtocol::CPointerConstraintsProtocol(const wl_interface* iface, const int& ver, const std::string& name) : IWaylandProtocol(iface, ver, name) {
+    ;
+}
+
+void CPointerConstraintsProtocol::bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id) {
+    const auto RESOURCE = m_vManagers.emplace_back(std::make_unique<CZwpPointerConstraintsV1>(client, ver, id)).get();
+    RESOURCE->setOnDestroy([this](CZwpPointerConstraintsV1* p) { this->onManagerResourceDestroy(p->resource()); });
+
+    RESOURCE->setDestroy([this](CZwpPointerConstraintsV1* pMgr) { this->onManagerResourceDestroy(pMgr->resource()); });
+    RESOURCE->setConfinePointer([this](CZwpPointerConstraintsV1* pMgr, uint32_t id, wl_resource* surface, wl_resource* pointer, wl_resource* region,
+                                       zwpPointerConstraintsV1Lifetime lifetime) { this->onConfinePointer(pMgr, id, surface, pointer, region, lifetime); });
+    RESOURCE->setLockPointer([this](CZwpPointerConstraintsV1* pMgr, uint32_t id, wl_resource* surface, wl_resource* pointer, wl_resource* region,
+                                    zwpPointerConstraintsV1Lifetime lifetime) { this->onLockPointer(pMgr, id, surface, pointer, region, lifetime); });
+}
+
+void CPointerConstraintsProtocol::onManagerResourceDestroy(wl_resource* res) {
+    std::erase_if(m_vManagers, [&](const auto& other) { return other->resource() == res; });
+}
+
+void CPointerConstraintsProtocol::destroyPointerConstraint(CPointerConstraint* hyprland🥚) {
+    std::erase_if(m_vConstraints, [&](const auto& other) { return other.get() == hyprland🥚; });
+}
+
+void CPointerConstraintsProtocol::onNewConstraint(SP<CPointerConstraint> constraint, CZwpPointerConstraintsV1* pMgr) {
+    if (!constraint->good()) {
+        LOGM(ERR, "Couldn't create constraint??");
+        wl_resource_post_no_memory(pMgr->resource());
+        m_vConstraints.pop_back();
+        return;
+    }
+
+    if (!constraint->owner()) {
+        LOGM(ERR, "New constraint has no CWLSurface owner??");
+        return;
+    }
+
+    const auto OWNER = constraint->owner();
+
+    const auto DUPES = std::count_if(m_vConstraints.begin(), m_vConstraints.end(), [OWNER](const auto& c) { return c->owner() == OWNER; });
+
+    if (DUPES > 1) {
+        LOGM(ERR, "Constraint for surface duped");
+        wl_resource_post_error(pMgr->resource(), ZWP_POINTER_CONSTRAINTS_V1_ERROR_ALREADY_CONSTRAINED, "Surface already confined");
+        m_vConstraints.pop_back();
+        return;
+    }
+
+    OWNER->appendConstraint(constraint);
+
+    g_pInputManager->m_vConstraints.push_back(constraint);
+}
+
+void CPointerConstraintsProtocol::onLockPointer(CZwpPointerConstraintsV1* pMgr, uint32_t id, wl_resource* surface, wl_resource* pointer, wl_resource* region,
+                                                zwpPointerConstraintsV1Lifetime lifetime) {
+    const auto CLIENT   = wl_resource_get_client(pMgr->resource());
+    const auto RESOURCE = m_vConstraints.emplace_back(std::make_shared<CPointerConstraint>(
+        std::make_shared<CZwpLockedPointerV1>(CLIENT, wl_resource_get_version(pMgr->resource()), id), wlr_surface_from_resource(surface), region, lifetime));
+
+    onNewConstraint(RESOURCE, pMgr);
+}
+
+void CPointerConstraintsProtocol::onConfinePointer(CZwpPointerConstraintsV1* pMgr, uint32_t id, wl_resource* surface, wl_resource* pointer, wl_resource* region,
+                                                   zwpPointerConstraintsV1Lifetime lifetime) {
+    const auto CLIENT   = wl_resource_get_client(pMgr->resource());
+    const auto RESOURCE = m_vConstraints.emplace_back(std::make_shared<CPointerConstraint>(
+        std::make_shared<CZwpConfinedPointerV1>(CLIENT, wl_resource_get_version(pMgr->resource()), id), wlr_surface_from_resource(surface), region, lifetime));
+
+    onNewConstraint(RESOURCE, pMgr);
+}