mirror of
				https://github.com/labwc/labwc.git
				synced 2025-11-03 09:01:51 -05:00 
			
		
		
		
	
		
			
				
	
	
		
			1652 lines
		
	
	
	
		
			45 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			1652 lines
		
	
	
	
		
			45 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
// SPDX-License-Identifier: GPL-2.0-only
 | 
						|
/*
 | 
						|
 * Based on https://github.com/swaywm/sway/tree/master/swaynag
 | 
						|
 *
 | 
						|
 * Copyright (C) 2016-2017 Drew DeVault
 | 
						|
 * Copyright (C) 2025 Johan Malm
 | 
						|
 */
 | 
						|
#define _POSIX_C_SOURCE 200809L
 | 
						|
#include <assert.h>
 | 
						|
#include <cairo.h>
 | 
						|
#include <ctype.h>
 | 
						|
#include <getopt.h>
 | 
						|
#include <glib.h>
 | 
						|
#include <pango/pangocairo.h>
 | 
						|
#include <poll.h>
 | 
						|
#include <stdio.h>
 | 
						|
#include <stdint.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#ifdef __FreeBSD__
 | 
						|
#include <sys/event.h> /* For signalfd() */
 | 
						|
#endif
 | 
						|
#include <sys/signalfd.h>
 | 
						|
#include <sys/timerfd.h>
 | 
						|
#include <sys/wait.h>
 | 
						|
#include <time.h>
 | 
						|
#include <unistd.h>
 | 
						|
#include <wayland-cursor.h>
 | 
						|
#include <wlr/util/log.h>
 | 
						|
#include "action-prompt-codes.h"
 | 
						|
#include "pool-buffer.h"
 | 
						|
#include "cursor-shape-v1-client-protocol.h"
 | 
						|
#include "wlr-layer-shell-unstable-v1-client-protocol.h"
 | 
						|
 | 
						|
#define LABNAG_MAX_HEIGHT 500
 | 
						|
 | 
						|
struct conf {
 | 
						|
	PangoFontDescription *font_description;
 | 
						|
	char *output;
 | 
						|
	uint32_t anchors;
 | 
						|
	int32_t layer; /* enum zwlr_layer_shell_v1_layer or -1 if unset */
 | 
						|
 | 
						|
	/* Colors */
 | 
						|
	uint32_t button_text;
 | 
						|
	uint32_t button_background;
 | 
						|
	uint32_t details_background;
 | 
						|
	uint32_t background;
 | 
						|
	uint32_t text;
 | 
						|
	uint32_t button_border;
 | 
						|
	uint32_t border_bottom;
 | 
						|
 | 
						|
	/* Sizing */
 | 
						|
	ssize_t bar_border_thickness;
 | 
						|
	ssize_t message_padding;
 | 
						|
	ssize_t details_border_thickness;
 | 
						|
	ssize_t button_border_thickness;
 | 
						|
	ssize_t button_gap;
 | 
						|
	ssize_t button_gap_close;
 | 
						|
	ssize_t button_margin_right;
 | 
						|
	ssize_t button_padding;
 | 
						|
};
 | 
						|
 | 
						|
struct pointer {
 | 
						|
	struct wl_pointer *pointer;
 | 
						|
	uint32_t serial;
 | 
						|
	struct wl_cursor_theme *cursor_theme;
 | 
						|
	struct wl_cursor_image *cursor_image;
 | 
						|
	struct wl_surface *cursor_surface;
 | 
						|
	int x;
 | 
						|
	int y;
 | 
						|
};
 | 
						|
 | 
						|
struct seat {
 | 
						|
	struct wl_seat *wl_seat;
 | 
						|
	uint32_t wl_name;
 | 
						|
	struct nag *nag;
 | 
						|
	struct pointer pointer;
 | 
						|
	struct wl_list link; /* nag.seats */
 | 
						|
};
 | 
						|
 | 
						|
struct output {
 | 
						|
	char *name;
 | 
						|
	struct wl_output *wl_output;
 | 
						|
	uint32_t wl_name;
 | 
						|
	uint32_t scale;
 | 
						|
	struct nag *nag;
 | 
						|
	struct wl_list link; /* nag.outputs */
 | 
						|
};
 | 
						|
 | 
						|
struct button {
 | 
						|
	char *text;
 | 
						|
	char *action;
 | 
						|
	int x;
 | 
						|
	int y;
 | 
						|
	int width;
 | 
						|
	int height;
 | 
						|
	bool expand;
 | 
						|
	bool dismiss;
 | 
						|
	struct wl_list link;
 | 
						|
};
 | 
						|
 | 
						|
enum {
 | 
						|
	FD_WAYLAND,
 | 
						|
	FD_TIMER,
 | 
						|
	FD_SIGNAL,
 | 
						|
 | 
						|
	NR_FDS,
 | 
						|
};
 | 
						|
 | 
						|
struct nag {
 | 
						|
	bool run_display;
 | 
						|
 | 
						|
	struct wl_display *display;
 | 
						|
	struct wl_compositor *compositor;
 | 
						|
	struct wl_seat *seat;
 | 
						|
	struct wl_shm *shm;
 | 
						|
	struct wl_list outputs;
 | 
						|
	struct wl_list seats;
 | 
						|
	struct output *output;
 | 
						|
	struct zwlr_layer_shell_v1 *layer_shell;
 | 
						|
	struct zwlr_layer_surface_v1 *layer_surface;
 | 
						|
	struct wp_cursor_shape_manager_v1 *cursor_shape_manager;
 | 
						|
	struct wl_surface *surface;
 | 
						|
 | 
						|
	uint32_t width;
 | 
						|
	uint32_t height;
 | 
						|
	int32_t scale;
 | 
						|
	struct pool_buffer buffers[2];
 | 
						|
	struct pool_buffer *current_buffer;
 | 
						|
 | 
						|
	struct conf *conf;
 | 
						|
	char *message;
 | 
						|
	struct wl_list buttons;
 | 
						|
	struct pollfd pollfds[NR_FDS];
 | 
						|
 | 
						|
	struct {
 | 
						|
		bool visible;
 | 
						|
		char *message;
 | 
						|
		char *details_text;
 | 
						|
		int close_timeout;
 | 
						|
		bool use_exclusive_zone;
 | 
						|
 | 
						|
		int x;
 | 
						|
		int y;
 | 
						|
		int width;
 | 
						|
		int height;
 | 
						|
 | 
						|
		int offset;
 | 
						|
		int visible_lines;
 | 
						|
		int total_lines;
 | 
						|
		struct button *button_details;
 | 
						|
		struct button button_up;
 | 
						|
		struct button button_down;
 | 
						|
	} details;
 | 
						|
};
 | 
						|
 | 
						|
static int exit_status = LAB_EXIT_FAILURE;
 | 
						|
 | 
						|
static void close_pollfd(struct pollfd *pollfd);
 | 
						|
 | 
						|
static PangoLayout *
 | 
						|
