init

2026-07-12 01:07:23 +08:00
commit 01a1dbec8e
39 changed files with 12038 additions and 0 deletions
@@ -0,0 +1,210 @@
+/* My shitty port of Mango's ext-workspace.h for my compositor; 
+ * thanks DreamMaoMao for doing most of the heavy lifting 
+ * by actually writing the thing, it didn't work the first 
+ * few tries and I barely have any idea how this code works anyway. 
+ * To say that it's the eighth wonder of the world 
+ * would be an understatement
+*/
+
+#include "wlr_ext_workspace_v1.h"
+
+#define EXT_WORKSPACE_CAPS \
+	(EXT_WORKSPACE_HANDLE_V1_WORKSPACE_CAPABILITIES_ACTIVATE | \
+	 EXT_WORKSPACE_HANDLE_V1_WORKSPACE_CAPABILITIES_DEACTIVATE)
+
+/* types */
+
+struct ext_workspace {
+	struct wl_list link;           
+	uint32_t tag;                  
+	Monitor *m;
+	struct wlr_ext_workspace_handle_v1 *handle;
+};
+
+/* global */
+
+static struct wlr_ext_workspace_manager_v1 *ext_workspace_manager;
+static struct wl_list ext_workspaces; 
+static struct wl_listener ext_commit_listener;
+static struct wl_listener ext_destroy_listener;
+
+/* helpers */
+
+
+static int
+get_tag_status(uint32_t tag, Monitor *m)
+{
+	Client *c;
+	uint32_t mask = 1 << (tag - 1);
+
+	wl_list_for_each(c, &clients, link) {
+		if (c->mon != m)
+			continue;
+		if (!(c->tags & mask))
+			continue;
+		if (c->isurgent)
+			return 2;
+		return 1;
+	}
+	return 0;
+}
+
+static const char *
+tag_name(uint32_t tag)
+{
+	static const char *names[] = {
+		"1", "2", "3", "4", "5", "6", "7", "8", "9",
+	};
+	if (tag >= 1 && tag <= LENGTH(names))
+		return names[tag - 1];
+	return "?";
+}
+
+/* workspace lifecycle */
+
+static void
+ext_workspace_destroy(struct ext_workspace *ws)
+{
+	wlr_ext_workspace_handle_v1_destroy(ws->handle);
+	wl_list_remove(&ws->link);
+	free(ws);
+}
+
+static void
+ext_workspace_create(uint32_t tag, Monitor *m)
+{
+	const char *name = tag_name(tag);
+
+	struct ext_workspace *ws = ecalloc(1, sizeof(*ws));
+	ws->tag = tag;
+	ws->m = m;
+	ws->handle = wlr_ext_workspace_handle_v1_create(
+			ext_workspace_manager, name, EXT_WORKSPACE_CAPS);
+	ws->handle->data = ws;
+
+	wlr_ext_workspace_handle_v1_set_group(ws->handle, m->ext_group);
+	wlr_ext_workspace_handle_v1_set_name(ws->handle, name);
+
+	wl_list_insert(ext_workspaces.prev, &ws->link);
+}
+
+/* monitor integration */
+
+static void
+ext_workspace_createmon(Monitor *m)
+{
+	uint32_t i;
+
+	m->ext_group = wlr_ext_workspace_group_handle_v1_create(
+			ext_workspace_manager, 0);
+	wlr_ext_workspace_group_handle_v1_output_enter(
+			m->ext_group, m->wlr_output);
+
+	for (i = 1; i <= TAGCOUNT; i++)
+		ext_workspace_create(i, m);
+}
+
+static void
+ext_workspace_cleanupmon(Monitor *m)
+{
+	struct ext_workspace *ws, *tmp;
+	wl_list_for_each_safe(ws, tmp, &ext_workspaces, link)
+		if (ws->m == m)
+			ext_workspace_destroy(ws);
+
+	wlr_ext_workspace_group_handle_v1_output_leave(
+			m->ext_group, m->wlr_output);
+	wlr_ext_workspace_group_handle_v1_destroy(m->ext_group);
+	m->ext_group = NULL;
+}
+
+/* status */
+
+static void
+ext_workspace_printstatus(Monitor *m)
+{
+	struct ext_workspace *ws;
+
+	wl_list_for_each(ws, &ext_workspaces, link) {
+		if (ws->m != m)
+			continue;
+
+		int status = get_tag_status(ws->tag, m);
+		int active = !!(m->tagset[m->seltags] & (1 << (ws->tag - 1)) & TAGMASK);
+
+		wlr_ext_workspace_handle_v1_set_urgent(ws->handle, status == 2);
+		wlr_ext_workspace_handle_v1_set_hidden(ws->handle, 0);
+		wlr_ext_workspace_handle_v1_set_active(ws->handle, active);
+	}
+}
+
+/* commit handler */
+
+static void
+handle_ext_commit(struct wl_listener *listener, void *data)
+{
+	struct wlr_ext_workspace_v1_commit_event *event = data;
+	struct wlr_ext_workspace_v1_request *req;
+
+	wl_list_for_each(req, event->requests, link) {
+		struct ext_workspace *ws = NULL;
+		struct ext_workspace *w;
+
+		switch (req->type) {
+		case WLR_EXT_WORKSPACE_V1_REQUEST_ACTIVATE:
+			if (!req->activate.workspace)
+				break;
+			wl_list_for_each(w, &ext_workspaces, link) {
+				if (w->handle == req->activate.workspace) {
+					ws = w;
+					break;
+				}
+			}
+			if (!ws)
+				break;
+			view(&(Arg){.ui = 1 << (ws->tag - 1)});
+			break;
+
+		case WLR_EXT_WORKSPACE_V1_REQUEST_DEACTIVATE:
+			if (!req->deactivate.workspace)
+				break;
+			wl_list_for_each(w, &ext_workspaces, link) {
+				if (w->handle == req->deactivate.workspace) {
+					ws = w;
+					break;
+				}
+			}
+			if (!ws)
+				break;
+			toggleview(&(Arg){.ui = 1 << (ws->tag - 1)});
+			break;
+
+		default:
+			break;
+		}
+	}
+}
+
+/* init */
+
+static void
+handle_ext_destroy(struct wl_listener *listener, void *data)
+{
+	wl_list_remove(&ext_commit_listener.link);
+	wl_list_remove(&ext_destroy_listener.link);
+}
+
+static void
+workspaces_init(void)
+{
+	ext_workspace_manager = wlr_ext_workspace_manager_v1_create(dpy, 1);
+	wl_list_init(&ext_workspaces);
+
+	ext_commit_listener.notify = handle_ext_commit;
+	wl_signal_add(&ext_workspace_manager->events.commit,
+			&ext_commit_listener);
+
+	ext_destroy_listener.notify = handle_ext_destroy;
+	wl_signal_add(&ext_workspace_manager->events.destroy,
+			&ext_destroy_listener);
+}