Blob


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 */
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>
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>
36 #include <X11/extensions/Xinerama.h>
38 #define RESNAME "MyMenu"
39 #define RESCLASS "mymenu"
41 #define SYM_BUF_SIZE 4
43 #define DEFFONT "monospace"
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:"
47 #define MIN(a, b) ((a) < (b) ? (a) : (b))
48 #define MAX(a, b) ((a) > (b) ? (a) : (b))
50 #define EXPANDBITS(x) (((x & 0xf0) * 0x100) | (x & 0x0f) * 0x10)
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])
55 /* The states of the event loop */
56 enum state { LOOPING, OK_LOOP, OK, ERR };
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 };
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 };
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 */
96 int offset; /* scroll offset */
98 short free_text;
99 short first_selected;
100 short multiple_select;
102 /* four border width */
103 int borders[4];
104 int p_borders[4];
105 int c_borders[4];
106 int ch_borders[4];
108 short horizontal_layout;
110 /* prompt */
111 char *ps1;
112 int ps1len;
113 int ps1w; /* ps1 width */
114 int ps1h; /* ps1 height */
116 int text_height; /* cache for the vertical layout */
118 XIC xic;
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 };
132 struct completion {
133 char *completion;
134 char *rcompletion;
136 /*
137 * The X (or Y, depending on the layour) at which the item is
138 * rendered
139 */
140 int offset;
141 };
143 /* Wrap the linked list of completions */
144 struct completions {
145 struct completion *completions;
146 ssize_t selected;
147 size_t length;
148 };
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;
161 /* Return a newly allocated (and empty) completion list */
162 struct completions *
163 compls_new(size_t length)
165 struct completions *cs = malloc(sizeof(struct completions));
167 if (cs == NULL)
168 return cs;
170 cs->completions = calloc(length, sizeof(struct completion));
171 if (cs->completions == NULL) {
172 free(cs);
173 return NULL;
176 cs->selected = -1;
177 cs->length = length;
178 return cs;
181 /* Delete the wrapper and the whole list */
182 void
183 compls_delete(struct completions *cs)
185 if (cs == NULL)
186 return;
188 free(cs->completions);
189 free(cs);
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)
200 size_t index = 0;
201 size_t matching = 0;
202 char *l;
204 if (vlines == NULL)
205 vlines = lines;
207 while (1) {
208 if (lines[index] == NULL)
209 break;
211 l = vlines[index] != NULL ? vlines[index] : lines[index];
213 if (strcasestr(l, text) != NULL) {
214 struct completion *c = &cs->completions[matching];
215 c->completion = l;
216 c->rcompletion = lines[index];
217 matching++;
220 index++;
222 cs->length = matching;
223 cs->selected = -1;
226 /* Update the given completion */
227 void
228 update_completions(struct completions *cs, char *text, char **lines,
229 char **vlines, short first_selected)
231 filter(cs, text, lines, vlines);
232 if (first_selected && cs->length > 0)
233 cs->selected = 0;
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)
246 struct completion *n;
247 int index;
249 if (cs == NULL || cs->length == 0)
250 return;
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;
266 *textlen = strlen(*text);
267 return;
270 index = cs->selected;
272 if (index == -1 && p)
273 index = 0;
274 index = cs->selected = (cs->length + (p ? index - 1 : index + 1))
275 % cs->length;
277 n = &cs->completions[cs->selected];
279 free(*text);
280 *text = strdup(n->completion);
281 if (text == NULL) {
282 fprintf(stderr, "Memory allocation error!\n");
283 *status = ERR;
284 return;
286 *textlen = strlen(*text);
289 /* Push the character c at the end of the string pointed by p */
290 int
291 pushc(char **p, int maxlen, char c)
293 int len;
295 len = strnlen(*p, maxlen);
296 if (!(len < maxlen - 2)) {
297 char *newptr;
299 maxlen += maxlen >> 1;
300 newptr = realloc(*p, maxlen);
301 if (newptr == NULL) /* bad */
302 return -1;
303 *p = newptr;
306 (*p)[len] = c;
307 (*p)[len + 1] = '\0';
308 return maxlen;
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)
319 int len = strlen(p);
320 char *e;
322 if (len == 0)
323 return p;
325 e = p + len - 1;
327 do {
328 char c = *e;
330 *e = '\0';
331 e -= 1;
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);
341 return e;
344 /* Remove the last word plus trailing white spaces from the given string */
345 void
346 popw(char *w)
348 int len;
349 short in_word = 1;
351 if ((len = strlen(w)) == 0)
352 return;
354 while (1) {
355 char *e = popc(w);
357 if (e < w)
358 return;
360 if (in_word && isspace(*e))
361 in_word = 0;
363 if (!in_word && !isspace(*e))
364 return;
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)
375 int len, p;
376 char *s;
378 if ((len = strlen(str)) == 0)
379 return NULL;
381 if ((s = calloc(len, sizeof(char))) == NULL)
382 err(1, "calloc");
383 p = 0;
385 while (*str) {
386 char c = *str;
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;
397 if (c == '"') {
398 str++; /* skip only this char */
399 continue;
401 s[p] = c;
402 p++;
403 str++;
406 return s;
409 char **
410 readlines(size_t *lineslen)
412 size_t len = 0, cap = 0;
413 size_t linesize = 0;
414 ssize_t linelen;
415 char *line = NULL, **lines = NULL;
417 while ((linelen = getline(&line, &linesize, stdin)) != -1) {
418 if (linelen != 0 && line[linelen-1] == '\n')
419 line[linelen-1] = '\0';
421 if (len == cap) {
422 size_t newcap;
423 void *t;
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;
433 if ((lines[len++] = strdup(line)) == NULL)
434 err(1, "strdup");
437 if (ferror(stdin))
438 err(1, "getline");
439 free(line);
441 *lineslen = len;
442 return lines;
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)
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;
459 if (ret_width != NULL)
460 *ret_width = width;
461 if (ret_height != NULL)
462 *ret_height = height;
463 return width;
466 void
467 draw_string(char *str, int len, int x, int y, struct rendering *r,
468 enum obj_type tt)
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];
478 XftDrawStringUtf8(r->xftdraw, &xftcolor, r->font, x, y, str, len);
481 /* Duplicate the string and substitute every space with a 'n` */
482 char *
483 strdupn(char *str)
485 char *t, *dup;
487 if (str == NULL || *str == '\0')
488 return NULL;
490 if ((dup = strdup(str)) == NULL)
491 return NULL;
493 for (t = dup; *t; ++t) {
494 if (*t == ' ')
495 *t = 'n';
498 return dup;
501 int
502 draw_v_box(struct rendering *r, int y, char *prefix, int prefix_width,
503 enum obj_type t, char *text)
505 GC *border_color, bg;
506 int *padding, *borders;
507 int ret = 0, inner_width, inner_height, x;
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;
530 ret = borders[0] + padding[0] + r->text_height + padding[2] + borders[2];
532 inner_width = INNER_WIDTH(r) - borders[1] - borders[3];
533 inner_height = padding[0] + r->text_height + padding[2];
535 /* Border top */
536 XFillRectangle(r->d, r->w, border_color[0], r->x_zero, y, r->width,
537 borders[0]);
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);
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]);
548 /* Border left */
549 XFillRectangle(r->d, r->w, border_color[3], r->x_zero, y, borders[3],
550 ret);
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);
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;
564 draw_string(text, strlen(text), x, y, r, t);
566 return ret;
569 int
570 draw_h_box(struct rendering *r, int x, char *prefix, int prefix_width,
571 enum obj_type t, char *text)
573 GC *border_color, bg;
574 int *padding, *borders;
575 int ret = 0, inner_width, inner_height, y, text_width;
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;
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;
603 padding[2] = padding[0];
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;
610 /* Get the text width */
611 text_extents(text, strlen(text), r, &text_width, NULL);
612 if (prefix != NULL)
613 text_width += prefix_width;
615 ret = borders[3] + padding[3] + text_width + padding[1] + borders[1];
617 inner_width = padding[3] + text_width + padding[1];
618 inner_height = INNER_HEIGHT(r) - borders[0] - borders[2];
620 /* Border top */
621 XFillRectangle(r->d, r->w, border_color[0], x, r->y_zero, ret,
622 borders[0]);
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));
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]);
634 /* Border left */
635 XFillRectangle(r->d, r->w, border_color[3], x, r->y_zero, borders[3],
636 INNER_HEIGHT(r));
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);
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;
650 draw_string(text, strlen(text), x, y, r, t);
652 return ret;
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)
663 size_t i;
664 int x = r->x_zero;
666 /* Draw the prompt */
667 x += draw_h_box(r, x, r->ps1, r->ps1w, PROMPT, text);
669 for (i = r->offset; i < cs->length; ++i) {
670 enum obj_type t;
672 if (cs->selected == (ssize_t)i)
673 t = COMPL_HIGH;
674 else
675 t = COMPL;
677 cs->completions[i].offset = x;
679 x += draw_h_box(r, x, NULL, 0, t,
680 cs->completions[i].completion);
682 if (x > INNER_WIDTH(r))
683 break;
686 for (i += 1; i < cs->length; ++i)
687 cs->completions[i].offset = -1;
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)
702 size_t i;
703 int y = r->y_zero;
705 y += draw_v_box(r, y, r->ps1, r->ps1w, PROMPT, text);
707 for (i = r->offset; i < cs->length; ++i) {
708 enum obj_type t;
710 if (cs->selected == (ssize_t)i)
711 t = COMPL_HIGH;
712 else
713 t = COMPL;
715 cs->completions[i].offset = y;
717 y += draw_v_box(r, y, NULL, 0, t,
718 cs->completions[i].completion);
720 if (y > INNER_HEIGHT(r))
721 break;
724 for (i += 1; i < cs->length; ++i)
725 cs->completions[i].offset = -1;
728 void
729 draw(struct rendering *r, char *text, struct completions *cs)
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));
735 /* Draw the contents */
736 if (r->horizontal_layout)
737 draw_horizontally(r, text, cs);
738 else
739 draw_vertically(r, text, cs);
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]);
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);
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]);
755 if (r->borders[3] != 0)
756 XFillRectangle(r->d, r->w, r->borders_bg[3], 0, 0,
757 r->borders[3], r->height);
759 /* render! */
760 XFlush(r->d);
763 /* Set some WM stuff */
764 void
765 set_win_atoms_hints(Display *d, Window w, int width, int height)
767 Atom type;
768 XClassHint *class_hint;
769 XSizeHints *size_hint;
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);
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);
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);
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);
794 class_hint->res_name = RESNAME;
795 class_hint->res_class = RESCLASS;
796 XSetClassHint(d, w, class_hint);
797 XFree(class_hint);
799 size_hint = XAllocSizeHints();
800 if (size_hint == NULL) {
801 fprintf(stderr, "Could not allocate memory for size hint\n");
802 exit(EX_UNAVAILABLE);
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;
811 XFlush(d);
814 /* Get the width and height of the window `w' */
815 void
816 get_wh(Display *d, Window *w, int *width, int *height)
818 XWindowAttributes win_attr;
820 XGetWindowAttributes(d, *w, &win_attr);
821 *height = win_attr.height;
822 *width = win_attr.width;
825 /* find the current xinerama monitor if possible */
826 void
827 findmonitor(Display *d, int *x, int *y, int *width, int *height)
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;
837 if (!XineramaIsActive(d))
838 return;
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;
849 if (!res)
850 return;
852 /* Now find in which monitor the mice is */
853 info = XineramaQueryScreens(d, &monitors);
854 if (info == NULL)
855 return;
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;
870 XFree(info);
873 int
874 grabfocus(Display *d, Window w)
876 int i;
877 for (i = 0; i < 100; ++i) {
878 Window focuswin;
879 int revert_to_win;
881 XGetInputFocus(d, &focuswin, &revert_to_win);
883 if (focuswin == w)
884 return 1;
886 XSetInputFocus(d, w, RevertToParent, CurrentTime);
887 usleep(1000);
889 return 0;
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)
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);
907 fprintf(stderr, "Cannot grab keyboard\n");
908 return 0;
911 unsigned long
912 parse_color(const char *str, const char *def)
914 size_t len;
915 rgba_t tmp;
916 char *ep;
918 if (str == NULL)
919 goto err;
921 len = strlen(str);
923 /* +1 for the # ath the start */
924 if (*str != '#' || len > 9 || len < 4)
925 goto err;
926 ++str; /* skip the # */
928 errno = 0;
929 tmp = (rgba_t)(uint32_t)strtoul(str, &ep, 16);
931 if (errno)
932 goto err;
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 */
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;
953 return 0U;
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;
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)
969 const char *errstr;
970 int i;
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;
978 return i;
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)
988 int len = strlen(str);
990 if (len > 0 && str[len - 1] == '%') {
991 int val;
992 char *cpy;
994 if ((cpy = strdup(str)) == NULL)
995 err(1, "strdup");
997 cpy[len - 1] = '\0';
998 val = parse_integer(cpy, default_value);
999 free(cpy);
1000 return val * max / 100;
1003 return parse_integer(str, default_value);
1006 void
1007 get_mouse_coords(Display *d, int *x, int *y)
1009 Window w, root;
1010 int i;
1011 unsigned int u;
1013 *x = *y = 0;
1014 root = DefaultRootWindow(d);
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;
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
1033 int
1034 parse_int_with_pos(Display *d, const char *str, int default_value, int max,
1035 int self)
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;
1046 get_mouse_coords(d, &x, &y);
1047 if (!strcmp(str, "mx"))
1048 return x - 1;
1049 else
1050 return y - 1;
1052 return parse_int_with_percentage(str, default_value, max);
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)
1060 int i, j;
1061 char *s, *token;
1062 short any_null;
1064 memset(ret, 0, 4 * sizeof(*ret));
1066 s = strdup(str);
1067 if (s == NULL)
1068 return -1;
1070 for (i = 0; (token = strsep(&s, " ")) != NULL && i < 4; ++i)
1071 ret[i] = strdup(token);
1073 if (i == 1)
1074 for (j = 1; j < 4; j++)
1075 ret[j] = strdup(ret[0]);
1077 if (i == 2) {
1078 ret[2] = strdup(ret[0]);
1079 ret[3] = strdup(ret[1]);
1082 if (i == 3)
1083 ret[3] = strdup(ret[1]);
1086 * before we didn't check for the return type of strdup, here
1087 * we will
1090 any_null = 0;
1091 for (i = 0; i < 4; ++i)
1092 any_null = ret[i] == NULL || any_null;
1094 if (any_null)
1095 for (i = 0; i < 4; ++i)
1096 free(ret[i]);
1098 if (i == 0 || any_null) {
1099 free(s);
1100 return -1;
1103 return 1;
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.
1111 enum action
1112 parse_event(Display *d, XKeyPressedEvent *ev, XIC xic, char **input)
1114 char str[SYM_BUF_SIZE] = { 0 };
1115 Status s;
1117 if (ev->keycode == XKeysymToKeycode(d, XK_BackSpace))
1118 return DEL_CHAR;
1120 if (ev->keycode == XKeysymToKeycode(d, XK_Tab))
1121 return ev->state & ShiftMask ? PREV_COMPL : NEXT_COMPL;
1123 if (ev->keycode == XKeysymToKeycode(d, XK_Return))
1124 return CONFIRM;
1126 if (ev->keycode == XKeysymToKeycode(d, XK_Escape))
1127 return EXIT;
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;
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;
1158 *input = strdup(str);
1159 if (*input == NULL) {
1160 fprintf(stderr, "Error while allocating memory for key.\n");
1161 return EXIT;
1164 return ADD_CHAR;
1167 void
1168 confirm(enum state *status, struct rendering *r, struct completions *cs,
1169 char **text, int *textlen)
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;
1177 while (1) {
1178 if (index == 0)
1179 break;
1180 c++;
1181 index--;
1184 t = c->rcompletion;
1185 free(*text);
1186 *text = strdup(t);
1188 if (*text == NULL) {
1189 fprintf(stderr, "Memory allocation error\n");
1190 *status = ERR;
1193 *textlen = strlen(*text);
1194 return;
1197 if (!r->free_text) /* cannot accept arbitrary text */
1198 *status = LOOPING;
1202 * cs: completion list
1203 * offset: the offset of the click
1204 * first: the first (rendered) item
1205 * def: the default action
1207 enum action
1208 select_clicked(struct completions *cs, size_t offset, size_t first,
1209 enum action def)
1211 ssize_t selected = first;
1212 int set = 0;
1214 if (cs->length == 0)
1215 return NO_OP;
1217 if (offset < cs->completions[selected].offset)
1218 return EXIT;
1220 /* skip the first entry */
1221 for (selected += 1; selected < cs->length; ++selected) {
1222 if (cs->completions[selected].offset == -1)
1223 break;
1225 if (offset < cs->completions[selected].offset) {
1226 cs->selected = selected - 1;
1227 set = 1;
1228 break;
1232 if (!set)
1233 cs->selected = selected - 1;
1235 return def;
1238 enum action
1239 handle_mouse(struct rendering *r, struct completions *cs,
1240 XButtonPressedEvent *e)
1242 size_t off;
1244 if (r->horizontal_layout)
1245 off = e->x;
1246 else
1247 off = e->y;
1249 switch (e->button) {
1250 case Button1:
1251 return select_clicked(cs, off, r->offset, CONFIRM);
1253 case Button3:
1254 return select_clicked(cs, off, r->offset, CONFIRM_CONTINUE);
1256 case Button4:
1257 return SCROLL_UP;
1259 case Button5:
1260 return SCROLL_DOWN;
1263 return NO_OP;
1266 /* event loop */
1267 enum state
1268 loop(struct rendering *r, char **text, int *textlen, struct completions *cs,
1269 char **lines, char **vlines)
1271 enum state status = LOOPING;
1273 while (status == LOOPING) {
1274 XEvent e;
1275 XNextEvent(r->d, &e);
1277 if (XFilterEvent(&e, r->w))
1278 continue;
1280 switch (e.type) {
1281 case KeymapNotify:
1282 XRefreshKeyboardMapping(&e.xmapping);
1283 break;
1285 case FocusIn:
1286 /* Re-grab focus */
1287 if (e.xfocus.window != r->w)
1288 grabfocus(r->d, r->w);
1289 break;
1291 case VisibilityNotify:
1292 if (e.xvisibility.state != VisibilityUnobscured)
1293 XRaiseWindow(r->d, r->w);
1294 break;
1296 case MapNotify:
1297 get_wh(r->d, &r->w, &r->width, &r->height);
1298 draw(r, *text, cs);
1299 break;
1301 case KeyPress:
1302 case ButtonPress: {
1303 enum action a;
1304 char *input = NULL;
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);
1313 switch (a) {
1314 case NO_OP:
1315 break;
1317 case EXIT:
1318 status = ERR;
1319 break;
1321 case CONFIRM: {
1322 status = OK;
1323 confirm(&status, r, cs, text, textlen);
1324 break;
1327 case CONFIRM_CONTINUE: {
1328 status = OK_LOOP;
1329 confirm(&status, r, cs, text, textlen);
1330 break;
1333 case PREV_COMPL: {
1334 complete(cs, r->first_selected, 1, text,
1335 textlen, &status);
1336 r->offset = cs->selected;
1337 break;
1340 case NEXT_COMPL: {
1341 complete(cs, r->first_selected, 0, text,
1342 textlen, &status);
1343 r->offset = cs->selected;
1344 break;
1347 case DEL_CHAR:
1348 popc(*text);
1349 update_completions(cs, *text, lines, vlines,
1350 r->first_selected);
1351 r->offset = 0;
1352 break;
1354 case DEL_WORD: {
1355 popw(*text);
1356 update_completions(cs, *text, lines, vlines,
1357 r->first_selected);
1358 break;
1361 case DEL_LINE: {
1362 int i;
1363 for (i = 0; i < *textlen; ++i)
1364 (*text)[i] = 0;
1365 update_completions(cs, *text, lines, vlines,
1366 r->first_selected);
1367 r->offset = 0;
1368 break;
1371 case ADD_CHAR: {
1372 int str_len, i;
1374 str_len = strlen(input);
1377 * sometimes a strange key is pressed
1378 * i.e. ctrl alone), so input will be
1379 * empty. Don't need to update
1380 * completion in that case
1382 if (str_len == 0)
1383 break;
1385 for (i = 0; i < str_len; ++i) {
1386 *textlen = pushc(text, *textlen,
1387 input[i]);
1388 if (*textlen == -1) {
1389 fprintf(stderr,
1390 "Memory allocation "
1391 "error\n");
1392 status = ERR;
1393 break;
1397 if (status != ERR) {
1398 update_completions(cs, *text, lines,
1399 vlines, r->first_selected);
1400 free(input);
1403 r->offset = 0;
1404 break;
1407 case TOGGLE_FIRST_SELECTED:
1408 r->first_selected = !r->first_selected;
1409 if (r->first_selected && cs->selected < 0)
1410 cs->selected = 0;
1411 if (!r->first_selected && cs->selected == 0)
1412 cs->selected = -1;
1413 break;
1415 case SCROLL_DOWN:
1416 r->offset = MIN(r->offset + 1, cs->length - 1);
1417 break;
1419 case SCROLL_UP:
1420 r->offset = MAX((ssize_t)r->offset - 1, 0);
1421 break;
1426 draw(r, *text, cs);
1429 return status;
1432 int
1433 load_font(struct rendering *r, const char *fontname)
1435 r->font = XftFontOpenName(r->d, DefaultScreen(r->d), fontname);
1436 return 0;
1439 void
1440 xim_init(struct rendering *r, XrmDatabase *xdb)
1442 XIMStyle best_match_style;
1443 XIMStyles *xis;
1444 int i;
1446 /* Open the X input method */
1447 if ((r->xim = XOpenIM(r->d, *xdb, RESNAME, RESCLASS)) == NULL)
1448 err(1, "XOpenIM");
1450 if (XGetIMValues(r->xim, XNQueryInputStyle, &xis, NULL) || !xis) {
1451 fprintf(stderr, "Input Styles could not be retrieved\n");
1452 exit(EX_UNAVAILABLE);
1455 best_match_style = 0;
1456 for (i = 0; i < xis->count_styles; ++i) {
1457 XIMStyle ts = xis->supported_styles[i];
1458 if (ts == (XIMPreeditNothing | XIMStatusNothing)) {
1459 best_match_style = ts;
1460 break;
1463 XFree(xis);
1465 if (!best_match_style)
1466 fprintf(stderr,
1467 "No matching input style could be determined\n");
1469 r->xic = XCreateIC(r->xim, XNInputStyle, best_match_style,
1470 XNClientWindow, r->w, XNFocusWindow, r->w, NULL);
1471 if (r->xic == NULL)
1472 err(1, "XCreateIC");
1475 void
1476 create_window(struct rendering *r, Window parent_window, Colormap cmap,
1477 XVisualInfo vinfo, int x, int y, int ox, int oy,
1478 unsigned long background_pixel)
1480 XSetWindowAttributes attr;
1481 unsigned long vmask;
1483 /* Create the window */
1484 attr.colormap = cmap;
1485 attr.override_redirect = 1;
1486 attr.border_pixel = 0;
1487 attr.background_pixel = background_pixel;
1488 attr.event_mask = StructureNotifyMask | ExposureMask | KeyPressMask
1489 | KeymapStateMask | ButtonPress | VisibilityChangeMask;
1491 vmask = CWBorderPixel | CWBackPixel | CWColormap | CWEventMask |
1492 CWOverrideRedirect;
1494 r->w = XCreateWindow(r->d, parent_window, x + ox, y + oy, r->width, r->height, 0,
1495 vinfo.depth, InputOutput, vinfo.visual, vmask, &attr);
1498 void
1499 ps1extents(struct rendering *r)
1501 char *dup;
1502 dup = strdupn(r->ps1);
1503 text_extents(dup == NULL ? r->ps1 : dup, r->ps1len, r, &r->ps1w, &r->ps1h);
1504 free(dup);
1507 void
1508 usage(char *prgname)
1510 fprintf(stderr,
1511 "%s [-Aahmv] [-B colors] [-b size] [-C color] [-c color]\n"
1512 " [-d separator] [-e window] [-f font] [-G color] [-g "
1513 "size]\n"
1514 " [-H height] [-I color] [-i size] [-J color] [-j "
1515 "size] [-l layout]\n"
1516 " [-P padding] [-p prompt] [-S color] [-s color] [-T "
1517 "color]\n"
1518 " [-t color] [-W width] [-x coord] [-y coord]\n",
1519 prgname);
1522 int
1523 main(int argc, char **argv)
1525 struct completions *cs;
1526 struct rendering r;
1527 XVisualInfo vinfo;
1528 Colormap cmap;
1529 size_t nlines, i;
1530 Window parent_window;
1531 XrmDatabase xdb;
1532 unsigned long fgs[3], bgs[3]; /* prompt, compl, compl_highlighted */
1533 unsigned long borders_bg[4], p_borders_bg[4], c_borders_bg[4],
1534 ch_borders_bg[4]; /* N E S W */
1535 enum state status = LOOPING;
1536 int ch;
1537 int offset_x = 0, offset_y = 0;
1538 int x = 0, y = 0;
1539 int textlen, d_width, d_height;
1540 short embed;
1541 const char *sep = NULL;
1542 const char *parent_window_id = NULL;
1543 char *tmp[4];
1544 char **lines, **vlines;
1545 char *fontname, *text, *xrm;
1547 setlocale(LC_ALL, getenv("LANG"));
1549 for (i = 0; i < 4; ++i) {
1550 /* default paddings */
1551 r.p_padding[i] = 10;
1552 r.c_padding[i] = 10;
1553 r.ch_padding[i] = 10;
1555 /* default borders */
1556 r.borders[i] = 0;
1557 r.p_borders[i] = 0;
1558 r.c_borders[i] = 0;
1559 r.ch_borders[i] = 0;
1562 r.first_selected = 0;
1563 r.free_text = 1;
1564 r.multiple_select = 0;
1565 r.offset = 0;
1567 /* default width and height */
1568 r.width = 400;
1569 r.height = 20;
1572 * The prompt. We duplicate the string so later is easy to
1573 * free (in the case it's been overwritten by the user)
1575 if ((r.ps1 = strdup("$ ")) == NULL)
1576 err(1, "strdup");
1578 /* same for the font name */
1579 if ((fontname = strdup(DEFFONT)) == NULL)
1580 err(1, "strdup");
1582 while ((ch = getopt(argc, argv, ARGS)) != -1) {
1583 switch (ch) {
1584 case 'h': /* help */
1585 usage(*argv);
1586 return 0;
1587 case 'v': /* version */
1588 fprintf(stderr, "%s version: %s\n", *argv, VERSION);
1589 return 0;
1590 case 'e': /* embed */
1591 if ((parent_window_id = strdup(optarg)) == NULL)
1592 err(1, "strdup");
1593 break;
1594 case 'd':
1595 if ((sep = strdup(optarg)) == NULL)
1596 err(1, "strdup");
1597 break;
1598 case 'A':
1599 r.free_text = 0;
1600 break;
1601 case 'm':
1602 r.multiple_select = 1;
1603 break;
1604 default:
1605 break;
1609 lines = readlines(&nlines);
1611 vlines = NULL;
1612 if (sep != NULL) {
1613 int l;
1614 l = strlen(sep);
1615 if ((vlines = calloc(nlines, sizeof(char *))) == NULL)
1616 err(1, "calloc");
1618 for (i = 0; i < nlines; i++) {
1619 char *t;
1620 t = strstr(lines[i], sep);
1621 if (t == NULL)
1622 vlines[i] = lines[i];
1623 else
1624 vlines[i] = t + l;
1628 textlen = 10;
1629 if ((text = malloc(textlen * sizeof(char))) == NULL)
1630 err(1, "malloc");
1632 /* struct completions *cs = filter(text, lines); */
1633 if ((cs = compls_new(nlines)) == NULL)
1634 err(1, "compls_new");
1636 /* start talking to xorg */
1637 r.d = XOpenDisplay(NULL);
1638 if (r.d == NULL) {
1639 fprintf(stderr, "Could not open display!\n");
1640 return EX_UNAVAILABLE;
1643 embed = 1;
1644 if (!(parent_window_id && (parent_window = strtol(parent_window_id, NULL, 0)))) {
1645 parent_window = DefaultRootWindow(r.d);
1646 embed = 0;
1649 /* get display size */
1650 get_wh(r.d, &parent_window, &d_width, &d_height);
1652 if (!embed)
1653 findmonitor(r.d, &offset_x, &offset_y, &d_width, &d_height);
1655 XMatchVisualInfo(r.d, DefaultScreen(r.d), 32, TrueColor, &vinfo);
1656 cmap = XCreateColormap(r.d, XDefaultRootWindow(r.d), vinfo.visual,
1657 AllocNone);
1659 fgs[0] = fgs[1] = parse_color("#fff", NULL);
1660 fgs[2] = parse_color("#000", NULL);
1662 bgs[0] = bgs[1] = parse_color("#000", NULL);
1663 bgs[2] = parse_color("#fff", NULL);
1665 borders_bg[0] = borders_bg[1] = borders_bg[2] = borders_bg[3] =
1666 parse_color("#000", NULL);
1668 p_borders_bg[0] = p_borders_bg[1] = p_borders_bg[2] = p_borders_bg[3]
1669 = parse_color("#000", NULL);
1670 c_borders_bg[0] = c_borders_bg[1] = c_borders_bg[2] = c_borders_bg[3]
1671 = parse_color("#000", NULL);
1672 ch_borders_bg[0] = ch_borders_bg[1] = ch_borders_bg[2] = ch_borders_bg[3]
1673 = parse_color("#000", NULL);
1675 r.horizontal_layout = 1;
1677 /* Read the resources */
1678 XrmInitialize();
1679 xrm = XResourceManagerString(r.d);
1680 xdb = NULL;
1681 if (xrm != NULL) {
1682 XrmValue value;
1683 char *datatype[20];
1685 xdb = XrmGetStringDatabase(xrm);
1687 if (XrmGetResource(xdb, "MyMenu.font", "*", datatype, &value)) {
1688 free(fontname);
1689 if ((fontname = strdup(value.addr)) == NULL)
1690 err(1, "strdup");
1691 } else {
1692 fprintf(stderr, "no font defined, using %s\n", fontname);
1695 if (XrmGetResource(xdb, "MyMenu.layout", "*", datatype, &value))
1696 r.horizontal_layout = !strcmp(value.addr, "horizontal");
1697 else
1698 fprintf(stderr, "no layout defined, using horizontal\n");
1700 if (XrmGetResource(xdb, "MyMenu.prompt", "*", datatype, &value)) {
1701 free(r.ps1);
1702 r.ps1 = normalize_str(value.addr);
1703 } else {
1704 fprintf(stderr,
1705 "no prompt defined, using \"%s\" as "
1706 "default\n",
1707 r.ps1);
1710 if (XrmGetResource(xdb, "MyMenu.prompt.border.size", "*", datatype, &value)) {
1711 if (parse_csslike(value.addr, tmp) == -1)
1712 err(1, "parse_csslike");
1713 for (i = 0; i < 4; ++i) {
1714 r.p_borders[i] = parse_integer(tmp[i], 0);
1715 free(tmp[i]);
1719 if (XrmGetResource(xdb, "MyMenu.prompt.border.color", "*", datatype, &value)) {
1720 if (parse_csslike(value.addr, tmp) == -1)
1721 err(1, "parse_csslike");
1723 for (i = 0; i < 4; ++i) {
1724 p_borders_bg[i] = parse_color(tmp[i], "#000");
1725 free(tmp[i]);
1729 if (XrmGetResource(xdb, "MyMenu.prompt.padding", "*", datatype, &value)) {
1730 if (parse_csslike(value.addr, tmp) == -1)
1731 err(1, "parse_csslike");
1733 for (i = 0; i < 4; ++i) {
1734 r.p_padding[i] = parse_integer(tmp[i], 0);
1735 free(tmp[i]);
1739 if (XrmGetResource(xdb, "MyMenu.width", "*", datatype, &value))
1740 r.width = parse_int_with_percentage(value.addr, r.width, d_width);
1741 else
1742 fprintf(stderr, "no width defined, using %d\n", r.width);
1744 if (XrmGetResource(xdb, "MyMenu.height", "*", datatype, &value))
1745 r.height = parse_int_with_percentage(value.addr, r.height, d_height);
1746 else
1747 fprintf(stderr, "no height defined, using %d\n", r.height);
1749 if (XrmGetResource(xdb, "MyMenu.x", "*", datatype, &value))
1750 x = parse_int_with_pos(r.d, value.addr, x, d_width, r.width);
1752 if (XrmGetResource(xdb, "MyMenu.y", "*", datatype, &value))
1753 y = parse_int_with_pos(r.d, value.addr, y, d_height, r.height);
1755 if (XrmGetResource(xdb, "MyMenu.border.size", "*", datatype, &value)) {
1756 if (parse_csslike(value.addr, tmp) == -1)
1757 err(1, "parse_csslike");
1759 for (i = 0; i < 4; ++i) {
1760 r.borders[i] = parse_int_with_percentage(tmp[i], 0,
1761 (i % 2) == 0 ? d_height : d_width);
1762 free(tmp[i]);
1766 /* Prompt */
1767 if (XrmGetResource(xdb, "MyMenu.prompt.foreground", "*", datatype, &value))
1768 fgs[0] = parse_color(value.addr, "#fff");
1770 if (XrmGetResource(xdb, "MyMenu.prompt.background", "*", datatype, &value))
1771 bgs[0] = parse_color(value.addr, "#000");
1773 /* Completions */
1774 if (XrmGetResource(xdb, "MyMenu.completion.foreground", "*", datatype, &value))
1775 fgs[1] = parse_color(value.addr, "#fff");
1777 if (XrmGetResource(xdb, "MyMenu.completion.background", "*", datatype, &value))
1778 bgs[1] = parse_color(value.addr, "#000");
1780 if (XrmGetResource(xdb, "MyMenu.completion.padding", "*", datatype, &value)) {
1781 if (parse_csslike(value.addr, tmp) == -1)
1782 err(1, "parse_csslike");
1784 for (i = 0; i < 4; ++i) {
1785 r.c_padding[i] = parse_integer(tmp[i], 0);
1786 free(tmp[i]);
1790 if (XrmGetResource(xdb, "MyMenu.completion.border.size", "*", datatype, &value)) {
1791 if (parse_csslike(value.addr, tmp) == -1)
1792 err(1, "parse_csslike");
1794 for (i = 0; i < 4; ++i) {
1795 r.c_borders[i] = parse_integer(tmp[i], 0);
1796 free(tmp[i]);
1800 if (XrmGetResource(xdb, "MyMenu.completion.border.color", "*", datatype, &value)) {
1801 if (parse_csslike(value.addr, tmp) == -1)
1802 err(1, "parse_csslike");
1804 for (i = 0; i < 4; ++i) {
1805 c_borders_bg[i] = parse_color(tmp[i], "#000");
1806 free(tmp[i]);
1810 /* Completion Highlighted */
1811 if (XrmGetResource(
1812 xdb, "MyMenu.completion_highlighted.foreground", "*", datatype, &value))
1813 fgs[2] = parse_color(value.addr, "#000");
1815 if (XrmGetResource(
1816 xdb, "MyMenu.completion_highlighted.background", "*", datatype, &value))
1817 bgs[2] = parse_color(value.addr, "#fff");
1819 if (XrmGetResource(
1820 xdb, "MyMenu.completion_highlighted.padding", "*", datatype, &value)) {
1821 if (parse_csslike(value.addr, tmp) == -1)
1822 err(1, "parse_csslike");
1824 for (i = 0; i < 4; ++i) {
1825 r.ch_padding[i] = parse_integer(tmp[i], 0);
1826 free(tmp[i]);
1830 if (XrmGetResource(xdb, "MyMenu.completion_highlighted.border.size", "*", datatype,
1831 &value)) {
1832 if (parse_csslike(value.addr, tmp) == -1)
1833 err(1, "parse_csslike");
1835 for (i = 0; i < 4; ++i) {
1836 r.ch_borders[i] = parse_integer(tmp[i], 0);
1837 free(tmp[i]);
1841 if (XrmGetResource(xdb, "MyMenu.completion_highlighted.border.color", "*", datatype,
1842 &value)) {
1843 if (parse_csslike(value.addr, tmp) == -1)
1844 err(1, "parse_csslike");
1846 for (i = 0; i < 4; ++i) {
1847 ch_borders_bg[i] = parse_color(tmp[i], "#000");
1848 free(tmp[i]);
1852 /* Border */
1853 if (XrmGetResource(xdb, "MyMenu.border.color", "*", datatype, &value)) {
1854 if (parse_csslike(value.addr, tmp) == -1)
1855 err(1, "parse_csslike");
1857 for (i = 0; i < 4; ++i) {
1858 borders_bg[i] = parse_color(tmp[i], "#000");
1859 free(tmp[i]);
1864 /* Second round of args parsing */
1865 optind = 0; /* reset the option index */
1866 while ((ch = getopt(argc, argv, ARGS)) != -1) {
1867 switch (ch) {
1868 case 'a':
1869 r.first_selected = 1;
1870 break;
1871 case 'A':
1872 /* free_text -- already catched */
1873 case 'd':
1874 /* separator -- this case was already catched */
1875 case 'e':
1876 /* embedding mymenu this case was already catched. */
1877 case 'm':
1878 /* multiple selection this case was already catched.
1880 break;
1881 case 'p': {
1882 char *newprompt;
1883 newprompt = strdup(optarg);
1884 if (newprompt != NULL) {
1885 free(r.ps1);
1886 r.ps1 = newprompt;
1888 break;
1890 case 'x':
1891 x = parse_int_with_pos(r.d, optarg, x, d_width, r.width);
1892 break;
1893 case 'y':
1894 y = parse_int_with_pos(r.d, optarg, y, d_height, r.height);
1895 break;
1896 case 'P':
1897 if (parse_csslike(optarg, tmp) == -1)
1898 err(1, "parse_csslike");
1899 for (i = 0; i < 4; ++i)
1900 r.p_padding[i] = parse_integer(tmp[i], 0);
1901 break;
1902 case 'G':
1903 if (parse_csslike(optarg, tmp) == -1)
1904 err(1, "parse_csslike");
1905 for (i = 0; i < 4; ++i)
1906 p_borders_bg[i] = parse_color(tmp[i], "#000");
1907 break;
1908 case 'g':
1909 if (parse_csslike(optarg, tmp) == -1)
1910 err(1, "parse_csslike");
1911 for (i = 0; i < 4; ++i)
1912 r.p_borders[i] = parse_integer(tmp[i], 0);
1913 break;
1914 case 'I':
1915 if (parse_csslike(optarg, tmp) == -1)
1916 err(1, "parse_csslike");
1917 for (i = 0; i < 4; ++i)
1918 c_borders_bg[i] = parse_color(tmp[i], "#000");
1919 break;
1920 case 'i':
1921 if (parse_csslike(optarg, tmp) == -1)
1922 err(1, "parse_csslike");
1923 for (i = 0; i < 4; ++i)
1924 r.c_borders[i] = parse_integer(tmp[i], 0);
1925 break;
1926 case 'J':
1927 if (parse_csslike(optarg, tmp) == -1)
1928 err(1, "parse_csslike");
1929 for (i = 0; i < 4; ++i)
1930 ch_borders_bg[i] = parse_color(tmp[i], "#000");
1931 break;
1932 case 'j':
1933 if (parse_csslike(optarg, tmp) == -1)
1934 err(1, "parse_csslike");
1935 for (i = 0; i < 4; ++i)
1936 r.ch_borders[i] = parse_integer(tmp[i], 0);
1937 break;
1938 case 'l':
1939 r.horizontal_layout = !strcmp(optarg, "horizontal");
1940 break;
1941 case 'f': {
1942 char *newfont;
1943 if ((newfont = strdup(optarg)) != NULL) {
1944 free(fontname);
1945 fontname = newfont;
1947 break;
1949 case 'W':
1950 r.width = parse_int_with_percentage(optarg, r.width, d_width);
1951 break;
1952 case 'H':
1953 r.height = parse_int_with_percentage(optarg, r.height, d_height);
1954 break;
1955 case 'b':
1956 if (parse_csslike(optarg, tmp) == -1)
1957 err(1, "parse_csslike");
1958 for (i = 0; i < 4; ++i)
1959 r.borders[i] = parse_integer(tmp[i], 0);
1960 break;
1961 case 'B':
1962 if (parse_csslike(optarg, tmp) == -1)
1963 err(1, "parse_csslike");
1964 for (i = 0; i < 4; ++i)
1965 borders_bg[i] = parse_color(tmp[i], "#000");
1966 break;
1967 case 't':
1968 fgs[0] = parse_color(optarg, NULL);
1969 break;
1970 case 'T':
1971 bgs[0] = parse_color(optarg, NULL);
1972 break;
1973 case 'c':
1974 fgs[1] = parse_color(optarg, NULL);
1975 break;
1976 case 'C':
1977 bgs[1] = parse_color(optarg, NULL);
1978 break;
1979 case 's':
1980 fgs[2] = parse_color(optarg, NULL);
1981 break;
1982 case 'S':
1983 bgs[2] = parse_color(optarg, NULL);
1984 break;
1985 default:
1986 fprintf(stderr, "Unrecognized option %c\n", ch);
1987 status = ERR;
1988 break;
1992 if (r.height < 0 || r.width < 0 || x < 0 || y < 0) {
1993 fprintf(stderr, "height, width, x or y are lesser than 0.");
1994 status = ERR;
1997 /* since only now we know if the first should be selected,
1998 * update the completion here */
1999 update_completions(cs, text, lines, vlines, r.first_selected);
2001 /* update the prompt lenght, only now we surely know the length of it
2003 r.ps1len = strlen(r.ps1);
2005 /* Create the window */
2006 create_window(&r, parent_window, cmap, vinfo, x, y, offset_x, offset_y, bgs[1]);
2007 set_win_atoms_hints(r.d, r.w, r.width, r.height);
2008 XMapRaised(r.d, r.w);
2010 /* If embed, listen for other events as well */
2011 if (embed) {
2012 Window *children, parent, root;
2013 unsigned int children_no;
2015 XSelectInput(r.d, parent_window, FocusChangeMask);
2016 if (XQueryTree(r.d, parent_window, &root, &parent, &children, &children_no)
2017 && children) {
2018 for (i = 0; i < children_no && children[i] != r.w; ++i)
2019 XSelectInput(r.d, children[i], FocusChangeMask);
2020 XFree(children);
2022 grabfocus(r.d, r.w);
2025 take_keyboard(r.d, r.w);
2027 r.x_zero = r.borders[3];
2028 r.y_zero = r.borders[0];
2031 XGCValues values;
2033 for (i = 0; i < 3; ++i) {
2034 r.fgs[i] = XCreateGC(r.d, r.w, 0, &values);
2035 r.bgs[i] = XCreateGC(r.d, r.w, 0, &values);
2038 for (i = 0; i < 4; ++i) {
2039 r.borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2040 r.p_borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2041 r.c_borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2042 r.ch_borders_bg[i] = XCreateGC(r.d, r.w, 0, &values);
2046 /* Load the colors in our GCs */
2047 for (i = 0; i < 3; ++i) {
2048 XSetForeground(r.d, r.fgs[i], fgs[i]);
2049 XSetForeground(r.d, r.bgs[i], bgs[i]);
2052 for (i = 0; i < 4; ++i) {
2053 XSetForeground(r.d, r.borders_bg[i], borders_bg[i]);
2054 XSetForeground(r.d, r.p_borders_bg[i], p_borders_bg[i]);
2055 XSetForeground(r.d, r.c_borders_bg[i], c_borders_bg[i]);
2056 XSetForeground(r.d, r.ch_borders_bg[i], ch_borders_bg[i]);
2059 if (load_font(&r, fontname) == -1)
2060 status = ERR;
2062 r.xftdraw = XftDrawCreate(r.d, r.w, vinfo.visual, cmap);
2064 for (i = 0; i < 3; ++i) {
2065 rgba_t c;
2066 XRenderColor xrcolor;
2068 c = *(rgba_t *)&fgs[i];
2069 xrcolor.red = EXPANDBITS(c.rgba.r);
2070 xrcolor.green = EXPANDBITS(c.rgba.g);
2071 xrcolor.blue = EXPANDBITS(c.rgba.b);
2072 xrcolor.alpha = EXPANDBITS(c.rgba.a);
2073 XftColorAllocValue(r.d, vinfo.visual, cmap, &xrcolor, &r.xft_colors[i]);
2076 /* compute prompt dimensions */
2077 ps1extents(&r);
2079 xim_init(&r, &xdb);
2081 #ifdef __OpenBSD__
2082 if (pledge("stdio", "") == -1)
2083 err(1, "pledge");
2084 #endif
2086 /* Cache text height */
2087 text_extents("fyjpgl", 6, &r, NULL, &r.text_height);
2089 /* Draw the window for the first time */
2090 draw(&r, text, cs);
2092 /* Main loop */
2093 while (status == LOOPING || status == OK_LOOP) {
2094 status = loop(&r, &text, &textlen, cs, lines, vlines);
2096 if (status != ERR)
2097 printf("%s\n", text);
2099 if (!r.multiple_select && status == OK_LOOP)
2100 status = OK;
2103 XUngrabKeyboard(r.d, CurrentTime);
2105 for (i = 0; i < 3; ++i)
2106 XftColorFree(r.d, vinfo.visual, cmap, &r.xft_colors[i]);
2108 for (i = 0; i < 3; ++i) {
2109 XFreeGC(r.d, r.fgs[i]);
2110 XFreeGC(r.d, r.bgs[i]);
2113 for (i = 0; i < 4; ++i) {
2114 XFreeGC(r.d, r.borders_bg[i]);
2115 XFreeGC(r.d, r.p_borders_bg[i]);
2116 XFreeGC(r.d, r.c_borders_bg[i]);
2117 XFreeGC(r.d, r.ch_borders_bg[i]);
2120 XDestroyIC(r.xic);
2121 XCloseIM(r.xim);
2123 for (i = 0; i < 3; ++i)
2124 XftColorFree(r.d, vinfo.visual, cmap, &r.xft_colors[i]);
2125 XftFontClose(r.d, r.font);
2126 XftDrawDestroy(r.xftdraw);
2128 free(r.ps1);
2129 free(fontname);
2130 free(text);
2132 free(lines);
2133 free(vlines);
2134 compls_delete(cs);
2136 XFreeColormap(r.d, cmap);
2138 XDestroyWindow(r.d, r.w);
2139 XCloseDisplay(r.d);
2141 return status != OK;