]> git.armaanb.net Git - dmenu.git/blobdiff - dmenu.c
Switch to dark theme
[dmenu.git] / dmenu.c
diff --git a/dmenu.c b/dmenu.c
index 7745947dd5aa8a2498a71b194e64ffe9f04070d0..1560b60466856fb86992f7f727bed73fd0dbb71e 100644 (file)
--- a/dmenu.c
+++ b/dmenu.c
@@ -1,11 +1,13 @@
 /* See LICENSE file for copyright and license details. */
 #include <ctype.h>
 #include <locale.h>
+#include <math.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <strings.h>
 #include <time.h>
+#include <unistd.h>
 
 #include <X11/Xlib.h>
 #include <X11/Xatom.h>
                              * MAX(0, MIN((y)+(h),(r).y_org+(r).height) - MAX((y),(r).y_org)))
 #define LENGTH(X)             (sizeof X / sizeof X[0])
 #define TEXTW(X)              (drw_fontset_getwidth(drw, (X)) + lrpad)
-
-#ifdef __OpenBSD__
-#include <unistd.h>
-#else
-#define pledge(a,b) 0
-#endif
+#define NUMBERSMAXDIGITS      100
+#define NUMBERSBUFSIZE        (NUMBERSMAXDIGITS * 2) + 1
 
 /* enums */
 enum { SchemeNorm, SchemeSel, SchemeOut, SchemeLast }; /* color schemes */
 
+
 struct item {
        char *text;
        struct item *left, *right;
        int out;
+       double distance;
 };
 
+static char numbers[NUMBERSBUFSIZE] = "";
 static char text[BUFSIZ] = "";
 static char *embed;
 static int bh, mw, mh;
-static int inputw = 0, promptw;
+static int inputw = 0, promptw, passwd = 0;
 static int lrpad; /* sum of left and right padding */
 static size_t cursor;
 static struct item *items = NULL;
@@ -118,9 +119,47 @@ cistrstr(const char *s, const char *sub)
        return NULL;
 }
 
+static void
+drawhighlights(struct item *item, int x, int y, int maxw)
+{
+       int i, indent;
+       char *highlight;
+       char c;
+
+       if (!(strlen(item->text) && strlen(text)))
+               return;
+
+       drw_setscheme(drw, scheme[item == sel ? SchemeSel : SchemeNorm]);
+       for (i = 0, highlight = item->text; *highlight && text[i];) {
+               if (*highlight == text[i]) {
+                       /* get indentation */
+                       c = *highlight;
+                       *highlight = '\0';
+                       indent = TEXTW(item->text);
+                       *highlight = c;
+
+                       /* highlight character */
+                       c = highlight[1];
+                       highlight[1] = '\0';
+                       drw_text(
+                               drw,
+                               x + indent - (lrpad / 2),
+                               y,
+                               MIN(maxw - indent, TEXTW(highlight) - lrpad),
+                               bh, 0, highlight, 0
+                       );
+                       highlight[1] = c;
+                       i++;
+               }
+               highlight++;
+       }
+}
+
+
 static int
 drawitem(struct item *item, int x, int y, int w)
 {
+       int r;
        if (item == sel)
                drw_setscheme(drw, scheme[SchemeSel]);
        else if (item->out)
@@ -128,7 +167,24 @@ drawitem(struct item *item, int x, int y, int w)
        else
                drw_setscheme(drw, scheme[SchemeNorm]);
 
-       return drw_text(drw, x, y, w, bh, lrpad / 2, item->text, 0);
+       r = drw_text(drw, x, y, w, bh, lrpad / 2, item->text, 0);
+       drawhighlights(item, x, y, w);
+       return r;
+}
+
+static void
+recalculatenumbers()
+{
+       unsigned int numer = 0, denom = 0;
+       struct item *item;
+       if (matchend) {
+               numer++;
+               for (item = matchend; item && item->left; item = item->left)
+                       numer++;
+       }
+       for (item = items; item && item->text; item++)
+               denom++;
+       snprintf(numbers, NUMBERSBUFSIZE, "%d/%d", numer, denom);
 }
 
 static void
@@ -137,6 +193,7 @@ drawmenu(void)
        unsigned int curpos;
        struct item *item;
        int x = 0, y = 0, w;
+       char *censort;
 
        drw_setscheme(drw, scheme[SchemeNorm]);
        drw_rect(drw, 0, 0, mw, mh, 1, 1);
@@ -148,7 +205,12 @@ drawmenu(void)
        /* draw input field */
        w = (lines > 0 || !matches) ? mw - x : inputw;
        drw_setscheme(drw, scheme[SchemeNorm]);
