hikari/src/mark_assign_mode.c

#include <hikari/mark_assign_mode.h>

#include <wlr/types/wlr_seat.h>

#include <hikari/configuration.h>
#include <hikari/indicator.h>
#include <hikari/indicator_frame.h>
#include <hikari/keyboard.h>
#include <hikari/mark.h>
#include <hikari/normal_mode.h>
#include <hikari/renderer.h>
#include <hikari/view.h>

static struct hikari_mark_assign_mode *
get_mode(void)
{
	struct hikari_mark_assign_mode *mode = &hikari_server.mark_assign_mode;

	assert(mode == (struct hikari_mark_assign_mode *)hikari_server.mode);

	return mode;
}

static void
clear_conflict(struct hikari_mark_assign_mode *mode, struct hikari_mark *mark)
{
	if (mark != NULL && mark->view != NULL) {
		struct hikari_view *view = mark->view;

		hikari_view_damage_border(view);
		hikari_indicator_damage(&mode->indicator, view);
	}
}

static void
update_state(struct hikari_mark *mark)
{
	struct hikari_mark_assign_mode *mode = get_mode();
	struct hikari_workspace *workspace = hikari_server.workspace;
	struct hikari_view *view = workspace->focus_view;
	struct hikari_output *output = workspace->output;
	struct wlr_box *geometry = hikari_view_border_geometry(view);
	struct hikari_indicator *indicator = &hikari_server.indicator;
	struct hikari_mark *pending_mark = mode->pending_mark;

	if (pending_mark != mark) {
		clear_conflict(mode, pending_mark);
	}

	if (mark == NULL) {
		hikari_indicator_set_color_mark(
				indicator, hikari_configuration->indicator_insert);
		hikari_indicator_update_mark(indicator, output, " ");
	} else {
		float *indicator_color;
		struct hikari_view *mark_view = mark->view;

		if (mark_view != NULL) {
			hikari_indicator_update(&mode->indicator, mark_view);

			hikari_view_damage_border(mark_view);
			hikari_indicator_damage(&mode->indicator, mark_view);

			indicator_color = hikari_configuration->indicator_conflict;
		} else {
			indicator_color = hikari_configuration->indicator_insert;
		}

		hikari_indicator_set_color_mark(indicator, indicator_color);
		hikari_indicator_update_mark(indicator, output, mark->name);
	}

	hikari_indicator_damage_mark(indicator, output, geometry);

	mode->pending_mark = mark;
}

static void
confirm_mark_assign(void)
{
	struct hikari_mark_assign_mode *mode = get_mode();
	struct hikari_workspace *workspace = hikari_server.workspace;
	struct hikari_view *view = workspace->focus_view;
	struct hikari_output *output = workspace->output;
	struct hikari_mark *pending_mark = mode->pending_mark;
	struct hikari_indicator *indicator = &hikari_server.indicator;

	hikari_indicator_set_color_mark(
			indicator, hikari_configuration->indicator_selected);

	if (pending_mark != NULL) {
		clear_conflict(mode, pending_mark);

		hikari_mark_set(pending_mark, view);
		hikari_indicator_update_mark(indicator, output, mode->pending_mark->name);

		mode->pending_mark = NULL;
	} else {
		if (view->mark != NULL) {
			hikari_mark_clear(view->mark);
		}

		hikari_indicator_update_mark(indicator, output, "");
	}

	hikari_server_enter_normal_mode(NULL);
}

static void
select_mark(struct hikari_keyboard *keyboard, uint32_t keycode)
{
	uint32_t codepoint = hikari_keyboard_get_codepoint(keyboard, keycode);

	struct hikari_mark *mark;
	if (hikari_mark_get(codepoint, &mark)) {
		update_state(mark);
	}
}