get_pango_layout(cairo_t *cairo, const PangoFontDescription *desc,
 | 
						|
		const char *text, double scale, bool markup)
 | 
						|
{
 | 
						|
	PangoLayout *layout = pango_cairo_create_layout(cairo);
 | 
						|
	pango_context_set_round_glyph_positions(pango_layout_get_context(layout), false);
 | 
						|
 | 
						|
	PangoAttrList *attrs;
 | 
						|
	if (markup) {
 | 
						|
		char *buf;
 | 
						|
		GError *error = NULL;
 | 
						|
		if (pango_parse_markup(text, -1, 0, &attrs, &buf, NULL, &error)) {
 | 
						|
			pango_layout_set_text(layout, buf, -1);
 | 
						|
			free(buf);
 | 
						|
		} else {
 | 
						|
			wlr_log(WLR_ERROR, "pango_parse_markup '%s' -> error %s",
 | 
						|
				text, error->message);
 | 
						|
			g_error_free(error);
 | 
						|
			markup = false; /* fallback to plain text */
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (!markup) {
 | 
						|
		attrs = pango_attr_list_new();
 | 
						|
		pango_layout_set_text(layout, text, -1);
 | 
						|
	}
 | 
						|
 | 
						|
	pango_attr_list_insert(attrs, pango_attr_scale_new(scale));
 | 
						|
	pango_layout_set_font_description(layout, desc);
 | 
						|
	pango_layout_set_single_paragraph_mode(layout, 1);
 | 
						|
	pango_layout_set_attributes(layout, attrs);
 | 
						|
	pango_attr_list_unref(attrs);
 | 
						|
	return layout;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
get_text_size(cairo_t *cairo, const PangoFontDescription *desc, int *width, int *height,
 | 
						|
		int *baseline, double scale, bool markup, const char *fmt, ...)
 | 
						|
{
 | 
						|
	va_list args;
 | 
						|
	va_start(args, fmt);
 | 
						|
	gchar *buf = g_strdup_vprintf(fmt, args);
 | 
						|
	va_end(args);
 | 
						|
	if (!buf) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	PangoLayout *layout = get_pango_layout(cairo, desc, buf, scale, markup);
 | 
						|
	pango_cairo_update_layout(cairo, layout);
 | 
						|
	pango_layout_get_pixel_size(layout, width, height);
 | 
						|
	if (baseline) {
 | 
						|
		*baseline = pango_layout_get_baseline(layout) / PANGO_SCALE;
 | 
						|
	}
 | 
						|
	g_object_unref(layout);
 | 
						|
 | 
						|
	g_free(buf);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
render_text(cairo_t *cairo, const PangoFontDescription *desc, double scale,
 | 
						|
		bool markup, const char *fmt, ...)
 | 
						|
{
 | 
						|
	va_list args;
 | 
						|
	va_start(args, fmt);
 | 
						|
	gchar *buf = g_strdup_vprintf(fmt, args);
 | 
						|
	va_end(args);
 | 
						|
	if (!buf) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	PangoLayout *layout = get_pango_layout(cairo, desc, buf, scale, markup);
 | 
						|
	cairo_font_options_t *fo = cairo_font_options_create();
 | 
						|
	cairo_get_font_options(cairo, fo);
 | 
						|
	pango_cairo_context_set_font_options(pango_layout_get_context(layout), fo);
 | 
						|
	cairo_font_options_destroy(fo);
 | 
						|
	pango_cairo_update_layout(cairo, layout);
 | 
						|
	pango_cairo_show_layout(cairo, layout);
 | 
						|
	g_object_unref(layout);
 | 
						|
 | 
						|
	g_free(buf);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
cairo_set_source_u32(cairo_t *cairo, uint32_t color)
 | 
						|
{
 | 
						|
	cairo_set_source_rgba(cairo,
 | 
						|
			(color >> (3*8) & 0xFF) / 255.0,
 | 
						|
			(color >> (2*8) & 0xFF) / 255.0,
 | 
						|
			(color >> (1*8) & 0xFF) / 255.0,
 | 
						|
			(color >> (0*8) & 0xFF) / 255.0);
 | 
						|
}
 | 
						|
 | 
						|
static uint32_t
 | 
						|
render_message(cairo_t *cairo, struct nag *nag)
 | 
						|
{
 | 
						|
	int text_width, text_height;
 | 
						|
	get_text_size(cairo, nag->conf->font_description, &text_width,
 | 
						|
		&text_height, NULL, 1, true, "%s", nag->message);
 | 
						|
 | 
						|
	int padding = nag->conf->message_padding;
 | 
						|
 | 
						|
	uint32_t ideal_height = text_height + padding * 2;
 | 
						|
	uint32_t ideal_surface_height = ideal_height;
 | 
						|
	if (nag->height < ideal_surface_height) {
 | 
						|
		return ideal_surface_height;
 | 
						|
	}
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->text);
 | 
						|
	cairo_move_to(cairo, padding, (int)(ideal_height - text_height) / 2);
 | 
						|
	render_text(cairo, nag->conf->font_description, 1, false,
 | 
						|
			"%s", nag->message);
 | 
						|
 | 
						|
	return ideal_surface_height;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
render_details_scroll_button(cairo_t *cairo, struct nag *nag,
 | 
						|
		struct button *button)
 | 
						|
{
 | 
						|
	int text_width, text_height;
 | 
						|
	get_text_size(cairo, nag->conf->font_description, &text_width,
 | 
						|
		&text_height, NULL, 1, true, "%s", button->text);
 | 
						|
 | 
						|
	int border = nag->conf->button_border_thickness;
 | 
						|
	int padding = nag->conf->button_padding;
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->details_background);
 | 
						|
	cairo_rectangle(cairo, button->x, button->y,
 | 
						|
			button->width, button->height);
 | 
						|
	cairo_fill(cairo);
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->button_background);
 | 
						|
	cairo_rectangle(cairo, button->x + border, button->y + border,
 | 
						|
			button->width - (border * 2),
 | 
						|
			button->height - (border * 2));
 | 
						|
	cairo_fill(cairo);
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->button_text);
 | 
						|
	cairo_move_to(cairo, button->x + border + padding,
 | 
						|
			button->y + border + (button->height - text_height) / 2);
 | 
						|
	render_text(cairo, nag->conf->font_description, 1, true,
 | 
						|
			"%s", button->text);
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
get_detailed_scroll_button_width(cairo_t *cairo, struct nag *nag)
 | 
						|
{
 | 
						|
	int up_width, down_width, temp_height;
 | 
						|
	get_text_size(cairo, nag->conf->font_description, &up_width, &temp_height,
 | 
						|
		NULL, 1, true, "%s", nag->details.button_up.text);
 | 
						|
	get_text_size(cairo, nag->conf->font_description, &down_width, &temp_height,
 | 
						|
		NULL, 1, true, "%s", nag->details.button_down.text);
 | 
						|
 | 
						|
	int text_width =  up_width > down_width ? up_width : down_width;
 | 
						|
	int border = nag->conf->button_border_thickness;
 | 
						|
	int padding = nag->conf->button_padding;
 | 
						|
 | 
						|
	return text_width + border * 2 + padding * 2;
 | 
						|
}
 | 
						|
 | 
						|
static uint32_t
 | 
						|
render_detailed(cairo_t *cairo, struct nag *nag, uint32_t y)
 | 
						|
{
 | 
						|
	uint32_t width = nag->width;
 | 
						|
 | 
						|
	int border = nag->conf->details_border_thickness;
 | 
						|
	int padding = nag->conf->message_padding;
 | 
						|
	int decor = padding + border;
 | 
						|
 | 
						|
	nag->details.x = decor;
 | 
						|
	nag->details.y = y + decor;
 | 
						|
	nag->details.width = width - decor * 2;
 | 
						|
 | 
						|
	PangoLayout *layout = get_pango_layout(cairo, nag->conf->font_description,
 | 
						|
			nag->details.message, 1, false);
 | 
						|
	pango_layout_set_width(layout,
 | 
						|
			(nag->details.width - padding * 2) * PANGO_SCALE);
 | 
						|
	pango_layout_set_wrap(layout, PANGO_WRAP_WORD_CHAR);
 | 
						|
	pango_layout_set_single_paragraph_mode(layout, false);
 | 
						|
	pango_cairo_update_layout(cairo, layout);
 | 
						|
	nag->details.total_lines = pango_layout_get_line_count(layout);
 | 
						|
 | 
						|
	PangoLayoutLine *line;
 | 
						|
	line = pango_layout_get_line_readonly(layout, nag->details.offset);
 | 
						|
	gint offset = line->start_index;
 | 
						|
	const char *text = pango_layout_get_text(layout);
 | 
						|
	pango_layout_set_text(layout, text + offset, strlen(text) - offset);
 | 
						|
 | 
						|
	int text_width, text_height;
 | 
						|
	pango_cairo_update_layout(cairo, layout);
 | 
						|
	pango_layout_get_pixel_size(layout, &text_width, &text_height);
 | 
						|
 | 
						|
	bool show_buttons = nag->details.offset > 0;
 | 
						|
	int button_width = get_detailed_scroll_button_width(cairo, nag);
 | 
						|
	if (show_buttons) {
 | 
						|
		nag->details.width -= button_width;
 | 
						|
		pango_layout_set_width(layout,
 | 
						|
				(nag->details.width - padding * 2) * PANGO_SCALE);
 | 
						|
	}
 | 
						|
 | 
						|
	uint32_t ideal_height;
 | 
						|
	do {
 | 
						|
		ideal_height = nag->details.y + text_height + decor + padding * 2;
 | 
						|
		if (ideal_height > LABNAG_MAX_HEIGHT) {
 | 
						|
			ideal_height = LABNAG_MAX_HEIGHT;
 | 
						|
 | 
						|
			if (!show_buttons) {
 | 
						|
				show_buttons = true;
 | 
						|
				nag->details.width -= button_width;
 | 
						|
				pango_layout_set_width(layout,
 | 
						|
					(nag->details.width - padding * 2) * PANGO_SCALE);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		nag->details.height = ideal_height - nag->details.y - decor;
 | 
						|
		pango_layout_set_height(layout,
 | 
						|
				(nag->details.height - padding * 2) * PANGO_SCALE);
 | 
						|
		pango_layout_set_ellipsize(layout, PANGO_ELLIPSIZE_END);
 | 
						|
		pango_cairo_update_layout(cairo, layout);
 | 
						|
		pango_layout_get_pixel_size(layout, &text_width, &text_height);
 | 
						|
	} while (text_height != (nag->details.height - padding * 2));
 | 
						|
 | 
						|
	nag->details.visible_lines = pango_layout_get_line_count(layout);
 | 
						|
 | 
						|
	if (show_buttons) {
 | 
						|
		nag->details.button_up.x = nag->details.x + nag->details.width;
 | 
						|
		nag->details.button_up.y = nag->details.y;
 | 
						|
		nag->details.button_up.width = button_width;
 | 
						|
		nag->details.button_up.height = nag->details.height / 2;
 | 
						|
		render_details_scroll_button(cairo, nag, &nag->details.button_up);
 | 
						|
 | 
						|
		nag->details.button_down.x = nag->details.x + nag->details.width;
 | 
						|
		nag->details.button_down.y =
 | 
						|
			nag->details.button_up.y + nag->details.button_up.height;
 | 
						|
		nag->details.button_down.width = button_width;
 | 
						|
		nag->details.button_down.height = nag->details.height / 2;
 | 
						|
		render_details_scroll_button(cairo, nag, &nag->details.button_down);
 | 
						|
	}
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->details_background);
 | 
						|
	cairo_rectangle(cairo, nag->details.x, nag->details.y,
 | 
						|
			nag->details.width, nag->details.height);
 | 
						|
	cairo_fill(cairo);
 | 
						|
 | 
						|
	cairo_move_to(cairo, nag->details.x + padding, nag->details.y + padding);
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->text);
 | 
						|
	pango_cairo_show_layout(cairo, layout);
 | 
						|
	g_object_unref(layout);
 | 
						|
 | 
						|
	return ideal_height;
 | 
						|
}
 | 
						|
 | 
						|
static uint32_t
 | 
						|
render_button(cairo_t *cairo, struct nag *nag, struct button *button, int *x)
 | 
						|
{
 | 
						|
	int text_width, text_height;
 | 
						|
	get_text_size(cairo, nag->conf->font_description, &text_width,
 | 
						|
		&text_height, NULL, 1, true, "%s", button->text);
 | 
						|
 | 
						|
	int border = nag->conf->button_border_thickness;
 | 
						|
	int padding = nag->conf->button_padding;
 | 
						|
 | 
						|
	uint32_t ideal_height = text_height + padding * 2 + border * 2;
 | 
						|
	uint32_t ideal_surface_height = ideal_height;
 | 
						|
	if (nag->height < ideal_surface_height) {
 | 
						|
		return ideal_surface_height;
 | 
						|
	}
 | 
						|
 | 
						|
	button->x = *x - border - text_width - padding * 2 + 1;
 | 
						|
	button->y = (int)(ideal_height - text_height) / 2 - padding + 1;
 | 
						|
	button->width = text_width + padding * 2;
 | 
						|
	button->height = text_height + padding * 2;
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->button_border);
 | 
						|
	cairo_rectangle(cairo, button->x - border, button->y - border,
 | 
						|
			button->width + border * 2, button->height + border * 2);
 | 
						|
	cairo_fill(cairo);
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->button_background);
 | 
						|
	cairo_rectangle(cairo, button->x, button->y,
 | 
						|
			button->width, button->height);
 | 
						|
	cairo_fill(cairo);
 | 
						|
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->button_text);
 | 
						|
	cairo_move_to(cairo, button->x + padding, button->y + padding);
 | 
						|
	render_text(cairo, nag->conf->font_description, 1, true,
 | 
						|
			"%s", button->text);
 | 
						|
 | 
						|
	*x = button->x - border;
 | 
						|
 | 
						|
	return ideal_surface_height;
 | 
						|
}
 | 
						|
 | 
						|
