/* See LICENSE file for copyright and license details. */
#include <assert.h>
#include <errno.h>
#include <getopt.h>
#include <linux/input-event-codes.h>
#include <poll.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/signalfd.h>
#include <sys/wait.h>
#include <unistd.h>
#include <wayland-client.h>

#include "drwl/drwl.h"
#include "drwl/bufpool.h"
#include "river-status-unstable-v1-protocol.h"
#include "river-control-unstable-v1-protocol.h"
#include "wlr-layer-shell-unstable-v1-protocol.h"

#define LENGTH(X)               (sizeof X / sizeof X[0])
#define END(A)                  ((A) + LENGTH(A))
#define TEXTW(bar, text)        (drwl_font_getwidth(bar->drw, text) + bar->lrpad)

enum { ClkTagBar, ClkLayout, ClkMode, ClkTitle, ClkStatus }; /* clicks */

enum { SchemeNorm, SchemeSel };

typedef struct {
	uint32_t wl_name;
	struct wl_output *wl_output;
	struct wl_surface *surface;
	struct wl_callback *frame_callback;
	bool configured;

	Drwl *drw;
	BufPool pool;
	uint32_t width, height, scale;
	int lrpad;

	struct zwlr_layer_surface_v1 *layer_surface;
	struct zriver_output_status_v1 *output_status;
	uint32_t mtags, ctags, urg;
	char *layout, *title;

	struct wl_list link;
} Bar;

typedef struct {
	unsigned int ui;
	char *s;
	const void *v;
} Arg;

typedef struct {
	unsigned int click;
	unsigned int button;
	void (*func)(const Arg *arg);
	const Arg arg;
} Button;

static void command(const Arg *arg);
static void spawn(const Arg *arg);

#include "config.h"

static struct wl_display *display;
static struct wl_registry *registry;
static struct wl_compositor *compositor;
static struct wl_shm *shm;
static struct zwlr_layer_shell_v1 *layer_shell;
static struct wl_seat *seat;
static struct zriver_control_v1 *control;
static struct zriver_status_manager_v1 *status_manager;

static struct zriver_seat_status_v1 *seat_status;
static struct wl_list bars;
static Bar *selbar;
static char stext[256];
static char river_version[22];
static char *mode;

static struct {
	struct wl_pointer *pointer;
	double x, y;
	uint32_t button;
} pointer;

static int signal_fd = -1;

static void
noop()
{
	/*
	 * :3c
	 */
}

static void
die(const char *fmt, ...)
{
	va_list ap;

	va_start(ap, fmt);
	vfprintf(stderr, fmt, ap);
	va_end(ap);

	if (fmt[0] && fmt[strlen(fmt)-1] == ':') {
		fputc(' ', stderr);
		perror(NULL);
	} else {
		fputc('\n', stderr);
	}

	exit(1);
}

static void
river_version_setup(void)
{
	char *p;
	FILE *fp;

	if (!(fp = popen("river -version", "r")))
		die("popen:");

	p = fgets(river_version, sizeof(river_version) - 1, fp);
	if (pclose(fp) < 0)
		die("pclose:");
	if (!p)
		die("no river version read");

	river_version[strcspn(river_version, "\n")] = 0;
	memmove(p + 6, p, strlen(p) + 1);
	memcpy(p, "river ", 6);
}

static void
parse_color(uint32_t *dest, const char *src)
{
	int len;

	if (src[0] == '#')
		src++;
	len = strlen(src);
	if (len != 6 && len != 8)
		die("bad color: %s", src);

	*dest = strtoul(src, NULL, 16);
	if (len == 6)
		*dest = (*dest << 8) | 0xFF;
}

static void
command_handle_message(void *data,
		struct zriver_command_callback_v1 *zriver_command_callback_v1,
		const char *message)
{
	if (message && message[0] != '\0')
		fprintf(stderr, "command message: %s\n", message);
}

static struct zriver_command_callback_v1_listener callback_listener = {
	.success = command_handle_message,
	.failure = command_handle_message,
};

static void
command_run(void)
{
	struct zriver_command_callback_v1 *callback;
	callback = zriver_control_v1_run_command(control, seat);
	zriver_command_callback_v1_add_listener(callback, &callback_listener, NULL);
}

