Xinqi Bao's Git

projects / dmenu.git / blob


688fabf26a4aa39d5c912f7276d6e29b6911cb59
[dmenu.git] / dmenu.c
1 /* See LICENSE file for copyright and license details. */
2 #include <ctype.h>
3 #include <locale.h>
4 #include <stdarg.h>
5 #include <stdlib.h>
6 #include <stdio.h>
7 #include <string.h>
8 #include <unistd.h>
9 #include <X11/Xlib.h>
10 #include <X11/Xutil.h>
11 #include <X11/keysym.h>
12
13 /* macros */
14 #define CLEANMASK(mask) (mask & ~(numlockmask | LockMask))
15
16 /* enums */
17 enum { ColFG, ColBG, ColLast };
18
19 /* typedefs */
20 typedef struct {
21 int x, y, w, h;
22 unsigned long norm[ColLast];
23 unsigned long sel[ColLast];
24 Drawable drawable;
25 GC gc;
26 struct {
27 XFontStruct *xfont;
28 XFontSet set;
29 int ascent;
30 int descent;
31 int height;
32 } font;
33 } DC; /* draw context */
34
35 typedef struct Item Item;
36 struct Item {
37 char *text;
38 Bool matched;
39 Item *next; /* traverses all items */
40 Item *left, *right; /* traverses items matching current search pattern */
41 };
42
43 /* forward declarations */
44 Item *appenditem(Item *i, Item *last);
45 void calcoffsets(void);
46 char *cistrstr(const char *s, const char *sub);
47 void cleanup(void);
48 void drawmenu(void);
49 void drawtext(const char *text, unsigned long col[ColLast]);
50 void *emalloc(unsigned int size);
51 void eprint(const char *errstr, ...);
52 char *estrdup(const char *str);
53 unsigned long getcolor(const char *colstr);
54 Bool grabkeyboard(void);
55 void initfont(const char *fontstr);
56 void kpress(XKeyEvent * e);
57 void match(char *pattern);
58 void readstdin(void);
59 void run(void);
60 void setup(int x, int y, int w);
61 unsigned int textnw(const char *text, unsigned int len);
62 unsigned int textw(const char *text);
63
64 #include "config.h"
65
66 /* variables */
67 char *font = FONT;
68 char *maxname = NULL;
69 char *normbg = NORMBGCOLOR;
70 char *normfg = NORMFGCOLOR;
71 char *prompt = NULL;
72 char *selbg = SELBGCOLOR;
73 char *selfg = SELFGCOLOR;
74 char text[4096];
75 int screen;
76 int ret = 0;
77 unsigned int cmdw = 0;
78 unsigned int mw, mh;
79 unsigned int promptw = 0;
80 unsigned int nitem = 0;
81 unsigned int numlockmask = 0;
82 Bool running = True;
83 Display *dpy;
84 DC dc = {0};
85 Item *allitems = NULL; /* first of all items */
86 Item *item = NULL; /* first of pattern matching items */
87 Item *sel = NULL;
88 Item *next = NULL;
89 Item *prev = NULL;
90 Item *curr = NULL;
91 Window root, win;
92 int (*fstrncmp)(const char *, const char *, size_t n) = strncmp;
93 char *(*fstrstr)(const char *, const char *) = strstr;
94
95 Item *
96 appenditem(Item *i, Item *last) {
97 if(!last)
98 item = i;
99 else
100 last->right = i;
101 i->left = last;
102 i->right = NULL;
103 last = i;
104 nitem++;
105 return last;
106 }
107
108 void
109 calcoffsets(void) {
110 unsigned int tw, w;
111
112 if(!curr)
113 return;
114 w = promptw + cmdw + 2 * SPACE;
115 for(next = curr; next; next=next->right) {
116 tw = textw(next->text);
117 if(tw > mw / 3)
118 tw = mw / 3;
119 w += tw;
120 if(w > mw)
121 break;
122 }
123 w = promptw + cmdw + 2 * SPACE;
124 for(prev = curr; prev && prev->left; prev=prev->left) {
125 tw = textw(prev->left->text);
126 if(tw > mw / 3)
127 tw = mw / 3;
128 w += tw;
129 if(w > mw)
130 break;
131 }
132 }
133
134 char *
135 cistrstr(const char *s, const char *sub) {
136 int c, csub;
137 unsigned int len;
138
139 if(!sub)
140 return (char *)s;
141 if((c = *sub++) != 0) {
142 c = tolower(c);
143 len = strlen(sub);
144 do {
145 do {
146 if((csub = *s++) == 0)
147 return (NULL);
148 }
149 while(tolower(csub) != c);
150 }
151 while(strncasecmp(s, sub, len) != 0);
152 s--;
153 }
154 return (char *)s;
155 }
156
157 void
158 cleanup(void) {
159 Item *itm;
160
161 while(allitems) {
162 itm = allitems->next;
163 free(allitems->text);
164 free(allitems);
165 allitems = itm;
166 }
167 if(dc.font.set)
168 XFreeFontSet(dpy, dc.font.set);
169 else
170 XFreeFont(dpy, dc.font.xfont);
171 XFreePixmap(dpy, dc.drawable);
172 XFreeGC(dpy, dc.gc);
173 XDestroyWindow(dpy, win);
174 XUngrabKeyboard(dpy, CurrentTime);
175 }
176
177 void
178 drawmenu(void) {
179 Item *i;
180
181 dc.x = 0;
182 dc.y = 0;
183 dc.w = mw;
184 dc.h = mh;
185 drawtext(NULL, dc.norm);
186 /* print prompt? */
187 if(promptw) {
188 dc.w = promptw;
189 drawtext(prompt, dc.sel);
190 }
191 dc.x += promptw;
192 dc.w = mw - promptw;
193 /* print command */
194 if(cmdw && item)
195 dc.w = cmdw;
196 drawtext(text[0] ? text : NULL, dc.norm);
197 dc.x += cmdw;
198 if(curr) {
199 dc.w = SPACE;
200 drawtext((curr && curr->left) ? "<" : NULL, dc.norm);
201 dc.x += dc.w;
202 /* determine maximum items */
203 for(i = curr; i != next; i=i->right) {
204 dc.w = textw(i->text);
205 if(dc.w > mw / 3)
206 dc.w = mw / 3;
207 drawtext(i->text, (sel == i) ? dc.sel : dc.norm);
208 dc.x += dc.w;
209 }
210 dc.x = mw - SPACE;
211 dc.w = SPACE;
212 drawtext(next ? ">" : NULL, dc.norm);
213 }
214 XCopyArea(dpy, dc.drawable, win, dc.gc, 0, 0, mw, mh, 0, 0);
215 XFlush(dpy);
216 }
217
218 void
219 drawtext(const char *text, unsigned long col[ColLast]) {
220 int x, y, w, h;
221 static char buf[256];
222 unsigned int len, olen;
223 XRectangle r = { dc.x, dc.y, dc.w, dc.h };
224
225 XSetForeground(dpy, dc.gc, col[ColBG]);
226 XFillRectangles(dpy, dc.drawable, dc.gc, &r, 1);
227 if(!text)
228 return;
229 w = 0;
230 olen = len = strlen(text);
231 if(len >= sizeof buf)
232 len = sizeof buf - 1;
233 memcpy(buf, text, len);
234 buf[len] = 0;
235 h = dc.font.ascent + dc.font.descent;
236 y = dc.y + (dc.h / 2) - (h / 2) + dc.font.ascent;
237 x = dc.x + (h / 2);
238 /* shorten text if necessary */
239 while(len && (w = textnw(buf, len)) > dc.w - h)
240 buf[--len] = 0;
241 if(len < olen) {
242 if(len > 1)
243 buf[len - 1] = '.';
244 if(len > 2)
245 buf[len - 2] = '.';
246 if(len > 3)
247 buf[len - 3] = '.';
248 }
249 if(w > dc.w)
250 return; /* too long */
251 XSetForeground(dpy, dc.gc, col[ColFG]);
252 if(dc.font.set)
253 XmbDrawString(dpy, dc.drawable, dc.font.set, dc.gc, x, y, buf, len);
254 else
255 XDrawString(dpy, dc.drawable, dc.gc, x, y, buf, len);
256 }
257
258 void *
259 emalloc(unsigned int size) {
260 void *res = malloc(size);
261
262 if(!res)
263 eprint("fatal: could not malloc() %u bytes\n", size);
264 return res;
265 }
266
267 void
268 eprint(const char *errstr, ...) {
269 va_list ap;
270
271 va_start(ap, errstr);
272 vfprintf(stderr, errstr, ap);
273 va_end(ap);
274 exit(EXIT_FAILURE);
275 }
276
277 char *
278 estrdup(const char *str) {
279 void *res = strdup(str);
280
281 if(!res)
282 eprint("fatal: could not malloc() %u bytes\n", strlen(str));
283 return res;
284 }
285
286 unsigned long
287 getcolor(const char *colstr) {
288 Colormap cmap = DefaultColormap(dpy, screen);
289 XColor color;
290
291 if(!XAllocNamedColor(dpy, cmap, colstr, &color, &color))
292 eprint("error, cannot allocate color '%s'\n", colstr);
293 return color.pixel;
294 }
295
296 Bool
297 grabkeyboard(void) {
298 unsigned int len;
299
300 for(len = 1000; len; len--) {
301 if(XGrabKeyboard(dpy, root, True, GrabModeAsync, GrabModeAsync, CurrentTime)
302 == GrabSuccess)
303 break;
304 usleep(1000);
305 }
306 return len > 0;
307 }
308
309 void
310 initfont(const char *fontstr) {
311 char *def, **missing;
312 int i, n;
313
314 if(!fontstr || fontstr[0] == '\0')
315 eprint("error, cannot load font: '%s'\n", fontstr);
316 missing = NULL;
317 if(dc.font.set)
318 XFreeFontSet(dpy, dc.font.set);
319 dc.font.set = XCreateFontSet(dpy, fontstr, &missing, &n, &def);
320 if(missing)
321 XFreeStringList(missing);
322 if(dc.font.set) {
323 XFontSetExtents *font_extents;
324 XFontStruct **xfonts;
325 char **font_names;
326 dc.font.ascent = dc.font.descent = 0;
327 font_extents = XExtentsOfFontSet(dc.font.set);
328 n = XFontsOfFontSet(dc.font.set, &xfonts, &font_names);
329 for(i = 0, dc.font.ascent = 0, dc.font.descent = 0; i < n; i++) {
330 if(dc.font.ascent < (*xfonts)->ascent)
331 dc.font.ascent = (*xfonts)->ascent;
332 if(dc.font.descent < (*xfonts)->descent)
333 dc.font.descent = (*xfonts)->descent;
334 xfonts++;
335 }
336 }
337 else {
338 if(dc.font.xfont)
339 XFreeFont(dpy, dc.font.xfont);
340 dc.font.xfont = NULL;
341 if(!(dc.font.xfont = XLoadQueryFont(dpy, fontstr))
342 && !(dc.font.xfont = XLoadQueryFont(dpy, "fixed")))
343 eprint("error, cannot load font: '%s'\n", fontstr);
344 dc.font.ascent = dc.font.xfont->ascent;
345 dc.font.descent = dc.font.xfont->descent;
346 }
347 dc.font.height = dc.font.ascent + dc.font.descent;
348 }
349
350 void
351 kpress(XKeyEvent * e) {
352 char buf[32];
353 int i, num;
354 unsigned int len;
355 KeySym ksym;
356
357 len = strlen(text);
358 buf[0] = 0;
359 num = XLookupString(e, buf, sizeof buf, &ksym, 0);
360 if(IsKeypadKey(ksym)) {
361 if(ksym == XK_KP_Enter) {
362 ksym = XK_Return;
363 } else if(ksym >= XK_KP_0 && ksym <= XK_KP_9) {
364 ksym = (ksym - XK_KP_0) + XK_0;
365 }
366 }
367 if(IsFunctionKey(ksym) || IsKeypadKey(ksym)
368 || IsMiscFunctionKey(ksym) || IsPFKey(ksym)
369 || IsPrivateKeypadKey(ksym))
370 return;
371 /* first check if a control mask is omitted */
372 if(e->state & ControlMask) {
373 switch (ksym) {
374 default: /* ignore other control sequences */
375 return;
376 case XK_bracketleft:
377 ksym = XK_Escape;
378 break;
379 case XK_h:
380 case XK_H:
381 ksym = XK_BackSpace;
382 break;
383 case XK_i:
384 case XK_I:
385 ksym = XK_Tab;
386 break;
387 case XK_j:
388 case XK_J:
389 ksym = XK_Return;
390 break;
391 case XK_u:
392 case XK_U:
393 text[0] = 0;
394 match(text);
395 drawmenu();
396 return;
397 case XK_w:
398 case XK_W:
399 if(len) {
400 i = len - 1;
401 while(i >= 0 && text[i] == ' ')
402 text[i--] = 0;
403 while(i >= 0 && text[i] != ' ')
404 text[i--] = 0;
405 match(text);
406 drawmenu();
407 }
408 return;
409 }
410 }
411 if(CLEANMASK(e->state) & Mod1Mask) {
412 switch(ksym) {
413 default: return;
414 case XK_h:
415 ksym = XK_Left;
416 break;
417 case XK_l:
418 ksym = XK_Right;
419 break;
420 case XK_j:
421 ksym = XK_Next;
422 break;
423 case XK_k:
424 ksym = XK_Prior;
425 break;
426 case XK_g:
427 ksym = XK_Home;
428 break;
429 case XK_G:
430 ksym = XK_End;
431 break;
432 }
433 }
434 switch(ksym) {
435 default:
436 if(num && !iscntrl((int) buf[0])) {
437 buf[num] = 0;
438 if(len > 0)
439 strncat(text, buf, sizeof text);
440 else
441 strncpy(text, buf, sizeof text);
442 match(text);
443 }
444 break;
445 case XK_BackSpace:
446 if(len) {
447 text[--len] = 0;
448 match(text);
449 }
450 break;
451 case XK_End:
452 if(!item)
453 return;
454 while(next) {
455 sel = curr = next;
456 calcoffsets();
457 }
458 while(sel && sel->right)
459 sel = sel->right;
460 break;
461 case XK_Escape:
462 ret = 1;
463 running = False;
464 break;
465 case XK_Home:
466 if(!item)
467 return;
468 sel = curr = item;
469 calcoffsets();
470 break;
471 case XK_Left:
472 if(!(sel && sel->left))
473 return;
474 sel=sel->left;
475 if(sel->right == curr) {
476 curr = prev;
477 calcoffsets();
478 }
479 break;
480 case XK_Next:
481 if(!next)
482 return;
483 sel = curr = next;
484 calcoffsets();
485 break;
486 case XK_Prior:
487 if(!prev)
488 return;
489 sel = curr = prev;
490 calcoffsets();
491 break;
492 case XK_Return:
493 if((e->state & ShiftMask) && text)
494 fprintf(stdout, "%s", text);
495 else if(sel)
496 fprintf(stdout, "%s", sel->text);
497 else if(text)
498 fprintf(stdout, "%s", text);
499 fflush(stdout);
500 running = False;
501 break;
502 case XK_Right:
503 if(!(sel && sel->right))
504 return;
505 sel=sel->right;
506 if(sel == next) {
507 curr = next;
508 calcoffsets();
509 }
510 break;
511 case XK_Tab:
512 if(!sel)
513 return;
514 strncpy(text, sel->text, sizeof text);
515 match(text);
516 break;
517 }
518 drawmenu();
519 }
520
521 void
522 match(char *pattern) {
523 unsigned int plen;
524 Item *i, *j;
525
526 if(!pattern)
527 return;
528 plen = strlen(pattern);
529 item = j = NULL;
530 nitem = 0;
531 for(i = allitems; i; i = i->next)
532 if((i->matched = !fstrncmp(pattern, i->text, plen)))
533 j = appenditem(i, j);
534 for(i = allitems; i; i = i->next)
535 if(!i->matched && fstrstr(i->text, pattern))
536 j = appenditem(i, j);
537 curr = prev = next = sel = item;
538 calcoffsets();
539 }
540
541 void
542 readstdin(void) {
543 char *p, buf[1024];
544 unsigned int len = 0, max = 0;
545 Item *i, *new;
546
547 i = 0;
548 while(fgets(buf, sizeof buf, stdin)) {
549 len = strlen(buf);
550 if (buf[len - 1] == '\n')
551 buf[len - 1] = 0;
552 p = estrdup(buf);
553 if(max < len) {
554 maxname = p;
555 max = len;
556 }
557 new = emalloc(sizeof(Item));
558 new->next = new->left = new->right = NULL;
559 new->text = p;
560 if(!i)
561 allitems = new;
562 else
563 i->next = new;
564 i = new;
565 }
566 }
567
568 void
569 run(void) {
570 XEvent ev;
571
572 /* main event loop */
573 while(running && !XNextEvent(dpy, &ev))
574 switch (ev.type) {
575 default: /* ignore all crap */
576 break;
577 case KeyPress:
578 kpress(&ev.xkey);
579 break;
580 case Expose:
581 if(ev.xexpose.count == 0)
582 drawmenu();
583 break;
584 }
585 }
586
587 void
588 setup(int x, int y, int w) {
589 unsigned int i, j;
590 XModifierKeymap *modmap;
591 XSetWindowAttributes wa;
592
593 /* init modifier map */
594 modmap = XGetModifierMapping(dpy);
595 for(i = 0; i < 8; i++)
596 for(j = 0; j < modmap->max_keypermod; j++) {
597 if(modmap->modifiermap[i * modmap->max_keypermod + j]
598 == XKeysymToKeycode(dpy, XK_Num_Lock))
599 numlockmask = (1 << i);
600 }
601 XFreeModifiermap(modmap);
602
603 /* style */
604 dc.norm[ColBG] = getcolor(normbg);
605 dc.norm[ColFG] = getcolor(normfg);
606 dc.sel[ColBG] = getcolor(selbg);
607 dc.sel[ColFG] = getcolor(selfg);
608 initfont(font);
609
610 /* menu window */
611 wa.override_redirect = 1;
612 wa.background_pixmap = ParentRelative;
613 wa.event_mask = ExposureMask | ButtonPressMask | KeyPressMask;
614 mw = w ? w : DisplayWidth(dpy, screen);
615 mh = dc.font.height + 2;
616 if(y < 0) {
617 if(y == (int)(unsigned int)-1)
618 y = DisplayHeight(dpy, screen) - mh;
619 else
620 y = (-1 * y) - mh;
621 }
622 win = XCreateWindow(dpy, root, x, y, mw, mh, 0,
623 DefaultDepth(dpy, screen), CopyFromParent,
624 DefaultVisual(dpy, screen),
625 CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
626
627 /* pixmap */
628 dc.drawable = XCreatePixmap(dpy, root, mw, mh, DefaultDepth(dpy, screen));
629 dc.gc = XCreateGC(dpy, root, 0, 0);
630 XSetLineAttributes(dpy, dc.gc, 1, LineSolid, CapButt, JoinMiter);
631 if(!dc.font.set)
632 XSetFont(dpy, dc.gc, dc.font.xfont->fid);
633 if(maxname)
634 cmdw = textw(maxname);
635 if(cmdw > mw / 3)
636 cmdw = mw / 3;
637 if(prompt)
638 promptw = textw(prompt);
639 if(promptw > mw / 5)
640 promptw = mw / 5;
641 text[0] = 0;
642 match(text);
643 XMapRaised(dpy, win);
644 }
645
646 unsigned int
647 textnw(const char *text, unsigned int len) {
648 XRectangle r;
649
650 if(dc.font.set) {
651 XmbTextExtents(dc.font.set, text, len, NULL, &r);
652 return r.width;
653 }
654 return XTextWidth(dc.font.xfont, text, len);
655 }
656
657 unsigned int
658 textw(const char *text) {
659 return textnw(text, strlen(text)) + dc.font.height;
660 }
661
662 int
663 main(int argc, char *argv[]) {
664 int x = 0, y = 0, w = 0;
665 unsigned int i;
666
667 /* command line args */
668 for(i = 1; i < argc; i++)
669 if(!strcmp(argv[i], "-i")) {
670 fstrncmp = strncasecmp;
671 fstrstr = cistrstr;
672 }
673 else if(!strcmp(argv[i], "-fn")) {
674 if(++i < argc) font = argv[i];
675 }
676 else if(!strcmp(argv[i], "-nb")) {
677 if(++i < argc) normbg = argv[i];
678 }
679 else if(!strcmp(argv[i], "-nf")) {
680 if(++i < argc) normfg = argv[i];
681 }
682 else if(!strcmp(argv[i], "-p")) {
683 if(++i < argc) prompt = argv[i];
684 }
685 else if(!strcmp(argv[i], "-sb")) {
686 if(++i < argc) selbg = argv[i];
687 }
688 else if(!strcmp(argv[i], "-sf")) {
689 if(++i < argc) selfg = argv[i];
690 }
691 else if(!strcmp(argv[i], "-x")) {
692 if(++i < argc) x = atoi(argv[i]);
693 }
694 else if(!strcmp(argv[i], "-y")) {
695 if(++i < argc) {
696 if(!strcmp(argv[i], "-0"))
697 y = (int)(unsigned int)-1;
698 else
699 y = atoi(argv[i]);
700 }
701 }
702 else if(!strcmp(argv[i], "-w")) {
703 if(++i < argc) w = atoi(argv[i]);
704 }
705 else if(!strcmp(argv[i], "-v"))
706 eprint("dmenu-"VERSION", © 2006-2008 dmenu engineers, see LICENSE for details\n");
707 else
708 eprint("usage: dmenu [-i] [-fn <font>] [-nb <color>] [-nf <color>]\n"
709 " [-p <prompt>] [-sb <color>] [-sf <color>]\n"
710 " [-x <x>] [-y <y>] [-w <w>] [-v]\n");
711 setlocale(LC_CTYPE, "");
712 dpy = XOpenDisplay(0);
713 if(!dpy)
714 eprint("dmenu: cannot open display\n");
715 screen = DefaultScreen(dpy);
716 root = RootWindow(dpy, screen);
717
718 if(isatty(STDIN_FILENO)) {
719 readstdin();
720 running = grabkeyboard();
721 }
722 else { /* prevent keypress loss */
723 running = grabkeyboard();
724 readstdin();
725 }
726
727 setup(x, y, w);
728 drawmenu();
729 XSync(dpy, False);
730 run();
731 cleanup();
732 XCloseDisplay(dpy);
733 return ret;
734 }