2 files changed, 145 insertions, 0 deletions

diff --git a/include/sway/tree/view.h b/include/sway/tree/view.h
index d4dace4a..f8e41652 100644
--- a/include/sway/tree/view.h
+++ b/include/sway/tree/view.h
@@ -45,6 +45,12 @@ struct sway_view {
 		struct wlr_xwayland_surface *wlr_xwayland_surface;
 		struct wlr_wl_shell_surface *wlr_wl_shell_surface;
 	};
+
+	struct {
+		struct wl_signal unmap;
+	} events;
+
+	struct wl_listener surface_new_subsurface;
 };
 
 struct sway_xdg_shell_v6_view {
@@ -95,6 +101,27 @@ struct sway_wl_shell_view {
 	int pending_width, pending_height;
 };
 
+struct sway_view_child;
+
+struct sway_view_child_impl {
+	void (*destroy)(struct sway_view_child *child);
+};
+
+/**
+ * A view child is a surface in the view tree, such as a subsurface or a popup.
+ */
+struct sway_view_child {
+	const struct sway_view_child_impl *impl;
+
+	struct sway_view *view;
+	struct wlr_surface *surface;
+
+	struct wl_listener surface_commit;
+	struct wl_listener surface_new_subsurface;
+	struct wl_listener surface_destroy;
+	struct wl_listener view_unmap;
+};
+
 const char *view_get_title(struct sway_view *view);
 
 const char *view_get_app_id(struct sway_view *view);
@@ -129,4 +156,10 @@ void view_update_position(struct sway_view *view, double ox, double oy);
 
 void view_update_size(struct sway_view *view, int width, int height);
 
+void view_child_init(struct sway_view_child *child,
+	const struct sway_view_child_impl *impl, struct sway_view *view,
+	struct wlr_surface *surface);
+
+void view_child_destroy(struct sway_view_child *child);
+
 #endif
diff --git a/sway/tree/view.c b/sway/tree/view.c
index 3927c195..85fd3093 100644
--- a/sway/tree/view.c
+++ b/sway/tree/view.c
@@ -11,6 +11,7 @@ void view_init(struct sway_view *view, enum sway_view_type type,
 		const struct sway_view_impl *impl) {
 	view->type = type;
 	view->impl = impl;
+	wl_signal_init(&view->events.unmap);
 }
 
 void view_destroy(struct sway_view *view) {
@@ -123,6 +124,20 @@ static void view_update_outputs(struct sway_view *view,
 	}
 }
 
+static void view_subsurface_create(struct sway_view *view,
+	struct wlr_subsurface *subsurface);
+
+static void view_init_subsurfaces(struct sway_view *view,
+	struct wlr_surface *surface);
+
+static void view_handle_surface_new_subsurface(struct wl_listener *listener,
+		void *data) {
+	struct sway_view *view =
+		wl_container_of(listener, view, surface_new_subsurface);
+	struct wlr_subsurface *subsurface = data;
+	view_subsurface_create(view, subsurface);
+}
+
 void view_map(struct sway_view *view, struct wlr_surface *wlr_surface) {
 	if (!sway_assert(view->surface == NULL, "cannot map mapped view")) {
 		return;
@@ -136,11 +151,18 @@ void view_map(struct sway_view *view, struct wlr_surface *wlr_surface) {
 	view->surface = wlr_surface;
 	view->swayc = cont;
 
+	view_init_subsurfaces(view, wlr_surface);
+	wl_signal_add(&wlr_surface->events.new_subsurface,
+		&view->surface_new_subsurface);
+	view->surface_new_subsurface.notify = view_handle_surface_new_subsurface;
+
 	arrange_windows(cont->parent, -1, -1);
 	input_manager_set_focus(input_manager, cont);
 
 	view_damage_whole(view);
 	view_update_outputs(view, NULL);
+
+	// TODO: create view children for subsurfaces
 }
 
 void view_unmap(struct sway_view *view) {
@@ -148,6 +170,8 @@ void view_unmap(struct sway_view *view) {
 		return;
 	}
 
+	wl_signal_emit(&view->events.unmap, view);
+
 	view_damage_whole(view);
 
 	struct sway_container *parent = container_destroy(view->swayc);
@@ -185,3 +209,91 @@ void view_update_size(struct sway_view *view, int width, int height) {
 	view_update_outputs(view, &box);
 	view_damage_whole(view);
 }
+
+
+static void view_subsurface_create(struct sway_view *view,
+		struct wlr_subsurface *subsurface) {
+	struct sway_view_child *child = calloc(1, sizeof(struct sway_view_child));
+	if (child == NULL) {
+		wlr_log(L_ERROR, "Allocation failed");
+		return;
+	}
+	view_child_init(child, NULL, view, subsurface->surface);
+}
+
+static void view_child_handle_surface_commit(struct wl_listener *listener,
+		void *data) {
+	struct sway_view_child *child =
+		wl_container_of(listener, child, surface_commit);
+	// TODO: only accumulate damage from the child
+	view_damage_from(child->view);
+}
+
+static void view_child_handle_surface_new_subsurface(
+		struct wl_listener *listener, void *data) {
+	struct sway_view_child *child =
+		wl_container_of(listener, child, surface_new_subsurface);
+	struct wlr_subsurface *subsurface = data;
+	view_subsurface_create(child->view, subsurface);
+}
+
+static void view_child_handle_surface_destroy(struct wl_listener *listener,
+		void *data) {
+	struct sway_view_child *child =
+		wl_container_of(listener, child, surface_destroy);
+	view_child_destroy(child);
+}
+
+static void view_child_handle_view_unmap(struct wl_listener *listener,
+		void *data) {
+	struct sway_view_child *child =
+		wl_container_of(listener, child, view_unmap);
+	view_child_destroy(child);
+}
+
+static void view_init_subsurfaces(struct sway_view *view,
+		struct wlr_surface *surface) {
+	struct wlr_subsurface *subsurface;
+	wl_list_for_each(subsurface, &surface->subsurface_list, parent_link) {
+		view_subsurface_create(view, subsurface);
+	}
+}
+
+void view_child_init(struct sway_view_child *child,
+		const struct sway_view_child_impl *impl, struct sway_view *view,
+		struct wlr_surface *surface) {
+	child->impl = impl;
+	child->view = view;
+	child->surface = surface;
+
+	wl_signal_add(&surface->events.commit, &child->surface_commit);
+	child->surface_commit.notify = view_child_handle_surface_commit;
+	wl_signal_add(&surface->events.new_subsurface,
+		&child->surface_new_subsurface);
+	child->surface_new_subsurface.notify =
+		view_child_handle_surface_new_subsurface;
+	wl_signal_add(&surface->events.destroy, &child->surface_destroy);
+	child->surface_destroy.notify = view_child_handle_surface_destroy;
+	wl_signal_add(&view->events.unmap, &child->view_unmap);
+	child->view_unmap.notify = view_child_handle_view_unmap;
+
+	view_init_subsurfaces(child->view, surface);
+
+	// TODO: only damage the whole child
+	view_damage_whole(child->view);
+}
+
+void view_child_destroy(struct sway_view_child *child) {
+	// TODO: only damage the whole child
+	view_damage_whole(child->view);
+
+	wl_list_remove(&child->surface_commit.link);
+	wl_list_remove(&child->surface_destroy.link);
+	wl_list_remove(&child->view_unmap.link);
+
+	if (child->impl && child->impl->destroy) {
+		child->impl->destroy(child);
+	} else {
+		free(child);
+	}
+}