void
command(const Arg *arg)
{
	char argbuf[4];
	zriver_control_v1_add_argument(control, arg->s);
	if (arg->ui) {
		snprintf(argbuf, sizeof(argbuf), "%d", arg->ui);
		zriver_control_v1_add_argument(control, argbuf);
	}
	command_run();
}

void
spawn(const Arg *arg)
{
	char *const *i;
	zriver_control_v1_add_argument(control, "spawn");
	for (i = (char *const *)arg->v; *i != NULL; i++)
		zriver_control_v1_add_argument(control, *i);
	command_run();
}

static void
bar_load_fonts(Bar *bar)
{
	char fontattrs[12];

	drwl_font_destroy(bar->drw->font);
	snprintf(fontattrs, sizeof(fontattrs), "dpi=%d", 96 * bar->scale);
	if (!(drwl_font_create(bar->drw, LENGTH(fonts), fonts, fontattrs)))
		die("failed to load fonts");

	bar->lrpad = bar->drw->font->height;
  bar->height = user_bh ? user_bh : bar->drw->font->height + 2;
	if (bar->layer_surface) {
		zwlr_layer_surface_v1_set_size(bar->layer_surface, 0, bar->height / bar->scale);
		zwlr_layer_surface_v1_set_exclusive_zone(bar->layer_surface, bar->height / bar->scale);
	}
}

static void
bar_draw(Bar *bar)
{
	int x = 0, w, tw = 0;
	int boxs = bar->drw->font->height / 9;
	int boxw = bar->drw->font->height / 6 + 2;
	unsigned int i;
	DrwBuf *buf;

	if (!bar->configured)
		return;

	errno = 0;
	if (!(buf = bufpool_getbuf(&bar->pool, shm, bar->width, bar->height)))
		die(errno ? "bufpool_getbuf:" : "no buffer available");
	drwl_setimage(bar->drw, buf->image);

	/* draw status first so it can be overdrawn by tags later */
	if (bar == selbar) { /* status is only drawn on selected monitor */
		drwl_setscheme(bar->drw, colors[SchemeNorm]);
		tw = TEXTW(bar, stext);
		drwl_text(bar->drw, bar->width - tw, 0, tw, bar->height, bar->lrpad / 2, stext, 0);
	}

	for (i = 0; i < LENGTH(tags); i++) {
		w = TEXTW(bar, tags[i]);
		drwl_setscheme(bar->drw, colors[bar->mtags & 1 << i ? SchemeSel : SchemeNorm]);
		drwl_text(bar->drw, x, 0, w, bar->height, bar->lrpad / 2, tags[i], bar->urg & 1 << i);
		if (bar->ctags & 1 << i)
			drwl_rect(bar->drw, x + boxs, boxs, boxw, boxw,
				bar->mtags & 1 << i && bar == selbar,
				bar->urg & 1 << i);
		x += w;
	}

	if (strcmp(mode, "normal")) {
		w = TEXTW(bar, mode);
		drwl_setscheme(bar->drw, colors[SchemeSel]);
		x = drwl_text(bar->drw, x, 0, w, bar->height, bar->lrpad / 2, mode, 0);
	}

	bar->layout = bar->layout ? bar->layout : strdup(layouts[LENGTH(layouts)-1][1]);
	w = TEXTW(bar, bar->layout);
	drwl_setscheme(bar->drw, colors[SchemeNorm]);
	x = drwl_text(bar->drw, x, 0, w, bar->height, bar->lrpad / 2, bar->layout, 0);

	if ((w = bar->width - tw - x) > bar->height) {
		drwl_setscheme(bar->drw, colors[bar == selbar && bar->title ? SchemeSel : SchemeNorm]);
		if (bar->title)
			drwl_text(bar->drw, x, 0, w, bar->height, bar->lrpad / 2, bar->title, 0);
		else
			drwl_rect(bar->drw, x, 0, w, bar->height, 1, 1);
	}

	drwl_setimage(bar->drw, NULL);
	wl_surface_set_buffer_scale(bar->surface, bar->scale);
	wl_surface_attach(bar->surface, buf->wl_buf, 0, 0);
	wl_surface_damage_buffer(bar->surface, 0, 0, bar->width, bar->height);
	wl_surface_commit(bar->surface);
}


static void
frame_callback_handle_done(void *data, struct wl_callback *callback,
		uint32_t time)
{
	Bar *bar = data;
	wl_callback_destroy(bar->frame_callback);
	bar->frame_callback = NULL;
	bar_draw(bar);
}

