fast grab patch (thanks Rob)
[dmenu.git] / dmenu.c
diff --git a/dmenu.c b/dmenu.c
index 384def6..2193f82 100644 (file)
--- a/dmenu.c
+++ b/dmenu.c
 #ifdef XINERAMA
 #include <X11/extensions/Xinerama.h>
 #endif
-#include <draw.h>
+#include "draw.h"
 
 #define INRECT(x,y,rx,ry,rw,rh) ((x) >= (rx) && (x) < (rx)+(rw) && (y) >= (ry) && (y) < (ry)+(rh))
-#define LINEH                   (dc->font.height + 2)
 #define MIN(a,b)                ((a) < (b) ? (a) : (b))
 #define MAX(a,b)                ((a) > (b) ? (a) : (b))
-#define UTF8_CODEPOINT(c)       (((c) & 0xc0) != 0x80)
 
 typedef struct Item Item;
 struct Item {
@@ -27,40 +25,103 @@ struct Item {
 
 static void appenditem(Item *item, Item **list, Item **last);
 static void calcoffsets(void);
-static char *cistrstr(const char *s, const char *sub);
 static void drawmenu(void);
+static char *fstrstr(const char *s, const char *sub);
 static void grabkeyboard(void);
 static void insert(const char *s, ssize_t n);
 static void keypress(XKeyEvent *ev);
 static void match(void);
+static size_t nextrune(int incr);
 static void paste(void);
 static void readstdin(void);
 static void run(void);
 static void setup(void);
-static void usage(void);
 
-static char text[BUFSIZ];
+static char text[BUFSIZ] = "";
+static int bh, mw, mh;
+static int inputw = 0;
+static int lines = 0;
+static int monitor = -1;
+static int promptw;
 static size_t cursor = 0;
+static const char *font = NULL;
 static const char *prompt = NULL;
 static const char *normbgcolor = "#cccccc";
 static const char *normfgcolor = "#000000";
 static const char *selbgcolor  = "#0066ff";
 static const char *selfgcolor  = "#ffffff";
-static unsigned int inputw = 0;
-static unsigned int lines = 0;
-static unsigned int mw, mh;
-static unsigned int promptw;
 static unsigned long normcol[ColLast];
 static unsigned long selcol[ColLast];
 static Atom utf8;
 static Bool topbar = True;
 static DC *dc;
-static Item *allitems, *matches;
-static Item *curr, *prev, *next, *sel;
+static Item *items = NULL;
+static Item *matches, *sel;
+static Item *prev, *curr, *next;
 static Window root, win;
 
 static int (*fstrncmp)(const char *, const char *, size_t) = strncmp;
-static char *(*fstrstr)(const char *, const char *) = strstr;
+
+int
+main(int argc, char *argv[]) {
+       Bool fast = False;
+       int i;
+
+       progname = "dmenu";
+       for(i = 1; i < argc; i++)
+               /* single flags */
+               if(!strcmp(argv[i], "-v")) {
+                       fputs("dmenu-"VERSION", © 2006-2011 dmenu engineers, see LICENSE for details\n", stdout);
+                       exit(EXIT_SUCCESS);
+               }
+               else if(!strcmp(argv[i], "-b"))
+                       topbar = False;
+               else if(!strcmp(argv[i], "-i"))
+                       fstrncmp = strncasecmp;
+               else if(!strcmp(argv[i], "-f"))
+                       fast = True;
+               else if(i == argc-1)
+                       goto usage;
+               /* double flags */
+               else if(!strcmp(argv[i], "-l"))
+                       lines = atoi(argv[++i]);
+               else if(!strcmp(argv[i], "-m"))
+                       monitor = atoi(argv[++i]);
+               else if(!strcmp(argv[i], "-p"))
+                       prompt = argv[++i];
+               else if(!strcmp(argv[i], "-fn"))
+                       font = argv[++i];
+               else if(!strcmp(argv[i], "-nb"))
+                       normbgcolor = argv[++i];
+               else if(!strcmp(argv[i], "-nf"))
+                       normfgcolor = argv[++i];
+               else if(!strcmp(argv[i], "-sb"))
+                       selbgcolor = argv[++i];
+               else if(!strcmp(argv[i], "-sf"))
+                       selfgcolor = argv[++i];
+               else
+                       goto usage;
+
+       dc = initdc();
+       initfont(dc, font);
+
+       if(fast) {
+               setup();
+               readstdin();
+       }
+       else {
+               readstdin();
+               setup();
+       }
+       match();
+       run();
+       return EXIT_FAILURE;
+
+usage:
+       fputs("usage: dmenu [-b] [-f] [-i] [-l lines] [-m monitor] [-p prompt] [-fn font]\n"
+             "             [-nb color] [-nf color] [-sb color] [-sf color] [-v]\n", stderr);
+       return EXIT_FAILURE;
+}
 
 void
 appenditem(Item *item, Item **list, Item **last) {
@@ -78,25 +139,16 @@ calcoffsets(void) {
        unsigned int i, n;
 
        if(lines > 0)
-               n = lines * LINEH;
+               n = lines * bh;
        else
                n = mw - (promptw + inputw + textw(dc, "<") + textw(dc, ">"));
 
-       for(i = 0, next = curr; i <= n && next; next = next->right)
-               i += (lines > 0) ? LINEH : MIN(textw(dc, next->text), mw/3);
-
-       for(i = 0, prev = curr; i <= n && prev && prev->left; prev = prev->left)
-               i += (lines > 0) ? LINEH : MIN(textw(dc, prev->left->text), mw/3);
-}
-
-char *
-cistrstr(const char *s, const char *sub) {
-       size_t len;
-
-       for(len = strlen(sub); *s; s++)
-               if(!strncasecmp(s, sub, len))
-                       return (char *)s;
-       return NULL;
+       for(i = 0, next = curr; next; next = next->right)
+               if((i += (lines > 0) ? bh : MIN(textw(dc, next->text), n)) > n)
+                       break;
+       for(i = 0, prev = curr; prev && prev->left; prev = prev->left)
+               if((i += (lines > 0) ? bh : MIN(textw(dc, prev->left->text), n)) > n)
+                       break;
 }
 
 void
@@ -106,8 +158,8 @@ drawmenu(void) {
 
        dc->x = 0;
        dc->y = 0;
-       dc->h = LINEH;
-       drawrect(dc, 0, 0, mw, mh, BG(dc, normcol));
+       dc->h = bh;
+       drawrect(dc, 0, 0, mw, mh, True, BG(dc, normcol));
 
        if(prompt) {
                dc->w = promptw;
@@ -117,7 +169,7 @@ drawmenu(void) {
        dc->w = (lines > 0 || !matches) ? mw - dc->x : inputw;
        drawtext(dc, text, normcol);
        if((curpos = textnw(dc, text, cursor) + dc->h/2 - 2) < dc->w)
-               drawrect(dc, curpos, 2, 1, dc->h - 4, FG(dc, normcol));
+               drawrect(dc, curpos, 2, 1, dc->h - 4, True, FG(dc, normcol));
 
        if(lines > 0) {
                dc->w = mw - dc->x;
@@ -133,7 +185,7 @@ drawmenu(void) {
                        drawtext(dc, "<", normcol);
                for(item = curr; item != next; item = item->right) {
                        dc->x += dc->w;
-                       dc->w = MIN(textw(dc, item->text), mw/3);
+                       dc->w = MIN(textw(dc, item->text), mw - dc->x - textw(dc, ">"));
                        drawtext(dc, item->text, (item == sel) ? selcol : normcol);
                }
                dc->w = textw(dc, ">");
@@ -141,7 +193,17 @@ drawmenu(void) {
                if(next)
                        drawtext(dc, ">", normcol);
        }
-       commitdraw(dc, win);
+       mapdc(dc, win, mw, mh);
+}
+
+char *
+fstrstr(const char *s, const char *sub) {
+       size_t len;
+
+       for(len = strlen(sub); *s; s++)
+               if(!fstrncmp(s, sub, len))
+                       return (char *)s;
+       return NULL;
 }
 
 void
@@ -158,7 +220,9 @@ grabkeyboard(void) {
 
 void
 insert(const char *s, ssize_t n) {
-       memmove(text + cursor + n, text + cursor, sizeof text - cursor - n);
+       if(strlen(text) + n > sizeof text - 1)
+               return;
+       memmove(text + cursor + n, text + cursor, sizeof text - cursor - MAX(n, 0));
        if(n > 0)
                memcpy(text + cursor, s, n);
        cursor += n;
@@ -168,13 +232,12 @@ insert(const char *s, ssize_t n) {
 void
 keypress(XKeyEvent *ev) {
        char buf[32];
-       int n;
        size_t len;
        KeySym ksym;
 
        len = strlen(text);
        XLookupString(ev, buf, sizeof buf, &ksym, NULL);
-       if(ev->state & ControlMask) {
+       if(ev->state & ControlMask)
                switch(tolower(ksym)) {
                default:
                        return;
@@ -216,38 +279,30 @@ keypress(XKeyEvent *ev) {
                        ksym = XK_Up;
                        break;
                case XK_u:  /* delete left */
-                       insert(NULL, -cursor);
+                       insert(NULL, 0 - cursor);
                        break;
                case XK_w:  /* delete word */
-                       if(cursor == 0)
-                               return;
-                       n = 0;
-                       while(cursor - n++ > 0 && text[cursor - n] == ' ');
-                       while(cursor - n++ > 0 && text[cursor - n] != ' ');
-                       insert(NULL, 1-n);
+                       while(cursor > 0 && text[nextrune(-1)] == ' ')
+                               insert(NULL, nextrune(-1) - cursor);
+                       while(cursor > 0 && text[nextrune(-1)] != ' ')
+                               insert(NULL, nextrune(-1) - cursor);
                        break;
                case XK_y:  /* paste selection */
-                       XConvertSelection(dc->dpy, XA_PRIMARY, utf8, None, win, CurrentTime);
+                       XConvertSelection(dc->dpy, XA_PRIMARY, utf8, utf8, win, CurrentTime);
                        return;
                }
-       }
        switch(ksym) {
        default:
-               if(isprint(*buf))
-                       insert(buf, MIN(strlen(buf), sizeof text - cursor));
-               break;
-       case XK_BackSpace:
-               if(cursor == 0)
-                       return;
-               for(n = 1; cursor - n > 0 && !UTF8_CODEPOINT(text[cursor - n]); n++);
-               insert(NULL, -n);
+               if(!iscntrl(*buf))
+                       insert(buf, strlen(buf));
                break;
        case XK_Delete:
                if(cursor == len)
                        return;
-               for(n = 1; cursor + n < len && !UTF8_CODEPOINT(text[cursor + n]); n++);
-               cursor += n;
-               insert(NULL, -n);
+               cursor = nextrune(+1);
+       case XK_BackSpace:
+               if(cursor > 0)
+                       insert(NULL, nextrune(-1) - cursor);
                break;
        case XK_End:
                if(cursor < len) {
@@ -273,15 +328,13 @@ keypress(XKeyEvent *ev) {
                break;
        case XK_Left:
                if(cursor > 0 && (!sel || !sel->left || lines > 0)) {
-                       while(cursor-- > 0 && !UTF8_CODEPOINT(text[cursor]));
+                       cursor = nextrune(-1);
                        break;
                }
                else if(lines > 0)
                        return;
        case XK_Up:
-               if(!sel || !sel->left)
-                       return;
-               if((sel = sel->left)->right == curr) {
+               if(sel && sel->left && (sel = sel->left)->right == curr) {
                        curr = prev;
                        calcoffsets();
                }
@@ -301,19 +354,16 @@ keypress(XKeyEvent *ev) {
        case XK_Return:
        case XK_KP_Enter:
                fputs((sel && !(ev->state & ShiftMask)) ? sel->text : text, stdout);
-               fflush(stdout);
                exit(EXIT_SUCCESS);
        case XK_Right:
                if(cursor < len) {
-                       while(cursor++ < len && !UTF8_CODEPOINT(text[cursor]));
+                       cursor = nextrune(+1);
                        break;
                }
                else if(lines > 0)
                        return;
        case XK_Down:
-               if(!sel || !sel->right)
-                       return;
-               if((sel = sel->right) == next) {
+               if(sel && sel->right && (sel = sel->right) == next) {
                        curr = next;
                        calcoffsets();
                }
@@ -336,13 +386,14 @@ match(void) {
 
        len = strlen(text);
        matches = lexact = lprefix = lsubstr = itemend = exactend = prefixend = substrend = NULL;
-       for(item = allitems; item; item = item->next)
+       for(item = items; item; item = item->next)
                if(!fstrncmp(text, item->text, len + 1))
                        appenditem(item, &lexact, &exactend);
                else if(!fstrncmp(text, item->text, len))
                        appenditem(item, &lprefix, &prefixend);
                else if(fstrstr(item->text, text))
                        appenditem(item, &lsubstr, &substrend);
+
        if(lexact) {
                matches = lexact;
                itemend = exactend;
@@ -364,10 +415,19 @@ match(void) {
                else
                        matches = lsubstr;
        }
-       curr = prev = next = sel = matches;
+       curr = sel = matches;
        calcoffsets();
 }
 
+size_t
+nextrune(int incr) {
+       size_t n, len;
+
+       len = strlen(text);
+       for(n = cursor + incr; n >= 0 && n < len && (text[n] & 0xc0) == 0x80; n += incr);
+       return n;
+}
+
 void
 paste(void) {
        char *p, *q;
@@ -375,7 +435,7 @@ paste(void) {
        unsigned long dl;
        Atom da;
 
-       XGetWindowProperty(dc->dpy, win, utf8, 0, sizeof text - cursor, True,
+       XGetWindowProperty(dc->dpy, win, utf8, 0, (sizeof text / 4) + 1, False,
                           utf8, &da, &di, &dl, &dl, (unsigned char **)&p);
        insert(p, (q = strchr(p, '\n')) ? q-p : strlen(p));
        XFree(p);
@@ -385,22 +445,16 @@ paste(void) {
 void
 readstdin(void) {
        char buf[sizeof text], *p;
-       Item *item, *new;
+       Item *item, **end;
 
-       allitems = NULL;
-       for(item = NULL; fgets(buf, sizeof buf, stdin); item = new) {
+       for(end = &items; fgets(buf, sizeof buf, stdin); *end = item, end = &item->next) {
                if((p = strchr(buf, '\n')))
                        *p = '\0';
-               if(!(new = malloc(sizeof *new)))
-                       eprintf("cannot malloc %u bytes\n", sizeof *new);
-               if(!(new->text = strdup(buf)))
-                       eprintf("cannot strdup %u bytes\n", strlen(buf)+1);
-               inputw = MAX(inputw, textw(dc, new->text));
-               new->next = new->left = new->right = NULL;
-               if(item)
-                       item->next = new;
-               else
-                       allitems = new;
+               if(!(item = calloc(1, sizeof *item)))
+                       eprintf("cannot malloc %u bytes:", sizeof *item);
+               if(!(item->text = strdup(buf)))
+                       eprintf("cannot strdup %u bytes:", strlen(buf)+1);
+               inputw = MAX(inputw, textw(dc, item->text));
        }
 }
 
@@ -447,7 +501,9 @@ setup(void) {
        selcol[ColFG] = getcolor(dc, selfgcolor);
 
        /* menu geometry */
-       mh = (lines + 1) * LINEH;
+       bh = dc->font.height + 2;
+       lines = MAX(lines, 0);
+       mh = (lines + 1) * bh;
 #ifdef XINERAMA
        if((info = XineramaQueryScreens(dc->dpy, &n))) {
                int i, di;
@@ -456,7 +512,8 @@ setup(void) {
 
                XQueryPointer(dc->dpy, root, &dw, &dw, &x, &y, &di, &di, &du);
                for(i = 0; i < n; i++)
-                       if(INRECT(x, y, info[i].x_org, info[i].y_org, info[i].width, info[i].height))
+                       if((monitor == info[i].screen_number)
+                       || (monitor < 0 && INRECT(x, y, info[i].x_org, info[i].y_org, info[i].width, info[i].height)))
                                break;
                x = info[i].x_org;
                y = info[i].y_org + (topbar ? 0 : info[i].height - mh);
@@ -480,63 +537,8 @@ setup(void) {
                            CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
 
        grabkeyboard();
-       setcanvas(dc, mw, mh);
+       resizedc(dc, mw, mh);
        inputw = MIN(inputw, mw/3);
-       promptw = prompt ? MIN(textw(dc, prompt), mw/5) : 0;
+       promptw = prompt ? textw(dc, prompt) : 0;
        XMapRaised(dc->dpy, win);
-       text[0] = '\0';
-       match();
-}
-
-void
-usage(void) {
-       fputs("usage: dmenu [-b] [-i] [-l lines] [-p prompt] [-fn font] [-nb color]\n"
-             "             [-nf color] [-sb color] [-sf color] [-v]\n", stderr);
-       exit(EXIT_FAILURE);
-}
-
-int
-main(int argc, char *argv[]) {
-       int i;
-
-       progname = "dmenu";
-       dc = initdraw();
-
-       for(i = 1; i < argc; i++)
-               /* single flags */
-               if(!strcmp(argv[i], "-v")) {
-                       fputs("dmenu-"VERSION", © 2006-2010 dmenu engineers, see LICENSE for details\n", stdout);
-                       exit(EXIT_SUCCESS);
-               }
-               else if(!strcmp(argv[i], "-b"))
-                       topbar = False;
-               else if(!strcmp(argv[i], "-i")) {
-                       fstrncmp = strncasecmp;
-                       fstrstr = cistrstr;
-               }
-               else if(i == argc-1)
-                       usage();
-               /* double flags */
-               else if(!strcmp(argv[i], "-l"))
-                       lines = atoi(argv[++i]);
-               else if(!strcmp(argv[i], "-p"))
-                       prompt = argv[++i];
-               else if(!strcmp(argv[i], "-fn"))
-                       initfont(dc, argv[++i]);
-               else if(!strcmp(argv[i], "-nb"))
-                       normbgcolor = argv[++i];
-               else if(!strcmp(argv[i], "-nf"))
-                       normfgcolor = argv[++i];
-               else if(!strcmp(argv[i], "-sb"))
-                       selbgcolor = argv[++i];
-               else if(!strcmp(argv[i], "-sf"))
-                       selfgcolor = argv[++i];
-               else
-                       usage();
-
-       readstdin();
-       setup();
-       run();
-
-       return EXIT_FAILURE;  /* should not reach */
 }