1 /* mymenu -- simple dmenu alternative */
3 /* Copyright (C) 2018 Omar Polo <omar.polo@europecom.net>
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <https://www.gnu.org/licenses/>.
21 #include <string.h> // strdup, strnlen, ...
22 #include <ctype.h> // isalnum
23 #include <locale.h> // setlocale
31 #include <X11/Xutil.h> // XLookupString
32 #include <X11/Xresource.h>
33 #include <X11/Xcms.h> // colors
34 #include <X11/keysym.h>
37 # include <X11/extensions/Xinerama.h>
41 #define resname "MyMenu"
42 #define resclass "mymenu"
44 #define MIN(X, Y) (((X) < (Y)) ? (X) : (Y))
45 #define MAX(a, b) ((a) > (b) ? (a) : (b))
46 #define OVERLAP(a,b,c,d) (((a)==(c) && (b)==(d)) || MIN((a)+(b), (c)+(d)) - MAX((a), (c)) > 0)
47 #define INTERSECT(x,y,w,h,x1,y1,w1,h1) (OVERLAP((x),(w),(x1),(w1)) && OVERLAP((y),(h),(y1),(h1)))
49 #define update_completions(cs, text, lines) { \
51 cs = filter(text, lines); \
55 #define INITIAL_ITEMS 64
58 fprintf(stderr, "TODO! " s "\n"); \
61 #define cannot_allocate_memory { \
62 fprintf(stderr, "Could not allocate memory\n"); \
63 exit(EX_UNAVAILABLE); \
66 #define check_allocation(a) { \
68 cannot_allocate_memory; \
71 enum state {LOOPING, OK, ERR};
80 GC completion_highlighted;
81 GC completion_highlighted_bg;
85 bool horizontal_layout;
91 struct completions *next;
94 struct completions *compl_new() {
95 struct completions *c = malloc(sizeof(struct completions));
106 void compl_delete(struct completions *c) {
110 struct completions *compl_select_next(struct completions *c, bool n) {
118 struct completions *orig = c;
122 if (c->next != nil) {
123 // the current one is selected and the next one exists
124 c->next->selected = true;
127 // the current one is selected and the next one is nill,
128 // select the first one
129 orig->selected = true;
138 struct completions *compl_select_prev(struct completions *c, bool n) {
142 struct completions *cc = c;
144 if (n) // select the last one
146 if (cc->next == nil) {
152 else // select the previous one
154 if (cc->next != nil && cc->next->selected) {
155 cc->next->selected = false;
165 struct completions *filter(char *text, char **lines) {
167 struct completions *root = compl_new();
168 struct completions *c = root;
175 if (strstr(l, text) != nil) {
176 c->next = compl_new();
184 struct completions *r = root->next;
189 // push the character c at the end of the string pointed by p
190 int pushc(char **p, int maxlen, char c) {
191 int len = strnlen(*p, maxlen);
193 if (!(len < maxlen -2)) {
194 maxlen += maxlen >> 1;
195 char *newptr = realloc(*p, maxlen);
196 if (newptr == nil) { // bad!
207 int utf8strnlen(char *s, int maxlen) {
209 while (*s && maxlen > 0) {
210 len += (*s++ & 0xc0) != 0x80;
216 // remove the last *glyph* from the *utf8* string!
217 // this is different from just setting the last byte to 0 (in some
218 // cases ofc). The actual implementation is quite inefficient because
219 // it remove the last char until the number of glyphs doesn't change
220 void popc(char *p, int maxlen) {
221 int len = strnlen(p, maxlen);
226 int ulen = utf8strnlen(p, maxlen);
227 while (len > 0 && utf8strnlen(p, maxlen) == ulen) {
233 // read an arbitrary long line from stdin and return a pointer to it
234 // TODO: resize the allocated memory to exactly fit the string once
236 char *readline(bool *eof) {
238 char *str = calloc(maxlen, sizeof(char));
240 fprintf(stderr, "Cannot allocate memory!\n");
241 exit(EX_UNAVAILABLE);
245 while((c = getchar()) != EOF) {
249 maxlen = pushc(&str, maxlen, c);
252 fprintf(stderr, "Cannot allocate memory!\n");
253 exit(EX_UNAVAILABLE);
260 int readlines (char ***lns, int items) {
261 bool finished = false;
265 lines[n] = readline(&finished);
267 if (strlen(lines[n]) == 0 || lines[n][0] == '\n')
268 --n; // forget about this line
275 if (n == items - 1) {
277 char **l = realloc(lines, sizeof(char*) * items);
289 // |------------------|----------------------------------------------|
290 // | 20 char text | completion | completion | completion | compl |
291 // |------------------|----------------------------------------------|
292 void draw_horizontally(struct rendering *r, char *text, struct completions *cs) {
293 // TODO: make these dynamic?
294 int prompt_width = 20; // char
296 /* int start_at = XTextWidth(r->font, " ", 1) * prompt_width + padding; */
299 int start_at = XmbTextExtents(*r->font, " ", 1, nil, &rect);
300 start_at = start_at * prompt_width + padding;
302 int texty = (rect.height + r->height) >>1;
304 XFillRectangle(r->d, r->w, r->prompt_bg, 0, 0, start_at, r->height);
306 int text_len = strlen(text);
307 if (text_len > prompt_width)
308 text = text + (text_len - prompt_width);
309 /* XDrawString(r->d, r->w, r->prompt, padding, texty, text, MIN(text_len, prompt_width)); */
310 Xutf8DrawString(r->d, r->w, *r->font, r->prompt, padding, texty, text, MIN(text_len, prompt_width));
312 XFillRectangle(r->d, r->w, r->completion_bg, start_at, 0, r->width, r->height);
315 GC g = cs->selected ? r->completion_highlighted : r->completion;
316 GC h = cs->selected ? r->completion_highlighted_bg : r->completion_bg;
318 int len = strlen(cs->completion);
319 /* int text_width = XTextWidth(r->font, cs->completion, len); */
320 int text_width = XmbTextExtents(*r->font, cs->completion, len, nil, nil);
322 XFillRectangle(r->d, r->w, h, start_at, 0, text_width + padding*2, r->height);
324 /* XDrawString(r->d, r->w, g, start_at + padding, texty, cs->completion, len); */
325 Xutf8DrawString(r->d, r->w, *r->font, g, start_at + padding, texty, cs->completion, len);
327 start_at += text_width + padding * 2;
335 // |-----------------------------------------------------------------|
337 // |-----------------------------------------------------------------|
339 // |-----------------------------------------------------------------|
341 // |-----------------------------------------------------------------|
342 // TODO: dunno why but in the call to Xutf8DrawString, by logic,
343 // should be padding and not padding*2, but the text doesn't seem to
344 // be vertically centered otherwise....
345 void draw_vertically(struct rendering *r, char *text, struct completions *cs) {
346 int padding = 10; // TODO make this dynamic
349 XmbTextExtents(*r->font, "fjpgl", 5, &rect, nil);
350 int start_at = rect.height + padding*2;
352 XFillRectangle(r->d, r->w, r->completion_bg, 0, 0, r->width, r->height);
353 XFillRectangle(r->d, r->w, r->prompt_bg, 0, 0, r->width, start_at);
354 Xutf8DrawString(r->d, r->w, *r->font, r->prompt, padding, padding*2, text, strlen(text));
357 GC g = cs->selected ? r->completion_highlighted : r->completion;
358 GC h = cs->selected ? r->completion_highlighted_bg : r->completion_bg;
360 int len = strlen(cs->completion);
361 XmbTextExtents(*r->font, cs->completion, len, &rect, nil);
362 XFillRectangle(r->d, r->w, h, 0, start_at, r->width, rect.height + padding*2);
363 Xutf8DrawString(r->d, r->w, *r->font, g, padding, start_at + padding*2, cs->completion, len);
365 start_at += rect.height + padding *2;
372 void draw(struct rendering *r, char *text, struct completions *cs) {
373 if (r->horizontal_layout)
374 draw_horizontally(r, text, cs);
376 draw_vertically(r, text, cs);
379 /* Set some WM stuff */
380 void set_win_atoms_hints(Display *d, Window w, int width, int height) {
382 type = XInternAtom(d, "_NET_WM_WINDOW_TYPE_DOCK", false);
386 XInternAtom(d, "_NET_WM_WINDOW_TYPE", false),
387 XInternAtom(d, "ATOM", false),
390 (unsigned char *)&type,
394 /* some window managers honor this properties */
395 type = XInternAtom(d, "_NET_WM_STATE_ABOVE", false);
398 XInternAtom(d, "_NET_WM_STATE", false),
399 XInternAtom(d, "ATOM", false),
402 (unsigned char *)&type,
406 type = XInternAtom(d, "_NET_WM_STATE_FOCUSED", false);
409 XInternAtom(d, "_NET_WM_STATE", false),
410 XInternAtom(d, "ATOM", false),
413 (unsigned char *)&type,
417 // setting window hints
418 XClassHint *class_hint = XAllocClassHint();
419 if (class_hint == nil) {
420 fprintf(stderr, "Could not allocate memory for class hint\n");
421 exit(EX_UNAVAILABLE);
423 class_hint->res_name = "mymenu";
424 class_hint->res_class = "mymenu";
425 XSetClassHint(d, w, class_hint);
428 XSizeHints *size_hint = XAllocSizeHints();
429 if (size_hint == nil) {
430 fprintf(stderr, "Could not allocate memory for size hint\n");
431 exit(EX_UNAVAILABLE);
433 size_hint->min_width = width;
434 size_hint->base_width = width;
435 size_hint->min_height = height;
436 size_hint->base_height = height;
441 void get_wh(Display *d, Window *w, int *width, int *height) {
442 XWindowAttributes win_attr;
443 XGetWindowAttributes(d, *w, &win_attr);
444 *height = win_attr.height;
445 *width = win_attr.width;
448 // I know this may seem a little hackish BUT is the only way I managed
449 // to actually grab that goddam keyboard. Only one call to
450 // XGrabKeyboard does not always end up with the keyboard grabbed!
451 int take_keyboard(Display *d, Window w) {
453 for (i = 0; i < 100; i++) {
454 if (XGrabKeyboard(d, w, True, GrabModeAsync, GrabModeAsync, CurrentTime) == GrabSuccess)
461 void release_keyboard(Display *d) {
462 XUngrabKeyboard(d, CurrentTime);
465 int parse_integer(const char *str, int default_value, int max) {
466 int len = strlen(str);
467 if (len > 0 && str[len-1] == '%') {
468 char *cpy = strdup(str);
469 check_allocation(cpy);
471 int val = parse_integer(cpy, default_value, max);
473 return val * max / 100;
478 long lval = strtol(str, &ep, 10);
479 if (str[0] == '\0' || *ep != '\0') { // NaN
480 fprintf(stderr, "'%s' is not a valid number! Using %d as default.\n", str, default_value);
481 return default_value;
483 if ((errno == ERANGE && (lval == LONG_MAX || lval == LONG_MIN)) ||
484 (lval > INT_MAX || lval < INT_MIN)) {
485 fprintf(stderr, "%s out of range! Using %d as default.\n", str, default_value);
486 return default_value;
491 int parse_int_with_middle(const char *str, int default_value, int max, int self) {
492 if (!strcmp(str, "middle")) {
493 return (max - self)/2;
495 return parse_integer(str, default_value, max);
499 /* char *lines[INITIAL_ITEMS] = {0}; */
500 char **lines = calloc(INITIAL_ITEMS, sizeof(char*));
501 readlines(&lines, INITIAL_ITEMS);
503 setlocale(LC_ALL, getenv("LANG"));
505 enum state status = LOOPING;
507 // where the monitor start (used only with xinerama)
511 // width and height of the window
515 // position on the screen
519 char *fontname = strdup("fixed");
520 check_allocation(fontname);
523 char *text = malloc(textlen * sizeof(char));
524 check_allocation(text);
526 struct completions *cs = filter(text, lines);
527 bool nothing_selected = true;
529 // start talking to xorg
530 Display *d = XOpenDisplay(nil);
532 fprintf(stderr, "Could not open display!\n");
533 return EX_UNAVAILABLE;
537 // XXX: is getting the default root window dimension correct?
538 XWindowAttributes xwa;
539 XGetWindowAttributes(d, DefaultRootWindow(d), &xwa);
540 int d_width = xwa.width;
541 int d_height = xwa.height;
544 // TODO: this bit still needs to be improved
545 if (XineramaIsActive(d)) {
547 XineramaScreenInfo *info = XineramaQueryScreens(d, &monitors);
549 for (int i = 0; i < monitors; ++i) {
550 if (INTERSECT(x, y, 1, 1, info[i].x_org, info[i].y_org, info[i].width, info[i].height)) {
551 offset_x = info[x].x_org;
552 offset_y = info[y].y_org;
553 d_width = info[i].width;
554 d_height = info[i].height;
561 /* fprintf(stderr, "offset_x:\t%d\n" */
562 /* "offset_y:\t%d\n" */
563 /* "d_width:\t%d\n" */
564 /* "d_height:\t%d\n" */
565 /* , offset_x, offset_y, d_width, d_height); */
567 Colormap cmap = DefaultColormap(d, DefaultScreen(d));
570 compl_highlighted_fg, compl_highlighted_bg;
572 bool horizontal_layout = true;
576 char *xrm = XResourceManagerString(d);
577 XrmDatabase xdb = nil;
579 xdb = XrmGetStringDatabase(xrm);
583 if (XrmGetResource(xdb, "MyMenu.font", "*", datatype, &value) == true) {
584 fontname = strdup(value.addr);
585 check_allocation(fontname);
588 fprintf(stderr, "no font defined, using %s\n", fontname);
590 if (XrmGetResource(xdb, "MyMenu.layout", "*", datatype, &value) == true) {
591 char *v = strdup(value.addr);
593 horizontal_layout = !strcmp(v, "horizontal");
597 fprintf(stderr, "no layout defined, using horizontal\n");
599 if (XrmGetResource(xdb, "MyMenu.width", "*", datatype, &value) == true)
600 width = parse_integer(value.addr, width, d_width);
602 fprintf(stderr, "no width defined, using %d\n", width);
604 if (XrmGetResource(xdb, "MyMenu.height", "*", datatype, &value) == true)
605 height = parse_integer(value.addr, height, d_height);
607 fprintf(stderr, "no height defined, using %d\n", height);
609 if (XrmGetResource(xdb, "MyMenu.x", "*", datatype, &value) == true)
610 x = parse_int_with_middle(value.addr, x, d_width, width);
612 fprintf(stderr, "no x defined, using %d\n", width);
614 if (XrmGetResource(xdb, "MyMenu.y", "*", datatype, &value) == true)
615 y = parse_int_with_middle(value.addr, y, d_height, height);
617 fprintf(stderr, "no y defined, using %d\n", height);
620 // TODO: tmp needs to be free'd after every allocation?
623 if (XrmGetResource(xdb, "MyMenu.prompt.foreground", "*", datatype, &value) == true)
624 XAllocNamedColor(d, cmap, value.addr, &p_fg, &tmp);
626 XAllocNamedColor(d, cmap, "white", &p_fg, &tmp);
628 if (XrmGetResource(xdb, "MyMenu.prompt.background", "*", datatype, &value) == true)
629 XAllocNamedColor(d, cmap, value.addr, &p_bg, &tmp);
631 XAllocNamedColor(d, cmap, "black", &p_bg, &tmp);
634 if (XrmGetResource(xdb, "MyMenu.completion.foreground", "*", datatype, &value) == true)
635 XAllocNamedColor(d, cmap, value.addr, &compl_fg, &tmp);
637 XAllocNamedColor(d, cmap, "white", &compl_fg, &tmp);
639 if (XrmGetResource(xdb, "MyMenu.completion.background", "*", datatype, &value) == true)
640 XAllocNamedColor(d, cmap, value.addr, &compl_bg, &tmp);
642 XAllocNamedColor(d, cmap, "black", &compl_bg, &tmp);
644 // completion highlighted
645 if (XrmGetResource(xdb, "MyMenu.completion_highlighted.foreground", "*", datatype, &value) == true)
646 XAllocNamedColor(d, cmap, value.addr, &compl_highlighted_fg, &tmp);
648 XAllocNamedColor(d, cmap, "black", &compl_highlighted_fg, &tmp);
650 if (XrmGetResource(xdb, "MyMenu.completion_highlighted.background", "*", datatype, &value) == true)
651 XAllocNamedColor(d, cmap, value.addr, &compl_highlighted_bg, &tmp);
653 XAllocNamedColor(d, cmap, "white", &compl_highlighted_bg, &tmp);
656 XAllocNamedColor(d, cmap, "white", &p_fg, &tmp);
657 XAllocNamedColor(d, cmap, "black", &p_bg, &tmp);
658 XAllocNamedColor(d, cmap, "white", &compl_fg, &tmp);
659 XAllocNamedColor(d, cmap, "black", &compl_bg, &tmp);
660 XAllocNamedColor(d, cmap, "black", &compl_highlighted_fg, &tmp);
661 XAllocNamedColor(d, cmap, "white", &compl_highlighted_bg, &tmp);
665 /* XFontStruct *font = XLoadQueryFont(d, fontname); */
666 /* if (font == nil) { */
667 /* fprintf(stderr, "Unable to load %s font\n", fontname); */
668 /* font = XLoadQueryFont(d, "fixed"); */
671 char **missing_charset_list;
672 int missing_charset_count;
673 XFontSet font = XCreateFontSet(d, fontname, &missing_charset_list, &missing_charset_count, nil);
675 fprintf(stderr, "Unable to load the font(s) %s\n", fontname);
676 return EX_UNAVAILABLE;
680 XSetWindowAttributes attr;
682 Window w = XCreateWindow(d, // display
683 DefaultRootWindow(d), // parent
684 x + offset_x, y + offset_y, // x y
685 width, height, // w h
687 DefaultDepth(d, DefaultScreen(d)), // depth
688 InputOutput, // class
689 DefaultVisual(d, DefaultScreen(d)), // visual
693 set_win_atoms_hints(d, w, width, height);
695 // we want some events
696 XSelectInput(d, w, StructureNotifyMask | KeyPressMask | KeyReleaseMask | KeymapStateMask);
698 // make the window appear on the screen
701 // wait for the MapNotify event (i.e. the event "window rendered")
705 if (e.type == MapNotify)
709 // get the *real* width & height after the window was rendered
710 get_wh(d, &w, &width, &height);
715 // Create some graphics contexts
717 /* values.font = font->fid; */
719 struct rendering r = {
722 .prompt = XCreateGC(d, w, 0, &values),
723 .prompt_bg = XCreateGC(d, w, 0, &values),
724 .completion = XCreateGC(d, w, 0, &values),
725 .completion_bg = XCreateGC(d, w, 0, &values),
726 .completion_highlighted = XCreateGC(d, w, 0, &values),
727 .completion_highlighted_bg = XCreateGC(d, w, 0, &values),
728 /* .prompt = XCreateGC(d, w, GCFont, &values), */
729 /* .prompt_bg = XCreateGC(d, w, GCFont, &values), */
730 /* .completion = XCreateGC(d, w, GCFont, &values), */
731 /* .completion_bg = XCreateGC(d, w, GCFont, &values), */
732 /* .completion_highlighted = XCreateGC(d, w, GCFont, &values), */
733 /* .completion_highlighted_bg = XCreateGC(d, w, GCFont, &values), */
737 .horizontal_layout = horizontal_layout
740 // load the colors in our GCs
741 XSetForeground(d, r.prompt, p_fg.pixel);
742 XSetForeground(d, r.prompt_bg, p_bg.pixel);
743 XSetForeground(d, r.completion, compl_fg.pixel);
744 XSetForeground(d, r.completion_bg, compl_bg.pixel);
745 XSetForeground(d, r.completion_highlighted, compl_highlighted_fg.pixel);
746 XSetForeground(d, r.completion_highlighted_bg, compl_highlighted_bg.pixel);
748 // open the X input method
749 XIM xim = XOpenIM(d, xdb, resname, resclass);
750 check_allocation(xim);
752 XIMStyles *xis = nil;
753 if (XGetIMValues(xim, XNQueryInputStyle, &xis, NULL) || !xis) {
754 fprintf(stderr, "Input Styles could not be retrieved\n");
755 return EX_UNAVAILABLE;
758 XIMStyle bestMatchStyle = 0;
759 for (int i = 0; i < xis->count_styles; ++i) {
760 XIMStyle ts = xis->supported_styles[i];
761 if (ts == (XIMPreeditNothing | XIMStatusNothing)) {
768 if (!bestMatchStyle) {
769 fprintf(stderr, "No matching input style could be determined\n");
772 XIC xic = XCreateIC(xim, XNInputStyle, bestMatchStyle, XNClientWindow, w, XNFocusWindow, w, NULL);
773 check_allocation(xic);
775 // draw the window for the first time
779 while (status == LOOPING) {
783 if (XFilterEvent(&e, w))
788 XRefreshKeyboardMapping(&e.xmapping);
791 case KeyRelease: break; // ignore this
794 XKeyPressedEvent *ev = (XKeyPressedEvent*)&e;
796 if (ev->keycode == XKeysymToKeycode(d, XK_Tab)) {
797 bool shift = (ev->state & ShiftMask);
798 struct completions *n = shift ? compl_select_prev(cs, nothing_selected)
799 : compl_select_next(cs, nothing_selected);
801 nothing_selected = false;
803 text = strdup(n->completion);
805 fprintf(stderr, "Memory allocation error!\n");
809 textlen = strlen(text);
815 if (ev->keycode == XKeysymToKeycode(d, XK_BackSpace)) {
816 nothing_selected = true;
818 update_completions(cs, text, lines);
823 if (ev->keycode == XKeysymToKeycode(d, XK_Return)) {
828 if (ev->keycode == XKeysymToKeycode(d, XK_Escape)) {
833 // try to read what the user pressed
836 Xutf8LookupString(xic, ev, (char*)&symbol, 4, 0, &s);
838 if (s == XBufferOverflow) {
839 // should not happen since there are no utf-8 characters
840 // larger than 24bits, but is something to be aware of when
841 // used to directly write to a string buffer
842 fprintf(stderr, "Buffer overflow when trying to create keyboard symbol map.\n");
845 char *str = (char*)&symbol;
847 if (ev->state & ControlMask) {
848 // check for some key bindings
849 if (!strcmp(str, "")) { // C-u
850 nothing_selected = true;
851 for (int i = 0; i < textlen; ++i)
853 update_completions(cs, text, lines);
855 if (!strcmp(str, "")) { // C-h
856 nothing_selected = true;
858 update_completions(cs, text, lines);
860 if (!strcmp(str, "")) { // C-w
861 nothing_selected = true;
863 // `textlen` is the length of the allocated string, not the
864 // length of the ACTUAL string
865 int p = strlen(text) - 1;
866 if (p >= 0) { // delete the current char
870 while (p >= 0 && isalnum(text[p])) {
874 // erase also trailing white space
875 while (p >= 0 && isspace(text[p])) {
879 update_completions(cs, text, lines);
881 if (!strcmp(str, "\r")) { // C-m
888 int str_len = strlen(str);
889 for (int i = 0; i < str_len; ++i) {
890 textlen = pushc(&text, textlen, str[i]);
892 fprintf(stderr, "Memory allocation error\n");
896 nothing_selected = true;
897 update_completions(cs, text, lines);
905 fprintf(stderr, "Unknown event %d\n", e.type);
912 printf("%s\n", text);
919 /* XDestroyWindow(d, w); */
922 return status == OK ? 0 : 1;