static const struct wl_callback_listener frame_callback_listener = {
	.done = frame_callback_handle_done,
};

static void
bar_frame(Bar *bar)
{
	/* note: river sends focused_view too early, before bar surface init */
	if (bar->frame_callback || !bar->configured)
		return;
	bar->frame_callback = wl_surface_frame(bar->surface);
	wl_callback_add_listener(bar->frame_callback, &frame_callback_listener, bar);
	wl_surface_commit(bar->surface);
}

static void
bars_draw(void)
{
	Bar *bar;

	if (stext[0] == '\0')
		strcpy(stext, river_version);

	wl_list_for_each(bar, &bars, link)
		if (bar->drw) /* called before initial bars setup */
			bar_draw(bar);
}

static void
output_status_handle_focused_tags(void *data,
		struct zriver_output_status_v1 *output_status, uint32_t tags)
{
	Bar *bar = data;
	bar->mtags = tags;
	bar_frame(bar);
}

static void
output_status_handle_urgent_tags(void *data,
		struct zriver_output_status_v1 *output_status, uint32_t tags)
{
	Bar *bar = data;
	bar->urg = tags;
	bar_frame(bar);
}

static void
output_status_handle_view_tags(void *data,
		struct zriver_output_status_v1 *output_status, struct wl_array *wl_array)
{
	uint32_t *vt;
	Bar *bar = data;
	bar->ctags = 0;
	wl_array_for_each(vt, wl_array)
		bar->ctags |= *vt;
	bar_frame(bar);
}

static void
output_status_handle_layout_name(void *data,
		struct zriver_output_status_v1 *output_status, const char *name)
{
	int i;
	Bar *bar = data;
	if (bar->layout)
		free(bar->layout);
	for (i = 0; i < LENGTH(layouts); i++)
		if (layouts[i][0] && !strcmp(name, layouts[i][0]))
			name = layouts[i][1];
	bar->layout = strdup(name);
	bar_frame(bar);
}

static void
output_status_handle_layout_name_clear(void *data,
		struct zriver_output_status_v1 *output_status)
{
	Bar *bar = data;
	if (bar->layout)
		free(bar->layout);
	bar->layout = NULL;
	bar_frame(bar);
}

static const struct zriver_output_status_v1_listener output_status_listener = {
	.focused_tags = output_status_handle_focused_tags,
	.urgent_tags = output_status_handle_urgent_tags,
	.view_tags = output_status_handle_view_tags,
	.layout_name = output_status_handle_layout_name,
	.layout_name_clear = output_status_handle_layout_name_clear
};

static void
bar_hide(Bar *bar)
{
	zwlr_layer_surface_v1_destroy(bar->layer_surface);
	wl_surface_destroy(bar->surface);
	bar->configured = false;
}

static void
bar_destroy(Bar *bar)
{
	if (bar == selbar)
		selbar = NULL;

	wl_list_remove(&bar->link);
	bufpool_cleanup(&bar->pool);
	if (bar->layout)
		free(bar->layout);
	if (bar->title)
		free(bar->title);
	drwl_setimage(bar->drw, NULL);
	drwl_destroy(bar->drw);
	zriver_output_status_v1_destroy(bar->output_status);
	bar_hide(bar);
	wl_output_destroy(bar->wl_output);
	free(bar);
}

static void
layer_surface_configure(void *data, struct zwlr_layer_surface_v1 *surface,
                        uint32_t serial, uint32_t w, uint32_t h)
{
	Bar *bar = data;

	if (bar->configured
		&& (bar->width / bar->scale == w && bar->height / bar->scale == h))
		return;
	bar->width = w * bar->scale;
	bar->height = h * bar->scale;
	bar->configured = true;
	zwlr_layer_surface_v1_ack_configure(bar->layer_surface, serial);
	bar_draw(bar);
}

static void
layer_surface_closed(void *data, struct zwlr_layer_surface_v1 *layer_surface)
{
	Bar *bar = data;
	bar_destroy(bar);
}

static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
    .configure = layer_surface_configure,
    .closed = layer_surface_closed,
};