static void
handle_keysym(
		struct hikari_keyboard *keyboard, uint32_t keycode, xkb_keysym_t sym)
{
	switch (sym) {
		case XKB_KEY_c:
		case XKB_KEY_d:
			if (!hikari_keyboard_check_modifier(keyboard, WLR_MODIFIER_CTRL)) {
				select_mark(keyboard, keycode);
				break;
			}
		case XKB_KEY_Escape:
			hikari_server_enter_normal_mode(NULL);
			break;

		case XKB_KEY_Return:
			confirm_mark_assign();
			break;

		case XKB_KEY_BackSpace:
			update_state(NULL);
			break;

		default:
			select_mark(keyboard, keycode);
			break;
	}
}

static void
assign_mark(
		struct wlr_event_keyboard_key *event, struct hikari_keyboard *keyboard)
{
	assert(hikari_server.workspace->focus_view != NULL);

	uint32_t keycode = event->keycode + 8;

	hikari_keyboard_for_keysym(keyboard, keycode, handle_keysym);
}

static void
key_handler(
		struct hikari_keyboard *keyboard, struct wlr_event_keyboard_key *event)
{
	if (event->state == WL_KEYBOARD_KEY_STATE_PRESSED) {
		assign_mark(event, keyboard);
	}
}

static void
modifiers_handler(struct hikari_keyboard *keyboard)
{}

static void
cancel(void)
{
	struct hikari_workspace *workspace = hikari_server.workspace;
	struct hikari_view *view = workspace->focus_view;
	struct hikari_mark_assign_mode *mode = get_mode();
	struct hikari_indicator *indicator = &hikari_server.indicator;

	clear_conflict(mode, mode->pending_mark);

	hikari_indicator_set_color_mark(
			indicator, hikari_configuration->indicator_selected);

	if (view != NULL) {
		struct hikari_output *output = view->output;

		if (view->mark != NULL) {
			hikari_indicator_update_mark(indicator, output, view->mark->name);
		} else {
			hikari_indicator_update_mark(indicator, output, "");
		}

		hikari_indicator_damage(indicator, view);
		hikari_view_damage_border(view);
	}

	hikari_server.mark_assign_mode.pending_mark = NULL;
}

static void
button_handler(
		struct hikari_cursor *cursor, struct wlr_event_pointer_button *event)
{}

static void
cursor_move(uint32_t time_msec)
{}

void
hikari_mark_assign_mode_init(struct hikari_mark_assign_mode *mark_assign_mode)
{
	mark_assign_mode->mode.key_handler = key_handler;
	mark_assign_mode->mode.button_handler = button_handler;
	mark_assign_mode->mode.modifiers_handler = modifiers_handler;
	mark_assign_mode->mode.render = hikari_renderer_mark_assign_mode;
	mark_assign_mode->mode.cancel = cancel;
	mark_assign_mode->mode.cursor_move = cursor_move;

	hikari_indicator_init(
			&mark_assign_mode->indicator, hikari_configuration->indicator_conflict);
}

void
hikari_mark_assign_mode_fini(struct hikari_mark_assign_mode *mark_assign_mode)
{
	hikari_indicator_fini(&mark_assign_mode->indicator);
}

void
hikari_mark_assign_mode_enter(struct hikari_view *view)
{
	assert(view == hikari_server.workspace->focus_view);

	struct wlr_box *geometry = hikari_view_border_geometry(view);
	struct hikari_output *output = view->output;
	struct hikari_mark *mark = view->mark;
	struct hikari_indicator *indicator = &hikari_server.indicator;

	hikari_server.mark_assign_mode.pending_mark = NULL;
	hikari_server.mode = (struct hikari_mode *)&hikari_server.mark_assign_mode;

	hikari_indicator_set_color_mark(
			indicator, hikari_configuration->indicator_insert);

	if (mark == NULL) {
		hikari_indicator_update_mark(indicator, output, " ");
	} else {
		hikari_indicator_update_mark(indicator, output, mark->name);
	}

	hikari_indicator_damage_mark(indicator, output, geometry);
}