protocols: add Hyprland toplevel mapping implementation (#9775)

1 files changed, 78 insertions, 0 deletions

diff --git a/src/protocols/ToplevelMapping.cpp b/src/protocols/ToplevelMapping.cpp
new file mode 100644
index 00000000..e65695c8
--- /dev/null
+++ b/src/protocols/ToplevelMapping.cpp
@@ -0,0 +1,78 @@
+#include "ToplevelMapping.hpp"
+#include "hyprland-toplevel-mapping-v1.hpp"
+#include "ForeignToplevelWlr.hpp"
+#include "ForeignToplevel.hpp"
+
+CToplevelWindowMappingHandle::CToplevelWindowMappingHandle(SP<CHyprlandToplevelWindowMappingHandleV1> resource_) : resource(resource_) {}
+
+CToplevelMappingManager::CToplevelMappingManager(SP<CHyprlandToplevelMappingManagerV1> resource_) : resource(resource_) {
+    if UNLIKELY (!resource_->resource())
+        return;
+
+    resource->setOnDestroy([this](CHyprlandToplevelMappingManagerV1* h) { PROTO::toplevelMapping->onManagerResourceDestroy(this); });
+    resource->setDestroy([this](CHyprlandToplevelMappingManagerV1* h) { PROTO::toplevelMapping->onManagerResourceDestroy(this); });
+
+    resource->setGetWindowForToplevel([this](CHyprlandToplevelMappingManagerV1* mgr, uint32_t handle, wl_resource* toplevel) {
+        const auto NEWHANDLE = PROTO::toplevelMapping->m_vHandles.emplace_back(
+            makeShared<CToplevelWindowMappingHandle>(makeShared<CHyprlandToplevelWindowMappingHandleV1>(resource->client(), resource->version(), handle)));
+
+        if UNLIKELY (!NEWHANDLE->resource->resource()) {
+            LOGM(ERR, "Couldn't alloc mapping handle! (no memory)");
+            resource->noMemory();
+            return;
+        }
+
+        NEWHANDLE->resource->setOnDestroy([](CHyprlandToplevelWindowMappingHandleV1* h) { PROTO::toplevelMapping->destroyHandle(h); });
+        NEWHANDLE->resource->setDestroy([](CHyprlandToplevelWindowMappingHandleV1* h) { PROTO::toplevelMapping->destroyHandle(h); });
+
+        const auto WINDOW = PROTO::foreignToplevel->windowFromHandleResource(toplevel);
+        if (!WINDOW)
+            NEWHANDLE->resource->sendFailed();
+        else
+            NEWHANDLE->resource->sendWindowAddress((uint64_t)WINDOW.get() >> 32 & 0xFFFFFFFF, (uint64_t)WINDOW.get() & 0xFFFFFFFF);
+    });
+    resource->setGetWindowForToplevelWlr([this](CHyprlandToplevelMappingManagerV1* mgr, uint32_t handle, wl_resource* toplevel) {
+        const auto NEWHANDLE = PROTO::toplevelMapping->m_vHandles.emplace_back(
+            makeShared<CToplevelWindowMappingHandle>(makeShared<CHyprlandToplevelWindowMappingHandleV1>(resource->client(), resource->version(), handle)));
+
+        if UNLIKELY (!NEWHANDLE->resource->resource()) {
+            LOGM(ERR, "Couldn't alloc mapping handle! (no memory)");
+            resource->noMemory();
+            return;
+        }
+
+        NEWHANDLE->resource->setOnDestroy([](CHyprlandToplevelWindowMappingHandleV1* h) { PROTO::toplevelMapping->destroyHandle(h); });
+        NEWHANDLE->resource->setDestroy([](CHyprlandToplevelWindowMappingHandleV1* h) { PROTO::toplevelMapping->destroyHandle(h); });
+
+        const auto WINDOW = PROTO::foreignToplevelWlr->windowFromHandleResource(toplevel);
+        if (!WINDOW)
+            NEWHANDLE->resource->sendFailed();
+        else
+            NEWHANDLE->resource->sendWindowAddress((uint64_t)WINDOW.get() >> 32 & 0xFFFFFFFF, (uint64_t)WINDOW.get() & 0xFFFFFFFF);
+    });
+}
+
+bool CToplevelMappingManager::good() const {
+    return resource->resource();
+}
+
+CToplevelMappingProtocol::CToplevelMappingProtocol(const wl_interface* iface, const int& ver, const std::string& name) : IWaylandProtocol(iface, ver, name) {}
+
+void CToplevelMappingProtocol::bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id) {
+    const auto RESOURCE = m_vManagers.emplace_back(makeUnique<CToplevelMappingManager>(makeShared<CHyprlandToplevelMappingManagerV1>(client, ver, id))).get();
+
+    if UNLIKELY (!RESOURCE->good()) {
+        LOGM(ERR, "Couldn't create a toplevel mapping manager");
+        wl_client_post_no_memory(client);
+        m_vManagers.pop_back();
+        return;
+    }
+}
+
+void CToplevelMappingProtocol::onManagerResourceDestroy(CToplevelMappingManager* mgr) {
+    std::erase_if(m_vManagers, [&](const auto& other) { return other.get() == mgr; });
+}
+
+void CToplevelMappingProtocol::destroyHandle(CHyprlandToplevelWindowMappingHandleV1* handle) {
+    std::erase_if(m_vHandles, [&](const auto& other) { return other->resource.get() == handle; });
+}
\ No newline at end of file