static void
bar_show(Bar *bar)
{
	bar->surface = wl_compositor_create_surface(compositor);

	bar->layer_surface = zwlr_layer_shell_v1_get_layer_surface(
		layer_shell, bar->surface, bar->wl_output, ZWLR_LAYER_SHELL_V1_LAYER_BOTTOM, "bar");
	zwlr_layer_surface_v1_add_listener(
		bar->layer_surface, &layer_surface_listener, bar);
	zwlr_layer_surface_v1_set_size(bar->layer_surface, 0, bar->height / bar->scale);
    zwlr_layer_surface_v1_set_exclusive_zone(bar->layer_surface, bar->height / bar->scale);
	zwlr_layer_surface_v1_set_anchor(bar->layer_surface,
		(topbar ? ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP : ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM)
		| ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT | ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT);
	wl_surface_commit(bar->surface);
}

static void
bars_toggle_selected(void)
{
	Bar *bar;
	wl_list_for_each(bar, &bars, link)
		if (bar == selbar && bar->configured)
			bar_hide(bar);
		else if (bar == selbar)
			bar_show(bar);
}

static void
output_handle_done(void *data, struct wl_output *wl_output)
{
	Bar *bar = data;

	if (bar->drw)
		return;
	if (!(bar->drw = drwl_create()))
		die("failed to create drwl context");
	bar_load_fonts(bar);
	bar->output_status = zriver_status_manager_v1_get_river_output_status(
		status_manager, bar->wl_output);
	zriver_output_status_v1_add_listener(bar->output_status,
		&output_status_listener, bar);

	if (showbar)
		bar_show(bar);
		
}

static void
output_handle_scale(void *data, struct wl_output *wl_output, int32_t factor)
{
	Bar *bar = data;
	bar->scale = factor;
	if (bar->drw)
		bar_load_fonts(bar);
}

static const struct wl_output_listener output_listener = {
	.geometry = noop,
	.mode = noop,
	.done = output_handle_done,
	.scale = output_handle_scale,
	.name = noop,
	.description = noop,
};

static void
seat_status_handle_focused_output(void *data,
		struct zriver_seat_status_v1 *seat_status, struct wl_output *wl_output)
{
	Bar *bar;

	wl_list_for_each(bar, &bars, link) {
		if (bar->wl_output != wl_output)
			continue;

		bar_frame((selbar = bar));
		break;
	}
}

static void
seat_status_handle_unfocused_output(void *data, struct zriver_seat_status_v1 *seat_status,
				   struct wl_output *wl_output)
{
	Bar *oldbar;
	if (!selbar)
		return;
	oldbar = selbar;
	selbar = NULL;
	bar_frame(oldbar);
}

static void
seat_status_handle_focused_view(void *data,
		struct zriver_seat_status_v1 *seat_status, const char *title)
{
	if (!selbar)
		return;
	if (selbar->title)
		free(selbar->title);
	selbar->title = NULL;
	if (title[0] != '\0')
		selbar->title = strdup(title);
	bar_frame(selbar);
}

static void
seat_status_handle_mode(void *data,
		struct zriver_seat_status_v1 *seat_status, const char *name)
{
	if (mode)
		free(mode);
	mode = strdup(name);
	bars_draw();
}

static const struct zriver_seat_status_v1_listener seat_status_listener = {
	.focused_output = seat_status_handle_focused_output,
	.unfocused_output = seat_status_handle_unfocused_output,
	.focused_view = seat_status_handle_focused_view,
	.mode = seat_status_handle_mode,
};

static void
pointer_motion(void *data, struct wl_pointer *wl_pointer, uint32_t time,
	       wl_fixed_t surface_x, wl_fixed_t surface_y)
{
	pointer.button = 0;
	if (!selbar)
		return;
	pointer.x = wl_fixed_to_int(surface_x) * selbar->scale;
	pointer.y = wl_fixed_to_int(surface_y) * selbar->scale;
}