-       drw_text(drw, x, 0, w, bh, lrpad / 2, text, 0);
+       if (passwd) {
+               censort = ecalloc(1, sizeof(text));
+               memset(censort, '*', strlen(text));
+               drw_text(drw, x, 0, w, bh, lrpad / 2, censort, 0);
+               free(censort);
+       } else drw_text(drw, x, 0, w, bh, lrpad / 2, text, 0);
 
        curpos = TEXTW(text) - TEXTW(&text[cursor]);
        if ((curpos += lrpad / 2 - 1) < w) {
@@ -156,6 +218,7 @@ drawmenu(void)
                drw_rect(drw, x + curpos, 2, 2, bh - 4, 1, 0);
        }
 
+       recalculatenumbers();
        if (lines > 0) {
                /* draw vertical list */
                for (item = curr; item != next; item = item->right)
@@ -170,13 +233,15 @@ drawmenu(void)
                }
                x += w;
                for (item = curr; item != next; item = item->right)
-                       x = drawitem(item, x, 0, MIN(TEXTW(item->text), mw - x - TEXTW(">")));
+                       x = drawitem(item, x, 0, MIN(TEXTW(item->text), mw - x - TEXTW(">") - TEXTW(numbers)));
                if (next) {
                        w = TEXTW(">");
                        drw_setscheme(drw, scheme[SchemeNorm]);
-                       drw_text(drw, mw - w, 0, w, bh, lrpad / 2, ">", 0);
+                       drw_text(drw, mw - w - TEXTW(numbers), 0, w, bh, lrpad / 2, ">", 0);
                }
        }
+       drw_setscheme(drw, scheme[SchemeNorm]);
+       drw_text(drw, mw - TEXTW(numbers), 0, TEXTW(numbers), bh, lrpad / 2, numbers, 0);
        drw_map(drw, win, 0, 0, mw, mh);
 }
 
@@ -215,9 +280,94 @@ grabkeyboard(void)
        die("cannot grab keyboard");
 }
 
