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