static void
pointer_handle_frame(void *data, struct wl_pointer *wl_pointer)
{
	int lw, mw = 0;
	Arg arg = {0};
	unsigned int i = 0, x = 0;
	unsigned int tag, click;

	if (!pointer.button)
		return;

	lw = TEXTW(selbar, selbar->layout);

	do
		x += TEXTW(selbar, tags[i]);
	while (pointer.x >= x && ++i < LENGTH(tags));

	if (i < LENGTH(tags)) {
		click = ClkTagBar;
		tag = 1 << i;
	} else if (pointer.x > x + lw && pointer.x < x + lw + mw)
		click = ClkMode;
	else if (pointer.x < x + lw)
		click = ClkLayout;
	else if (pointer.x > selbar->width - (int)TEXTW(selbar, stext))
		click = ClkStatus;
	else
		click = ClkTitle;

	for (i = 0; i < LENGTH(buttons); i++)
		if (buttons[i].click == click && buttons[i].func && buttons[i].button == pointer.button) {
			arg = buttons[i].arg;
			if (click == ClkTagBar && !arg.ui)
				arg.ui = tag;
			buttons[i].func(&arg);
        }
}

static void
pointer_handle_button(void *data, struct wl_pointer *wl_pointer, uint32_t serial,
		uint32_t time, uint32_t button, uint32_t state)
{
	pointer.button = state == WL_POINTER_BUTTON_STATE_PRESSED ? button : 0;
}

static const struct wl_pointer_listener pointer_listener = {
	.enter = noop,
	.leave = noop,
	.motion = pointer_motion,
	.axis = noop,
	.frame = pointer_handle_frame,
	.axis_discrete = noop,
	.axis_source = noop,
	.axis_stop = noop,
	.button = pointer_handle_button,
};

static void
seat_handle_capabilities(void *data, struct wl_seat *wl_seat,
		enum wl_seat_capability caps)
{
	if (!(caps & WL_SEAT_CAPABILITY_POINTER))
		return;

	pointer.pointer = wl_seat_get_pointer(seat);
	wl_pointer_add_listener(pointer.pointer, &pointer_listener, NULL);
}


static const struct wl_seat_listener seat_listener = {
	.capabilities = seat_handle_capabilities,
	.name = noop,
};

static void
registry_handle_global(void *data, struct wl_registry *wl_registry,
		uint32_t name, const char *interface, uint32_t version)
{
	if (!strcmp(interface, wl_compositor_interface.name))
		compositor = wl_registry_bind(registry, name, &wl_compositor_interface, 4);
	else if (!strcmp(interface, wl_shm_interface.name))
		shm = wl_registry_bind(registry, name, &wl_shm_interface, 1);
	else if (!strcmp(interface, zwlr_layer_shell_v1_interface.name))
		layer_shell = wl_registry_bind(wl_registry, name, &zwlr_layer_shell_v1_interface, 3);
	else if (!strcmp(interface, wl_seat_interface.name)) {
		seat = wl_registry_bind(registry, name, &wl_seat_interface, 5);
		wl_seat_add_listener(seat, &seat_listener, NULL);
	} else if (!strcmp(interface, zriver_control_v1_interface.name))
		control = wl_registry_bind(registry, name, &zriver_control_v1_interface, 1);
	else if (!strcmp(interface, zriver_status_manager_v1_interface.name))
		status_manager = wl_registry_bind(registry, name, &zriver_status_manager_v1_interface, 4);
	else if (!strcmp(interface, wl_output_interface.name)) {
		Bar *bar = calloc(1, sizeof(Bar));
		bar->scale = 1;
		bar->wl_name = name;
		bar->wl_output = wl_registry_bind(registry, name, &wl_output_interface, 2);
		wl_output_add_listener(bar->wl_output, &output_listener, bar);
		wl_list_insert(&bars, &bar->link);

		/* https://codeberg.org/river/river/issues/1160 */
		if (wl_list_length(&bars) == 1)
			selbar = bar;
	}
}

static void
registry_handle_global_remove(void *data, struct wl_registry *registry, uint32_t name)
{
	Bar *bar, *bar_tmp;
	wl_list_for_each_safe(bar, bar_tmp, &bars, link)
		if (bar->wl_name == name) {
			bar_destroy(bar);
			return;
		}
}

static const struct wl_registry_listener registry_listener = {
	.global = registry_handle_global,
	.global_remove = registry_handle_global_remove,
};

static void
readstdin(void)
{
	ssize_t n;

	n = read(STDIN_FILENO, stext, sizeof(stext) - 1);
	if (n < 0)
		die("read:");
	stext[n] = '\0';
	stext[strcspn(stext, "\n")] = '\0';

	bars_draw();
}

