From 6284eea24b44d05260f96fe842fa9dd752185942 Mon Sep 17 00:00:00 2001
From: adnano <me@adnano.co>
Date: Thu, 2 May 2024 17:03:07 -0400
Subject: Separate menu state from Wayland state
---
main.c | 331 ++--------------------------------------
menu.c | 179 +++++-----------------
menu.h | 51 +------
meson.build | 1 +
render.c | 16 +-
wayland.c | 492 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
wayland.h | 17 +++
7 files changed, 578 insertions(+), 509 deletions(-)
create mode 100644 wayland.c
create mode 100644 wayland.h
diff --git a/main.c b/main.c
index 710f75f..be21e33 100644
--- a/main.c
+++ b/main.c
@@ -1,329 +1,24 @@
#define _POSIX_C_SOURCE 200809L
-#include <assert.h>
-#include <errno.h>
-#include <poll.h>
-#include <stdbool.h>
-#include <signal.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <strings.h>
-#include <time.h>
-#include <unistd.h>
-#include <sys/mman.h>
-#include <sys/timerfd.h>
-#include <wayland-client.h>
-#include <wayland-client-protocol.h>
-#include <xkbcommon/xkbcommon.h>
#include "menu.h"
-#include "render.h"
-#include "xdg-activation-v1-client-protocol.h"
-#include "wlr-layer-shell-unstable-v1-client-protocol.h"
-
-static void noop() {
- // Do nothing
-}
-
-static void surface_enter(void *data, struct wl_surface *surface, struct wl_output *wl_output) {
- struct menu *menu = data;
- menu->output = wl_output_get_user_data(wl_output);
-}
-
-static const struct wl_surface_listener surface_listener = {
- .enter = surface_enter,
- .leave = noop,
-};
-
-static void layer_surface_configure(void *data,
- struct zwlr_layer_surface_v1 *surface,
- uint32_t serial, uint32_t width, uint32_t height) {
- struct menu *menu = data;
- menu->width = width;
- menu->height = height;
- zwlr_layer_surface_v1_ack_configure(surface, serial);
-}
-
-static void layer_surface_closed(void *data, struct zwlr_layer_surface_v1 *surface) {
- struct menu *menu = data;
- menu->exit = true;
-}
-
-static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
- .configure = layer_surface_configure,
- .closed = layer_surface_closed,
-};
-
-static void output_scale(void *data, struct wl_output *wl_output, int32_t factor) {
- struct output *output = data;
- output->scale = factor;
-}
-
-static void output_name(void *data, struct wl_output *wl_output, const char *name) {
- struct output *output = data;
- output->name = name;
-
- struct menu *menu = output->menu;
- if (menu->output_name && strcmp(menu->output_name, name) == 0) {
- menu->output = output;
- }
-}
-
-static const struct wl_output_listener output_listener = {
- .geometry = noop,
- .mode = noop,
- .done = noop,
- .scale = output_scale,
- .name = output_name,
- .description = noop,
-};
-
-static void keyboard_keymap(void *data, struct wl_keyboard *wl_keyboard,
- uint32_t format, int32_t fd, uint32_t size) {
- struct keyboard *keyboard = data;
- assert(format == WL_KEYBOARD_KEYMAP_FORMAT_XKB_V1);
-
- char *map_shm = mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0);
- assert(map_shm != MAP_FAILED);
-
- keyboard->keymap = xkb_keymap_new_from_string(keyboard->context,
- map_shm, XKB_KEYMAP_FORMAT_TEXT_V1, 0);
- munmap(map_shm, size);
- close(fd);
-
- keyboard->state = xkb_state_new(keyboard->keymap);
-}
-
-static void keyboard_repeat(struct keyboard *keyboard) {
- menu_keypress(keyboard->menu, keyboard->repeat_key_state, keyboard->repeat_sym);
- struct itimerspec spec = { 0 };
- spec.it_value.tv_sec = keyboard->repeat_period / 1000;
- spec.it_value.tv_nsec = (keyboard->repeat_period % 1000) * 1000000l;
- timerfd_settime(keyboard->repeat_timer, 0, &spec, NULL);
-}
-
-static void keyboard_key(void *data, struct wl_keyboard *wl_keyboard,
- uint32_t serial, uint32_t time, uint32_t key, uint32_t _key_state) {
- struct keyboard *keyboard = data;
-
- enum wl_keyboard_key_state key_state = _key_state;
- xkb_keysym_t sym = xkb_state_key_get_one_sym(keyboard->state, key + 8);
- menu_keypress(keyboard->menu, key_state, sym);
-
- if (key_state == WL_KEYBOARD_KEY_STATE_PRESSED && keyboard->repeat_period >= 0) {
- keyboard->repeat_key_state = key_state;
- keyboard->repeat_sym = sym;
-
- struct itimerspec spec = { 0 };
- spec.it_value.tv_sec = keyboard->repeat_delay / 1000;
- spec.it_value.tv_nsec = (keyboard->repeat_delay % 1000) * 1000000l;
- timerfd_settime(keyboard->repeat_timer, 0, &spec, NULL);
- } else if (key_state == WL_KEYBOARD_KEY_STATE_RELEASED) {
- struct itimerspec spec = { 0 };
- timerfd_settime(keyboard->repeat_timer, 0, &spec, NULL);
- }
-}
-
-static void keyboard_repeat_info(void *data, struct wl_keyboard *wl_keyboard,
- int32_t rate, int32_t delay) {
- struct keyboard *keyboard = data;
- keyboard->repeat_delay = delay;
- if (rate > 0) {
- keyboard->repeat_period = 1000 / rate;
- } else {
- keyboard->repeat_period = -1;
- }
-}
-
-static void keyboard_modifiers(void *data, struct wl_keyboard *wl_keyboard,
- uint32_t serial, uint32_t mods_depressed,
- uint32_t mods_latched, uint32_t mods_locked,
- uint32_t group) {
- struct keyboard *keyboard = data;
- xkb_state_update_mask(keyboard->state, mods_depressed, mods_latched,
- mods_locked, 0, 0, group);
-}
-
-static const struct wl_keyboard_listener keyboard_listener = {
- .keymap = keyboard_keymap,
- .enter = noop,
- .leave = noop,
- .key = keyboard_key,
- .modifiers = keyboard_modifiers,
- .repeat_info = keyboard_repeat_info,
-};
-
-static void seat_capabilities(void *data, struct wl_seat *seat,
- enum wl_seat_capability caps) {
- struct menu *menu = data;
- if (caps & WL_SEAT_CAPABILITY_KEYBOARD) {
- struct wl_keyboard *wl_keyboard = wl_seat_get_keyboard(seat);
- struct keyboard *keyboard = keyboard_create(menu, wl_keyboard);
- wl_keyboard_add_listener(wl_keyboard, &keyboard_listener, keyboard);
- menu_set_keyboard(menu, keyboard);
- }
-}
-
-static const struct wl_seat_listener seat_listener = {
- .capabilities = seat_capabilities,
- .name = noop,
-};
-
-static void data_device_selection(void *data, struct wl_data_device *data_device,
- struct wl_data_offer *data_offer) {
- struct menu *menu = data;
- menu->data_offer = data_offer;
-}
-
-static const struct wl_data_device_listener data_device_listener = {
- .data_offer = noop,
- .enter = noop,
- .leave = noop,
- .motion = noop,
- .drop = noop,
- .selection = data_device_selection,
-};
-
-static void handle_global(void *data, struct wl_registry *registry,
- uint32_t name, const char *interface, uint32_t version) {
- struct menu *menu = data;
- if (strcmp(interface, wl_compositor_interface.name) == 0) {
- menu->compositor = wl_registry_bind(registry, name, &wl_compositor_interface, 4);
- } else if (strcmp(interface, wl_shm_interface.name) == 0) {
- menu->shm = wl_registry_bind(registry, name, &wl_shm_interface, 1);
- } else if (strcmp(interface, wl_seat_interface.name) == 0) {
- menu->seat = wl_registry_bind(registry, name, &wl_seat_interface, 4);
- wl_seat_add_listener(menu->seat, &seat_listener, menu);
- } else if (strcmp(interface, wl_data_device_manager_interface.name) == 0) {
- menu->data_device_manager = wl_registry_bind(registry, name, &wl_data_device_manager_interface, 3);
- } else if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
- menu->layer_shell = wl_registry_bind(registry, name, &zwlr_layer_shell_v1_interface, 1);
- } else if (strcmp(interface, wl_output_interface.name) == 0) {
- struct wl_output *wl_output = wl_registry_bind(registry, name, &wl_output_interface, 4);
- struct output *output = output_create(menu, wl_output);
- wl_output_set_user_data(wl_output, output);
- wl_output_add_listener(wl_output, &output_listener, output);
- menu_add_output(menu, output);
- } else if (strcmp(interface, xdg_activation_v1_interface.name) == 0) {
- menu->activation = wl_registry_bind(registry, name, &xdg_activation_v1_interface, 1);
- }
-}
-
-static const struct wl_registry_listener registry_listener = {
- .global = handle_global,
- .global_remove = noop,
-};
-
-// Connect to the Wayland display.
-static void menu_connect(struct menu *menu) {
- menu->display = wl_display_connect(NULL);
- if (!menu->display) {
- fprintf(stderr, "Failed to connect to display.\n");
- exit(EXIT_FAILURE);
- }
-
- struct wl_registry *registry = wl_display_get_registry(menu->display);
- wl_registry_add_listener(registry, ®istry_listener, menu);
- wl_display_roundtrip(menu->display);
- assert(menu->compositor != NULL);
- assert(menu->shm != NULL);
- assert(menu->seat != NULL);
- assert(menu->data_device_manager != NULL);
- assert(menu->layer_shell != NULL);
- assert(menu->activation != NULL);
- menu->registry = registry;
-
- // Get data device for seat
- struct wl_data_device *data_device = wl_data_device_manager_get_data_device(
- menu->data_device_manager, menu->seat);
- wl_data_device_add_listener(data_device, &data_device_listener, menu);
- menu->data_device = data_device;
-
- // Second roundtrip for seat and output listeners
- wl_display_roundtrip(menu->display);
- assert(menu->keyboard != NULL);
-
- if (menu->output_name && !menu->output) {
- fprintf(stderr, "Output %s not found\n", menu->output_name);
- exit(EXIT_FAILURE);
- }
-
- menu->surface = wl_compositor_create_surface(menu->compositor);
- wl_surface_add_listener(menu->surface, &surface_listener, menu);
-
- struct zwlr_layer_surface_v1 *layer_surface = zwlr_layer_shell_v1_get_layer_surface(
- menu->layer_shell,
- menu->surface,
- menu->output ? menu->output->output : NULL,
- ZWLR_LAYER_SHELL_V1_LAYER_TOP,
- "menu"
- );
- assert(layer_surface != NULL);
- menu->layer_surface = layer_surface;
-
- uint32_t anchor = ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT |
- ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
- if (menu->bottom) {
- anchor |= ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM;
- } else {
- anchor |= ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP;
+#include "wayland.h"
+
+static void read_items(struct menu *menu) {
+ char buf[sizeof menu->input];
+ while (fgets(buf, sizeof buf, stdin)) {
+ char *p = strchr(buf, '\n');
+ if (p) {
+ *p = '\0';
+ }
+ menu_add_item(menu, strdup(buf));
}
-
- zwlr_layer_surface_v1_set_anchor(layer_surface, anchor);
- zwlr_layer_surface_v1_set_size(layer_surface, 0, menu->height);
- zwlr_layer_surface_v1_set_exclusive_zone(layer_surface, -1);
- zwlr_layer_surface_v1_set_keyboard_interactivity(layer_surface, true);
- zwlr_layer_surface_v1_add_listener(layer_surface, &layer_surface_listener, menu);
-
- wl_surface_commit(menu->surface);
- wl_display_roundtrip(menu->display);
}
int main(int argc, char *argv[]) {
struct menu *menu = menu_create();
menu_getopts(menu, argc, argv);
- menu_connect(menu);
- render_menu(menu);
-
- read_menu_items(menu);
- render_menu(menu);
-
- struct pollfd fds[] = {
- { wl_display_get_fd(menu->display), POLLIN },
- { menu->keyboard->repeat_timer, POLLIN },
- };
- const size_t nfds = sizeof(fds) / sizeof(*fds);
-
- while (!menu->exit) {
- errno = 0;
- do {
- if (wl_display_flush(menu->display) == -1 && errno != EAGAIN) {
- fprintf(stderr, "wl_display_flush: %s\n", strerror(errno));
- break;
- }
- } while (errno == EAGAIN);
-
- if (poll(fds, nfds, -1) < 0) {
- fprintf(stderr, "poll: %s\n", strerror(errno));
- break;
- }
-
- if (fds[0].revents & POLLIN) {
- if (wl_display_dispatch(menu->display) < 0) {
- menu->exit = true;
- }
- }
-
- if (fds[1].revents & POLLIN) {
- keyboard_repeat(menu->keyboard);
- }
- }
-
- bool failure = menu->failure;
- menu_destroy(menu);
-
- if (failure) {
- return EXIT_FAILURE;
+ if (!menu->passwd) {
+ read_items(menu);
}
- return EXIT_SUCCESS;
+ return menu_run(menu);
}
diff --git a/menu.c b/menu.c
index 72d903f..eecf1a0 100644
--- a/menu.c
+++ b/menu.c
@@ -1,5 +1,4 @@
#define _POSIX_C_SOURCE 200809L
-#include <assert.h>
#include <ctype.h>
#include <poll.h>
#include <stdbool.h>
@@ -19,10 +18,8 @@
#include "menu.h"
#include "pango.h"
-#include "pool-buffer.h"
#include "render.h"
-#include "xdg-activation-v1-client-protocol.h"
-#include "wlr-layer-shell-unstable-v1-client-protocol.h"
+#include "wayland.h"
// Creates and returns a new menu.
struct menu *menu_create() {
@@ -38,38 +35,6 @@ struct menu *menu_create() {
return menu;
}
-// Creates and returns a new keyboard.
-struct keyboard *keyboard_create(struct menu *menu, struct wl_keyboard *wl_keyboard) {
- struct keyboard *keyboard = calloc(1, sizeof(struct keyboard));
- keyboard->menu = menu;
- keyboard->keyboard = wl_keyboard;
- keyboard->context = xkb_context_new(XKB_CONTEXT_NO_FLAGS);
- assert(keyboard->context != NULL);
- keyboard->repeat_timer = timerfd_create(CLOCK_MONOTONIC, 0);
- assert(keyboard->repeat_timer != -1);
- return keyboard;
-}
-
-// Sets the current keyboard.
-void menu_set_keyboard(struct menu *menu, struct keyboard *keyboard) {
- menu->keyboard = keyboard;
-}
-
-// Creates and returns a new output.
-struct output *output_create(struct menu *menu, struct wl_output *wl_output) {
- struct output *output = calloc(1, sizeof(struct output));
- output->menu = menu;
- output->output = wl_output;
- output->scale = 1;
- return output;
-}
-
-// Adds an output to the output list.
-void menu_add_output(struct menu *menu, struct output *output) {
- output->next = menu->output_list;
- menu->output_list = output;
-}
-
static bool parse_color(const char *color, uint32_t *result) {
if (color[0] == '#') {
++color;
@@ -170,6 +135,22 @@ void menu_getopts(struct menu *menu, int argc, char *argv[]) {
menu->padding = height / 2;
}
+// Add an item to the menu.
+void menu_add_item(struct menu *menu, char *text) {
+ struct item *item = calloc(1, sizeof *item);
+ if (!item) {
+ return;
+ }
+ item->text = text;
+
+ if (menu->lastitem) {
+ menu->lastitem->next = item;
+ } else {
+ menu->items = item;
+ }
+ menu->lastitem = item;
+}
+
static void append_page(struct page *page, struct page **first, struct page **last) {
if (*last) {
(*last)->next = page;
@@ -246,7 +227,7 @@ static const char *fstrstr(struct menu *menu, const char *s, const char *sub) {
return NULL;
}
-static void append_item(struct item *item, struct item **first, struct item **last) {
+static void append_match(struct item *item, struct item **first, struct item **last) {
if (*last) {
(*last)->next_match = item;
} else {
@@ -300,11 +281,11 @@ static void match_items(struct menu *menu) {
continue;
}
if (!tokc || !menu->strncmp(menu->input, item->text, input_len + 1)) {
- append_item(item, &lexact, &exactend);
+ append_match(item, &lexact, &exactend);
} else if (!menu->strncmp(tokv[0], item->text, tok_len)) {
- append_item(item, &lprefix, &prefixend);
+ append_match(item, &lprefix, &prefixend);
} else {
- append_item(item, &lsubstr, &substrend);
+ append_match(item, &lsubstr, &substrend);
}
}
@@ -339,45 +320,27 @@ static void match_items(struct menu *menu) {
}
}
-// Read menu items from standard input.
-void read_menu_items(struct menu *menu) {
- if (menu->passwd) {
- // Don't read standard input in password mode
- calc_widths(menu);
- return;
- }
-
- char buf[sizeof menu->input];
- struct item **next = &menu->items;
- while (fgets(buf, sizeof buf, stdin)) {
- char *p = strchr(buf, '\n');
- if (p) {
- *p = '\0';
- }
- struct item *item = calloc(1, sizeof *item);
- if (!item) {
- return;
- }
- item->text = strdup(buf);
-
- *next = item;
- next = &item->next;
- }
-
+// Process menu items.
+void menu_process_items(struct menu *menu) {
calc_widths(menu);
match_items(menu);
}
-static void insert(struct menu *menu, const char *s, ssize_t n) {
- if (strlen(menu->input) + n > sizeof menu->input - 1) {
+static void insert(struct menu *menu, const char *text, ssize_t len) {
+ if (strlen(menu->input) + len > sizeof menu->input - 1) {
return;
}
- memmove(menu->input + menu->cursor + n, menu->input + menu->cursor,
- sizeof menu->input - menu->cursor - MAX(n, 0));
- if (n > 0 && s != NULL) {
- memcpy(menu->input + menu->cursor, s, n);
+ memmove(menu->input + menu->cursor + len, menu->input + menu->cursor,
+ sizeof menu->input - menu->cursor - MAX(len, 0));
+ if (len > 0 && text != NULL) {
+ memcpy(menu->input + menu->cursor, text, len);
}
- menu->cursor += n;
+ menu->cursor += len;
+}
+
+// Add pasted text to the menu input.
+void menu_paste(struct menu *menu, const char *text, ssize_t len) {
+ insert(menu, text, len);
}
static size_t nextrune(struct menu *menu, int incr) {
@@ -417,14 +380,12 @@ void menu_keypress(struct menu *menu, enum wl_keyboard_key_state key_state,
return;
}
- bool ctrl = xkb_state_mod_name_is_active(menu->keyboard->state,
- XKB_MOD_NAME_CTRL,
+ struct xkb_state *state = context_get_xkb_state(menu->context);
+ bool ctrl = xkb_state_mod_name_is_active(state, XKB_MOD_NAME_CTRL,
XKB_STATE_MODS_DEPRESSED | XKB_STATE_MODS_LATCHED);
- bool meta = xkb_state_mod_name_is_active(menu->keyboard->state,
- XKB_MOD_NAME_ALT,
+ bool meta = xkb_state_mod_name_is_active(state, XKB_MOD_NAME_ALT,
XKB_STATE_MODS_DEPRESSED | XKB_STATE_MODS_LATCHED);
- bool shift = xkb_state_mod_name_is_active(menu->keyboard->state,
- XKB_MOD_NAME_SHIFT,
+ bool shift = xkb_state_mod_name_is_active(state, XKB_MOD_NAME_SHIFT,
XKB_STATE_MODS_DEPRESSED | XKB_STATE_MODS_LATCHED);
size_t len = strlen(menu->input);
@@ -501,32 +462,9 @@ void menu_keypress(struct menu *menu, enum wl_keyboard_key_state key_state,
return;
case XKB_KEY_Y:
// Paste clipboard
- if (!menu->data_offer) {
+ if (!context_paste(menu->context)) {
return;
}
-
- int fds[2];
- if (pipe(fds) == -1) {
- // Pipe failed
- return;
- }
- wl_data_offer_receive(menu->data_offer, "text/plain", fds[1]);
- close(fds[1]);
-
- wl_display_roundtrip(menu->display);
-
- while (true) {
- char buf[1024];
- ssize_t n = read(fds[0], buf, sizeof(buf));
- if (n <= 0) {
- break;
- }
- insert(menu, buf, n);
- }
- close(fds[0]);
-
- wl_data_offer_destroy(menu->data_offer);
- menu->data_offer = NULL;
match_items(menu);
render_menu(menu);
return;
@@ -696,26 +634,6 @@ void menu_keypress(struct menu *menu, enum wl_keyboard_key_state key_state,
}
}
-// Frees the keyboard.
-static void free_keyboard(struct keyboard *keyboard) {
- wl_keyboard_release(keyboard->keyboard);
- xkb_state_unref(keyboard->state);
- xkb_keymap_unref(keyboard->keymap);
- xkb_context_unref(keyboard->context);
- free(keyboard);
-}
-
-// Frees the outputs.
-static void free_outputs(struct menu *menu) {
- struct output *next = menu->output_list;
- while (next) {
- struct output *output = next;
- next = output->next;
- wl_output_destroy(output->output);
- free(output);
- }
-}
-
// Frees menu pages.
static void free_pages(struct menu *menu) {
struct page *next = menu->pages;
@@ -739,26 +657,9 @@ static void free_items(struct menu *menu) {
// Destroys the menu, freeing memory associated with it.
void menu_destroy(struct menu *menu) {
- wl_registry_destroy(menu->registry);
- wl_compositor_destroy(menu->compositor);
- wl_shm_destroy(menu->shm);
- wl_seat_destroy(menu->seat);
- wl_data_device_manager_destroy(menu->data_device_manager);
- zwlr_layer_shell_v1_destroy(menu->layer_shell);
- free_outputs(menu);
-
- free_keyboard(menu->keyboard);
- wl_data_device_destroy(menu->data_device);
- wl_surface_destroy(menu->surface);
- zwlr_layer_surface_v1_destroy(menu->layer_surface);
- xdg_activation_v1_destroy(menu->activation);
-
free_pages(menu);
free_items(menu);
destroy_buffer(&menu->buffers[0]);
destroy_buffer(&menu->buffers[1]);
-
- wl_display_disconnect(menu->display);
- free(menu);
}
diff --git a/menu.h b/menu.h
index 0143405..6afdd7e 100644
--- a/menu.h
+++ b/menu.h
@@ -4,7 +4,6 @@
#include <xkbcommon/xkbcommon.h>
#include "pool-buffer.h"
-#include "xdg-activation-v1-client-protocol.h"
// A menu item.
struct item {
@@ -24,30 +23,6 @@ struct page {
struct page *next; // next page
};
-// A Wayland output.
-struct output {
- struct menu *menu;
- struct wl_output *output;
- const char *name; // output name
- int32_t scale; // output scale
- struct output *next; // next output
-};
-
-// Keyboard state.
-struct keyboard {
- struct menu *menu;
- struct wl_keyboard *keyboard;
- struct xkb_context *context;
- struct xkb_keymap *keymap;
- struct xkb_state *state;
-
- int repeat_timer;
- int repeat_delay;
- int repeat_period;
- enum wl_keyboard_key_state repeat_key_state;
- xkb_keysym_t repeat_sym;
-};
-
// Menu state.
struct menu {
// Whether the menu appears at the bottom of the screen
@@ -71,22 +46,7 @@ struct menu {
// Selection colors
uint32_t selectionbg, selectionfg;
- struct wl_display *display;
- struct wl_registry *registry;
- struct wl_compositor *compositor;
- struct wl_shm *shm;
- struct wl_seat *seat;
- struct wl_data_device_manager *data_device_manager;
- struct zwlr_layer_shell_v1 *layer_shell;
- struct output *output_list;
- struct xdg_activation_v1 *activation;
-
- struct keyboard *keyboard;
- struct wl_data_device *data_device;
- struct wl_surface *surface;
- struct zwlr_layer_surface_v1 *layer_surface;
- struct wl_data_offer *data_offer;
- struct output *output;
+ struct wl_context *context;
struct pool_buffer buffers[2];
struct pool_buffer *current;
@@ -104,6 +64,7 @@ struct menu {
size_t cursor;
struct item *items; // list of all items
+ struct item *lastitem; // last item in the list
struct item *matches; // list of matching items
struct item *matches_end; // last matching item
struct item *sel; // selected item
@@ -114,12 +75,10 @@ struct menu {
};
struct menu *menu_create();
-struct keyboard *keyboard_create(struct menu *menu, struct wl_keyboard *wl_keyboard);
-void menu_set_keyboard(struct menu *menu, struct keyboard *keyboard);
-struct output *output_create(struct menu *menu, struct wl_output *wl_output);
-void menu_add_output(struct menu *menu, struct output *output);
void menu_getopts(struct menu *menu, int argc, char *argv[]);
-void read_menu_items(struct menu *menu);
+void menu_add_item(struct menu *menu, char *text);
+void menu_process_items(struct menu *menu);
+void menu_paste(struct menu *menu, const char *text, ssize_t len);
void menu_keypress(struct menu *menu, enum wl_keyboard_key_state key_state,
xkb_keysym_t sym);
void menu_destroy(struct menu *menu);
diff --git a/meson.build b/meson.build
index 56fe511..fc4a2c0 100644
--- a/meson.build
+++ b/meson.build
@@ -43,6 +43,7 @@ executable(
'pango.c',
'pool-buffer.c',
'render.c',
+ 'wayland.c',
),
dependencies: [
cairo,
diff --git a/render.c b/render.c
index a0a2a8c..7142518 100644
--- a/render.c
+++ b/render.c
@@ -7,6 +7,7 @@
#include "menu.h"
#include "pango.h"
+#include "wayland.h"
// Calculate text widths.
void calc_widths(struct menu *menu) {
@@ -177,6 +178,8 @@ static void render_to_cairo(struct menu *menu, cairo_t *cairo) {
// Renders a single frame of the menu.
void render_menu(struct menu *menu) {
+ struct wl_context *context = menu->context;
+
cairo_surface_t *recorder = cairo_recording_surface_create(
CAIRO_CONTENT_COLOR_ALPHA, NULL);
cairo_t *cairo = cairo_create(recorder);
@@ -191,8 +194,8 @@ void render_menu(struct menu *menu) {
render_to_cairo(menu, cairo);
- int scale = menu->output ? menu->output->scale : 1;
- menu->current = get_next_buffer(menu->shm,
+ int scale = context_get_scale(context);
+ menu->current = get_next_buffer(context_get_shm(context),
menu->buffers, menu->width, menu->height, scale);
if (!menu->current) {
goto cleanup;
@@ -206,10 +209,11 @@ void render_menu(struct menu *menu) {
cairo_set_source_surface(shm, recorder, 0, 0);
cairo_paint(shm);
- wl_surface_set_buffer_scale(menu->surface, scale);
- wl_surface_attach(menu->surface, menu->current->buffer, 0, 0);
- wl_surface_damage(menu->surface, 0, 0, menu->width, menu->height);
- wl_surface_commit(menu->surface);
+ struct wl_surface *surface = context_get_surface(context);
+ wl_surface_set_buffer_scale(surface, scale);
+ wl_surface_attach(surface, menu->current->buffer, 0, 0);
+ wl_surface_damage(surface, 0, 0, menu->width, menu->height);
+ wl_surface_commit(surface);
cleanup:
cairo_destroy(cairo);
diff --git a/wayland.c b/wayland.c
new file mode 100644
index 0000000..efcad50
--- /dev/null
+++ b/wayland.c
@@ -0,0 +1,492 @@
+#define _POSIX_C_SOURCE 200809L
+#include <assert.h>
+#include <errno.h>
+#include <poll.h>
+#include <stdbool.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <strings.h>
+#include <time.h>
+#include <unistd.h>
+#include <sys/mman.h>
+#include <sys/timerfd.h>
+#include <wayland-client.h>
+#include <wayland-client-protocol.h>
+#include <xkbcommon/xkbcommon.h>
+
+#include "menu.h"
+#include "render.h"
+#include "wayland.h"
+#include "xdg-activation-v1-client-protocol.h"
+#include "wlr-layer-shell-unstable-v1-client-protocol.h"
+
+// A Wayland output.
+struct output {
+ struct wl_context *context;
+ struct wl_output *output;
+ const char *name; // output name
+ int32_t scale; // output scale
+ struct output *next; // next output
+};
+
+// Creates and returns a new output.
+static struct output *output_create(struct wl_context *context, struct wl_output *wl_output) {
+ struct output *output = calloc(1, sizeof(struct output));
+ output->context = context;
+ output->output = wl_output;
+ output->scale = 1;
+ return output;
+}
+
+// Keyboard state.
+struct keyboard {
+ struct menu *menu;
+ struct wl_keyboard *keyboard;
+ struct xkb_context *context;
+ struct xkb_keymap *keymap;
+ struct xkb_state *state;
+
+ int repeat_timer;
+ int repeat_delay;
+ int repeat_period;
+ enum wl_keyboard_key_state repeat_key_state;
+ xkb_keysym_t repeat_sym;
+};
+
+// Creates and returns a new keyboard.
+static struct keyboard *keyboard_create(struct menu *menu, struct wl_keyboard *wl_keyboard) {
+ struct keyboard *keyboard = calloc(1, sizeof(struct keyboard));
+ keyboard->menu = menu;
+ keyboard->keyboard = wl_keyboard;
+ keyboard->context = xkb_context_new(XKB_CONTEXT_NO_FLAGS);
+ assert(keyboard->context != NULL);
+ keyboard->repeat_timer = timerfd_create(CLOCK_MONOTONIC, 0);
+ assert(keyboard->repeat_timer != -1);
+ return keyboard;
+}
+
+// Frees the keyboard.
+static void free_keyboard(struct keyboard *keyboard) {
+ wl_keyboard_release(keyboard->keyboard);
+ xkb_state_unref(keyboard->state);
+ xkb_keymap_unref(keyboard->keymap);
+ xkb_context_unref(keyboard->context);
+ free(keyboard);
+}
+
+// Wayland context.
+struct wl_context {
+ struct menu *menu;
+
+ struct wl_display *display;
+ struct wl_registry *registry;
+ struct wl_compositor *compositor;
+ struct wl_shm *shm;
+ struct wl_seat *seat;
+ struct wl_data_device_manager *data_device_manager;
+ struct zwlr_layer_shell_v1 *layer_shell;
+ struct output *output_list;
+ struct xdg_activation_v1 *activation;
+
+ struct keyboard *keyboard;
+ struct wl_data_device *data_device;
+ struct wl_surface *surface;
+ struct zwlr_layer_surface_v1 *layer_surface;
+ struct wl_data_offer *data_offer;
+ struct output *output;
+};
+
+// Returns the current output_scale.
+int context_get_scale(struct wl_context *context) {
+ return context->output ? context->output->scale : 1;
+}
+
+// Returns the shared memory for the context.
+struct wl_shm *context_get_shm(struct wl_context *context) {
+ return context->shm;
+}
+
+// Returns the Wayland surface for the context.
+struct wl_surface *context_get_surface(struct wl_context *context) {
+ return context->surface;
+}
+
+// Returns the XKB state for the context.
+struct xkb_state *context_get_xkb_state(struct wl_context *context) {
+ return context->keyboard->state;
+}
+
+// Retrieves pasted text from a Wayland data offer.
+bool context_paste(struct wl_context *context) {
+ if (!context->data_offer) {
+ return false;
+ }
+
+ int fds[2];
+ if (pipe(fds) == -1) {
+ // Pipe failed
+ return false;
+ }
+ wl_data_offer_receive(context->data_offer, "text/plain", fds[1]);
+ close(fds[1]);
+
+ wl_display_roundtrip(context->display);
+
+ while (true) {
+ char buf[1024];
+ ssize_t n = read(fds[0], buf, sizeof(buf));
+ if (n <= 0) {
+ break;
+ }
+ menu_paste(context->menu, buf, n);
+ }
+ close(fds[0]);
+
+ wl_data_offer_destroy(context->data_offer);
+ context->data_offer = NULL;
+ return true;
+}
+
+// Adds an output to the output list.
+static void context_add_output(struct wl_context *context, struct output *output) {
+ output->next = context->output_list;
+ context->output_list = output;
+}
+
+// Frees the outputs.
+static void free_outputs(struct wl_context *context) {
+ struct output *next = context->output_list;
+ while (next) {
+ struct output *output = next;
+ next = output->next;
+ wl_output_destroy(output->output);
+ free(output);
+ }
+}
+
+// Destroys the Wayland context, freeing memory associated with it.
+static void context_destroy(struct wl_context *context) {
+ wl_registry_destroy(context->registry);
+ wl_compositor_destroy(context->compositor);
+ wl_shm_destroy(context->shm);
+ wl_seat_destroy(context->seat);
+ wl_data_device_manager_destroy(context->data_device_manager);
+ zwlr_layer_shell_v1_destroy(context->layer_shell);
+ free_outputs(context);
+
+ free_keyboard(context->keyboard);
+ wl_data_device_destroy(context->data_device);
+ wl_surface_destroy(context->surface);
+ zwlr_layer_surface_v1_destroy(context->layer_surface);
+ xdg_activation_v1_destroy(context->activation);
+
+ wl_display_disconnect(context->display);
+ free(context);
+}
+
+static void noop() {
+ // Do nothing
+}
+
+static void surface_enter(void *data, struct wl_surface *surface, struct wl_output *wl_output) {
+ struct wl_context *context = data;
+ context->output = wl_output_get_user_data(wl_output);
+}
+
+static const struct wl_surface_listener surface_listener = {
+ .enter = surface_enter,
+ .leave = noop,
+};
+
+static void layer_surface_configure(void *data,
+ struct zwlr_layer_surface_v1 *surface,
+ uint32_t serial, uint32_t width, uint32_t height) {
+ struct wl_context *context = data;
+ context->menu->width = width;
+ context->menu->height = height;
+ zwlr_layer_surface_v1_ack_configure(surface, serial);
+}
+
+static void layer_surface_closed(void *data, struct zwlr_layer_surface_v1 *surface) {
+ struct wl_context *context = data;
+ context->menu->exit = true;
+}
+
+static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
+ .configure = layer_surface_configure,
+ .closed = layer_surface_closed,
+};
+
+static void output_scale(void *data, struct wl_output *wl_output, int32_t factor) {
+ struct output *output = data;
+ output->scale = factor;
+}
+
+static void output_name(void *data, struct wl_output *wl_output, const char *name) {
+ struct output *output = data;
+ output->name = name;
+
+ struct wl_context *context = output->context;
+ if (context->menu->output_name && strcmp(context->menu->output_name, name) == 0) {
+ context->output = output;
+ }
+}
+
+static const struct wl_output_listener output_listener = {
+ .geometry = noop,
+ .mode = noop,
+ .done = noop,
+ .scale = output_scale,
+ .name = output_name,
+ .description = noop,
+};
+
+static void keyboard_keymap(void *data, struct wl_keyboard *wl_keyboard,
+ uint32_t format, int32_t fd, uint32_t size) {
+ struct keyboard *keyboard = data;
+ assert(format == WL_KEYBOARD_KEYMAP_FORMAT_XKB_V1);
+
+ char *map_shm = mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0);
+ assert(map_shm != MAP_FAILED);
+
+ keyboard->keymap = xkb_keymap_new_from_string(keyboard->context,
+ map_shm, XKB_KEYMAP_FORMAT_TEXT_V1, 0);
+ munmap(map_shm, size);
+ close(fd);
+
+ keyboard->state = xkb_state_new(keyboard->keymap);
+}
+
+static void keyboard_repeat(struct keyboard *keyboard) {
+ menu_keypress(keyboard->menu, keyboard->repeat_key_state, keyboard->repeat_sym);
+ struct itimerspec spec = { 0 };
+ spec.it_value.tv_sec = keyboard->repeat_period / 1000;
+ spec.it_value.tv_nsec = (keyboard->repeat_period % 1000) * 1000000l;
+ timerfd_settime(keyboard->repeat_timer, 0, &spec, NULL);
+}
+
+static void keyboard_key(void *data, struct wl_keyboard *wl_keyboard,
+ uint32_t serial, uint32_t time, uint32_t key, uint32_t _key_state) {
+ struct keyboard *keyboard = data;
+
+ enum wl_keyboard_key_state key_state = _key_state;
+ xkb_keysym_t sym = xkb_state_key_get_one_sym(keyboard->state, key + 8);
+ menu_keypress(keyboard->menu, key_state, sym);
+
+ if (key_state == WL_KEYBOARD_KEY_STATE_PRESSED && keyboard->repeat_period >= 0) {
+ keyboard->repeat_key_state = key_state;
+ keyboard->repeat_sym = sym;
+
+ struct itimerspec spec = { 0 };
+ spec.it_value.tv_sec = keyboard->repeat_delay / 1000;
+ spec.it_value.tv_nsec = (keyboard->repeat_delay % 1000) * 1000000l;
+ timerfd_settime(keyboard->repeat_timer, 0, &spec, NULL);
+ } else if (key_state == WL_KEYBOARD_KEY_STATE_RELEASED) {
+ struct itimerspec spec = { 0 };
+ timerfd_settime(keyboard->repeat_timer, 0, &spec, NULL);
+ }
+}
+
+static void keyboard_repeat_info(void *data, struct wl_keyboard *wl_keyboard,
+ int32_t rate, int32_t delay) {
+ struct keyboard *keyboard = data;
+ keyboard->repeat_delay = delay;
+ if (rate > 0) {
+ keyboard->repeat_period = 1000 / rate;
+ } else {
+ keyboard->repeat_period = -1;
+ }
+}
+
+static void keyboard_modifiers(void *data, struct wl_keyboard *wl_keyboard,
+ uint32_t serial, uint32_t mods_depressed,
+ uint32_t mods_latched, uint32_t mods_locked,
+ uint32_t group) {
+ struct keyboard *keyboard = data;
+ xkb_state_update_mask(keyboard->state, mods_depressed, mods_latched,
+ mods_locked, 0, 0, group);
+}
+
+static const struct wl_keyboard_listener keyboard_listener = {
+ .keymap = keyboard_keymap,
+ .enter = noop,
+ .leave = noop,
+ .key = keyboard_key,
+ .modifiers = keyboard_modifiers,
+ .repeat_info = keyboard_repeat_info,
+};
+
+static void seat_capabilities(void *data, struct wl_seat *seat,
+ enum wl_seat_capability caps) {
+ struct wl_context *context = data;
+ if (caps & WL_SEAT_CAPABILITY_KEYBOARD) {
+ struct wl_keyboard *wl_keyboard = wl_seat_get_keyboard(seat);
+ struct keyboard *keyboard = keyboard_create(context->menu, wl_keyboard);
+ wl_keyboard_add_listener(wl_keyboard, &keyboard_listener, keyboard);
+ context->keyboard = keyboard;
+ }
+}
+
+static const struct wl_seat_listener seat_listener = {
+ .capabilities = seat_capabilities,
+ .name = noop,
+};
+
+static void data_device_selection(void *data, struct wl_data_device *data_device,
+ struct wl_data_offer *data_offer) {
+ struct wl_context *context = data;
+ context->data_offer = data_offer;
+}
+
+static const struct wl_data_device_listener data_device_listener = {
+ .data_offer = noop,
+ .enter = noop,
+ .leave = noop,
+ .motion = noop,
+ .drop = noop,
+ .selection = data_device_selection,
+};
+
+static void handle_global(void *data, struct wl_registry *registry,
+ uint32_t name, const char *interface, uint32_t version) {
+ struct wl_context *context = data;
+ if (strcmp(interface, wl_compositor_interface.name) == 0) {
+ context->compositor = wl_registry_bind(registry, name, &wl_compositor_interface, 4);
+ } else if (strcmp(interface, wl_shm_interface.name) == 0) {
+ context->shm = wl_registry_bind(registry, name, &wl_shm_interface, 1);
+ } else if (strcmp(interface, wl_seat_interface.name) == 0) {
+ context->seat = wl_registry_bind(registry, name, &wl_seat_interface, 4);
+ wl_seat_add_listener(context->seat, &seat_listener, data);
+ } else if (strcmp(interface, wl_data_device_manager_interface.name) == 0) {
+ context->data_device_manager = wl_registry_bind(registry, name, &wl_data_device_manager_interface, 3);
+ } else if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
+ context->layer_shell = wl_registry_bind(registry, name, &zwlr_layer_shell_v1_interface, 1);
+ } else if (strcmp(interface, wl_output_interface.name) == 0) {
+ struct wl_output *wl_output = wl_registry_bind(registry, name, &wl_output_interface, 4);
+ struct output *output = output_create(context, wl_output);
+ wl_output_set_user_data(wl_output, output);
+ wl_output_add_listener(wl_output, &output_listener, output);
+ context_add_output(context, output);
+ } else if (strcmp(interface, xdg_activation_v1_interface.name) == 0) {
+ context->activation = wl_registry_bind(registry, name, &xdg_activation_v1_interface, 1);
+ }
+}
+
+static const struct wl_registry_listener registry_listener = {
+ .global = handle_global,
+ .global_remove = noop,
+};
+
+// Connect to the Wayland display and run the menu.
+int menu_run(struct menu *menu) {
+ struct wl_context *context = calloc(1, sizeof(struct wl_context));
+ context->menu = menu;
+ menu->context = context;
+
+ context->display = wl_display_connect(NULL);
+ if (!context->display) {
+ fprintf(stderr, "Failed to connect to display.\n");
+ exit(EXIT_FAILURE);
+ }
+
+ struct wl_registry *registry = wl_display_get_registry(context->display);
+ wl_registry_add_listener(registry, ®istry_listener, context);
+ wl_display_roundtrip(context->display);
+ assert(context->compositor != NULL);
+ assert(context->shm != NULL);
+ assert(context->seat != NULL);
+ assert(context->data_device_manager != NULL);
+ assert(context->layer_shell != NULL);
+ assert(context->activation != NULL);
+ context->registry = registry;
+
+ // Get data device for seat
+ struct wl_data_device *data_device = wl_data_device_manager_get_data_device(
+ context->data_device_manager, context->seat);
+ wl_data_device_add_listener(data_device, &data_device_listener, context);
+ context->data_device = data_device;
+
+ // Second roundtrip for seat and output listeners
+ wl_display_roundtrip(context->display);
+ assert(context->keyboard != NULL);
+
+ if (menu->output_name && !context->output) {
+ fprintf(stderr, "Output %s not found\n", menu->output_name);
+ exit(EXIT_FAILURE);
+ }
+
+ context->surface = wl_compositor_create_surface(context->compositor);
+ wl_surface_add_listener(context->surface, &surface_listener, context);
+
+ struct zwlr_layer_surface_v1 *layer_surface = zwlr_layer_shell_v1_get_layer_surface(
+ context->layer_shell,
+ context->surface,
+ context->output ? context->output->output : NULL,
+ ZWLR_LAYER_SHELL_V1_LAYER_TOP,
+ "menu"
+ );
+ assert(layer_surface != NULL);
+ context->layer_surface = layer_surface;
+
+ uint32_t anchor = ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT |
+ ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
+ if (menu->bottom) {
+ anchor |= ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM;
+ } else {
+ anchor |= ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP;
+ }
+
+ zwlr_layer_surface_v1_set_anchor(layer_surface, anchor);
+ zwlr_layer_surface_v1_set_size(layer_surface, 0, menu->height);
+ zwlr_layer_surface_v1_set_exclusive_zone(layer_surface, -1);
+ zwlr_layer_surface_v1_set_keyboard_interactivity(layer_surface, true);
+ zwlr_layer_surface_v1_add_listener(layer_surface, &layer_surface_listener, context);
+
+ wl_surface_commit(context->surface);
+ wl_display_roundtrip(context->display);
+ render_menu(menu);
+ menu_process_items(menu);
+ render_menu(menu);
+
+ struct pollfd fds[] = {
+ { wl_display_get_fd(context->display), POLLIN },
+ { context->keyboard->repeat_timer, POLLIN },
+ };
+ const size_t nfds = sizeof(fds) / sizeof(*fds);
+
+ while (!menu->exit) {
+ errno = 0;
+ do {
+ if (wl_display_flush(context->display) == -1 && errno != EAGAIN) {
+ fprintf(stderr, "wl_display_flush: %s\n", strerror(errno));
+ break;
+ }
+ } while (errno == EAGAIN);
+
+ if (poll(fds, nfds, -1) < 0) {
+ fprintf(stderr, "poll: %s\n", strerror(errno));
+ break;
+ }
+
+ if (fds[0].revents & POLLIN) {
+ if (wl_display_dispatch(context->display) < 0) {
+ menu->exit = true;
+ }
+ }
+
+ if (fds[1].revents & POLLIN) {
+ keyboard_repeat(context->keyboard);
+ }
+ }
+
+ bool failure = menu->failure;
+ menu_destroy(menu);
+ context_destroy(context);
+
+ if (failure) {
+ return EXIT_FAILURE;
+ }
+ return EXIT_SUCCESS;
+}
diff --git a/wayland.h b/wayland.h
new file mode 100644
index 0000000..bca663f
--- /dev/null
+++ b/wayland.h
@@ -0,0 +1,17 @@
+#ifndef WMENU_WAYLAND_H
+#define WMENU_WAYLAND_H
+
+#include "menu.h"
+#include <wayland-client-protocol.h>
+
+struct wl_context;
+
+int menu_run(struct menu *menu);
+
+int context_get_scale(struct wl_context *context);
+struct wl_shm *context_get_shm(struct wl_context *context);
+struct wl_surface *context_get_surface(struct wl_context *context);
+struct xkb_state *context_get_xkb_state(struct wl_context *context);
+bool context_paste(struct wl_context *context);
+
+#endif
--
cgit v1.2.3