Fix xwayland unmanaged surfaces

4 files changed, 56 insertions, 15 deletions

diff --git a/include/sway/tree/view.h b/include/sway/tree/view.h
index 2eca7a3e..e52bee66 100644
--- a/include/sway/tree/view.h
+++ b/include/sway/tree/view.h
@@ -87,6 +87,9 @@ struct sway_xwayland_unmanaged {
         struct wlr_xwayland_surface *wlr_xwayland_surface;
         struct wl_list link;
 
+        struct wl_listener commit;
+        struct wl_listener map;
+        struct wl_listener unmap;
         struct wl_listener destroy;
 };
 
diff --git a/sway/desktop/wl_shell.c b/sway/desktop/wl_shell.c
index 5955fa9d..fff31da8 100644
--- a/sway/desktop/wl_shell.c
+++ b/sway/desktop/wl_shell.c
@@ -68,6 +68,7 @@ static const struct sway_view_impl view_impl = {
         .get_prop = get_prop,
         .configure = configure,
         .close = _close,
+        .destroy = destroy,
 };
 
 static void handle_commit(struct wl_listener *listener, void *data) {
diff --git a/sway/desktop/xdg_shell_v6.c b/sway/desktop/xdg_shell_v6.c
index baf07a1e..c66cc39a 100644
--- a/sway/desktop/xdg_shell_v6.c
+++ b/sway/desktop/xdg_shell_v6.c
@@ -147,6 +147,7 @@ static const struct sway_view_impl view_impl = {
         .configure = configure,
         .set_activated = set_activated,
         .close = _close,
+        .destroy = destroy,
 };
 
 static void handle_commit(struct wl_listener *listener, void *data) {
diff --git a/sway/desktop/xwayland.c b/sway/desktop/xwayland.c
index 384f4236..543c914e 100644
--- a/sway/desktop/xwayland.c
+++ b/sway/desktop/xwayland.c
@@ -14,30 +14,65 @@
 #include "sway/input/input-manager.h"
 #include "log.h"
 
+static void unmanaged_handle_commit(struct wl_listener *listener, void *data) {
+        struct sway_xwayland_unmanaged *surface =
+                wl_container_of(listener, surface, commit);
+        // TODO: damage tracking
+}
+
+static void unmanaged_handle_map(struct wl_listener *listener, void *data) {
+        struct sway_xwayland_unmanaged *surface =
+                wl_container_of(listener, surface, map);
+        struct wlr_xwayland_surface *xsurface = surface->wlr_xwayland_surface;
+        wl_list_insert(&root_container.sway_root->xwayland_unmanaged,
+                &surface->link);
+        wl_signal_add(&xsurface->surface->events.commit, &surface->commit);
+        surface->commit.notify = unmanaged_handle_commit;
+        // TODO: damage tracking
+}
+
+static void unmanaged_handle_unmap(struct wl_listener *listener, void *data) {
+        struct sway_xwayland_unmanaged *surface =
+                wl_container_of(listener, surface, unmap);
+        wl_list_remove(&surface->link);
+        wl_list_remove(&surface->commit.link);
+        // TODO: damage tracking
+}
+
 static void unmanaged_handle_destroy(struct wl_listener *listener, void *data) {
-        struct sway_xwayland_unmanaged *sway_surface =
-                wl_container_of(listener, sway_surface, destroy);
-        wl_list_remove(&sway_surface->destroy.link);
-        wl_list_remove(&sway_surface->link);
-        free(sway_surface);
+        struct sway_xwayland_unmanaged *surface =
+                wl_container_of(listener, surface, destroy);
+        struct wlr_xwayland_surface *xsurface = surface->wlr_xwayland_surface;
+        if (xsurface->mapped) {
+                unmanaged_handle_unmap(&surface->unmap, xsurface);
+        }
+        wl_list_remove(&surface->map.link);
+        wl_list_remove(&surface->unmap.link);
+        wl_list_remove(&surface->destroy.link);
+        free(surface);
 }
 
-static void create_unmanaged(struct wlr_xwayland_surface *xsurface) {
-        struct sway_xwayland_unmanaged *sway_surface =
+static struct sway_xwayland_unmanaged *create_unmanaged(
+                struct wlr_xwayland_surface *xsurface) {
+        struct sway_xwayland_unmanaged *surface =
                 calloc(1, sizeof(struct sway_xwayland_unmanaged));
-        if (!sway_assert(sway_surface, "Failed to allocate surface")) {
-                return;
+        if (surface == NULL) {
+                wlr_log(L_ERROR, "Allocation failed");
+                return NULL;
         }
 
-        sway_surface->wlr_xwayland_surface = xsurface;
+        surface->wlr_xwayland_surface = xsurface;
 
-        wl_signal_add(&xsurface->events.destroy, &sway_surface->destroy);
-        sway_surface->destroy.notify = unmanaged_handle_destroy;
+        wl_signal_add(&xsurface->events.map, &surface->map);
+        surface->map.notify = unmanaged_handle_map;
+        wl_signal_add(&xsurface->events.unmap, &surface->unmap);
+        surface->unmap.notify = unmanaged_handle_unmap;
+        wl_signal_add(&xsurface->events.destroy, &surface->destroy);
+        surface->destroy.notify = unmanaged_handle_destroy;
 
-        wl_list_insert(&root_container.sway_root->xwayland_unmanaged,
-                &sway_surface->link);
+        unmanaged_handle_map(&surface->map, xsurface);
 
-        // TODO: damage tracking
+        return surface;
 }
 
 
@@ -127,6 +162,7 @@ static const struct sway_view_impl view_impl = {
         .configure = configure,
         .set_activated = set_activated,
         .close = _close,
+        .destroy = destroy,
 };
 
 static void handle_commit(struct wl_listener *listener, void *data) {