static void
setup(void)
{
	sigset_t mask;

	if (!(display = wl_display_connect(NULL)))
		die("failed to connect to wayland");

	wl_list_init(&bars);

	registry = wl_display_get_registry(display);
	wl_registry_add_listener(registry, &registry_listener, NULL);
	wl_display_roundtrip(display);

	if (!compositor || !shm || !layer_shell || !status_manager || !control || !seat)
		die("unsupported compositor");

	river_version_setup();

	sigemptyset(&mask);
	sigaddset(&mask, SIGUSR1);
	sigaddset(&mask, SIGUSR2);
	sigaddset(&mask, SIGTERM);
	sigaddset(&mask, SIGINT);

	if (sigprocmask(SIG_BLOCK, &mask, NULL) < 0)
		die("sigprocmask:");
	if ((signal_fd = signalfd(-1, &mask, SFD_NONBLOCK)) < 0)
		die("signalfd:");

	drwl_init();

	seat_status = zriver_status_manager_v1_get_river_seat_status(
		status_manager, seat);
	zriver_seat_status_v1_add_listener(seat_status,
		&seat_status_listener, NULL);
}

static void
run(void)
{
	struct pollfd pfds[] = {
		{ .fd = wl_display_get_fd(display), .events = POLLIN },
		{ .fd = STDIN_FILENO,               .events = POLLIN },
		{ .fd = signal_fd,                  .events = POLLIN },
	};

	for (;;) {
		wl_display_flush(display);

		while (poll(pfds, 3, -1) < 0)
			die("poll:");

		if (pfds[0].revents & POLLIN)
			if (wl_display_dispatch(display) < 0)
				die("display dispatch failed");

		if (pfds[1].revents & POLLHUP) { /* stdin closed */
			pfds[1].fd = -1;
			stext[0] = '\0';
			bars_draw();
		}
		if (pfds[1].revents & POLLIN)
			readstdin();

		if (pfds[2].revents & POLLIN) {
			struct signalfd_siginfo si;
			ssize_t n = read(signal_fd, &si, sizeof(si));
			if (n != sizeof(si))
				die("signalfd/read:");
			if (si.ssi_signo == SIGUSR1)
				bars_toggle_selected();
			else if (si.ssi_signo == SIGTERM ||
			         si.ssi_signo == SIGINT)
				break;
		}
	}
}

static void
cleanup(void)
{
	Bar *bar, *bar_tmp;

	wl_pointer_destroy(pointer.pointer);
	if (mode)
		free(mode);
	wl_list_for_each_safe(bar, bar_tmp, &bars, link)
		bar_destroy(bar);
	zriver_seat_status_v1_destroy(seat_status);
	drwl_fini();
	zriver_status_manager_v1_destroy(status_manager);
	zriver_control_v1_destroy(control);
	wl_seat_destroy(seat);
	zwlr_layer_shell_v1_destroy(layer_shell);
	wl_shm_destroy(shm);
	wl_compositor_destroy(compositor);
	wl_registry_destroy(registry);
	wl_display_disconnect(display);
	close(signal_fd); /* ignore error */
}

static void
usage(void)
{
	die("usage: dam [-st] [-f font] [-nb color] [-nf color] [-sb color] [-sf color]");
}

int
main(int argc, char *argv[])
{
	int i;

	for (i = 1; i < argc; i++) {
		/* these options take no arguments */
		if (!strcmp(argv[i], "-v")) {
			puts("dam-"VERSION);
			return EXIT_SUCCESS;
		} else if (!strcmp(argv[i], "-s"))
			showbar = !showbar;
		else if (!strcmp(argv[i], "-t"))
			topbar = !showbar;
		else if (i + 1 == argc)
			usage();
		else if (!strcmp(argv[i], "-f"))
			fonts[0] = argv[++i];
		else if (!strcmp(argv[i], "-nb"))
			parse_color(&colors[SchemeNorm][ColBg], argv[++i]);
		else if (!strcmp(argv[i], "-nf"))
			parse_color(&colors[SchemeNorm][ColFg], argv[++i]);
		else if (!strcmp(argv[i], "-sb"))
			parse_color(&colors[SchemeSel][ColBg], argv[++i]);
		else if (!strcmp(argv[i], "-sf"))
			parse_color(&colors[SchemeSel][ColFg], argv[++i]);
		else
			usage();
	}

	setup();
	run();
	cleanup();

	return EXIT_SUCCESS;
}