static uint32_t
 | 
						|
render_to_cairo(cairo_t *cairo, struct nag *nag)
 | 
						|
{
 | 
						|
	uint32_t max_height = 0;
 | 
						|
 | 
						|
	cairo_set_operator(cairo, CAIRO_OPERATOR_SOURCE);
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->background);
 | 
						|
	cairo_paint(cairo);
 | 
						|
 | 
						|
	uint32_t h = render_message(cairo, nag);
 | 
						|
	max_height = h > max_height ? h : max_height;
 | 
						|
 | 
						|
	int x = nag->width - nag->conf->button_margin_right;
 | 
						|
	x -= nag->conf->button_gap_close;
 | 
						|
 | 
						|
	struct button *button;
 | 
						|
	wl_list_for_each(button, &nag->buttons, link) {
 | 
						|
		h = render_button(cairo, nag, button, &x);
 | 
						|
		max_height = h > max_height ? h : max_height;
 | 
						|
		x -= nag->conf->button_gap;
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->details.visible) {
 | 
						|
		h = render_detailed(cairo, nag, max_height);
 | 
						|
		max_height = h > max_height ? h : max_height;
 | 
						|
	}
 | 
						|
 | 
						|
	int border = nag->conf->bar_border_thickness;
 | 
						|
	if (max_height > nag->height) {
 | 
						|
		max_height += border;
 | 
						|
	}
 | 
						|
	cairo_set_source_u32(cairo, nag->conf->border_bottom);
 | 
						|
	cairo_rectangle(cairo, 0,
 | 
						|
			nag->height - border,
 | 
						|
			nag->width,
 | 
						|
			border);
 | 
						|
	cairo_fill(cairo);
 | 
						|
 | 
						|
	return max_height;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