+int
+compare_distance(const void *a, const void *b)
+{
+       struct item *da = *(struct item **) a;
+       struct item *db = *(struct item **) b;
+
+       if (!db)
+               return 1;
+       if (!da)
+               return -1;
+
+       return da->distance == db->distance ? 0 : da->distance < db->distance ? -1 : 1;
+}
+
+void
+fuzzymatch(void)
+{
+       /* bang - we have so much memory */
+       struct item *it;
+       struct item **fuzzymatches = NULL;
+       char c;
+       int number_of_matches = 0, i, pidx, sidx, eidx;
+       int text_len = strlen(text), itext_len;
+
+       matches = matchend = NULL;
+
+       /* walk through all items */
+       for (it = items; it && it->text; it++) {
+               if (text_len) {
+                       itext_len = strlen(it->text);
+                       pidx = 0; /* pointer */
+                       sidx = eidx = -1; /* start of match, end of match */
+                       /* walk through item text */
+                       for (i = 0; i < itext_len && (c = it->text[i]); i++) {
+                               /* fuzzy match pattern */
+                               if (!fstrncmp(&text[pidx], &c, 1)) {
+                                       if(sidx == -1)
+                                               sidx = i;
+                                       pidx++;
+                                       if (pidx == text_len) {
+                                               eidx = i;
+                                               break;
+                                       }
+                               }
+                       }
+                       /* build list of matches */
+                       if (eidx != -1) {
+                               /* compute distance */
+                               /* add penalty if match starts late (log(sidx+2))
+                                * add penalty for long a match without many matching characters */
+                               it->distance = log(sidx + 2) + (double)(eidx - sidx - text_len);
+                               /* fprintf(stderr, "distance %s %f\n", it->text, it->distance); */
+                               appenditem(it, &matches, &matchend);
+                               number_of_matches++;
+                       }
+               } else {
+                       appenditem(it, &matches, &matchend);
+               }
+       }
+
+       if (number_of_matches) {
+               /* initialize array with matches */
+               if (!(fuzzymatches = realloc(fuzzymatches, number_of_matches * sizeof(struct item*))))
+                       die("cannot realloc %u bytes:", number_of_matches * sizeof(struct item*));
+               for (i = 0, it = matches; it && i < number_of_matches; i++, it = it->right) {
+                       fuzzymatches[i] = it;
+               }
+               /* sort matches according to distance */
+               qsort(fuzzymatches, number_of_matches, sizeof(struct item*), compare_distance);
+               /* rebuild list of matches */
+               matches = matchend = NULL;
+               for (i = 0, it = fuzzymatches[i];  i < number_of_matches && it && \
+                               it->text; i++, it = fuzzymatches[i]) {
+                       appenditem(it, &matches, &matchend);
+               }
+               free(fuzzymatches);
+       }
+       curr = sel = matches;
+       calcoffsets();
+}
+
 static void
 match(void)
 {
+       if (fuzzy) {
+               fuzzymatch();
+               return;
+       }
        static char **tokv = NULL;
        static int tokn = 0;
 
@@ -359,8 +509,8 @@ keypress(XKeyEvent *ev)
                        while (cursor > 0 && !strchr(worddelimiters, text[nextrune(-1)]))
                                insert(NULL, nextrune(-1) - cursor);
                        break;
-               case XK_y: /* paste selection */
-               case XK_Y:
+               case XK_v: /* paste selection */
+               case XK_V:
                        XConvertSelection(dpy, (ev->state & ShiftMask) ? clip : XA_PRIMARY,
                                          utf8, utf8, win, CurrentTime);
                        return;
@@ -529,6 +679,11 @@ readstdin(void)
        char buf[sizeof text], *p;
        size_t i, imax = 0, size = 0;
        unsigned int tmpmax = 0;
+       if(passwd){
+       inputw = lines = 0;
+       return;
+       }
+
 
        /* read each line from stdin and add it to the item list */
        for (i = 0; fgets(buf, sizeof buf, stdin); i++) {
@@ -558,9 +713,14 @@ run(void)
        XEvent ev;
 
        while (!XNextEvent(dpy, &ev)) {
-               if (XFilterEvent(&ev, None))
+               if (XFilterEvent(&ev, win))
                        continue;
                switch(ev.type) {
+               case DestroyNotify:
+                       if (ev.xdestroywindow.window != win)
+                               break;
+                       cleanup();
+                       exit(1);
                case Expose:
                        if (ev.xexpose.count == 0)
                                drw_map(drw, win, 0, 0, mw, mh);
@@ -664,15 +824,17 @@ setup(void)
                            CWOverrideRedirect | CWBackPixel | CWEventMask, &swa);
        XSetClassHint(dpy, win, &ch);
 
-       /* open input methods */
-       xim = XOpenIM(dpy, NULL, NULL, NULL);
+
+       /* input methods */
+       if ((xim = XOpenIM(dpy, NULL, NULL, NULL)) == NULL)
+               die("XOpenIM failed: could not open input device");
+
        xic = XCreateIC(xim, XNInputStyle, XIMPreeditNothing | XIMStatusNothing,
                        XNClientWindow, win, XNFocusWindow, win, NULL);
 
        XMapRaised(dpy, win);
-       XSetInputFocus(dpy, win, RevertToParent, CurrentTime);
        if (embed) {
-               XSelectInput(dpy, parentwin, FocusChangeMask);
+               XSelectInput(dpy, parentwin, FocusChangeMask | SubstructureNotifyMask);
                if (XQueryTree(dpy, parentwin, &dw, &w, &dws, &du) && dws) {
                        for (i = 0; i < du && dws[i] != win; ++i)
                                XSelectInput(dpy, dws[i], FocusChangeMask);
@@ -687,8 +849,9 @@ setup(void)
 static void
 usage(void)
 {
-       fputs("usage: dmenu [-bfiv] [-l lines] [-p prompt] [-fn font] [-m monitor]\n"
-             "             [-nb color] [-nf color] [-sb color] [-sf color] [-w windowid]\n", stderr);
+       fputs("usage: dmenu [-bfivP] [-l lines] [-p prompt] [-fn font] [-m monitor]\n"
+             "             [-nb color] [-nf color] [-sb color] [-sf color]\n"
+             "             [-w windowid]\n", stderr);
        exit(1);
 }
 
@@ -707,10 +870,14 @@ main(int argc, char *argv[])
                        topbar = 0;
                else if (!strcmp(argv[i], "-f"))   /* grabs keyboard before reading stdin */
                        fast = 1;
+               else if (!strcmp(argv[i], "-F"))   /* grabs keyboard before reading stdin */
+                       fuzzy = 0;
                else if (!strcmp(argv[i], "-i")) { /* case-insensitive item matching */
                        fstrncmp = strncasecmp;
                        fstrstr = cistrstr;
-               } else if (i + 1 == argc)
+               } else if (!strcmp(argv[i], "-P"))   /* is the input a password */
+                       passwd = 1;
+               else if (i + 1 == argc)
                        usage();
                /* these options take one argument */
                else if (!strcmp(argv[i], "-l"))   /* number of lines in vertical list */
@@ -736,8 +903,6 @@ main(int argc, char *argv[])
 
        if (!setlocale(LC_CTYPE, "") || !XSupportsLocale())
                fputs("warning: no locale support\n", stderr);
-       if (!XSetLocaleModifiers(""))
-               fputs("warning: no locale modifiers support\n", stderr);
        if (!(dpy = XOpenDisplay(NULL)))
                die("cannot open display");
        screen = DefaultScreen(dpy);
@@ -752,10 +917,12 @@ main(int argc, char *argv[])
                die("no fonts could be loaded.");
        lrpad = drw->fonts->h;
 
-       if (pledge("stdio rpath", NULL) < 0)
+#ifdef __OpenBSD__
+       if (pledge("stdio rpath", NULL) == -1)
                die("pledge");
+#endif
 
-       if (fast) {
+       if (fast && !isatty(0)) {
                grabkeyboard();
                readstdin();
        } else {