From 064edeb5a3a30b2671e9b17e925048a282c94c9a Mon Sep 17 00:00:00 2001
From: outfoxxed <outfoxxed@outfoxxed.me>
Date: Fri, 31 Jan 2025 17:49:59 -0800
Subject: [PATCH] protocols: implement hyprland-ctm-control v2

---
 src/managers/ProtocolManager.cpp |  2 +-
 src/protocols/CTMControl.cpp     | 22 +++++++++++++++++++++-
 src/protocols/CTMControl.hpp     |  5 ++++-
 3 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/src/managers/ProtocolManager.cpp b/src/managers/ProtocolManager.cpp
index d3270fe2bd0..62736ae5fe0 100644
--- a/src/managers/ProtocolManager.cpp
+++ b/src/managers/ProtocolManager.cpp
@@ -167,7 +167,7 @@ CProtocolManager::CProtocolManager() {
     PROTO::xdgDialog           = makeUnique<CXDGDialogProtocol>(&xdg_dialog_v1_interface, 1, "XDGDialog");
     PROTO::singlePixel         = makeUnique<CSinglePixelProtocol>(&wp_single_pixel_buffer_manager_v1_interface, 1, "SinglePixel");
     PROTO::securityContext     = makeUnique<CSecurityContextProtocol>(&wp_security_context_manager_v1_interface, 1, "SecurityContext");
-    PROTO::ctm                 = makeUnique<CHyprlandCTMControlProtocol>(&hyprland_ctm_control_manager_v1_interface, 1, "CTMControl");
+    PROTO::ctm                 = makeUnique<CHyprlandCTMControlProtocol>(&hyprland_ctm_control_manager_v1_interface, 2, "CTMControl");
     PROTO::hyprlandSurface     = makeUnique<CHyprlandSurfaceProtocol>(&hyprland_surface_manager_v1_interface, 2, "HyprlandSurface");
 
     if (*PENABLEXXCM) {
diff --git a/src/protocols/CTMControl.cpp b/src/protocols/CTMControl.cpp
index 5560271f688..322f95e2d0e 100644
--- a/src/protocols/CTMControl.cpp
+++ b/src/protocols/CTMControl.cpp
@@ -15,6 +15,9 @@ CHyprlandCTMControlResource::CHyprlandCTMControlResource(SP<CHyprlandCtmControlM
 
     resource->setSetCtmForOutput([this](CHyprlandCtmControlManagerV1* r, wl_resource* output, wl_fixed_t mat0, wl_fixed_t mat1, wl_fixed_t mat2, wl_fixed_t mat3, wl_fixed_t mat4,
                                         wl_fixed_t mat5, wl_fixed_t mat6, wl_fixed_t mat7, wl_fixed_t mat8) {
+        if (blocked)
+            return;
+
         const auto OUTPUTRESOURCE = CWLOutputResource::fromResource(output);
 
         if UNLIKELY (!OUTPUTRESOURCE)
@@ -41,6 +44,9 @@ CHyprlandCTMControlResource::CHyprlandCTMControlResource(SP<CHyprlandCtmControlM
     });
 
     resource->setCommit([this](CHyprlandCtmControlManagerV1* r) {
+        if (blocked)
+            return;
+
         LOGM(LOG, "Committing ctms to outputs");
 
         for (auto& m : g_pCompositor->m_vMonitors) {
@@ -54,7 +60,17 @@ CHyprlandCTMControlResource::CHyprlandCTMControlResource(SP<CHyprlandCtmControlM
     });
 }
 
+void CHyprlandCTMControlResource::block() {
+    blocked = true;
+
+    if (resource->version() >= 2)
+        resource->sendBlocked();
+}
+
 CHyprlandCTMControlResource::~CHyprlandCTMControlResource() {
+    if (blocked)
+        return;
+
     for (auto& m : g_pCompositor->m_vMonitors) {
         PROTO::ctm->setCTM(m, Mat3x3::identity());
     }
@@ -69,7 +85,6 @@ CHyprlandCTMControlProtocol::CHyprlandCTMControlProtocol(const wl_interface* ifa
 }
 
 void CHyprlandCTMControlProtocol::bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id) {
-
     const auto RESOURCE = m_vManagers.emplace_back(makeShared<CHyprlandCTMControlResource>(makeShared<CHyprlandCtmControlManagerV1>(client, ver, id)));
 
     if UNLIKELY (!RESOURCE->good()) {
@@ -78,6 +93,11 @@ void CHyprlandCTMControlProtocol::bindManager(wl_client* client, void* data, uin
         return;
     }
 
+    if (m_pManager)
+        RESOURCE->block();
+    else
+        m_pManager = RESOURCE;
+
     LOGM(LOG, "New CTM Manager at 0x{:x}", (uintptr_t)RESOURCE.get());
 }
 
diff --git a/src/protocols/CTMControl.hpp b/src/protocols/CTMControl.hpp
index 2c168acd76e..eb54a3aae7f 100644
--- a/src/protocols/CTMControl.hpp
+++ b/src/protocols/CTMControl.hpp
@@ -16,11 +16,13 @@ class CHyprlandCTMControlResource {
     ~CHyprlandCTMControlResource();
 
     bool good();
+    void block();
 
   private:
     SP<CHyprlandCtmControlManagerV1>        resource;
 
     std::unordered_map<std::string, Mat3x3> ctms;
+    bool                                    blocked = false;
 };
 
 class CHyprlandCTMControlProtocol : public IWaylandProtocol {
@@ -37,6 +39,7 @@ class CHyprlandCTMControlProtocol : public IWaylandProtocol {
 
     //
     std::vector<SP<CHyprlandCTMControlResource>> m_vManagers;
+    WP<CHyprlandCTMControlResource>              m_pManager;
 
     //
     struct SCTMData {
@@ -51,4 +54,4 @@ class CHyprlandCTMControlProtocol : public IWaylandProtocol {
 
 namespace PROTO {
     inline UP<CHyprlandCTMControlProtocol> ctm;
-};
\ No newline at end of file
+};