render_frame(struct nag *nag)
 | 
						|
{
 | 
						|
	if (!nag->run_display) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	cairo_surface_t *recorder = cairo_recording_surface_create(
 | 
						|
			CAIRO_CONTENT_COLOR_ALPHA, NULL);
 | 
						|
	cairo_t *cairo = cairo_create(recorder);
 | 
						|
	cairo_scale(cairo, nag->scale, nag->scale);
 | 
						|
	cairo_save(cairo);
 | 
						|
	cairo_set_operator(cairo, CAIRO_OPERATOR_CLEAR);
 | 
						|
	cairo_paint(cairo);
 | 
						|
	cairo_restore(cairo);
 | 
						|
	uint32_t height = render_to_cairo(cairo, nag);
 | 
						|
	if (height != nag->height) {
 | 
						|
		zwlr_layer_surface_v1_set_size(nag->layer_surface, 0, height);
 | 
						|
		if (nag->details.use_exclusive_zone) {
 | 
						|
			zwlr_layer_surface_v1_set_exclusive_zone(
 | 
						|
				nag->layer_surface, height);
 | 
						|
		}
 | 
						|
		wl_surface_commit(nag->surface);
 | 
						|
		wl_display_roundtrip(nag->display);
 | 
						|
	} else {
 | 
						|
		nag->current_buffer = get_next_buffer(nag->shm,
 | 
						|
				nag->buffers,
 | 
						|
				nag->width * nag->scale,
 | 
						|
				nag->height * nag->scale);
 | 
						|
		if (!nag->current_buffer) {
 | 
						|
			wlr_log(WLR_DEBUG, "Failed to get buffer. Skipping frame.");
 | 
						|
			goto cleanup;
 | 
						|
		}
 | 
						|
 | 
						|
		cairo_t *shm = nag->current_buffer->cairo;
 | 
						|
		cairo_save(shm);
 | 
						|
		cairo_set_operator(shm, CAIRO_OPERATOR_CLEAR);
 | 
						|
		cairo_paint(shm);
 | 
						|
		cairo_restore(shm);
 | 
						|
		cairo_set_source_surface(shm, recorder, 0.0, 0.0);
 | 
						|
		cairo_paint(shm);
 | 
						|
 | 
						|
		wl_surface_set_buffer_scale(nag->surface, nag->scale);
 | 
						|
		wl_surface_attach(nag->surface,
 | 
						|
				nag->current_buffer->buffer, 0, 0);
 | 
						|
		wl_surface_damage(nag->surface, 0, 0,
 | 
						|
				nag->width, nag->height);
 | 
						|
		wl_surface_commit(nag->surface);
 | 
						|
		wl_display_roundtrip(nag->display);
 | 
						|
	}
 | 
						|
 | 
						|
cleanup:
 | 
						|
	cairo_surface_destroy(recorder);
 | 
						|
	cairo_destroy(cairo);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
seat_destroy(struct seat *seat)
 | 
						|
{
 | 
						|
	if (seat->pointer.cursor_theme) {
 | 
						|
		wl_cursor_theme_destroy(seat->pointer.cursor_theme);
 | 
						|
	}
 | 
						|
	if (seat->pointer.pointer) {
 | 
						|
		wl_pointer_destroy(seat->pointer.pointer);
 | 
						|
	}
 | 
						|
	wl_seat_destroy(seat->wl_seat);
 | 
						|
	wl_list_remove(&seat->link);
 | 
						|
	free(seat);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
nag_destroy(struct nag *nag)
 | 
						|
{
 | 
						|
	nag->run_display = false;
 | 
						|
 | 
						|
	struct button *button, *next;
 | 
						|
	wl_list_for_each_safe(button, next, &nag->buttons, link) {
 | 
						|
		wl_list_remove(&button->link);
 | 
						|
		free(button);
 | 
						|
	}
 | 
						|
	free(nag->details.message);
 | 
						|
 | 
						|
	pango_font_description_free(nag->conf->font_description);
 | 
						|
 | 
						|
	if (nag->layer_surface) {
 | 
						|
		zwlr_layer_surface_v1_destroy(nag->layer_surface);
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->surface) {
 | 
						|
		wl_surface_destroy(nag->surface);
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->layer_shell) {
 | 
						|
		zwlr_layer_shell_v1_destroy(nag->layer_shell);
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->cursor_shape_manager) {
 | 
						|
		wp_cursor_shape_manager_v1_destroy(nag->cursor_shape_manager);
 | 
						|
	}
 | 
						|
 | 
						|
	struct seat *seat, *tmpseat;
 | 
						|
	wl_list_for_each_safe(seat, tmpseat, &nag->seats, link) {
 | 
						|
		seat_destroy(seat);
 | 
						|
	}
 | 
						|
 | 
						|
	destroy_buffer(&nag->buffers[0]);
 | 
						|
	destroy_buffer(&nag->buffers[1]);
 | 
						|
 | 
						|
	if (nag->outputs.prev || nag->outputs.next) {
 | 
						|
		struct output *output, *temp;
 | 
						|
		wl_list_for_each_safe(output, temp, &nag->outputs, link) {
 | 
						|
			wl_output_destroy(output->wl_output);
 | 
						|
			free(output->name);
 | 
						|
			wl_list_remove(&output->link);
 | 
						|
			free(output);
 | 
						|
		};
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->compositor) {
 | 
						|
		wl_compositor_destroy(nag->compositor);
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->shm) {
 | 
						|
		wl_shm_destroy(nag->shm);
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->display) {
 | 
						|
		wl_display_disconnect(nag->display);
 | 
						|
	}
 | 
						|
	pango_cairo_font_map_set_default(NULL);
 | 
						|
 | 
						|
	close_pollfd(&nag->pollfds[FD_TIMER]);
 | 
						|
	close_pollfd(&nag->pollfds[FD_SIGNAL]);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
button_execute(struct nag *nag, struct button *button)
 | 
						|
{
 | 
						|
	wlr_log(WLR_DEBUG, "Executing [%s]: %s", button->text, button->action);
 | 
						|
	if (button->expand) {
 | 
						|
		nag->details.visible = !nag->details.visible;
 | 
						|
		render_frame(nag);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	if (button->dismiss) {
 | 
						|
		nag->run_display = false;
 | 
						|
	}
 | 
						|
	if (button->action) {
 | 
						|
		pid_t pid = fork();
 | 
						|
		if (pid < 0) {
 | 
						|
			wlr_log_errno(WLR_DEBUG, "Failed to fork");
 | 
						|
			return;
 | 
						|
		} else if (pid == 0) {
 | 
						|
			/*
 | 
						|
			 * Child process. Will be used to prevent zombie
 | 
						|
			 * processes
 | 
						|
			 */
 | 
						|
			pid = fork();
 | 
						|
			if (pid < 0) {
 | 
						|
				wlr_log_errno(WLR_DEBUG, "Failed to fork");
 | 
						|
				return;
 | 
						|
			} else if (pid == 0) {
 | 
						|
				/*
 | 
						|
				 * Child of the child. Will be reparented to the
 | 
						|
				 * init process
 | 
						|
				 */
 | 
						|
				execlp("sh", "sh", "-c", button->action, NULL);
 | 
						|
				wlr_log_errno(WLR_DEBUG, "execlp failed");
 | 
						|
				_exit(LAB_EXIT_FAILURE);
 | 
						|
			}
 | 
						|
			_exit(EXIT_SUCCESS);
 | 
						|
		}
 | 
						|
 | 
						|
		if (waitpid(pid, NULL, 0) < 0) {
 | 
						|
			wlr_log_errno(WLR_DEBUG, "waitpid failed");
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
layer_surface_configure(void *data, struct zwlr_layer_surface_v1 *surface,
 | 
						|
		uint32_t serial, uint32_t width, uint32_t height)
 | 
						|
{
 | 
						|
	struct nag *nag = data;
 | 
						|
	nag->width = width;
 | 
						|
	nag->height = height;
 | 
						|
	zwlr_layer_surface_v1_ack_configure(surface, serial);
 | 
						|
	render_frame(nag);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
layer_surface_closed(void *data, struct zwlr_layer_surface_v1 *surface)
 | 
						|
{
 | 
						|
	struct nag *nag = data;
 | 
						|
	nag_destroy(nag);
 | 
						|
}
 | 
						|
 | 
						|
static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
 | 
						|
	.configure = layer_surface_configure,
 | 
						|
	.closed = layer_surface_closed,
 | 
						|
};
 | 
						|
 | 
						|
static void
 | 
						|
surface_enter(void *data, struct wl_surface *surface, struct wl_output *output)
 | 
						|
{
 | 
						|
	struct nag *nag = data;
 | 
						|
	struct output *nag_output;
 | 
						|
	wl_list_for_each(nag_output, &nag->outputs, link) {
 | 
						|
		if (nag_output->wl_output == output) {
 | 
						|
			wlr_log(WLR_DEBUG, "Surface enter on output %s",
 | 
						|
					nag_output->name);
 | 
						|
			nag->output = nag_output;
 | 
						|
			nag->scale = nag->output->scale;
 | 
						|
			render_frame(nag);
 | 
						|
			break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
surface_leave(void *data, struct wl_surface *wl_surface, struct wl_output *output)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_surface_listener surface_listener = {
 | 
						|
	.enter = surface_enter,
 | 
						|
	.leave = surface_leave,
 | 
						|
};
 | 
						|
 | 
						|
static void
 | 
						|
update_cursor(struct seat *seat)
 | 
						|
{
 | 
						|
	struct pointer *pointer = &seat->pointer;
 | 
						|
	struct nag *nag = seat->nag;
 | 
						|
	if (pointer->cursor_theme) {
 | 
						|
		wl_cursor_theme_destroy(pointer->cursor_theme);
 | 
						|
	}
 | 
						|
	const char *cursor_theme = getenv("XCURSOR_THEME");
 | 
						|
	unsigned int cursor_size = 24;
 | 
						|
	const char *env_cursor_size = getenv("XCURSOR_SIZE");
 | 
						|
	if (env_cursor_size && *env_cursor_size) {
 | 
						|
		errno = 0;
 | 
						|
		char *end;
 | 
						|
		unsigned int size = strtoul(env_cursor_size, &end, 10);
 | 
						|
		if (!*end && errno == 0) {
 | 
						|
			cursor_size = size;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	pointer->cursor_theme = wl_cursor_theme_load(
 | 
						|
		cursor_theme, cursor_size * nag->scale, nag->shm);
 | 
						|
	if (!pointer->cursor_theme) {
 | 
						|
		wlr_log(WLR_ERROR, "Failed to load cursor theme");
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	struct wl_cursor *cursor =
 | 
						|
		wl_cursor_theme_get_cursor(pointer->cursor_theme, "default");
 | 
						|
	if (!cursor) {
 | 
						|
		wlr_log(WLR_ERROR, "Failed to get default cursor from theme");
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	pointer->cursor_image = cursor->images[0];
 | 
						|
	wl_surface_set_buffer_scale(pointer->cursor_surface,
 | 
						|
			nag->scale);
 | 
						|
	wl_surface_attach(pointer->cursor_surface,
 | 
						|
			wl_cursor_image_get_buffer(pointer->cursor_image), 0, 0);
 | 
						|
	wl_pointer_set_cursor(pointer->pointer, pointer->serial,
 | 
						|
			pointer->cursor_surface,
 | 
						|
			pointer->cursor_image->hotspot_x / nag->scale,
 | 
						|
			pointer->cursor_image->hotspot_y / nag->scale);
 | 
						|
	wl_surface_damage_buffer(pointer->cursor_surface, 0, 0,
 | 
						|
			INT32_MAX, INT32_MAX);
 | 
						|
	wl_surface_commit(pointer->cursor_surface);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
update_all_cursors(struct nag *nag)
 | 
						|
{
 | 
						|
	struct seat *seat;
 | 
						|
	wl_list_for_each(seat, &nag->seats, link) {
 | 
						|
		if (seat->pointer.pointer) {
 | 
						|
			update_cursor(seat);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_enter(void *data, struct wl_pointer *wl_pointer, uint32_t serial,
 | 
						|
		struct wl_surface *surface, wl_fixed_t surface_x,
 | 
						|
		wl_fixed_t surface_y)
 | 
						|
{
 | 
						|
	struct seat *seat = data;
 | 
						|
 | 
						|
	struct pointer *pointer = &seat->pointer;
 | 
						|
	pointer->x = wl_fixed_to_int(surface_x);
 | 
						|
	pointer->y = wl_fixed_to_int(surface_y);
 | 
						|
 | 
						|
	if (seat->nag->cursor_shape_manager) {
 | 
						|
		struct wp_cursor_shape_device_v1 *device =
 | 
						|
			wp_cursor_shape_manager_v1_get_pointer(
 | 
						|
				seat->nag->cursor_shape_manager, wl_pointer);
 | 
						|
		wp_cursor_shape_device_v1_set_shape(device, serial,
 | 
						|
			WP_CURSOR_SHAPE_DEVICE_V1_SHAPE_DEFAULT);
 | 
						|
		wp_cursor_shape_device_v1_destroy(device);
 | 
						|
	} else {
 | 
						|
		pointer->serial = serial;
 | 
						|
		update_cursor(seat);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_leave(void *data, struct wl_pointer *wl_pointer,
 | 
						|
		uint32_t serial, struct wl_surface *surface)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_motion(void *data, struct wl_pointer *wl_pointer, uint32_t time,
 | 
						|
		wl_fixed_t surface_x, wl_fixed_t surface_y)
 | 
						|
{
 | 
						|
	struct seat *seat = data;
 | 
						|
	seat->pointer.x = wl_fixed_to_int(surface_x);
 | 
						|
	seat->pointer.y = wl_fixed_to_int(surface_y);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_button(void *data, struct wl_pointer *wl_pointer, uint32_t serial,
 | 
						|
		uint32_t time, uint32_t button, uint32_t state)
 | 
						|
{
 | 
						|
	struct seat *seat = data;
 | 
						|
	struct nag *nag = seat->nag;
 | 
						|
 | 
						|
	if (state != WL_POINTER_BUTTON_STATE_PRESSED) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	double x = seat->pointer.x;
 | 
						|
	double y = seat->pointer.y;
 | 
						|
 | 
						|
	int index = 0;
 | 
						|
	struct button *nagbutton;
 | 
						|
	wl_list_for_each(nagbutton, &nag->buttons, link) {
 | 
						|
		if (x >= nagbutton->x
 | 
						|
				&& y >= nagbutton->y
 | 
						|
				&& x < nagbutton->x + nagbutton->width
 | 
						|
				&& y < nagbutton->y + nagbutton->height) {
 | 
						|
			button_execute(nag, nagbutton);
 | 
						|
			exit_status = index;
 | 
						|
			return;
 | 
						|
		}
 | 
						|
		++index;
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag->details.visible &&
 | 
						|
			nag->details.total_lines != nag->details.visible_lines) {
 | 
						|
		struct button button_up = nag->details.button_up;
 | 
						|
		if (x >= button_up.x
 | 
						|
				&& y >= button_up.y
 | 
						|
				&& x < button_up.x + button_up.width
 | 
						|
				&& y < button_up.y + button_up.height
 | 
						|
				&& nag->details.offset > 0) {
 | 
						|
			nag->details.offset--;
 | 
						|
			render_frame(nag);
 | 
						|
			return;
 | 
						|
		}
 | 
						|
 | 
						|
		struct button button_down = nag->details.button_down;
 | 
						|
		int bot = nag->details.total_lines;
 | 
						|
		bot -= nag->details.visible_lines;
 | 
						|
		if (x >= button_down.x
 | 
						|
				&& y >= button_down.y
 | 
						|
				&& x < button_down.x + button_down.width
 | 
						|
				&& y < button_down.y + button_down.height
 | 
						|
				&& nag->details.offset < bot) {
 | 
						|
			nag->details.offset++;
 | 
						|
			render_frame(nag);
 | 
						|
			return;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_axis(void *data, struct wl_pointer *wl_pointer, uint32_t time,
 | 
						|
		uint32_t axis, wl_fixed_t value)
 | 
						|
{
 | 
						|
	struct seat *seat = data;
 | 
						|
	struct nag *nag = seat->nag;
 | 
						|
	if (!nag->details.visible
 | 
						|
			|| seat->pointer.x < nag->details.x
 | 
						|
			|| seat->pointer.y < nag->details.y
 | 
						|
			|| seat->pointer.x >= nag->details.x + nag->details.width
 | 
						|
			|| seat->pointer.y >= nag->details.y + nag->details.height
 | 
						|
			|| nag->details.total_lines == nag->details.visible_lines) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	int direction = wl_fixed_to_int(value);
 | 
						|
	int bot = nag->details.total_lines - nag->details.visible_lines;
 | 
						|
	if (direction < 0 && nag->details.offset > 0) {
 | 
						|
		nag->details.offset--;
 | 
						|
	} else if (direction > 0 && nag->details.offset < bot) {
 | 
						|
		nag->details.offset++;
 | 
						|
	}
 | 
						|
 | 
						|
	render_frame(nag);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_frame(void *data, struct wl_pointer *wl_pointer)
 | 
						|
{
 | 
						|
	struct seat *seat = data;
 | 
						|
	/* pointer inputs clears timer for auto-closing */
 | 
						|
	close_pollfd(&seat->nag->pollfds[FD_TIMER]);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_axis_source(void *data, struct wl_pointer *wl_pointer,
 | 
						|
		uint32_t axis_source)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_axis_stop(void *data, struct wl_pointer *wl_pointer,
 | 
						|
		uint32_t time, uint32_t axis)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
wl_pointer_axis_discrete(void *data, struct wl_pointer *wl_pointer,
 | 
						|
		uint32_t axis, int32_t discrete)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_pointer_listener pointer_listener = {
 | 
						|
	.enter = wl_pointer_enter,
 | 
						|
	.leave = wl_pointer_leave,
 | 
						|
	.motion = wl_pointer_motion,
 | 
						|
	.button = wl_pointer_button,
 | 
						|
	.axis = wl_pointer_axis,
 | 
						|
	.frame = wl_pointer_frame,
 | 
						|
	.axis_source = wl_pointer_axis_source,
 | 
						|
	.axis_stop = wl_pointer_axis_stop,
 | 
						|
	.axis_discrete = wl_pointer_axis_discrete,
 | 
						|
};
 | 
						|
 | 
						|
static void
 | 
						|
seat_handle_capabilities(void *data, struct wl_seat *wl_seat,
 | 
						|
		enum wl_seat_capability caps)
 | 
						|
{
 | 
						|
	struct seat *seat = data;
 | 
						|
	bool cap_pointer = caps & WL_SEAT_CAPABILITY_POINTER;
 | 
						|
	if (cap_pointer && !seat->pointer.pointer) {
 | 
						|
		seat->pointer.pointer = wl_seat_get_pointer(wl_seat);
 | 
						|
		wl_pointer_add_listener(seat->pointer.pointer,
 | 
						|
				&pointer_listener, seat);
 | 
						|
	} else if (!cap_pointer && seat->pointer.pointer) {
 | 
						|
		wl_pointer_destroy(seat->pointer.pointer);
 | 
						|
		seat->pointer.pointer = NULL;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
seat_handle_name(void *data, struct wl_seat *wl_seat, const char *name)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_seat_listener seat_listener = {
 | 
						|
	.capabilities = seat_handle_capabilities,
 | 
						|
	.name = seat_handle_name,
 | 
						|
};
 | 
						|
 | 
						|
static void
 | 
						|
output_geometry(void *data, struct wl_output *wl_output, int32_t x, int32_t y,
 | 
						|
		int32_t physical_width, int32_t physical_height,
 | 
						|
		int32_t subpixel, const char *make, const char *model,
 | 
						|
		int32_t transform)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
output_mode(void *data, struct wl_output *wl_output, uint32_t flags,
 | 
						|
		int32_t width, int32_t height, int32_t refresh)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
output_done(void *data, struct wl_output *output)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
output_scale(void *data, struct wl_output *output, int32_t factor)
 | 
						|
{
 | 
						|
	struct output *nag_output = data;
 | 
						|
	nag_output->scale = factor;
 | 
						|
	if (nag_output->nag->output == nag_output) {
 | 
						|
		nag_output->nag->scale = nag_output->scale;
 | 
						|
		if (!nag_output->nag->cursor_shape_manager) {
 | 
						|
			update_all_cursors(nag_output->nag);
 | 
						|
		}
 | 
						|
		render_frame(nag_output->nag);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
output_name(void *data, struct wl_output *output, const char *name)
 | 
						|
{
 | 
						|
	struct output *nag_output = data;
 | 
						|
	nag_output->name = strdup(name);
 | 
						|
 | 
						|
	const char *outname = nag_output->nag->conf->output;
 | 
						|
	if (!nag_output->nag->output && outname &&
 | 
						|
			strcmp(outname, name) == 0) {
 | 
						|
		wlr_log(WLR_DEBUG, "Using output %s", name);
 | 
						|
		nag_output->nag->output = nag_output;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
output_description(void *data, struct wl_output *wl_output,
 | 
						|
		const char *description)
 | 
						|
{
 | 
						|
	/* nop */
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_output_listener output_listener = {
 | 
						|
	.geometry = output_geometry,
 | 
						|
	.mode = output_mode,
 | 
						|
	.done = output_done,
 | 
						|
	.scale = output_scale,
 | 
						|
	.name = output_name,
 | 
						|
	.description = output_description,
 | 
						|
};
 | 
						|
 | 
						|
static void
 | 
						|
handle_global(void *data, struct wl_registry *registry, uint32_t name,
 | 
						|
		const char *interface, uint32_t version)
 | 
						|
{
 | 
						|
	struct nag *nag = data;
 | 
						|
	if (strcmp(interface, wl_compositor_interface.name) == 0) {
 | 
						|
		nag->compositor = wl_registry_bind(registry, name,
 | 
						|
				&wl_compositor_interface, 4);
 | 
						|
	} else if (strcmp(interface, wl_seat_interface.name) == 0) {
 | 
						|
		struct seat *seat = calloc(1, sizeof(*seat));
 | 
						|
		if (!seat) {
 | 
						|
			perror("calloc");
 | 
						|
			return;
 | 
						|
		}
 | 
						|
 | 
						|
		seat->nag = nag;
 | 
						|
		seat->wl_name = name;
 | 
						|
		seat->wl_seat =
 | 
						|
			wl_registry_bind(registry, name, &wl_seat_interface, 5);
 | 
						|
 | 
						|
		wl_seat_add_listener(seat->wl_seat, &seat_listener, seat);
 | 
						|
 | 
						|
		wl_list_insert(&nag->seats, &seat->link);
 | 
						|
	} else if (strcmp(interface, wl_shm_interface.name) == 0) {
 | 
						|
		nag->shm = wl_registry_bind(registry, name, &wl_shm_interface, 1);
 | 
						|
	} else if (strcmp(interface, wl_output_interface.name) == 0) {
 | 
						|
		if (!nag->output) {
 | 
						|
			struct output *output = calloc(1, sizeof(*output));
 | 
						|
			if (!output) {
 | 
						|
				perror("calloc");
 | 
						|
				return;
 | 
						|
			}
 | 
						|
			output->wl_output = wl_registry_bind(registry, name,
 | 
						|
					&wl_output_interface, 4);
 | 
						|
			output->wl_name = name;
 | 
						|
			output->scale = 1;
 | 
						|
			output->nag = nag;
 | 
						|
			wl_list_insert(&nag->outputs, &output->link);
 | 
						|
			wl_output_add_listener(output->wl_output,
 | 
						|
					&output_listener, output);
 | 
						|
		}
 | 
						|
	} else if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
 | 
						|
		nag->layer_shell = wl_registry_bind(
 | 
						|
				registry, name, &zwlr_layer_shell_v1_interface, 1);
 | 
						|
	} else if (strcmp(interface, wp_cursor_shape_manager_v1_interface.name) == 0) {
 | 
						|
		nag->cursor_shape_manager = wl_registry_bind(
 | 
						|
				registry, name, &wp_cursor_shape_manager_v1_interface, 1);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
handle_global_remove(void *data, struct wl_registry *registry, uint32_t name)
 | 
						|
{
 | 
						|
	struct nag *nag = data;
 | 
						|
	if (nag->output->wl_name == name) {
 | 
						|
		nag->run_display = false;
 | 
						|
	}
 | 
						|
 | 
						|
	struct seat *seat, *tmpseat;
 | 
						|
	wl_list_for_each_safe(seat, tmpseat, &nag->seats, link) {
 | 
						|
		if (seat->wl_name == name) {
 | 
						|
			seat_destroy(seat);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_registry_listener registry_listener = {
 | 
						|
	.global = handle_global,
 | 
						|
	.global_remove = handle_global_remove,
 | 
						|
};
 | 
						|
 | 
						|
static void
 | 
						|
nag_setup_cursors(struct nag *nag)
 | 
						|
{
 | 
						|
	struct seat *seat;
 | 
						|
 | 
						|
	wl_list_for_each(seat, &nag->seats, link) {
 | 
						|
		struct pointer *p = &seat->pointer;
 | 
						|
 | 
						|
		p->cursor_surface =
 | 
						|
			wl_compositor_create_surface(nag->compositor);
 | 
						|
		assert(p->cursor_surface);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
nag_setup(struct nag *nag)
 | 
						|
{
 | 
						|
	nag->display = wl_display_connect(NULL);
 | 
						|
	if (!nag->display) {
 | 
						|
		wlr_log(WLR_ERROR, "Unable to connect to the compositor. "
 | 
						|
				"If your compositor is running, check or set the "
 | 
						|
				"WAYLAND_DISPLAY environment variable.");
 | 
						|
		exit(LAB_EXIT_FAILURE);
 | 
						|
	}
 | 
						|
 | 
						|
	nag->scale = 1;
 | 
						|
 | 
						|
	struct wl_registry *registry = wl_display_get_registry(nag->display);
 | 
						|
	wl_registry_add_listener(registry, ®istry_listener, nag);
 | 
						|
	if (wl_display_roundtrip(nag->display) < 0) {
 | 
						|
		wlr_log(WLR_ERROR, "failed to register with the wayland display");
 | 
						|
		exit(LAB_EXIT_FAILURE);
 | 
						|
	}
 | 
						|
 | 
						|
	assert(nag->compositor && nag->layer_shell && nag->shm);
 | 
						|
 | 
						|
	/* Second roundtrip to get wl_output properties */
 | 
						|
	if (wl_display_roundtrip(nag->display) < 0) {
 | 
						|
		wlr_log(WLR_ERROR, "Error during outputs init.");
 | 
						|
		nag_destroy(nag);
 | 
						|
		exit(LAB_EXIT_FAILURE);
 | 
						|
	}
 | 
						|
 | 
						|
	if (!nag->output && nag->conf->output) {
 | 
						|
		wlr_log(WLR_ERROR, "Output '%s' not found", nag->conf->output);
 | 
						|
		nag_destroy(nag);
 | 
						|
		exit(LAB_EXIT_FAILURE);
 | 
						|
	}
 | 
						|
 | 
						|
	if (!nag->cursor_shape_manager) {
 | 
						|
		nag_setup_cursors(nag);
 | 
						|
	}
 | 
						|
 | 
						|
	nag->surface = wl_compositor_create_surface(nag->compositor);
 | 
						|
	assert(nag->surface);
 | 
						|
	wl_surface_add_listener(nag->surface, &surface_listener, nag);
 | 
						|
 | 
						|
	nag->layer_surface = zwlr_layer_shell_v1_get_layer_surface(
 | 
						|
			nag->layer_shell, nag->surface,
 | 
						|
			nag->output ? nag->output->wl_output : NULL,
 | 
						|
			nag->conf->layer,
 | 
						|
			"nag");
 | 
						|
	assert(nag->layer_surface);
 | 
						|
	zwlr_layer_surface_v1_add_listener(nag->layer_surface,
 | 
						|
			&layer_surface_listener, nag);
 | 
						|
	zwlr_layer_surface_v1_set_anchor(nag->layer_surface,
 | 
						|
			nag->conf->anchors);
 | 
						|
 | 
						|
	wl_registry_destroy(registry);
 | 
						|
 | 
						|
	nag->pollfds[FD_WAYLAND].fd = wl_display_get_fd(nag->display);
 | 
						|
	nag->pollfds[FD_WAYLAND].events = POLLIN;
 | 
						|
 | 
						|
	if (nag->details.close_timeout != 0) {
 | 
						|
		nag->pollfds[FD_TIMER].fd = timerfd_create(CLOCK_MONOTONIC, TFD_CLOEXEC);
 | 
						|
		nag->pollfds[FD_TIMER].events = POLLIN;
 | 
						|
		struct itimerspec timeout = {
 | 
						|
			.it_value.tv_sec = nag->details.close_timeout,
 | 
						|
		};
 | 
						|
		timerfd_settime(nag->pollfds[FD_TIMER].fd, 0, &timeout, NULL);
 | 
						|
	} else {
 | 
						|
		nag->pollfds[FD_TIMER].fd = -1;
 | 
						|
	}
 | 
						|
 | 
						|
	sigset_t mask;
 | 
						|
	sigaddset(&mask, SIGINT);
 | 
						|
	sigaddset(&mask, SIGTERM);
 | 
						|
	sigprocmask(SIG_BLOCK, &mask, NULL);
 | 
						|
	nag->pollfds[FD_SIGNAL].fd = signalfd(-1, &mask, SFD_CLOEXEC | SFD_NONBLOCK);
 | 
						|
	nag->pollfds[FD_SIGNAL].events = POLLIN;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
close_pollfd(struct pollfd *pollfd)
 | 
						|
{
 | 
						|
	if (pollfd->fd == -1) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	close(pollfd->fd);
 | 
						|
	pollfd->fd = -1;
 | 
						|
	pollfd->events = 0;
 | 
						|
	pollfd->revents = 0;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
nag_run(struct nag *nag)
 | 
						|
{
 | 
						|
	nag->run_display = true;
 | 
						|
	render_frame(nag);
 | 
						|
	while (nag->run_display) {
 | 
						|
		while (wl_display_prepare_read(nag->display) != 0) {
 | 
						|
			wl_display_dispatch_pending(nag->display);
 | 
						|
		}
 | 
						|
 | 
						|
		errno = 0;
 | 
						|
		if (wl_display_flush(nag->display) == -1 && errno != EAGAIN) {
 | 
						|
			break;
 | 
						|
		}
 | 
						|
 | 
						|
		if (!nag->run_display) {
 | 
						|
			break;
 | 
						|
		}
 | 
						|
 | 
						|
		poll(nag->pollfds, NR_FDS, -1);
 | 
						|
		if (nag->pollfds[FD_WAYLAND].revents & POLLIN) {
 | 
						|
			wl_display_read_events(nag->display);
 | 
						|
		} else {
 | 
						|
			wl_display_cancel_read(nag->display);
 | 
						|
		}
 | 
						|
		if (nag->pollfds[FD_TIMER].revents & POLLIN) {
 | 
						|
			exit_status = LAB_EXIT_TIMEOUT;
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		if (nag->pollfds[FD_SIGNAL].revents & POLLIN) {
 | 
						|
			break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
conf_init(struct conf *conf)
 | 
						|
{
 | 
						|
	conf->font_description = pango_font_description_from_string("pango:Sans 10");
 | 
						|
	conf->anchors = ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP
 | 
						|
		| ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT
 | 
						|
		| ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
 | 
						|
	conf->layer = ZWLR_LAYER_SHELL_V1_LAYER_TOP;
 | 
						|
	conf->button_background = 0x333333FF;
 | 
						|
	conf->details_background = 0x333333FF;
 | 
						|
	conf->background = 0x323232FF;
 | 
						|
	conf->text = 0xFFFFFFFF;
 | 
						|
	conf->button_text = 0xFFFFFFFF;
 | 
						|
	conf->button_border = 0x222222FF;
 | 
						|
	conf->border_bottom = 0x444444FF;
 | 
						|
	conf->bar_border_thickness = 2;
 | 
						|
	conf->message_padding = 8;
 | 
						|
	conf->details_border_thickness = 3;
 | 
						|
	conf->button_border_thickness = 3;
 | 
						|
	conf->button_gap = 20;
 | 
						|
	conf->button_gap_close = 15;
 | 
						|
	conf->button_margin_right = 2;
 | 
						|
	conf->button_padding = 3;
 | 
						|
	conf->button_background = 0x680A0AFF;
 | 
						|
	conf->details_background = 0x680A0AFF;
 | 
						|
	conf->background = 0x900000FF;
 | 
						|
	conf->text = 0xFFFFFFFF;
 | 
						|
	conf->button_text = 0xFFFFFFFF;
 | 
						|
	conf->button_border = 0xD92424FF;
 | 
						|
	conf->border_bottom = 0x470909FF;
 | 
						|
}
 | 
						|
 | 
						|
static bool
 | 
						|
parse_color(const char *color, uint32_t *result)
 | 
						|
{
 | 
						|
	if (color[0] == '#') {
 | 
						|
		++color;
 | 
						|
	}
 | 
						|
	int len = strlen(color);
 | 
						|
	if ((len != 6 && len != 8) || !isxdigit(color[0]) || !isxdigit(color[1])) {
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
	char *ptr;
 | 
						|
	uint32_t parsed = strtoul(color, &ptr, 16);
 | 
						|
	if (*ptr != '\0') {
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
	*result = len == 6 ? ((parsed << 8) | 0xFF) : parsed;
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
/*
 | 
						|
 * As labnag is slow for large "detailed messages" we curtail stdin at an
 | 
						|
 * arbitrary size to avoid hogging the CPU.
 | 
						|
 */
 | 
						|
#define MAX_STDIN_LINES 200
 | 
						|
static char *
 | 
						|
read_and_trim_stdin(void)
 | 
						|
{
 | 
						|
	char *buffer = NULL, *line = NULL;
 | 
						|
	size_t buffer_len = 0, line_size = 0;
 | 
						|
	int line_count = 0;
 | 
						|
	while (line_count < MAX_STDIN_LINES) {
 | 
						|
		ssize_t nread = getline(&line, &line_size, stdin);
 | 
						|
		if (nread == -1) {
 | 
						|
			if (feof(stdin)) {
 | 
						|
				break;
 | 
						|
			} else {
 | 
						|
				perror("getline");
 | 
						|
				goto freeline;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		buffer = realloc(buffer, buffer_len + nread + 1);
 | 
						|
		if (!buffer) {
 | 
						|
			perror("realloc");
 | 
						|
			return NULL;
 | 
						|
		}
 | 
						|
		memcpy(&buffer[buffer_len], line, nread + 1);
 | 
						|
		buffer_len += nread;
 | 
						|
		++line_count;
 | 
						|
	}
 | 
						|
	free(line);
 | 
						|
 | 
						|
	while (buffer_len && buffer[buffer_len - 1] == '\n') {
 | 
						|
		buffer[--buffer_len] = '\0';
 | 
						|
	}
 | 
						|
 | 
						|
	return buffer;
 | 
						|
 | 
						|
freeline:
 | 
						|
	free(line);
 | 
						|
	return NULL;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
nag_parse_options(int argc, char **argv, struct nag *nag,
 | 
						|
		struct conf *conf, bool *debug)
 | 
						|
{
 | 
						|
	enum type_options {
 | 
						|
		TO_COLOR_BACKGROUND = 256,
 | 
						|
		TO_COLOR_BUTTON_BORDER,
 | 
						|
		TO_COLOR_BORDER_BOTTOM,
 | 
						|
		TO_COLOR_BUTTON_BG,
 | 
						|
		TO_COLOR_DETAILS,
 | 
						|
		TO_COLOR_TEXT,
 | 
						|
		TO_COLOR_BUTTON_TEXT,
 | 
						|
		TO_THICK_BAR_BORDER,
 | 
						|
		TO_PADDING_MESSAGE,
 | 
						|
		TO_THICK_DET_BORDER,
 | 
						|
		TO_THICK_BTN_BORDER,
 | 
						|
		TO_GAP_BTN,
 | 
						|
		TO_GAP_BTN_DISMISS,
 | 
						|
		TO_MARGIN_BTN_RIGHT,
 | 
						|
		TO_PADDING_BTN,
 | 
						|
	};
 | 
						|
 | 
						|
	static const struct option opts[] = {
 | 
						|
		{"button", required_argument, NULL, 'B'},
 | 
						|
		{"button-dismiss", required_argument, NULL, 'Z'},
 | 
						|
		{"debug", no_argument, NULL, 'd'},
 | 
						|
		{"edge", required_argument, NULL, 'e'},
 | 
						|
		{"layer", required_argument, NULL, 'y'},
 | 
						|
		{"font", required_argument, NULL, 'f'},
 | 
						|
		{"help", no_argument, NULL, 'h'},
 | 
						|
		{"detailed-message", no_argument, NULL, 'l'},
 | 
						|
		{"detailed-button", required_argument, NULL, 'L'},
 | 
						|
		{"message", required_argument, NULL, 'm'},
 | 
						|
		{"output", required_argument, NULL, 'o'},
 | 
						|
		{"timeout", required_argument, NULL, 't'},
 | 
						|
		{"version", no_argument, NULL, 'v'},
 | 
						|
 | 
						|
		{"background-color", required_argument, NULL, TO_COLOR_BACKGROUND},
 | 
						|
		{"button-border-color", required_argument, NULL, TO_COLOR_BUTTON_BORDER},
 | 
						|
		{"border-bottom-color", required_argument, NULL, TO_COLOR_BORDER_BOTTOM},
 | 
						|
		{"button-background-color", required_argument, NULL, TO_COLOR_BUTTON_BG},
 | 
						|
		{"text-color", required_argument, NULL, TO_COLOR_TEXT},
 | 
						|
		{"button-text-color", required_argument, NULL, TO_COLOR_BUTTON_TEXT},
 | 
						|
		{"border-bottom-size", required_argument, NULL, TO_THICK_BAR_BORDER},
 | 
						|
		{"message-padding", required_argument, NULL, TO_PADDING_MESSAGE},
 | 
						|
		{"details-border-size", required_argument, NULL, TO_THICK_DET_BORDER},
 | 
						|
		{"details-background-color", required_argument, NULL, TO_COLOR_DETAILS},
 | 
						|
		{"button-border-size", required_argument, NULL, TO_THICK_BTN_BORDER},
 | 
						|
		{"button-gap", required_argument, NULL, TO_GAP_BTN},
 | 
						|
		{"button-dismiss-gap", required_argument, NULL, TO_GAP_BTN_DISMISS},
 | 
						|
		{"button-margin-right", required_argument, NULL, TO_MARGIN_BTN_RIGHT},
 | 
						|
		{"button-padding", required_argument, NULL, TO_PADDING_BTN},
 | 
						|
 | 
						|
		{0, 0, 0, 0}
 | 
						|
	};
 | 
						|
 | 
						|
	const char *usage =
 | 
						|
		"Usage: labnag [options...]\n"
 | 
						|
		"\n"
 | 
						|
		"  -B, --button <text> [<action>]  Create a button with text\n"
 | 
						|
		"  -Z, --button-dismiss <text> [<action>]\n"
 | 
						|
		"                                  Like -B but dismiss nag when pressed\n"
 | 
						|
		"  -d, --debug                     Enable debugging.\n"
 | 
						|
		"  -e, --edge top|bottom           Set the edge to use.\n"
 | 
						|
		"  -y, --layer overlay|top|bottom|background\n"
 | 
						|
		"                                  Set the layer to use.\n"
 | 
						|
		"  -f, --font <font>               Set the font to use.\n"
 | 
						|
		"  -h, --help                      Show help message and quit.\n"
 | 
						|
		"  -l, --detailed-message          Read a detailed message from stdin.\n"
 | 
						|
		"  -L, --detailed-button <text>    Set the text of the detail button.\n"
 | 
						|
		"  -m, --message <msg>             Set the message text.\n"
 | 
						|
		"  -o, --output <output>           Set the output to use.\n"
 | 
						|
		"  -t, --timeout <seconds>         Set duration to close dialog.\n"
 | 
						|
		"  -x, --exclusive-zone            Use exclusive zone.\n"
 | 
						|
		"  -v, --version                   Show the version number and quit.\n"
 | 
						|
		"\n"
 | 
						|
		"The following appearance options can also be given:\n"
 | 
						|
		"  --background-color RRGGBB[AA]    Background color.\n"
 | 
						|
		"  --button-border-color RRGGBB[AA] Button border color.\n"
 | 
						|
		"  --border-bottom-color RRGGBB[AA] Bottom border color.\n"
 | 
						|
		"  --button-background-color RRGGBB[AA]\n"
 | 
						|
		"                                   Button background color.\n"
 | 
						|
		"  --text-color RRGGBB[AA]          Text color.\n"
 | 
						|
		"  --button-text-color RRGGBB[AA]   Button text color.\n"
 | 
						|
		"  --border-bottom-size size        Thickness of the bar border.\n"
 | 
						|
		"  --message-padding padding        Padding for the message.\n"
 | 
						|
		"  --details-border-size size       Thickness for the details border.\n"
 | 
						|
		"  --details-background-color RRGGBB[AA]\n"
 | 
						|
		"                                   Details background color.\n"
 | 
						|
		"  --button-border-size size        Thickness for the button border.\n"
 | 
						|
		"  --button-gap gap                 Size of the gap between buttons\n"
 | 
						|
		"  --button-dismiss-gap gap         Size of the gap for dismiss button.\n"
 | 
						|
		"  --button-margin-right margin     Margin from dismiss button to edge.\n"
 | 
						|
		"  --button-padding padding         Padding for the button text.\n";
 | 
						|
 | 
						|
	optind = 1;
 | 
						|
	while (1) {
 | 
						|
		int c = getopt_long(argc, argv, "B:Z:c:de:y:f:hlL:m:o:s:t:vx", opts, NULL);
 | 
						|
		if (c == -1) {
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		switch (c) {
 | 
						|
		case 'B': /* Button */
 | 
						|
		case 'Z': /* Button (Dismiss) */ {
 | 
						|
			struct button *button = calloc(1, sizeof(*button));
 | 
						|
			if (!button) {
 | 
						|
				perror("calloc");
 | 
						|
				return LAB_EXIT_FAILURE;
 | 
						|
			}
 | 
						|
			if (argv[optind] && argv[optind][0] != '-') {
 | 
						|
				button->action = argv[optind];
 | 
						|
				optind++;
 | 
						|
			}
 | 
						|
			button->text = optarg;
 | 
						|
			button->dismiss = c == 'Z';
 | 
						|
			wl_list_insert(&nag->buttons, &button->link);
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		case 'd': /* Debug */
 | 
						|
			*debug = true;
 | 
						|
			break;
 | 
						|
		case 'e': /* Edge */
 | 
						|
			if (strcmp(optarg, "top") == 0) {
 | 
						|
				conf->anchors = ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP
 | 
						|
					| ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT
 | 
						|
					| ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
 | 
						|
			} else if (strcmp(optarg, "bottom") == 0) {
 | 
						|
				conf->anchors = ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM
 | 
						|
					| ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT
 | 
						|
					| ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
 | 
						|
			} else {
 | 
						|
				fprintf(stderr, "Invalid edge: %s\n", optarg);
 | 
						|
				return LAB_EXIT_FAILURE;
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case 'y': /* Layer */
 | 
						|
			if (strcmp(optarg, "background") == 0) {
 | 
						|
				conf->layer = ZWLR_LAYER_SHELL_V1_LAYER_BACKGROUND;
 | 
						|
			} else if (strcmp(optarg, "bottom") == 0) {
 | 
						|
				conf->layer = ZWLR_LAYER_SHELL_V1_LAYER_BOTTOM;
 | 
						|
			} else if (strcmp(optarg, "top") == 0) {
 | 
						|
				conf->layer = ZWLR_LAYER_SHELL_V1_LAYER_TOP;
 | 
						|
			} else if (strcmp(optarg, "overlay") == 0) {
 | 
						|
				conf->layer = ZWLR_LAYER_SHELL_V1_LAYER_OVERLAY;
 | 
						|
			} else {
 | 
						|
				fprintf(stderr, "Invalid layer: %s\n"
 | 
						|
						"Usage: --layer overlay|top|bottom|background\n",
 | 
						|
						optarg);
 | 
						|
				return LAB_EXIT_FAILURE;
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case 'f': /* Font */
 | 
						|
			pango_font_description_free(conf->font_description);
 | 
						|
			conf->font_description = pango_font_description_from_string(optarg);
 | 
						|
			break;
 | 
						|
		case 'l': /* Detailed Message */
 | 
						|
			free(nag->details.message);
 | 
						|
			nag->details.message = read_and_trim_stdin();
 | 
						|
			if (!nag->details.message) {
 | 
						|
				return LAB_EXIT_FAILURE;
 | 
						|
			}
 | 
						|
			nag->details.button_up.text = "▲";
 | 
						|
			nag->details.button_down.text = "▼";
 | 
						|
			break;
 | 
						|
		case 'L': /* Detailed Button Text */
 | 
						|
			nag->details.details_text = optarg;
 | 
						|
			break;
 | 
						|
		case 'm': /* Message */
 | 
						|
			nag->message = optarg;
 | 
						|
			break;
 | 
						|
		case 'o': /* Output */
 | 
						|
			free(conf->output);
 | 
						|
			conf->output = optarg;
 | 
						|
			break;
 | 
						|
		case 't':
 | 
						|
			nag->details.close_timeout = atoi(optarg);
 | 
						|
			break;
 | 
						|
		case 'x':
 | 
						|
			nag->details.use_exclusive_zone = true;
 | 
						|
			break;
 | 
						|
		case 'v': /* Version */
 | 
						|
			printf("labnag " LABWC_VERSION "\n");
 | 
						|
			return LAB_EXIT_FAILURE;
 | 
						|
		case TO_COLOR_BACKGROUND: /* Background color */
 | 
						|
			if (!parse_color(optarg, &conf->background)) {
 | 
						|
				fprintf(stderr, "Invalid background color: %s\n", optarg);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case TO_COLOR_BUTTON_BORDER: /* Border color */
 | 
						|
			if (!parse_color(optarg, &conf->button_border)) {
 | 
						|
				fprintf(stderr, "Invalid border color: %s\n", optarg);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case TO_COLOR_BORDER_BOTTOM: /* Bottom border color */
 | 
						|
			if (!parse_color(optarg, &conf->border_bottom)) {
 | 
						|
				fprintf(stderr, "Invalid border bottom color: %s\n", optarg);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case TO_COLOR_BUTTON_BG: /* Button background color */
 | 
						|
			if (!parse_color(optarg, &conf->button_background)) {
 | 
						|
				fprintf(stderr, "Invalid button background color: %s\n", optarg);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case TO_COLOR_DETAILS: /* Details background color */
 | 
						|
			if (!parse_color(optarg, &conf->details_background)) {
 | 
						|
				fprintf(stderr, "Invalid details background color: %s\n", optarg);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case TO_COLOR_TEXT: /* Text color */
 | 
						|
			if (!parse_color(optarg, &conf->text)) {
 | 
						|
				fprintf(stderr, "Invalid text color: %s\n", optarg);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case TO_COLOR_BUTTON_TEXT: /* Button text color */
 | 
						|
			if (!parse_color(optarg, &conf->button_text)) {
 | 
						|
				fprintf(stderr, "Invalid button text color: %s\n", optarg);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		case TO_THICK_BAR_BORDER: /* Bottom border thickness */
 | 
						|
			conf->bar_border_thickness = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		case TO_PADDING_MESSAGE: /* Message padding */
 | 
						|
			conf->message_padding = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		case TO_THICK_DET_BORDER: /* Details border thickness */
 | 
						|
			conf->details_border_thickness = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		case TO_THICK_BTN_BORDER: /* Button border thickness */
 | 
						|
			conf->button_border_thickness = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		case TO_GAP_BTN: /* Gap between buttons */
 | 
						|
			conf->button_gap = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		case TO_GAP_BTN_DISMISS: /* Gap between dismiss button */
 | 
						|
			conf->button_gap_close = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		case TO_MARGIN_BTN_RIGHT: /* Margin on the right side of button area */
 | 
						|
			conf->button_margin_right = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		case TO_PADDING_BTN: /* Padding for the button text */
 | 
						|
			conf->button_padding = strtol(optarg, NULL, 0);
 | 
						|
			break;
 | 
						|
		default: /* Help or unknown flag */
 | 
						|
			fprintf(c == 'h' ? stdout : stderr, "%s", usage);
 | 
						|
			return LAB_EXIT_FAILURE;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return LAB_EXIT_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
main(int argc, char **argv)
 | 
						|
{
 | 
						|
	struct conf conf = { 0 };
 | 
						|
	conf_init(&conf);
 | 
						|
 | 
						|
	struct nag nag = {
 | 
						|
		.conf = &conf,
 | 
						|
	};
 | 
						|
 | 
						|
	wl_list_init(&nag.buttons);
 | 
						|
	wl_list_init(&nag.outputs);
 | 
						|
	wl_list_init(&nag.seats);
 | 
						|
 | 
						|
	nag.details.details_text = "Toggle details";
 | 
						|
	nag.details.close_timeout = 5;
 | 
						|
	nag.details.use_exclusive_zone = false;
 | 
						|
 | 
						|
	bool debug = false;
 | 
						|
	if (argc > 1) {
 | 
						|
		exit_status = nag_parse_options(argc, argv, &nag, &conf, &debug);
 | 
						|
		if (exit_status == LAB_EXIT_FAILURE) {
 | 
						|
			goto cleanup;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	wlr_log_init(debug ? WLR_DEBUG : WLR_ERROR, NULL);
 | 
						|
 | 
						|
	if (!nag.message) {
 | 
						|
		wlr_log(WLR_ERROR, "No message passed. Please provide --message/-m");
 | 
						|
		exit_status = LAB_EXIT_FAILURE;
 | 
						|
		goto cleanup;
 | 
						|
	}
 | 
						|
 | 
						|
	if (nag.details.message) {
 | 
						|
		nag.details.button_details = calloc(1, sizeof(struct button));
 | 
						|
		assert(nag.details.button_details);
 | 
						|
		nag.details.button_details->text = nag.details.details_text;
 | 
						|
		assert(nag.details.button_details->text);
 | 
						|
		nag.details.button_details->expand = true;
 | 
						|
		wl_list_insert(nag.buttons.prev, &nag.details.button_details->link);
 | 
						|
	}
 | 
						|
 | 
						|
	wlr_log(WLR_DEBUG, "Output: %s", nag.conf->output);
 | 
						|
	wlr_log(WLR_DEBUG, "Anchors: %lu", (unsigned long)nag.conf->anchors);
 | 
						|
	wlr_log(WLR_DEBUG, "Message: %s", nag.message);
 | 
						|
	char *font = pango_font_description_to_string(nag.conf->font_description);
 | 
						|
	wlr_log(WLR_DEBUG, "Font: %s", font);
 | 
						|
	free(font);
 | 
						|
	wlr_log(WLR_DEBUG, "Buttons");
 | 
						|
 | 
						|
	struct button *button;
 | 
						|
	wl_list_for_each(button, &nag.buttons, link) {
 | 
						|
		wlr_log(WLR_DEBUG, "\t[%s] `%s`", button->text, button->action);
 | 
						|
	}
 | 
						|
 | 
						|
	nag_setup(&nag);
 | 
						|
 | 
						|
	nag_run(&nag);
 | 
						|
 | 
						|
cleanup:
 | 
						|
	nag_destroy(&nag);
 | 
						|
	return exit_status;
 | 
						|
}
 |