1 files changed, 150 insertions, 0 deletions
diff --git a/plugins/wayfire/ueberzugpp.cpp b/plugins/wayfire/ueberzugpp.cpp
new file mode 100644
index 0000000..cc70a16
--- /dev/null
+++ b/plugins/wayfire/ueberzugpp.cpp
@@ -0,0 +1,150 @@
+#include <memory>
+#include <unordered_map>
+#include <wayfire/core.hpp>
+#include <wayfire/scene-render.hpp>
+#include <wayfire/util.hpp>
+#include <wayfire/view.hpp>
+#include <wayfire/output.hpp>
+#include <wayfire/plugin.hpp>
+#include <wayfire/scene-operations.hpp>
+#include <wayfire/plugins/ipc/ipc-method-repository.hpp>
+#include <wayfire/plugins/common/shared-core-data.hpp>
+#include <wayfire/unstable/wlr-view-events.hpp>
+#include <wayfire/unstable/wlr-surface-node.hpp>
+#include <wayfire/unstable/translation-node.hpp>
+
+namespace wf
+{
+class ueberzugpp_surface_t
+{
+    std::weak_ptr<wf::view_interface_t> terminal;
+    std::shared_ptr<scene::wlr_surface_node_t> surface;
+    std::shared_ptr<scene::translation_node_t> translation;
+    wlr_xdg_toplevel *toplevel;
+
+    wf::wl_listener_wrapper on_toplevel_destroy;
+
+  public:
+
+    ueberzugpp_surface_t(const std::weak_ptr<wf::view_interface_t>& _terminal,
+        wlr_xdg_toplevel *_toplevel):
+        terminal(_terminal),
+        toplevel(_toplevel)
+    {
+        // We create a custom surface node for the ueberzugpp window and wrap it in a translation node, so that
+        // we can control its position.
+        // We add the translation node as a subsurface of the terminal.
+        surface = std::make_shared<scene::wlr_surface_node_t>(_toplevel->base->surface, true);
+        translation = std::make_shared<scene::translation_node_t>();
+
+        translation->set_children_list({surface});
+
+        auto term = _terminal.lock();
+        wf::scene::add_front(term->get_surface_root_node(), translation);
+
+        // Finally, wait for the toplevel to be destroyed
+        on_toplevel_destroy.set_callback([this] (auto)
+        {
+            destroy_callback();
+        });
+        on_toplevel_destroy.connect(&toplevel->base->events.destroy);
+    }
+
+    std::function<void()> destroy_callback;
+
+    ~ueberzugpp_surface_t()
+    {
+        wf::scene::remove_child(translation);
+    }
+
+    void set_offset(int xcoord, int ycoord)
+    {
+        translation->set_offset({xcoord, ycoord});
+        const auto term = terminal.lock();
+        if (!term) {
+            return;
+        }
+        term->damage();
+    }
+};
+
+
+class ueberzugpp_mapper : public wf::plugin_interface_t
+{
+  public:
+    void init() override
+    {
+        ipc_repo->register_method("ueberzugpp/set_offset", set_offset);
+        wf::get_core().connect(&on_new_xdg_surface);
+    }
+
+    void fini() override
+    {
+        ipc_repo->unregister_method("ueberzugpp/set_offset");
+    }
+
+    ipc::method_callback set_offset = [this] (nlohmann::json json)
+    {
+        // NOLINTBEGIN
+        WFJSON_EXPECT_FIELD(json, "app-id", string);
+        WFJSON_EXPECT_FIELD(json, "x", number_integer);
+        WFJSON_EXPECT_FIELD(json, "y", number_integer);
+        // NOLINTEND
+
+        const std::string app_id = json["app-id"];
+        const int xcoord = json["x"];
+        const int ycoord = json["y"];
+
+        const auto search = surfaces.find(app_id);
+        if (search == surfaces.end())
+        {
+            return ipc::json_error("Unknown ueberzugpp window with appid " + app_id);
+        }
+
+        search->second->set_offset(xcoord, ycoord);
+        return ipc::json_ok();
+    };
+
+    shared_data::ref_ptr_t<ipc::method_repository_t> ipc_repo;
+
+    // When a new xdg_toplevel is created, we need to check whether it is an ueberzugpp window by looking at
+    // its app-id. If it is indeed an ueberzugpp window, we take over the toplevel (by setting
+    // use_default_implementation=false) and create our own ueberzugpp_surface. In addition, we directly map
+    // the surface to the currently focused view, if it exists.
+    wf::signal::connection_t<new_xdg_surface_signal> on_new_xdg_surface = [this] (new_xdg_surface_signal *event)
+    {
+        if (!event->use_default_implementation) {
+            return;
+        }
+        if (event->surface->role != WLR_XDG_SURFACE_ROLE_TOPLEVEL) {
+            return;
+        }
+
+        const auto* toplevel_title = event->surface->toplevel->title;
+        const std::string appid = toplevel_title != nullptr ? toplevel_title : "null";
+        const std::string search_for = "ueberzugpp_";
+        if (appid.find(search_for) != std::string::npos)
+        {
+            auto terminal = wf::get_core().get_active_output()->get_active_view();
+            if (!terminal)
+            {
+                LOGE("Which window to map ueberzugpp to????");
+                return;
+            }
+
+            event->use_default_implementation = false;
+            const auto [iter, was_inserted] = surfaces.insert_or_assign(appid,
+                    std::make_unique<ueberzugpp_surface_t>(terminal->weak_from_this(), event->surface->toplevel));
+            iter->second->destroy_callback = [this, appid]
+            {
+                surfaces.erase(appid);
+            };
+        }
+    };
+
+    std::unordered_map<std::string, std::unique_ptr<ueberzugpp_surface_t>> surfaces;
+};
+} // end namespace wf
+
+// NOLINTNEXTLINE
+DECLARE_WAYFIRE_PLUGIN(wf::ueberzugpp_mapper);