1 /*
2  * Copyright (c) 2018, 2019, 2020, 2022 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <ctype.h> /* isalnum */
18 #include <err.h>
19 #include <errno.h>
20 #include <limits.h>
21 #include <locale.h> /* setlocale */
22 #include <stdint.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h> /* strdup, strlen */
26 #include <sysexits.h>
27 #include <unistd.h>
28 
29 #include <X11/Xcms.h>
30 #include <X11/Xlib.h>
31 #include <X11/Xresource.h>
32 #include <X11/Xutil.h>
33 #include <X11/keysym.h>
34 #include <X11/Xft/Xft.h>
35 
36 #include <X11/extensions/Xinerama.h>
37 
38 #define RESNAME "MyMenu"
39 #define RESCLASS "mymenu"
40 
41 #define SYM_BUF_SIZE 4
42 
43 #define DEFFONT "monospace"
44 
45 #define ARGS "Aahmve:p:P:l:f:W:H:x:y:b:B:t:T:c:C:s:S:d:G:g:I:i:J:j:"
46 
47 #define MIN(a, b) ((a) < (b) ? (a) : (b))
48 #define MAX(a, b) ((a) > (b) ? (a) : (b))
49 
50 #define EXPANDBITS(x) (((x & 0xf0) * 0x100) | (x & 0x0f) * 0x10)
51 
52 #define INNER_HEIGHT(r) (r->height - r->borders[0] - r->borders[2])
53 #define INNER_WIDTH(r) (r->width - r->borders[1] - r->borders[3])
54 
55 /* The states of the event loop */
56 enum state { LOOPING, OK_LOOP, OK, ERR };
57 
58 /*
59  * For the drawing-related function. The text to be rendere could be
60  * the prompt, a completion or a highlighted completion
61  */
62 enum obj_type { PROMPT, COMPL, COMPL_HIGH };
63 
64 /* These are the possible action to be performed after user input. */
65 enum action {
66 	NO_OP,
67 	EXIT,
68 	CONFIRM,
69 	CONFIRM_CONTINUE,
70 	NEXT_COMPL,
71 	PREV_COMPL,
72 	DEL_CHAR,
73 	DEL_WORD,
74 	DEL_LINE,
75 	ADD_CHAR,
76 	TOGGLE_FIRST_SELECTED,
77 	SCROLL_DOWN,
78 	SCROLL_UP,
79 };
80 
81 /* A big set of values that needs to be carried around for drawing. A
82  * big struct to rule them all */
83 struct rendering {
84 	Display *d; /* Connection to xorg */
85 	Window w;
86 	XIM xim;
87 	int width;
88 	int height;
89 	int p_padding[4];
90 	int c_padding[4];
91 	int ch_padding[4];
92 	int x_zero; /* the "zero" on the x axis (may not be exactly 0 'cause
93 		       the borders) */
94 	int y_zero; /* like x_zero but for the y axis */
95 
96 	int offset; /* scroll offset */
97 
98 	short free_text;
99 	short first_selected;
100 	short multiple_select;
101 
102 	/* four border width */
103 	int borders[4];
104 	int p_borders[4];
105 	int c_borders[4];
106 	int ch_borders[4];
107 
108 	short horizontal_layout;
109 
110 	/* prompt */
111 	char *ps1;
112 	int ps1len;
113 	int ps1w; /* ps1 width */
114 	int ps1h; /* ps1 height */
115 
116 	int text_height; /* cache for the vertical layout */
117 
118 	XIC xic;
119 
120 	/* colors */
121 	GC fgs[4];
122 	GC bgs[4];
123 	GC borders_bg[4];
124 	GC p_borders_bg[4];
125 	GC c_borders_bg[4];
126 	GC ch_borders_bg[4];
127 	XftFont *font;
128 	XftDraw *xftdraw;
129 	XftColor xft_colors[3];
130 };
131 
132 struct completion {
133 	char *completion;
134 	char *rcompletion;
135 
136 	/*
137 	 * The X (or Y, depending on the layour) at which the item is
138 	 * rendered
139 	 */
140 	int offset;
141 };
142 
143 /* Wrap the linked list of completions */
144 struct completions {
145 	struct completion *completions;
146 	ssize_t selected;
147 	size_t length;
148 };
149 
150 /* idea stolen from lemonbar;  ty lemonboy */
151 typedef union {
152 	struct {
153 		uint8_t b;
154 		uint8_t g;
155 		uint8_t r;
156 		uint8_t a;
157 	} rgba;
158 	uint32_t v;
159 } rgba_t;
160 
161 /* Return a newly allocated (and empty) completion list */
162 struct completions *
163 compls_new(size_t length)
164 {
165 	struct completions *cs = malloc(sizeof(struct completions));
166 
167 	if (cs == NULL)
168 		return cs;
169 
170 	cs->completions = calloc(length, sizeof(struct completion));
171 	if (cs->completions == NULL) {
172 		free(cs);
173 		return NULL;
174 	}
175 
176 	cs->selected = -1;
177 	cs->length = length;
178 	return cs;
179 }
180 
181 /* Delete the wrapper and the whole list */
182 void
183 compls_delete(struct completions *cs)
184 {
185 	if (cs == NULL)
186 		return;
187 
188 	free(cs->completions);
189 	free(cs);
190 }
191 
192 /*
193  * Create a completion list from a text and the list of possible
194  * completions (null terminated). Expects a non-null `cs'. `lines' and
195  * `vlines' should have the same length OR `vlines' is NULL.
196  */
197 void
198 filter(struct completions *cs, char *text, char **lines, char **vlines)
199 {
200 	size_t index = 0;
201 	size_t matching = 0;
202 	char *l;
203 
204 	if (vlines == NULL)
205 		vlines = lines;
206 
207 	while (1) {
208 		if (lines[index] == NULL)
209 			break;
210 
211 		l = vlines[index] != NULL ? vlines[index] : lines[index];
212 
213 		if (strcasestr(l, text) != NULL) {
214 			struct completion *c = &cs->completions[matching];
215 			c->completion = l;
216 			c->rcompletion = lines[index];
217 			matching++;
218 		}
219 
220 		index++;
221 	}
222 	cs->length = matching;
223 	cs->selected = -1;
224 }
225 
226 /* Update the given completion */
227 void
228 update_completions(struct completions *cs, char *text, char **lines,
229     char **vlines, short first_selected)
230 {
231 	filter(cs, text, lines, vlines);
232 	if (first_selected && cs->length > 0)
233 		cs->selected = 0;
234 }
235 
236 /*
237  * Select the next or previous selection and update some state. `text'
238  * will be updated with the text of the completion and `textlen' with
239  * the new length. If the memory cannot be allocated `status' will be
240  * set to `ERR'.
241  */
242 void
243 complete(struct completions *cs, short first_selected, short p,
244     char **text, int *textlen, enum state *status)
245 {
246 	struct completion *n;
247 	int index;
248 
249 	if (cs == NULL || cs->length == 0)
250 		return;
251 
252 	/*
253 	 * If the first is always selected and the first entry is
254 	 * different from the text, expand the text and return
255 	 */
256 	if (first_selected &&
257 	    cs->selected == 0 &&
258 	    strcmp(cs->completions->completion, *text) != 0 &&
259 	    !p) {
260 		free(*text);
261 		*text = strdup(cs->completions->completion);
262 		if (text == NULL) {
263 			*status = ERR;
264 			return;
265 		}
266 		*textlen = strlen(*text);
267 		return;
268 	}
269 
270 	index = cs->selected;
271 
272 	if (index == -1 && p)
273 		index = 0;
274 	index = cs->selected = (cs->length + (p ? index - 1 : index + 1))
275 		% cs->length;
276 
277 	n = &cs->completions[cs->selected];
278 
279 	free(*text);
280 	*text = strdup(n->completion);
281 	if (text == NULL) {
282 		fprintf(stderr, "Memory allocation error!\n");
283 		*status = ERR;
284 		return;
285 	}
286 	*textlen = strlen(*text);
287 }
288 
289 /* Push the character c at the end of the string pointed by p */
290 int
291 pushc(char **p, int maxlen, char c)
292 {
293 	int len;
294 
295 	len = strnlen(*p, maxlen);
296 	if (!(len < maxlen - 2)) {
297 		char *newptr;
298 
299 		maxlen += maxlen >> 1;
300 		newptr = realloc(*p, maxlen);
301 		if (newptr == NULL) /* bad */
302 			return -1;
303 		*p = newptr;
304 	}
305 
306 	(*p)[len] = c;
307 	(*p)[len + 1] = '\0';
308 	return maxlen;
309 }
310 
311 /*
312  * Remove the last rune from the *UTF-8* string! This is different
313  * from just setting the last byte to 0 (in some cases ofc). Return a
314  * pointer (e) to the last nonzero char. If e < p then p is empty!
315  */
316 char *
317 popc(char *p)
318 {
319 	int len = strlen(p);
320 	char *e;
321 
322 	if (len == 0)
323 		return p;
324 
325 	e = p + len - 1;
326 
327 	do {
328 		char c = *e;
329 
330 		*e = '\0';
331 		e -= 1;
332 
333 		/*
334 		 * If c is a starting byte (11......) or is under
335 		 * U+007F we're done.
336 		 */
337 		if (((c & 0x80) && (c & 0x40)) || !(c & 0x80))
338 			break;
339 	} while (e >= p);
340 
341 	return e;
342 }
343 
344 /* Remove the last word plus trailing white spaces from the given string */
345 void
346 popw(char *w)
347 {
348 	int len;
349 	short in_word = 1;
350 
351 	if ((len = strlen(w)) == 0)
352 		return;
353 
354 	while (1) {
355 		char *e = popc(w);
356 
357 		if (e < w)
358 			return;
359 
360 		if (in_word && isspace(*e))
361 			in_word = 0;
362 
363 		if (!in_word && !isspace(*e))
364 			return;
365 	}
366 }
367 
368 /*
369  * If the string is surrounded by quates (`"') remove them and replace
370  * every `\"' in the string with a single double-quote.
371  */
372 char *
373 normalize_str(const char *str)
374 {
375 	int len, p;
376 	char *s;
377 
378 	if ((len = strlen(str)) == 0)
379 		return NULL;
380 
381 	if ((s = calloc(len, sizeof(char))) == NULL)
382 		err(1, "calloc");
383 	p = 0;
384 
385 	while (*str) {
386 		char c = *str;
387 
388 		if (*str == '\\') {
389 			if (*(str + 1)) {
390 				s[p] = *(str + 1);
391 				p++;
392 				str += 2; /* skip this and the next char */
393 				continue;
394 			} else
395 				break;
396 		}
397 		if (c == '"') {
398 			str++; /* skip only this char */
399 			continue;
400 		}
401 		s[p] = c;
402 		p++;
403 		str++;
404 	}
405 
406 	return s;
407 }
408 
409 char **
410 readlines(size_t *lineslen)
411 {
412 	size_t len = 0, cap = 0;
413 	size_t linesize = 0;
414 	ssize_t linelen;
415 	char *line = NULL, **lines = NULL;
416 
417 	while ((linelen = getline(&line, &linesize, stdin)) != -1) {
418 		if (linelen != 0 && line[linelen-1] == '\n')
419 			line[linelen-1] = '\0';
420 
421 		if (len == cap) {
422 			size_t newcap;
423 			void *t;
424 
425 			newcap = MAX(cap * 1.5, 32);
426 			t = recallocarray(lines, cap, newcap, sizeof(char *));
427 			if (t == NULL)
428 				err(1, "recallocarray");
429 			cap = newcap;
430 			lines = t;
431 		}
432 
433 		if ((lines[len++] = strdup(line)) == NULL)
434 			err(1, "strdup");
435 	}
436 
437 	if (ferror(stdin))
438 		err(1, "getline");
439 	free(line);
440 
441 	*lineslen = len;
442 	return lines;
443 }
444 
445 /*
446  * Compute the dimensions of the string str once rendered.
447  * It'll return the width and set ret_width and ret_height if not NULL
448  */
449 int
450 text_extents(char *str, int len, struct rendering *r, int *ret_width,
451     int *ret_height)
452 {
453 	int height, width;
454 	XGlyphInfo gi;
455 	XftTextExtentsUtf8(r->d, r->font, str, len, &gi);
456 	height = r->font->ascent - r->font->descent;
457 	width = gi.width - gi.x;
458 
459 	if (ret_width != NULL)
460 		*ret_width = width;
461 	if (ret_height != NULL)
462 		*ret_height = height;
463 	return width;
464 }
465 
466 void
467 draw_string(char *str, int len, int x, int y, struct rendering *r,
468     enum obj_type tt)
469 {
470 	XftColor xftcolor;
471 	if (tt == PROMPT)
472 		xftcolor = r->xft_colors[0];
473 	if (tt == COMPL)
474 		xftcolor = r->xft_colors[1];
475 	if (tt == COMPL_HIGH)
476 		xftcolor = r->xft_colors[2];
477 
478 	XftDrawStringUtf8(r->xftdraw, &xftcolor, r->font, x, y, str, len);
479 }
480 
481 /* Duplicate the string and substitute every space with a 'n` */
482 char *
483 strdupn(char *str)
484 {
485 	char *t, *dup;
486 
487 	if (str == NULL || *str == '\0')
488 		return NULL;
489 
490 	if ((dup = strdup(str)) == NULL)
491 		return NULL;
492 
493 	for (t = dup; *t; ++t) {
494 		if (*t == ' ')
495 			*t = 'n';
496 	}
497 
498 	return dup;
499 }
500 
501 int
502 draw_v_box(struct rendering *r, int y, char *prefix, int prefix_width,
503     enum obj_type t, char *text)
504 {
505 	GC *border_color, bg;
506 	int *padding, *borders;
507 	int ret = 0, inner_width, inner_height, x;
508 
509 	switch (t) {
510 	case PROMPT:
511 		border_color = r->p_borders_bg;
512 		padding = r->p_padding;
513 		borders = r->p_borders;
514 		bg = r->bgs[0];
515 		break;
516 	case COMPL:
517 		border_color = r->c_borders_bg;
518 		padding = r->c_padding;
519 		borders = r->c_borders;
520 		bg = r->bgs[1];
521 		break;
522 	case COMPL_HIGH:
523 		border_color = r->ch_borders_bg;
524 		padding = r->ch_padding;
525 		borders = r->ch_borders;
526 		bg = r->bgs[2];
527 		break;
528 	}
529 
530 	ret = borders[0] + padding[0] + r->text_height + padding[2] + borders[2];
531 
532 	inner_width = INNER_WIDTH(r) - borders[1] - borders[3];
533 	inner_height = padding[0] + r->text_height + padding[2];
534 
535 	/* Border top */
536 	XFillRectangle(r->d, r->w, border_color[0], r->x_zero, y, r->width,
537 	    borders[0]);
538 
539 	/* Border right */
540 	XFillRectangle(r->d, r->w, border_color[1],
541 	    r->x_zero + INNER_WIDTH(r) - borders[1], y, borders[1], ret);
542 
543 	/* Border bottom */
544 	XFillRectangle(r->d, r->w, border_color[2], r->x_zero,
545 	    y + borders[0] + padding[0] + r->text_height + padding[2],
546 	    r->width, borders[2]);
547 
548 	/* Border left */
549 	XFillRectangle(r->d, r->w, border_color[3], r->x_zero, y, borders[3],
550 	    ret);
551 
552 	/* bg */
553 	x = r->x_zero + borders[3];
554 	y += borders[0];
555 	XFillRectangle(r->d, r->w, bg, x, y, inner_width, inner_height);
556 
557 	/* content */
558 	y += padding[0] + r->text_height;
559 	x += padding[3];
560 	if (prefix != NULL) {
561 		draw_string(prefix, strlen(prefix), x, y, r, t);
562 		x += prefix_width;
563 	}
564 	draw_string(text, strlen(text), x, y, r, t);
565 
566 	return ret;
567 }
568 
569 int
570 draw_h_box(struct rendering *r, int x, char *prefix, int prefix_width,
571     enum obj_type t, char *text)
572 {
573 	GC *border_color, bg;
574 	int *padding, *borders;
575 	int ret = 0, inner_width, inner_height, y, text_width;
576 
577 	switch (t) {
578 	case PROMPT:
579 		border_color = r->p_borders_bg;
580 		padding = r->p_padding;
581 		borders = r->p_borders;
582 		bg = r->bgs[0];
583 		break;
584 	case COMPL:
585 		border_color = r->c_borders_bg;
586 		padding = r->c_padding;
587 		borders = r->c_borders;
588 		bg = r->bgs[1];
589 		break;
590 	case COMPL_HIGH:
591 		border_color = r->ch_borders_bg;
592 		padding = r->ch_padding;
593 		borders = r->ch_borders;
594 		bg = r->bgs[2];
595 		break;
596 	}
597 
598 	if (padding[0] < 0 || padding[2] < 0) {
599 		padding[0] = INNER_HEIGHT(r) - borders[0] - borders[2]
600 			- r->text_height;
601 		padding[0] /= 2;
602 
603 		padding[2] = padding[0];
604 	}
605 
606 	/* If they are still lesser than 0, set 'em to 0 */
607 	if (padding[0] < 0 || padding[2] < 0)
608 		padding[0] = padding[2] = 0;
609 
610 	/* Get the text width */
611 	text_extents(text, strlen(text), r, &text_width, NULL);
612 	if (prefix != NULL)
613 		text_width += prefix_width;
614 
615 	ret = borders[3] + padding[3] + text_width + padding[1] + borders[1];
616 
617 	inner_width = padding[3] + text_width + padding[1];
618 	inner_height = INNER_HEIGHT(r) - borders[0] - borders[2];
619 
620 	/* Border top */
621 	XFillRectangle(r->d, r->w, border_color[0], x, r->y_zero, ret,
622 	    borders[0]);
623 
624 	/* Border right */
625 	XFillRectangle(r->d, r->w, border_color[1],
626 	    x + borders[3] + inner_width, r->y_zero, borders[1],
627 	    INNER_HEIGHT(r));
628 
629 	/* Border bottom */
630 	XFillRectangle(r->d, r->w, border_color[2], x,
631 	    r->y_zero + INNER_HEIGHT(r) - borders[2], ret,
632 	    borders[2]);
633 
634 	/* Border left */
635 	XFillRectangle(r->d, r->w, border_color[3], x, r->y_zero, borders[3],
636 	    INNER_HEIGHT(r));
637 
638 	/* bg */
639 	x += borders[3];
640 	y = r->y_zero + borders[0];
641 	XFillRectangle(r->d, r->w, bg, x, y, inner_width, inner_height);
642 
643 	/* content */
644 	y += padding[0] + r->text_height;
645 	x += padding[3];
646 	if (prefix != NULL) {
647 		draw_string(prefix, strlen(prefix), x, y, r, t);
648 		x += prefix_width;
649 	}
650 	draw_string(text, strlen(text), x, y, r, t);
651 
652 	return ret;
653 }
654 
655 /*
656  * ,-----------------------------------------------------------------,
657  * | 20 char text     | completion | completion | completion | compl |
658  *  `-----------------------------------------------------------------'
659  */
660 void
661 draw_horizontally(struct rendering *r, char *text, struct completions *cs)
662 {
663 	size_t i;
664 	int x = r->x_zero;
665 
666 	/* Draw the prompt */
667 	x += draw_h_box(r, x, r->ps1, r->ps1w, PROMPT, text);
668 
669 	for (i = r->offset; i < cs->length; ++i) {
670 		enum obj_type t;
671 
672 		if (cs->selected == (ssize_t)i)
673 			t = COMPL_HIGH;
674 		else
675 			t = COMPL;
676 
677 		cs->completions[i].offset = x;
678 
679 		x += draw_h_box(r, x, NULL, 0, t,
680 		    cs->completions[i].completion);
681 
682 		if (x > INNER_WIDTH(r))
683 			break;
684 	}
685 
686 	for (i += 1; i < cs->length; ++i)
687 		cs->completions[i].offset = -1;
688 }
689 
690 /*
691  * ,-----------------------------------------------------------------,
692  * |  prompt                                                         |
693  * |-----------------------------------------------------------------|
694  * |  completion                                                     |
695  * |-----------------------------------------------------------------|
696  * |  completion                                                     |
697  * `-----------------------------------------------------------------'
698  */
699 void
700 draw_vertically(struct rendering *r, char *text, struct completions *cs)
701 {
702 	size_t i;
703 	int y = r->y_zero;
704 
705 	y += draw_v_box(r, y, r->ps1, r->ps1w, PROMPT, text);
706 
707 	for (i = r->offset; i < cs->length; ++i) {
708 		enum obj_type t;
709 
710 		if (cs->selected == (ssize_t)i)
711 			t = COMPL_HIGH;
712 		else
713 			t = COMPL;
714 
715 		cs->completions[i].offset = y;
716 
717 		y += draw_v_box(r, y, NULL, 0, t,
718 		    cs->completions[i].completion);
719 
720 		if (y > INNER_HEIGHT(r))
721 			break;
722 	}
723 
724 	for (i += 1; i < cs->length; ++i)
725 		cs->completions[i].offset = -1;
726 }
727 
728 void
729 draw(struct rendering *r, char *text, struct completions *cs)
730 {
731 	/* Draw the background */
732 	XFillRectangle(r->d, r->w, r->bgs[1], r->x_zero, r->y_zero,
733 	    INNER_WIDTH(r), INNER_HEIGHT(r));
734 
735 	/* Draw the contents */
736 	if (r->horizontal_layout)
737 		draw_horizontally(r, text, cs);
738 	else
739 		draw_vertically(r, text, cs);
740 
741 	/* Draw the borders */
742 	if (r->borders[0] != 0)
743 		XFillRectangle(r->d, r->w, r->borders_bg[0], 0, 0, r->width,
744 		    r->borders[0]);
745 
746 	if (r->borders[1] != 0)
747 		XFillRectangle(r->d, r->w, r->borders_bg[1],
748 		    r->width - r->borders[1], 0, r->borders[1],
749 		    r->height);
750 
751 	if (r->borders[2] != 0)
752 		XFillRectangle(r->d, r->w, r->borders_bg[2], 0,
753 		    r->height - r->borders[2], r->width, r->borders[2]);
754 
755 	if (r->borders[3] != 0)
756 		XFillRectangle(r->d, r->w, r->borders_bg[3], 0, 0,
757 		    r->borders[3], r->height);
758 
759 	/* render! */
760 	XFlush(r->d);
761 }
762 
763 /* Set some WM stuff */
764 void
765 set_win_atoms_hints(Display *d, Window w, int width, int height)
766 {
767 	Atom type;
768 	XClassHint *class_hint;
769 	XSizeHints *size_hint;
770 
771 	type = XInternAtom(d, "_NET_WM_WINDOW_TYPE_DOCK", 0);
772 	XChangeProperty(d, w, XInternAtom(d, "_NET_WM_WINDOW_TYPE", 0),
773 	    XInternAtom(d, "ATOM", 0), 32, PropModeReplace,
774 	    (unsigned char *)&type, 1);
775 
776 	/* some window managers honor this properties */
777 	type = XInternAtom(d, "_NET_WM_STATE_ABOVE", 0);
778 	XChangeProperty(d, w, XInternAtom(d, "_NET_WM_STATE", 0),
779 	    XInternAtom(d, "ATOM", 0), 32, PropModeReplace,
780 	    (unsigned char *)&type, 1);
781 
782 	type = XInternAtom(d, "_NET_WM_STATE_FOCUSED", 0);
783 	XChangeProperty(d, w, XInternAtom(d, "_NET_WM_STATE", 0),
784 	    XInternAtom(d, "ATOM", 0), 32, PropModeAppend,
785 	    (unsigned char *)&type, 1);
786 
787 	/* Setting window hints */
788 	class_hint = XAllocClassHint();
789 	if (class_hint == NULL) {
790 		fprintf(stderr, "Could not allocate memory for class hint\n");
791 		exit(EX_UNAVAILABLE);
792 	}
793 
794 	class_hint->res_name = RESNAME;
795 	class_hint->res_class = RESCLASS;
796 	XSetClassHint(d, w, class_hint);
797 	XFree(class_hint);
798 
799 	size_hint = XAllocSizeHints();
800 	if (size_hint == NULL) {
801 		fprintf(stderr, "Could not allocate memory for size hint\n");
802 		exit(EX_UNAVAILABLE);
803 	}
804 
805 	size_hint->flags = PMinSize | PBaseSize;
806 	size_hint->min_width = width;
807 	size_hint->base_width = width;
808 	size_hint->min_height = height;
809 	size_hint->base_height = height;
810 
811 	XFlush(d);
812 }
813 
814 /* Get the width and height of the window `w' */
815 void
816 get_wh(Display *d, Window *w, int *width, int *height)
817 {
818 	XWindowAttributes win_attr;
819 
820 	XGetWindowAttributes(d, *w, &win_attr);
821 	*height = win_attr.height;
822 	*width = win_attr.width;
823 }
824 
825 /* find the current xinerama monitor if possible */
826 void
827 findmonitor(Display *d, int *x, int *y, int *width, int *height)
828 {
829 	XineramaScreenInfo *info;
830 	Window rr;
831 	Window root;
832 	int screens, monitors, i;
833 	int rootx, rooty, winx, winy;
834 	unsigned int mask;
835 	short res;
836 
837 	if (!XineramaIsActive(d))
838 		return;
839 
840 	screens = XScreenCount(d);
841 	for (i = 0; i < screens; ++i) {
842 		root = XRootWindow(d, i);
843 		res = XQueryPointer(d, root, &rr, &rr, &rootx, &rooty, &winx,
844 		    &winy, &mask);
845 		if (res)
846 			break;
847 	}
848 
849 	if (!res)
850 		return;
851 
852 	/* Now find in which monitor the mice is */
853 	info = XineramaQueryScreens(d, &monitors);
854 	if (info == NULL)
855 		return;
856 
857 	for (i = 0; i < monitors; ++i) {
858 		if (info[i].x_org <= rootx &&
859 		    rootx <= (info[i].x_org + info[i].width) &&
860 		    info[i].y_org <= rooty &&
861 		    rooty <= (info[i].y_org + info[i].height)) {
862 			*x = info[i].x_org;
863 			*y = info[i].y_org;
864 			*width = info[i].width;
865 			*height = info[i].height;
866 			break;
867 		}
868 	}
869 
870 	XFree(info);
871 }
872 
873 int
874 grabfocus(Display *d, Window w)
875 {
876 	int i;
877 	for (i = 0; i < 100; ++i) {
878 		Window focuswin;
879 		int revert_to_win;
880 
881 		XGetInputFocus(d, &focuswin, &revert_to_win);
882 
883 		if (focuswin == w)
884 			return 1;
885 
886 		XSetInputFocus(d, w, RevertToParent, CurrentTime);
887 		usleep(1000);
888 	}
889 	return 0;
890 }
891 
892 /*
893  * I know this may seem a little hackish BUT is the only way I managed
894  * to actually grab that goddam keyboard. Only one call to
895  * XGrabKeyboard does not always end up with the keyboard grabbed!
896  */
897 int
898 take_keyboard(Display *d, Window w)
899 {
900 	int i;
901 	for (i = 0; i < 100; i++) {
902 		if (XGrabKeyboard(d, w, 1, GrabModeAsync, GrabModeAsync,
903 		    CurrentTime) == GrabSuccess)
904 			return 1;
905 		usleep(1000);
906 	}
907 	fprintf(stderr, "Cannot grab keyboard\n");
908 	return 0;
909 }
910 
911 unsigned long
912 parse_color(const char *str, const char *def)
913 {
914 	size_t len;
915 	rgba_t tmp;
916 	char *ep;
917 
918 	if (str == NULL)
919 		goto err;
920 
921 	len = strlen(str);
922 
923 	/* +1 for the # ath the start */
924 	if (*str != '#' || len > 9 || len < 4)
925 		goto err;
926 	++str; /* skip the # */
927 
928 	errno = 0;
929 	tmp = (rgba_t)(uint32_t)strtoul(str, &ep, 16);
930 
931 	if (errno)
932 		goto err;
933 
934 	switch (len - 1) {
935 	case 3:
936 		/* expand #rgb -> #rrggbb */
937 		tmp.v = (tmp.v & 0xf00) * 0x1100 | (tmp.v & 0x0f0) * 0x0110
938 			| (tmp.v & 0x00f) * 0x0011;
939 	case 6:
940 		/* assume 0xff opacity */
941 		tmp.rgba.a = 0xff;
942 		break;
943 	} /* colors in #aarrggbb need no adjustments */
944 
945 	/* premultiply the alpha */
946 	if (tmp.rgba.a) {
947 		tmp.rgba.r = (tmp.rgba.r * tmp.rgba.a) / 255;
948 		tmp.rgba.g = (tmp.rgba.g * tmp.rgba.a) / 255;
949 		tmp.rgba.b = (tmp.rgba.b * tmp.rgba.a) / 255;
950 		return tmp.v;
951 	}
952 
953 	return 0U;
954 
955 err:
956 	fprintf(stderr, "Invalid color: \"%s\".\n", str);
957 	if (def != NULL)
958 		return parse_color(def, NULL);
959 	else
960 		return 0U;
961 }
962 
963 /*
964  * Given a string try to parse it as a number or return `def'.
965  */
966 int
967 parse_integer(const char *str, int def)
968 {
969 	const char *errstr;
970 	int i;
971 
972 	i = strtonum(str, INT_MIN, INT_MAX, &errstr);
973 	if (errstr != NULL) {
974 		warnx("'%s' is %s; using %d as default", str, errstr, def);
975 		return def;
976 	}
977 
978 	return i;
979 }
980 
981 /*
982  * Like parse_integer but recognize the percentages (i.e. strings
983  * ending with `%')
984  */
985 int
986 parse_int_with_percentage(const char *str, int default_value, int max)
987 {
988 	int len = strlen(str);
989 
990 	if (len > 0 && str[len - 1] == '%') {
991 		int val;
992 		char *cpy;
993 
994 		if ((cpy = strdup(str)) == NULL)
995 			err(1, "strdup");
996 
997 		cpy[len - 1] = '\0';
998 		val = parse_integer(cpy, default_value);
999 		free(cpy);
1000 		return val * max / 100;
1001 	}
1002 
1003 	return parse_integer(str, default_value);
1004 }
1005 
1006 void
1007 get_mouse_coords(Display *d, int *x, int *y)
1008 {
1009 	Window w, root;
1010 	int i;
1011 	unsigned int u;
1012 
1013 	*x = *y = 0;
1014 	root = DefaultRootWindow(d);
1015 
1016 	if (!XQueryPointer(d, root, &root, &w, x, y, &i, &i, &u)) {
1017 		for (i = 0; i < ScreenCount(d); ++i) {
1018 			if (root == RootWindow(d, i))
1019 				break;
1020 		}
1021 	}
1022 }
1023 
1024 /*
1025  * Like parse_int_with_percentage but understands some special values:
1026  * - middle that is (max-self)/2
1027  * - center = middle
1028  * - start  that is 0
1029  * - end    that is (max-self)
1030  * - mx     x coordinate of the mouse
1031  * - my     y coordinate of the mouse
1032  */
1033 int
1034 parse_int_with_pos(Display *d, const char *str, int default_value, int max,
1035     int self)
1036 {
1037 	if (!strcmp(str, "start"))
1038 		return 0;
1039 	if (!strcmp(str, "middle") || !strcmp(str, "center"))
1040 		return (max - self) / 2;
1041 	if (!strcmp(str, "end"))
1042 		return max - self;
1043 	if (!strcmp(str, "mx") || !strcmp(str, "my")) {
1044 		int x, y;
1045 
1046 		get_mouse_coords(d, &x, &y);
1047 		if (!strcmp(str, "mx"))
1048 			return x - 1;
1049 		else
1050 			return y - 1;
1051 	}
1052 	return parse_int_with_percentage(str, default_value, max);
1053 }
1054 
1055 /* Parse a string like a CSS value. */
1056 /* TODO: harden a bit this function */
1057 int
1058 parse_csslike(const char *str, char **ret)
1059 {
1060 	int i, j;
1061 	char *s, *token;
1062 	short any_null;
1063 
1064 	memset(ret, 0, 4 * sizeof(*ret));
1065 
1066 	s = strdup(str);
1067 	if (s == NULL)
1068 		return -1;
1069 
1070 	for (i = 0; (token = strsep(&s, " ")) != NULL && i < 4; ++i)
1071 		ret[i] = strdup(token);
1072 
1073 	if (i == 1)
1074 		for (j = 1; j < 4; j++)
1075 			ret[j] = strdup(ret[0]);
1076 
1077 	if (i == 2) {
1078 		ret[2] = strdup(ret[0]);
1079 		ret[3] = strdup(ret[1]);
1080 	}
1081 
1082 	if (i == 3)
1083 		ret[3] = strdup(ret[1]);
1084 
1085 	/*
1086 	 * before we didn't check for the return type of strdup, here
1087 	 * we will
1088 	 */
1089 
1090 	any_null = 0;
1091 	for (i = 0; i < 4; ++i)
1092 		any_null = ret[i] == NULL || any_null;
1093 
1094 	if (any_null)
1095 		for (i = 0; i < 4; ++i)
1096 			free(ret[i]);
1097 
1098 	if (i == 0 || any_null) {
1099 		free(s);
1100 		return -1;
1101 	}
1102 
1103 	return 1;
1104 }
1105 
1106 /*
1107  * Given an event, try to understand what the users wants. If the
1108  * return value is ADD_CHAR then `input' is a pointer to a string that
1109  * will need to be free'ed later.
1110  */
1111 enum action
1112 parse_event(Display *d, XKeyPressedEvent *ev, XIC xic, char **input)
1113 {
1114 	char str[SYM_BUF_SIZE] = { 0 };
1115 	Status s;
1116 
1117 	if (ev->keycode == XKeysymToKeycode(d, XK_BackSpace))
1118 		return DEL_CHAR;
1119 
1120 	if (ev->keycode == XKeysymToKeycode(d, XK_Tab))
1121 		return ev->state & ShiftMask ? PREV_COMPL : NEXT_COMPL;
1122 
1123 	if (ev->keycode == XKeysymToKeycode(d, XK_Return))
1124 		return CONFIRM;
1125 
1126 	if (ev->keycode == XKeysymToKeycode(d, XK_Escape))
1127 		return EXIT;
1128 
1129 	/* Try to read what key was pressed */
1130 	s = 0;
1131 	Xutf8LookupString(xic, ev, str, SYM_BUF_SIZE, 0, &s);
1132 	if (s == XBufferOverflow) {
1133 		fprintf(stderr,
1134 		    "Buffer overflow when trying to create keyboard "
1135 		    "symbol map.\n");
1136 		return EXIT;
1137 	}
1138 
1139 	if (ev->state & ControlMask) {
1140 		if (!strcmp(str, "")) /* C-u */
1141 			return DEL_LINE;
1142 		if (!strcmp(str, "")) /* C-w */
1143 			return DEL_WORD;
1144 		if (!strcmp(str, "")) /* C-h */
1145 			return DEL_CHAR;
1146 		if (!strcmp(str, "\r")) /* C-m */
1147 			return CONFIRM_CONTINUE;
1148 		if (!strcmp(str, "")) /* C-p */
1149 			return PREV_COMPL;
1150 		if (!strcmp(str, "")) /* C-n */
1151 			return NEXT_COMPL;
1152 		if (!strcmp(str, "")) /* C-c */
1153 			return EXIT;
1154 		if (!strcmp(str, "\t")) /* C-i */
1155 			return TOGGLE_FIRST_SELECTED;
1156 	}
1157 
1158 	*input = strdup(str);
1159 	if (*input == NULL) {
1160 		fprintf(stderr, "Error while allocating memory for key.\n");
1161 		return EXIT;
1162 	}
1163 
1164 	return ADD_CHAR;
1165 }
1166 
1167 void
1168 confirm(enum state *status, struct rendering *r, struct completions *cs,
1169     char **text, int *textlen)
1170 {
1171 	if ((cs->selected != -1) || (cs->length > 0 && r->first_selected)) {
1172 		/* if there is something selected expand it and return */
1173 		int index = cs->selected == -1 ? 0 : cs->selected;
1174 		struct completion *c = cs->completions;
1175 		char *t;
1176 
1177 		while (1) {
1178 			if (index == 0)
1179 				break;
1180 			c++;
1181 			index--;
1182 		}
1183 
1184 		t = c->rcompletion;
1185 		free(*text);
1186 		*text = strdup(t);
1187 
1188 		if (*text == NULL) {
1189 			fprintf(stderr, "Memory allocation error\n");
1190 			*status = ERR;
1191 		}
1192 
1193 		*textlen = strlen(*text);
1194 		return;
1195 	}
1196 
1197 	if (!r->free_text) /* cannot accept arbitrary text */
1198 		*status = LOOPING;
1199 }
1200 
1201 /*
1202  * cs: completion list
1203  * offset: the offset of the click
1204  * first: the first (rendered) item
1205  * def: the default action
1206  */
1207 enum action
1208 select_clicked(struct completions *cs, size_t offset, size_t first,
1209     enum action def)
1210 {
1211 	ssize_t selected = first;
1212 	int set = 0;
1213 
1214 	if (cs->length == 0)
1215 		return NO_OP;
1216 
1217 	if (offset < cs->completions[selected].offset)
1218 		return EXIT;
1219 
1220 	/* skip the first entry */
1221 	for (selected += 1; selected < cs->length; ++selected) {
1222 		if (cs->completions[selected].offset == -1)
1223 			break;
1224 
1225 		if (offset < cs->completions[selected].offset) {
1226 			cs->selected = selected - 1;
1227 			set = 1;
1228 			break;
1229 		}
1230 	}
1231 
1232 	if (!set)
1233 		cs->selected = selected - 1;
1234 
1235 	return def;
1236 }
1237 
1238 enum action
1239 handle_mouse(struct rendering *r, struct completions *cs,
1240     XButtonPressedEvent *e)
1241 {
1242 	size_t off;
1243 
1244 	if (r->horizontal_layout)
1245 		off = e->x;
1246 	else
1247 		off = e->y;
1248 
1249 	switch (e->button) {
1250 	case Button1:
1251 		return select_clicked(cs, off, r->offset, CONFIRM);
1252 
1253 	case Button3:
1254 		return select_clicked(cs, off, r->offset, CONFIRM_CONTINUE);
1255 
1256 	case Button4:
1257 		return SCROLL_UP;
1258 
1259 	case Button5:
1260 		return SCROLL_DOWN;
1261 	}
1262 
1263 	return NO_OP;
1264 }
1265 
1266 /* event loop */
1267 enum state
1268 loop(struct rendering *r, char **text, int *textlen, struct completions *cs,
1269     char **lines, char **vlines)
1270 {
1271 	enum action a;
1272 	char *input = NULL;
1273 	enum state status = LOOPING;
1274 	int i;
1275 
1276 	while (status == LOOPING) {
1277 		XEvent e;
1278 		XNextEvent(r->d, &e);
1279 
1280 		if (XFilterEvent(&e, r->w))
1281 			continue;
1282 
1283 		switch (e.type) {
1284 		case KeymapNotify:
1285 			XRefreshKeyboardMapping(&e.xmapping);
1286 			break;
1287 
1288 		case FocusIn:
1289 			/* Re-grab focus */
1290 			if (e.xfocus.window != r->w)
1291 				grabfocus(r->d, r->w);
1292 			break;
1293 
1294 		case VisibilityNotify:
1295 			if (e.xvisibility.state != VisibilityUnobscured)
1296 				XRaiseWindow(r->d, r->w);
1297 			break;
1298 
1299 		case MapNotify:
1300 			get_wh(r->d, &r->w, &r->width, &r->height);
1301 			draw(r, *text, cs);
1302 			break;
1303 
1304 		case KeyPress:
1305 		case ButtonPress:
1306 			if (e.type == KeyPress)
1307 				a = parse_event(r->d, (XKeyPressedEvent *)&e,
1308 				    r->xic, &input);
1309 			else
1310 				a = handle_mouse(r, cs,
1311 				    (XButtonPressedEvent *)&e);
1312 
1313 			switch (a) {
1314 			case NO_OP:
1315 				break;
1316 
1317 			case EXIT:
1318 				status = ERR;
1319 				break;
1320 
1321 			case CONFIRM:
1322 				status = OK;
1323 				confirm(&status, r, cs, text, textlen);
1324 				break;
1325 
1326 			case CONFIRM_CONTINUE:
1327 				status = OK_LOOP;
1328 				confirm(&status, r, cs, text, textlen);
1329 				break;
1330 
1331 			case PREV_COMPL:
1332 				complete(cs, r->first_selected, 1, text,
1333 				    textlen, &status);
1334 				r->offset = cs->selected;
1335 				break;
1336 
1337 			case NEXT_COMPL:
1338 				complete(cs, r->first_selected, 0, text,
1339 				    textlen, &status);
1340 				r->offset = cs->selected;
1341 				break;
1342 
1343 			case DEL_CHAR:
1344 				popc(*text);
1345 				update_completions(cs, *text, lines, vlines,
1346 				    r->first_selected);
1347 				r->offset = 0;
1348 				break;
1349 
1350 			case DEL_WORD:
1351 				popw(*text);
1352 				update_completions(cs, *text, lines, vlines,
1353 				    r->first_selected);
1354 				break;
1355 
1356 			case DEL_LINE:
1357 				for (i = 0; i < *textlen; ++i)
1358 					(*text)[i] = 0;
1359 				update_completions(cs, *text, lines, vlines,
1360 				    r->first_selected);
1361 				r->offset = 0;
1362 				break;
1363 
1364 			case ADD_CHAR: {
1365 				int str_len;
1366 
1367 				str_len = strlen(input);
1368 
1369 				/*
1370 				 * sometimes a strange key is pressed
1371 				 * i.e. ctrl alone), so input will be
1372 				 * empty. Don't need to update
1373 				 * completion in that case
1374 				 */
1375 				if (str_len == 0)
1376 					break;
1377 
1378 				for (i = 0; i < str_len; ++i) {
1379 					*textlen = pushc(text, *textlen,
1380 					    input[i]);
1381 					if (*textlen == -1) {
1382 						fprintf(stderr,
1383 						    "Memory allocation "
1384 						    "error\n");
1385 						status = ERR;
1386 						break;
1387 					}
1388 				}
1389 
1390 				if (status != ERR) {
1391 					update_completions(cs, *text, lines,
1392 					    vlines, r->first_selected);
1393 					free(input);
1394 				}
1395 
1396 				r->offset = 0;
1397 				break;
1398 			}
1399 
1400 			case TOGGLE_FIRST_SELECTED:
1401 				r->first_selected = !r->first_selected;
1402 				if (r->first_selected && cs->selected < 0)
1403 					cs->selected = 0;
1404 				if (!r->first_selected && cs->selected == 0)
1405 					cs->selected = -1;
1406 				break;
1407 
1408 			case SCROLL_DOWN:
1409 				r->offset = MIN(r->offset + 1, cs->length - 1);
1410 				break;
1411 
1412 			case SCROLL_UP:
1413 				r->offset = MAX((ssize_t)r->offset - 1, 0);
1414 				break;
1415 			}
1416 		}
1417 
1418 		draw(r, *text, cs);
1419 	}
1420 
1421 	return status;
1422 }
1423 
1424 int
1425 load_font(struct rendering *r, const char *fontname)
1426 {
1427 	r->font = XftFontOpenName(r->d, DefaultScreen(r->d), fontname);
1428 	return 0;
1429 }
1430 
1431 void
1432 xim_init(struct rendering *r, XrmDatabase *xdb)
1433 {
1434 	XIMStyle best_match_style;
1435 	XIMStyles *xis;
1436 	int i;
1437 
1438 	/* Open the X input method */
1439 	if ((r->xim = XOpenIM(r->d, *xdb, RESNAME, RESCLASS)) == NULL)
1440 		err(1, "XOpenIM");
1441 
1442 	if (XGetIMValues(r->xim, XNQueryInputStyle, &xis, NULL) || !xis) {
1443 		fprintf(stderr, "Input Styles could not be retrieved\n");
1444 		exit(EX_UNAVAILABLE);
1445 	}
1446 
1447 	best_match_style = 0;
1448 	for (i = 0; i < xis->count_styles; ++i) {
1449 		XIMStyle ts = xis->supported_styles[i];
1450 		if (ts == (XIMPreeditNothing | XIMStatusNothing)) {
1451 			best_match_style = ts;
1452 			break;
1453 		}
1454 	}
1455 	XFree(xis);
1456 
1457 	if (!best_match_style)
1458 		fprintf(stderr,
1459 		    "No matching input style could be determined\n");
1460 
1461 	r->xic = XCreateIC(r->xim, XNInputStyle, best_match_style,
1462 	    XNClientWindow, r->w, XNFocusWindow, r->w, NULL);
1463 	if (r->xic == NULL)
1464 		err(1, "XCreateIC");
1465 }
1466 
1467 void
1468 create_window(struct rendering *r, Window parent_window, Colormap cmap,
1469     XVisualInfo vinfo, int x, int y, int ox, int oy,
1470     unsigned long background_pixel)
1471 {
1472 	XSetWindowAttributes attr;
1473 	unsigned long vmask;
1474 
1475 	/* Create the window */
1476 	attr.colormap = cmap;
1477 	attr.override_redirect = 1;
1478 	attr.border_pixel = 0;
1479 	attr.background_pixel = background_pixel;
1480 	attr.event_mask = StructureNotifyMask | ExposureMask | KeyPressMask
1481 		| KeymapStateMask | ButtonPress | VisibilityChangeMask;
1482 
1483 	vmask = CWBorderPixel | CWBackPixel | CWColormap | CWEventMask |
1484 		CWOverrideRedirect;
1485 
1486 	r->w = XCreateWindow(r->d, parent_window, x + ox, y + oy, r->width, r->height, 0,
1487 	    vinfo.depth, InputOutput, vinfo.visual, vmask, &attr);
1488 }
1489 
1490 void
1491 ps1extents(struct rendering *r)
1492 {
1493 	char *dup;
1494 	dup = strdupn(r->ps1);
1495 	text_extents(dup == NULL ? r->ps1 : dup, r->ps1len, r, &r->ps1w, &r->ps1h);
1496 	free(dup);
1497 }
1498 
1499 void
1500 usage(char *prgname)
1501 {
1502 	fprintf(stderr,
1503 	    "%s [-Aahmv] [-B colors] [-b size] [-C color] [-c color]\n"
1504 	    "       [-d separator] [-e window] [-f font] [-G color] [-g "
1505 	    "size]\n"
1506 	    "       [-H height] [-I color] [-i size] [-J color] [-j "
1507 	    "size] [-l layout]\n"
1508 	    "       [-P padding] [-p prompt] [-S color] [-s color] [-T "
1509 	    "color]\n"
1510 	    "       [-t color] [-W width] [-x coord] [-y coord]\n",
1511 	    prgname);
1512 }
1513 
1514 int
1515 main(int argc, char **argv)
1516 {
1517 	struct completions *cs;
1518 	struct rendering r;
1519 	XVisualInfo vinfo;
1520 	Colormap cmap;
1521 	size_t nlines, i;
1522 	Window parent_window;
1523 	XrmDatabase xdb;
1524 	unsigned long fgs[3], bgs[3]; /* prompt, compl, compl_highlighted */
1525 	unsigned long borders_bg[4], p_borders_bg[4], c_borders_bg[4],
1526 		ch_borders_bg[4]; /* N E S W */
1527 	enum state status = LOOPING;
1528 	int ch;
1529 	int offset_x = 0, offset_y = 0;
1530 	int x = 0, y = 0;
1531 	int textlen, d_width, d_height;
1532 	short embed;
1533 	const char *sep = NULL;
1534 	const char *parent_window_id = NULL;
1535 	char *tmp[4];
1536 	char **lines, **vlines;
1537 	char *fontname, *text, *xrm;
1538 
1539 	setlocale(LC_ALL, getenv("LANG"));
1540 
1541 	for (i = 0; i < 4; ++i) {
1542 		/* default paddings */
1543 		r.p_padding[i] = 10;
1544 		r.c_padding[i] = 10;
1545 		r.ch_padding[i] = 10;
1546 
1547 		/* default borders */
1548 		r.borders[i] = 0;
1549 		r.p_borders[i] = 0;
1550 		r.c_borders[i] = 0;
1551 		r.ch_borders[i] = 0;
1552 	}
1553 
1554 	r.first_selected = 0;
1555 	r.free_text = 1;
1556 	r.multiple_select = 0;
1557 	r.offset = 0;
1558 
1559 	/* default width and height */
1560 	r.width = 400;
1561 	r.height = 20;
1562 
1563 	/*
1564 	 * The prompt. We duplicate the string so later is easy to
1565 	 * free (in the case it's been overwritten by the user)
1566 	 */
1567 	if ((r.ps1 = strdup("$ ")) == NULL)
1568 		err(1, "strdup");
1569 
1570 	/* same for the font name */
1571 	if ((fontname = strdup(DEFFONT)) == NULL)
1572 		err(1, "strdup");
1573 
1574 	while ((ch = getopt(argc, argv, ARGS)) != -1) {
1575 		switch (ch) {
1576 		case 'h': /* help */
1577 			usage(*argv);
1578 			return 0;
1579 		case 'v': /* version */
1580 			fprintf(stderr, "%s version: %s\n", *argv, VERSION);
1581 			return 0;
1582 		case 'e': /* embed */
1583 			if ((parent_window_id = strdup(optarg)) == NULL)
1584 				err(1, "strdup");
1585 			break;
1586 		case 'd':
1587 			if ((sep = strdup(optarg)) == NULL)
1588 				err(1, "strdup");
1589 			break;
1590 		case 'A':
1591 			r.free_text = 0;
1592 			break;
1593 		case 'm':
1594 			r.multiple_select = 1;
1595 			break;
1596 		default:
1597 			break;
1598 		}
1599 	}
1600 
1601 	lines = readlines(&nlines);
1602 
1603 	vlines = NULL;
1604 	if (sep != NULL) {
1605 		int l;
1606 		l = strlen(sep);
1607 		if ((vlines = calloc(nlines, sizeof(char *))) == NULL)
1608 			err(1, "calloc");
1609 
1610 		for (i = 0; i < nlines; i++) {
1611 			char *t;
1612 			t = strstr(lines[i], sep);
1613 			if (t == NULL)
1614 				vlines[i] = lines[i];
1615 			else
1616 				vlines[i] = t + l;
1617 		}
1618 	}
1619 
1620 	textlen = 10;
1621 	if ((text = malloc(textlen * sizeof(char))) == NULL)
1622 		err(1, "malloc");
1623 
1624 	/* struct completions *cs = filter(text, lines); */
1625 	if ((cs = compls_new(nlines)) == NULL)
1626 		err(1, "compls_new");
1627 
1628 	/* start talking to xorg */
1629 	r.d = XOpenDisplay(NULL);
1630 	if (r.d == NULL) {
1631 		fprintf(stderr, "Could not open display!\n");
1632 		return EX_UNAVAILABLE;
1633 	}
1634 
1635 	embed = 1;
1636 	if (!(parent_window_id && (parent_window = strtol(parent_window_id, NULL, 0)))) {
1637 		parent_window = DefaultRootWindow(r.d);
1638 		embed = 0;
1639 	}
1640 
1641 	/* get display size */
1642 	get_wh(r.d, &parent_window, &d_width, &d_height);
1643 
1644 	if (!embed)
1645 		findmonitor(r.d, &offset_x, &offset_y, &d_width, &d_height);
1646 
1647 	XMatchVisualInfo(r.d, DefaultScreen(r.d), 32, TrueColor, &vinfo);
1648 	cmap = XCreateColormap(r.d, XDefaultRootWindow(r.d), vinfo.visual,
1649 	    AllocNone);
1650 
1651 	fgs[0] = fgs[1] = parse_color("#fff", NULL);
1652 	fgs[2] = parse_color("#000", NULL);
1653 
1654 	bgs[0] = bgs[1] = parse_color("#000", NULL);
1655 	bgs[2] = parse_color("#fff", NULL);
1656 
1657 	borders_bg[0] = borders_bg[1] = borders_bg[2] = borders_bg[3] =
1658 		parse_color("#000", NULL);
1659 
1660 	p_borders_bg[0] = p_borders_bg[1] = p_borders_bg[2] = p_borders_bg[3]
1661 		= parse_color("#000", NULL);
1662 	c_borders_bg[0] = c_borders_bg[1] = c_borders_bg[2] = c_borders_bg[3]
1663 		= parse_color("#000", NULL);
1664 	ch_borders_bg[0] = ch_borders_bg[1] = ch_borders_bg[2] = ch_borders_bg[3]
1665 		= parse_color("#000", NULL);
1666 
1667 	r.horizontal_layout = 1;
1668 
1669 	/* Read the resources */
1670 	XrmInitialize();
1671 	xrm = XResourceManagerString(r.d);
1672 	xdb = NULL;
1673 	if (xrm != NULL) {
1674 		XrmValue value;
1675 		char *datatype[20];
1676 
1677 		xdb = XrmGetStringDatabase(xrm);
1678 
1679 		if (XrmGetResource(xdb, "MyMenu.font", "*", datatype, &value)) {
1680 			free(fontname);
1681 			if ((fontname = strdup(value.addr)) == NULL)
1682 				err(1, "strdup");
1683 		} else {
1684 			fprintf(stderr, "no font defined, using %s\n", fontname);
1685 		}
1686 
1687 		if (XrmGetResource(xdb, "MyMenu.layout", "*", datatype, &value))
1688 			r.horizontal_layout = !strcmp(value.addr, "horizontal");
1689 		else
1690 			fprintf(stderr, "no layout defined, using horizontal\n");
1691 
1692 		if (XrmGetResource(xdb, "MyMenu.prompt", "*", datatype, &value)) {
1693 			free(r.ps1);
1694 			r.ps1 = normalize_str(value.addr);
1695 		} else {
1696 			fprintf(stderr,
1697 				"no prompt defined, using \"%s\" as "
1698 				"default\n",
1699 				r.ps1);
1700 		}
1701 
1702 		if (XrmGetResource(xdb, "MyMenu.prompt.border.size", "*", datatype, &value)) {
1703 			if (parse_csslike(value.addr, tmp) == -1)
1704 				err(1, "parse_csslike");
1705 			for (i = 0; i < 4; ++i) {
1706 				r.p_borders[i] = parse_integer(tmp[i], 0);
1707 				free(tmp[i]);
1708 			}
1709 		}
1710 
1711 		if (XrmGetResource(xdb, "MyMenu.prompt.border.color", "*", datatype, &value)) {
1712 			if (parse_csslike(value.addr, tmp) == -1)
1713 				err(1, "parse_csslike");
1714 
1715 			for (i = 0; i < 4; ++i) {
1716 				p_borders_bg[i] = parse_color(tmp[i], "#000");
1717 				free(tmp[i]);
1718 			}
1719 		}
1720 
1721 		if (XrmGetResource(xdb, "MyMenu.prompt.padding", "*", datatype, &value)) {
1722 			if (parse_csslike(value.addr, tmp) == -1)
1723 				err(1, "parse_csslike");
1724 
1725 			for (i = 0; i < 4; ++i) {
1726 				r.p_padding[i] = parse_integer(tmp[i], 0);
1727 				free(tmp[i]);
1728 			}
1729 		}
1730 
1731 		if (XrmGetResource(xdb, "MyMenu.width", "*", datatype, &value))
1732 			r.width = parse_int_with_percentage(value.addr, r.width, d_width);
1733 		else
1734 			fprintf(stderr, "no width defined, using %d\n", r.width);
1735 
1736 		if (XrmGetResource(xdb, "MyMenu.height", "*", datatype, &value))
1737 			r.height = parse_int_with_percentage(value.addr, r.height, d_height);
1738 		else
1739 			fprintf(stderr, "no height defined, using %d\n", r.height);
1740 
1741 		if (XrmGetResource(xdb, "MyMenu.x", "*", datatype, &value))
1742 			x = parse_int_with_pos(r.d, value.addr, x, d_width, r.width);
1743 
1744 		if (XrmGetResource(xdb, "MyMenu.y", "*", datatype, &value))
1745 			y = parse_int_with_pos(r.d, value.addr, y, d_height, r.height);
1746 
1747 		if (XrmGetResource(xdb, "MyMenu.border.size", "*", datatype, &value)) {
1748 			if (parse_csslike(value.addr, tmp) == -1)
1749 				err(1, "parse_csslike");
1750 
1751 			for (i = 0; i < 4; ++i) {
1752 				r.borders[i] = parse_int_with_percentage(tmp[i], 0,
1753 				    (i % 2) == 0 ? d_height : d_width);
1754 				free(tmp[i]);
1755 			}
1756 		}
1757 
1758 		/* Prompt */
1759 		if (XrmGetResource(xdb, "MyMenu.prompt.foreground", "*", datatype, &value))
1760 			fgs[0] = parse_color(value.addr, "#fff");
1761 
1762 		if (XrmGetResource(xdb, "MyMenu.prompt.background", "*", datatype, &value))
1763 			bgs[0] = parse_color(value.addr, "#000");
1764 
1765 		/* Completions */
1766 		if (XrmGetResource(xdb, "MyMenu.completion.foreground", "*", datatype, &value))
1767 			fgs[1] = parse_color(value.addr, "#fff");
1768 
1769 		if (XrmGetResource(xdb, "MyMenu.completion.background", "*", datatype, &value))
1770 			bgs[1] = parse_color(value.addr, "#000");
1771 
1772 		if (XrmGetResource(xdb, "MyMenu.completion.padding", "*", datatype, &value)) {
1773 			if (parse_csslike(value.addr, tmp) == -1)
1774 				err(1, "parse_csslike");
1775 
1776 			for (i = 0; i < 4; ++i) {
1777 				r.c_padding[i] = parse_integer(tmp[i], 0);
1778 				free(tmp[i]);
1779 			}
1780 		}
1781 
1782 		if (XrmGetResource(xdb, "MyMenu.completion.border.size", "*", datatype, &value)) {
1783 			if (parse_csslike(value.addr, tmp) == -1)
1784 				err(1, "parse_csslike");
1785 
1786 			for (i = 0; i < 4; ++i) {
1787 				r.c_borders[i] = parse_integer(tmp[i], 0);
1788 				free(tmp[i]);
1789 			}
1790 		}
1791 
1792 		if (XrmGetResource(xdb, "MyMenu.completion.border.color", "*", datatype, &value)) {
1793 			if (parse_csslike(value.addr, tmp) == -1)
1794 				err(1, "parse_csslike");
1795 
1796 			for (i = 0; i < 4; ++i) {
1797 				c_borders_bg[i] = parse_color(tmp[i], "#000");
1798 				free(tmp[i]);
1799 			}
1800 		}
1801 
1802 		/* Completion Highlighted */
1803 		if (XrmGetResource(
1804 			    xdb, "MyMenu.completion_highlighted.foreground", "*", datatype, &value))
1805 			fgs[2] = parse_color(value.addr, "#000");
1806 
1807 		if (XrmGetResource(
1808 			    xdb, "MyMenu.completion_highlighted.background", "*", datatype, &value))
1809 			bgs[2] = parse_color(value.addr, "#fff");
1810 
1811 		if (XrmGetResource(
1812 			    xdb, "MyMenu.completion_highlighted.padding", "*", datatype, &value)) {
1813 			if (parse_csslike(value.addr, tmp) == -1)
1814 				err(1, "parse_csslike");
1815 
1816 			for (i = 0; i < 4; ++i) {
1817 				r.ch_padding[i] = parse_integer(tmp[i], 0);
1818 				free(tmp[i]);
1819 			}
1820 		}
1821 
1822 		if (XrmGetResource(xdb, "MyMenu.completion_highlighted.border.size", "*", datatype,
1823 		    &value)) {
1824 			if (parse_csslike(value.addr, tmp) == -1)
1825 				err(1, "parse_csslike");
1826 
1827 			for (i = 0; i < 4; ++i) {
1828 				r.ch_borders[i] = parse_integer(tmp[i], 0);
1829 				free(tmp[i]);
1830 			}
1831 		}
1832 
1833 		if (XrmGetResource(xdb, "MyMenu.completion_highlighted.border.color", "*", datatype,
1834 			    &value)) {
1835 			if (parse_csslike(value.addr, tmp) == -1)
1836 				err(1, "parse_csslike");
1837 
1838 			for (i = 0; i < 4; ++i) {
1839 				ch_borders_bg[i] = parse_color(tmp[i], "#000");
1840 				free(tmp[i]);
1841 			}
1842 		}
1843 
1844 		/* Border */
1845 		if (XrmGetResource(xdb, "MyMenu.border.color", "*", datatype, &value)) {
1846 			if (parse_csslike(value.addr, tmp) == -1)
1847 				err(1, "parse_csslike");
1848 
1849 			for (i = 0; i < 4; ++i) {
1850 				borders_bg[i] = parse_color(tmp[i], "#000");
1851 				free(tmp[i]);
1852 			}
1853 		}
1854 	}
1855 
1856 	/* Second round of args parsing */
1857 	optind = 0; /* reset the option index */
1858 	while ((ch = getopt(argc, argv, ARGS)) != -1) {
1859 		switch (ch) {
1860 		case 'a':
1861 			r.first_selected = 1;
1862 			break;
1863 		case 'A':
1864 			/* free_text -- already catched */
1865 		case 'd':
1866 			/* separator -- this case was already catched */
1867 		case 'e':
1868 			/* embedding mymenu this case was already catched. */
1869 		case 'm':
1870 			/* multiple selection this case was already catched.
1871 			 */
1872 			break;
1873 		case 'p': {
1874 			char *newprompt;
1875 			newprompt = strdup(optarg);
1876 			if (newprompt != NULL) {
1877 				free(r.ps1);
1878 				r.ps1 = newprompt;
1879 			}
1880 			break;
1881 		}
1882 		case 'x':
1883 			x = parse_int_with_pos(r.d, optarg, x, d_width, r.width);
1884 			break;
1885 		case 'y':
1886 			y = parse_int_with_pos(r.d, optarg, y, d_height, r.height);
1887 			break;
1888 		case 'P':
1889 			if (parse_csslike(optarg, tmp) == -1)
1890 				err(1, "parse_csslike");
1891 			for (i = 0; i < 4; ++i)
1892 				r.p_padding[i] = parse_integer(tmp[i], 0);
1893 			break;
1894 		case 'G':
1895 			if (parse_csslike(optarg, tmp) == -1)
1896 				err(1, "parse_csslike");
1897 			for (i = 0; i < 4; ++i)
1898 				p_borders_bg[i] = parse_color(tmp[i], "#000");
1899 			break;
1900 		case 'g':
1901 			if (parse_csslike(optarg, tmp) == -1)
1902 				err(1, "parse_csslike");
1903 			for (i = 0; i < 4; ++i)
1904 				r.p_borders[i] = parse_integer(tmp[i], 0);
1905 			break;
1906 		case 'I':
1907 			if (parse_csslike(optarg, tmp) == -1)
1908 				err(1, "parse_csslike");
1909 			for (i = 0; i < 4; ++i)
1910 				c_borders_bg[i] = parse_color(tmp[i], "#000");
1911 			break;
1912 		case 'i':
1913 			if (parse_csslike(optarg, tmp) == -1)
1914 				err(1, "parse_csslike");
1915 			for (i = 0; i < 4; ++i)
1916 				r.c_borders[i] = parse_integer(tmp[i], 0);
1917 			break;
1918 		case 'J':
1919 			if (parse_csslike(optarg, tmp) == -1)
1920 				err(1, "parse_csslike");
1921 			for (i = 0; i < 4; ++i)
1922 				ch_borders_bg[i] = parse_color(tmp[i], "#000");
1923 			break;
1924 		case 'j':
1925 			if (parse_csslike(optarg, tmp) == -1)
1926 				err(1, "parse_csslike");
1927 			for (i = 0; i < 4; ++i)
1928 				r.ch_borders[i] = parse_integer(tmp[i], 0);
1929 			break;
1930 		case 'l':
1931 			r.horizontal_layout = !strcmp(optarg, "horizontal");
1932 			break;
1933 		case 'f': {
1934 			char *newfont;
1935 			if ((newfont = strdup(optarg)) != NULL) {
1936 				free(fontname);
1937 				fontname = newfont;
1938 			}
1939 			break;
1940 		}
1941 		case 'W':
1942 			r.width = parse_int_with_percentage(optarg, r.width, d_width);
1943 			break;
1944 		case 'H':
1945 			r.height = parse_int_with_percentage(optarg, r.height, d_height);
1946 			break;
1947 		case 'b':
1948 			if (parse_csslike(optarg, tmp) == -1)
1949 				err(1, "parse_csslike");
1950 			for (i = 0; i < 4; ++i)
1951 				r.borders[i] = parse_integer(tmp[i], 0);
1952 			break;
1953 		case 'B':
1954 			if (parse_csslike(optarg, tmp) == -1)
1955 				err(1, "parse_csslike");
1956 			for (i = 0; i < 4; ++i)
1957 				borders_bg[i] = parse_color(tmp[i], "#000");
1958 			break;
1959 		case 't':
1960 			fgs[0] = parse_color(optarg, NULL);
1961 			break;
1962 		case 'T':
1963 			bgs[0] = parse_color(optarg, NULL);
1964 			break;
1965 		case 'c':
1966 			fgs[1] = parse_color(optarg, NULL);
1967 			break;
1968 		case 'C':
1969 			bgs[1] = parse_color(optarg, NULL);
1970 			break;
1971 		case 's':
1972 			fgs[2] = parse_color(optarg, NULL);
1973 			break;
1974 		case 'S':
1975 			bgs[2] = parse_color(optarg, NULL);
1976 			break;
1977 		default:
1978 			fprintf(stderr, "Unrecognized option %c\n", ch);
1979 			status = ERR;
1980 			break;
1981 		}
1982 	}
1983 
1984 	if (r.height < 0 || r.width < 0 || x < 0 || y < 0) {
1985 		fprintf(stderr, "height, width, x or y are lesser than 0.");
1986 		status = ERR;
1987 	}
1988 
1989 	/* since only now we know if the first should be selected,
1990 	 * update the completion here */
1991 	update_completions(cs, text, lines, vlines, r.first_selected);
1992 
1993 	/* update the prompt lenght, only now we surely know the length of it
1994 	 */
1995 	r.ps1len = strlen(r.ps1);
1996 
1997 	/* Create the window */
1998 	create_window(&r, parent_window, cmap, vinfo, x, y, offset_x, offset_y, bgs[1]);
1999 	set_win_atoms_hints(r.d, r.w, r.width, r.height);
2000 	XMapRaised(r.d, r.w);
2001 
2002 	/* If embed, listen for other events as well */
2003 	if (embed) {
2004 		Window *children, parent, root;
2005 		unsigned int children_no;
2006 
2007 		XSelectInput(r.d, parent_window, FocusChangeMask);
2008 		if (XQueryTree(r.d, parent_window, &root, &parent, &children, &children_no)
2009 			&& children) {
2010 			for (i = 0; i < children_no && children[i] != r.w; ++i)
2011 				XSelectInput(r.d, children[i], FocusChangeMask);
2012 			XFree(children);
2013 		}
2014 		grabfocus(r.d, r.w);
2015 	}
2016 
2017 	take_keyboard(r.d, r.w);
2018 
2019 	r.x_zero = r.borders[3];
2020 	r.y_zero = r.borders[0];
2021 
2022 	{
2023 		XGCValues values;
2024 
2025 		for (i = 0; i < 3; ++i) {
2026 			r.fgs[i] = XCreateGC(r.d, r.w, 0, &values);
2027 			r.bgs[i] = XCreateGC(r.d, r.w, 0, &values);
2028 		}
2029 
2030 		for (i = 0; i < 4; ++i) {
2031 			r.borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2032 			r.p_borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2033 			r.c_borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2034 			r.ch_borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2035 		}
2036 	}
2037 
2038 	/* Load the colors in our GCs */
2039 	for (i = 0; i < 3; ++i) {
2040 		XSetForeground(r.d, r.fgs[i], fgs[i]);
2041 		XSetForeground(r.d, r.bgs[i], bgs[i]);
2042 	}
2043 
2044 	for (i = 0; i < 4; ++i) {
2045 		XSetForeground(r.d, r.borders_bg[i], borders_bg[i]);
2046 		XSetForeground(r.d, r.p_borders_bg[i], p_borders_bg[i]);
2047 		XSetForeground(r.d, r.c_borders_bg[i], c_borders_bg[i]);
2048 		XSetForeground(r.d, r.ch_borders_bg[i], ch_borders_bg[i]);
2049 	}
2050 
2051 	if (load_font(&r, fontname) == -1)
2052 		status = ERR;
2053 
2054 	r.xftdraw = XftDrawCreate(r.d, r.w, vinfo.visual, cmap);
2055 
2056 	for (i = 0; i < 3; ++i) {
2057 		rgba_t c;
2058 		XRenderColor xrcolor;
2059 
2060 		c = *(rgba_t *)&fgs[i];
2061 		xrcolor.red = EXPANDBITS(c.rgba.r);
2062 		xrcolor.green = EXPANDBITS(c.rgba.g);
2063 		xrcolor.blue = EXPANDBITS(c.rgba.b);
2064 		xrcolor.alpha = EXPANDBITS(c.rgba.a);
2065 		XftColorAllocValue(r.d, vinfo.visual, cmap, &xrcolor, &r.xft_colors[i]);
2066 	}
2067 
2068 	/* compute prompt dimensions */
2069 	ps1extents(&r);
2070 
2071 	xim_init(&r, &xdb);
2072 
2073 #ifdef __OpenBSD__
2074 	if (pledge("stdio", "") == -1)
2075 		err(1, "pledge");
2076 #endif
2077 
2078 	/* Cache text height */
2079 	text_extents("fyjpgl", 6, &r, NULL, &r.text_height);
2080 
2081 	/* Draw the window for the first time */
2082 	draw(&r, text, cs);
2083 
2084 	/* Main loop */
2085 	while (status == LOOPING || status == OK_LOOP) {
2086 		status = loop(&r, &text, &textlen, cs, lines, vlines);
2087 
2088 		if (status != ERR)
2089 			printf("%s\n", text);
2090 
2091 		if (!r.multiple_select && status == OK_LOOP)
2092 			status = OK;
2093 	}
2094 
2095 	XUngrabKeyboard(r.d, CurrentTime);
2096 
2097 	for (i = 0; i < 3; ++i)
2098 		XftColorFree(r.d, vinfo.visual, cmap, &r.xft_colors[i]);
2099 
2100 	for (i = 0; i < 3; ++i) {
2101 		XFreeGC(r.d, r.fgs[i]);
2102 		XFreeGC(r.d, r.bgs[i]);
2103 	}
2104 
2105 	for (i = 0; i < 4; ++i) {
2106 		XFreeGC(r.d, r.borders_bg[i]);
2107 		XFreeGC(r.d, r.p_borders_bg[i]);
2108 		XFreeGC(r.d, r.c_borders_bg[i]);
2109 		XFreeGC(r.d, r.ch_borders_bg[i]);
2110 	}
2111 
2112 	XDestroyIC(r.xic);
2113 	XCloseIM(r.xim);
2114 
2115 	for (i = 0; i < 3; ++i)
2116 		XftColorFree(r.d, vinfo.visual, cmap, &r.xft_colors[i]);
2117 	XftFontClose(r.d, r.font);
2118 	XftDrawDestroy(r.xftdraw);
2119 
2120 	free(r.ps1);
2121 	free(fontname);
2122 	free(text);
2123 
2124 	free(lines);
2125 	free(vlines);
2126 	compls_delete(cs);
2127 
2128 	XFreeColormap(r.d, cmap);
2129 
2130 	XDestroyWindow(r.d, r.w);
2131 	XCloseDisplay(r.d);
2132 
2133 	return status != OK;
2134 }