Xinqi Bao's Git

projects / dmenu.git / blob


cee73b25254c944eb4e3f8d714290c8bad2268e9
[dmenu.git] / dmenu.c
1 /* See LICENSE file for copyright and license details. */
2 #include <ctype.h>
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <string.h>
6 #include <strings.h>
7 #include <unistd.h>
8 #include <X11/Xlib.h>
9 #include <X11/Xatom.h>
10 #include <X11/Xutil.h>
11 #ifdef XINERAMA
12 #include <X11/extensions/Xinerama.h>
13 #endif
14 #include "draw.h"
15
16 #define INRECT(x,y,rx,ry,rw,rh) ((x) >= (rx) && (x) < (rx)+(rw) && (y) >= (ry) && (y) < (ry)+(rh))
17 #define MIN(a,b) ((a) < (b) ? (a) : (b))
18 #define MAX(a,b) ((a) > (b) ? (a) : (b))
19
20 typedef struct Item Item;
21 struct Item {
22 char *text;
23 Item *left, *right;
24 };
25
26 static void appenditem(Item *item, Item **list, Item **last);
27 static void calcoffsets(void);
28 static void drawmenu(void);
29 static char *fstrstr(const char *s, const char *sub);
30 static void grabkeyboard(void);
31 static void insert(const char *str, ssize_t n);
32 static void keypress(XKeyEvent *ev);
33 static void match(Bool sub);
34 static size_t nextrune(int incr);
35 static void paste(void);
36 static void readstdin(void);
37 static void run(void);
38 static void setup(void);
39
40 static char text[BUFSIZ] = "";
41 static int bh, mw, mh;
42 static int inputw;
43 static int lines = 0;
44 static int monitor = -1;
45 static int promptw;
46 static size_t cursor = 0;
47 static const char *font = NULL;
48 static const char *prompt = NULL;
49 static const char *normbgcolor = "#cccccc";
50 static const char *normfgcolor = "#000000";
51 static const char *selbgcolor = "#0066ff";
52 static const char *selfgcolor = "#ffffff";
53 static unsigned long normcol[ColLast];
54 static unsigned long selcol[ColLast];
55 static Atom utf8;
56 static Bool topbar = True;
57 static DC *dc;
58 static Item *items = NULL;
59 static Item *matches, *matchend;
60 static Item *prev, *curr, *next, *sel;
61 static Window win;
62
63 static int (*fstrncmp)(const char *, const char *, size_t) = strncmp;
64
65 int
66 main(int argc, char *argv[]) {
67 Bool fast = False;
68 int i;
69
70 for(i = 1; i < argc; i++)
71 /* single flags */
72 if(!strcmp(argv[i], "-v")) {
73 fputs("dmenu-"VERSION", © 2006-2011 dmenu engineers, see LICENSE for details\n", stdout);
74 exit(EXIT_SUCCESS);
75 }
76 else if(!strcmp(argv[i], "-b"))
77 topbar = False;
78 else if(!strcmp(argv[i], "-f"))
79 fast = True;
80 else if(!strcmp(argv[i], "-i"))
81 fstrncmp = strncasecmp;
82 else if(i+1 == argc)
83 goto usage;
84 /* double flags */
85 else if(!strcmp(argv[i], "-l"))
86 lines = atoi(argv[++i]);
87 else if(!strcmp(argv[i], "-m"))
88 monitor = atoi(argv[++i]);
89 else if(!strcmp(argv[i], "-p"))
90 prompt = argv[++i];
91 else if(!strcmp(argv[i], "-fn"))
92 font = argv[++i];
93 else if(!strcmp(argv[i], "-nb"))
94 normbgcolor = argv[++i];
95 else if(!strcmp(argv[i], "-nf"))
96 normfgcolor = argv[++i];
97 else if(!strcmp(argv[i], "-sb"))
98 selbgcolor = argv[++i];
99 else if(!strcmp(argv[i], "-sf"))
100 selfgcolor = argv[++i];
101 else
102 goto usage;
103
104 dc = initdc();
105 initfont(dc, font);
106
107 if(fast) {
108 grabkeyboard();
109 readstdin();
110 }
111 else {
112 readstdin();
113 grabkeyboard();
114 }
115 setup();
116 run();
117 return EXIT_FAILURE;
118
119 usage:
120 fputs("usage: dmenu [-b] [-f] [-i] [-l lines] [-m monitor] [-p prompt] [-fn font]\n"
121 " [-nb color] [-nf color] [-sb color] [-sf color] [-v]\n", stderr);
122 return EXIT_FAILURE;
123 }
124
125 void
126 appenditem(Item *item, Item **list, Item **last) {
127 if(!*last)
128 *list = item;
129 else
130 (*last)->right = item;
131 item->left = *last;
132 item->right = NULL;
133 *last = item;
134 }
135
136 void
137 calcoffsets(void) {
138 unsigned int i, n;
139
140 if(lines > 0)
141 n = lines * bh;
142 else
143 n = mw - (promptw + inputw + textw(dc, "<") + textw(dc, ">"));
144
145 for(i = 0, next = curr; next; next = next->right)
146 if((i += (lines > 0) ? bh : MIN(textw(dc, next->text), n)) > n)
147 break;
148 for(i = 0, prev = curr; prev && prev->left; prev = prev->left)
149 if((i += (lines > 0) ? bh : MIN(textw(dc, prev->left->text), n)) > n)
150 break;
151 }
152
153 void
154 drawmenu(void) {
155 int curpos;
156 Item *item;
157
158 dc->x = 0;
159 dc->y = 0;
160 dc->h = bh;
161 drawrect(dc, 0, 0, mw, mh, True, BG(dc, normcol));
162
163 if(prompt) {
164 dc->w = promptw;
165 drawtext(dc, prompt, selcol);
166 dc->x = dc->w;
167 }
168 dc->w = (lines > 0 || !matches) ? mw - dc->x : inputw;
169 drawtext(dc, text, normcol);
170 if((curpos = textnw(dc, text, cursor) + dc->h/2 - 2) < dc->w)
171 drawrect(dc, curpos, 2, 1, dc->h - 4, True, FG(dc, normcol));
172
173 if(lines > 0) {
174 dc->w = mw - dc->x;
175 for(item = curr; item != next; item = item->right) {
176 dc->y += dc->h;
177 drawtext(dc, item->text, (item == sel) ? selcol : normcol);
178 }
179 }
180 else if(matches) {
181 dc->x += inputw;
182 dc->w = textw(dc, "<");
183 if(curr->left)
184 drawtext(dc, "<", normcol);
185 for(item = curr; item != next; item = item->right) {
186 dc->x += dc->w;
187 dc->w = MIN(textw(dc, item->text), mw - dc->x - textw(dc, ">"));
188 drawtext(dc, item->text, (item == sel) ? selcol : normcol);
189 }
190 dc->w = textw(dc, ">");
191 dc->x = mw - dc->w;
192 if(next)
193 drawtext(dc, ">", normcol);
194 }
195 mapdc(dc, win, mw, mh);
196 }
197
198 char *
199 fstrstr(const char *s, const char *sub) {
200 size_t len;
201
202 for(len = strlen(sub); *s; s++)
203 if(!fstrncmp(s, sub, len))
204 return (char *)s;
205 return NULL;
206 }
207
208 void
209 grabkeyboard(void) {
210 int i;
211
212 for(i = 0; i < 1000; i++) {
213 if(XGrabKeyboard(dc->dpy, DefaultRootWindow(dc->dpy), True,
214 GrabModeAsync, GrabModeAsync, CurrentTime) == GrabSuccess)
215 return;
216 usleep(1000);
217 }
218 eprintf("cannot grab keyboard\n");
219 }
220
221 void
222 insert(const char *str, ssize_t n) {
223 if(strlen(text) + n > sizeof text - 1)
224 return;
225 memmove(&text[cursor + n], &text[cursor], sizeof text - cursor - MAX(n, 0));
226 if(n > 0)
227 memcpy(&text[cursor], str, n);
228 cursor += n;
229 match(n > 0);
230 }
231
232 void
233 keypress(XKeyEvent *ev) {
234 char buf[32];
235 KeySym ksym;
236
237 XLookupString(ev, buf, sizeof buf, &ksym, NULL);
238 if(ev->state & ControlMask) {
239 KeySym lower, upper;
240
241 XConvertCase(ksym, &lower, &upper);
242 switch(lower) {
243 default:
244 return;
245 case XK_a:
246 ksym = XK_Home;
247 break;
248 case XK_b:
249 ksym = XK_Left;
250 break;
251 case XK_c:
252 ksym = XK_Escape;
253 break;
254 case XK_d:
255 ksym = XK_Delete;
256 break;
257 case XK_e:
258 ksym = XK_End;
259 break;
260 case XK_f:
261 ksym = XK_Right;
262 break;
263 case XK_h:
264 ksym = XK_BackSpace;
265 break;
266 case XK_i:
267 ksym = XK_Tab;
268 break;
269 case XK_j:
270 ksym = XK_Return;
271 break;
272 case XK_k: /* delete right */
273 text[cursor] = '\0';
274 match(False);
275 break;
276 case XK_n:
277 ksym = XK_Down;
278 break;
279 case XK_p:
280 ksym = XK_Up;
281 break;
282 case XK_u: /* delete left */
283 insert(NULL, 0 - cursor);
284 break;
285 case XK_w: /* delete word */
286 while(cursor > 0 && text[nextrune(-1)] == ' ')
287 insert(NULL, nextrune(-1) - cursor);
288 while(cursor > 0 && text[nextrune(-1)] != ' ')
289 insert(NULL, nextrune(-1) - cursor);
290 break;
291 case XK_y: /* paste selection */
292 XConvertSelection(dc->dpy, XA_PRIMARY, utf8, utf8, win, CurrentTime);
293 return;
294 }
295 }
296 switch(ksym) {
297 default:
298 if(!iscntrl(*buf))
299 insert(buf, strlen(buf));
300 break;
301 case XK_Delete:
302 if(text[cursor] == '\0')
303 return;
304 cursor = nextrune(+1);
305 case XK_BackSpace:
306 if(cursor > 0)
307 insert(NULL, nextrune(-1) - cursor);
308 break;
309 case XK_End:
310 if(text[cursor] != '\0') {
311 cursor = strlen(text);
312 break;
313 }
314 if(next) {
315 curr = matchend;
316 calcoffsets();
317 curr = prev;
318 calcoffsets();
319 while(next && (curr = curr->right))
320 calcoffsets();
321 }
322 sel = matchend;
323 break;
324 case XK_Escape:
325 exit(EXIT_FAILURE);
326 case XK_Home:
327 if(sel == matches) {
328 cursor = 0;
329 break;
330 }
331 sel = curr = matches;
332 calcoffsets();
333 break;
334 case XK_Left:
335 if(cursor > 0 && (!sel || !sel->left || lines > 0)) {
336 cursor = nextrune(-1);
337 break;
338 }
339 else if(lines > 0)
340 return;
341 case XK_Up:
342 if(sel && sel->left && (sel = sel->left)->right == curr) {
343 curr = prev;
344 calcoffsets();
345 }
346 break;
347 case XK_Next:
348 if(!next)
349 return;
350 sel = curr = next;
351 calcoffsets();
352 break;
353 case XK_Prior:
354 if(!prev)
355 return;
356 sel = curr = prev;
357 calcoffsets();
358 break;
359 case XK_Return:
360 case XK_KP_Enter:
361 fputs((sel && !(ev->state & ShiftMask)) ? sel->text : text, stdout);
362 exit(EXIT_SUCCESS);
363 case XK_Right:
364 if(text[cursor] != '\0') {
365 cursor = nextrune(+1);
366 break;
367 }
368 else if(lines > 0)
369 return;
370 case XK_Down:
371 if(sel && sel->right && (sel = sel->right) == next) {
372 curr = next;
373 calcoffsets();
374 }
375 break;
376 case XK_Tab:
377 if(!sel)
378 return;
379 strncpy(text, sel->text, sizeof text);
380 cursor = strlen(text);
381 match(True);
382 break;
383 }
384 drawmenu();
385 }
386
387 void
388 match(Bool sub) {
389 size_t len = strlen(text);
390 Item *lexact, *lprefix, *lsubstr, *exactend, *prefixend, *substrend;
391 Item *item, *next;
392
393 lexact = lprefix = lsubstr = exactend = prefixend = substrend = NULL;
394 for(item = sub ? matches : items; item && item->text; item = next) {
395 next = sub ? item->right : item + 1;
396 if(!fstrncmp(text, item->text, len + 1))
397 appenditem(item, &lexact, &exactend);
398 else if(!fstrncmp(text, item->text, len))
399 appenditem(item, &lprefix, &prefixend);
400 else if(fstrstr(item->text, text))
401 appenditem(item, &lsubstr, &substrend);
402 }
403 matches = lexact;
404 matchend = exactend;
405
406 if(lprefix) {
407 if(matchend) {
408 matchend->right = lprefix;
409 lprefix->left = matchend;
410 }
411 else
412 matches = lprefix;
413 matchend = prefixend;
414 }
415 if(lsubstr) {
416 if(matchend) {
417 matchend->right = lsubstr;
418 lsubstr->left = matchend;
419 }
420 else
421 matches = lsubstr;
422 matchend = substrend;
423 }
424 curr = sel = matches;
425 calcoffsets();
426 }
427
428 size_t
429 nextrune(int incr) {
430 size_t n, len = strlen(text);
431
432 for(n = cursor + incr; n >= 0 && n < len && (text[n] & 0xc0) == 0x80; n += incr);
433 return n;
434 }
435
436 void
437 paste(void) {
438 char *p, *q;
439 int di;
440 unsigned long dl;
441 Atom da;
442
443 XGetWindowProperty(dc->dpy, win, utf8, 0, (sizeof text / 4) + 1, False,
444 utf8, &da, &di, &dl, &dl, (unsigned char **)&p);
445 insert(p, (q = strchr(p, '\n')) ? q-p : strlen(p));
446 XFree(p);
447 drawmenu();
448 }
449
450 void
451 readstdin(void) {
452 char buf[sizeof text], *p, *maxstr = NULL;
453 size_t i, max = 0, size = 0;
454
455 for(i = 0; fgets(buf, sizeof buf, stdin); items[++i].text = NULL) {
456 if(i+1 >= size / sizeof *items)
457 if(!(items = realloc(items, (size += BUFSIZ))))
458 eprintf("cannot realloc %u bytes:", size);
459 if((p = strchr(buf, '\n')))
460 *p = '\0';
461 if(!(items[i].text = strdup(buf)))
462 eprintf("cannot strdup %u bytes:", strlen(buf)+1);
463 if(strlen(items[i].text) > max)
464 max = strlen(maxstr = items[i].text);
465 }
466 inputw = maxstr ? textw(dc, maxstr) : 0;
467 }
468
469 void
470 run(void) {
471 XEvent ev;
472
473 while(!XNextEvent(dc->dpy, &ev))
474 switch(ev.type) {
475 case Expose:
476 if(ev.xexpose.count == 0)
477 drawmenu();
478 break;
479 case KeyPress:
480 keypress(&ev.xkey);
481 break;
482 case SelectionNotify:
483 if(ev.xselection.property == utf8)
484 paste();
485 break;
486 case VisibilityNotify:
487 if(ev.xvisibility.state != VisibilityUnobscured)
488 XRaiseWindow(dc->dpy, win);
489 break;
490 }
491 }
492
493 void
494 setup(void) {
495 int x, y, screen = DefaultScreen(dc->dpy);
496 Window root = RootWindow(dc->dpy, screen);
497 XSetWindowAttributes wa;
498 #ifdef XINERAMA
499 int n;
500 XineramaScreenInfo *info;
501 #endif
502
503 normcol[ColBG] = getcolor(dc, normbgcolor);
504 normcol[ColFG] = getcolor(dc, normfgcolor);
505 selcol[ColBG] = getcolor(dc, selbgcolor);
506 selcol[ColFG] = getcolor(dc, selfgcolor);
507
508 utf8 = XInternAtom(dc->dpy, "UTF8_STRING", False);
509
510 /* menu geometry */
511 bh = dc->font.height + 2;
512 lines = MAX(lines, 0);
513 mh = (lines + 1) * bh;
514 #ifdef XINERAMA
515 if((info = XineramaQueryScreens(dc->dpy, &n))) {
516 int i, di;
517 unsigned int du;
518 Window dw;
519
520 XQueryPointer(dc->dpy, root, &dw, &dw, &x, &y, &di, &di, &du);
521 for(i = 0; i < n-1; i++)
522 if((monitor == info[i].screen_number)
523 || (monitor < 0 && INRECT(x, y, info[i].x_org, info[i].y_org, info[i].width, info[i].height)))
524 break;
525 x = info[i].x_org;
526 y = info[i].y_org + (topbar ? 0 : info[i].height - mh);
527 mw = info[i].width;
528 XFree(info);
529 }
530 else
531 #endif
532 {
533 x = 0;
534 y = topbar ? 0 : DisplayHeight(dc->dpy, screen) - mh;
535 mw = DisplayWidth(dc->dpy, screen);
536 }
537 inputw = MIN(inputw, mw/3);
538 promptw = prompt ? textw(dc, prompt) : 0;
539 match(False);
540
541 /* menu window */
542 wa.override_redirect = True;
543 wa.background_pixmap = ParentRelative;
544 wa.event_mask = ExposureMask | KeyPressMask | VisibilityChangeMask;
545 win = XCreateWindow(dc->dpy, root, x, y, mw, mh, 0,
546 DefaultDepth(dc->dpy, screen), CopyFromParent,
547 DefaultVisual(dc->dpy, screen),
548 CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
549
550 XMapRaised(dc->dpy, win);
551 resizedc(dc, mw, mh);
552 drawmenu();
553 }