You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
650 lines
14 KiB
650 lines
14 KiB
/* See LICENSE file for copyright and license details. */ |
|
#include <ctype.h> |
|
#include <locale.h> |
|
#include <stdio.h> |
|
#include <stdlib.h> |
|
#include <string.h> |
|
#include <unistd.h> |
|
#include <X11/keysym.h> |
|
#include <X11/Xlib.h> |
|
#include <X11/Xutil.h> |
|
#ifdef XINERAMA |
|
#include <X11/extensions/Xinerama.h> |
|
#endif |
|
#include <draw.h> |
|
#include "config.h" |
|
|
|
#define INRECT(x,y,rx,ry,rw,rh) ((rx) < (x) && (x) < (rx)+(rw) && (ry) < (y) && (y) < (ry)+(rh)) |
|
#define MIN(a,b) ((a) < (b) ? (a) : (b)) |
|
#define MAX(a,b) ((a) > (b) ? (a) : (b)) |
|
#define IS_UTF8_1ST_CHAR(c) (((c) & 0xc0) == 0xc0 || ((c) & 0x80) == 0x00) |
|
|
|
typedef struct Item Item; |
|
struct Item { |
|
char *text; |
|
Item *next; /* traverses all items */ |
|
Item *left, *right; /* traverses items matching current search pattern */ |
|
}; |
|
|
|
static void appenditem(Item *i, Item **list, Item **last); |
|
static void calcoffsetsh(void); |
|
static void calcoffsetsv(void); |
|
static char *cistrstr(const char *s, const char *sub); |
|
static void cleanup(void); |
|
static void drawitem(const char *s, unsigned long col[ColLast]); |
|
static void drawmenu(void); |
|
static void drawmenuh(void); |
|
static void drawmenuv(void); |
|
static void grabkeyboard(void); |
|
static void keypress(XKeyEvent *e); |
|
static void match(void); |
|
static void readstdin(void); |
|
static void run(void); |
|
static void setup(void); |
|
|
|
static char **argp = NULL; |
|
static char *maxname = NULL; |
|
static char *prompt; |
|
static char text[4096]; |
|
static int promptw; |
|
static int screen; |
|
static size_t cur = 0; |
|
static unsigned int cmdw = 0; |
|
static unsigned int lines = 0; |
|
static unsigned int numlockmask; |
|
static unsigned int mw, mh; |
|
static unsigned long normcol[ColLast]; |
|
static unsigned long selcol[ColLast]; |
|
static Bool topbar = True; |
|
static DC dc; |
|
static Display *dpy; |
|
static Item *allitems = NULL; /* first of all items */ |
|
static Item *item = NULL; /* first of pattern matching items */ |
|
static Item *sel = NULL; |
|
static Item *next = NULL; |
|
static Item *prev = NULL; |
|
static Item *curr = NULL; |
|
static Window win, root; |
|
|
|
static int (*fstrncmp)(const char *, const char *, size_t) = strncmp; |
|
static char *(*fstrstr)(const char *, const char *) = strstr; |
|
static void (*calcoffsets)(void) = calcoffsetsh; |
|
|
|
void |
|
appenditem(Item *i, Item **list, Item **last) { |
|
if(!(*last)) |
|
*list = i; |
|
else |
|
(*last)->right = i; |
|
i->left = *last; |
|
i->right = NULL; |
|
*last = i; |
|
} |
|
|
|
void |
|
calcoffsetsh(void) { |
|
unsigned int w, x; |
|
|
|
w = promptw + cmdw + textw(&dc, "<") + textw(&dc, ">"); |
|
for(x = w, next = curr; next; next = next->right) |
|
if((x += MIN(textw(&dc, next->text), mw / 3)) > mw) |
|
break; |
|
for(x = w, prev = curr; prev && prev->left; prev = prev->left) |
|
if((x += MIN(textw(&dc, prev->left->text), mw / 3)) > mw) |
|
break; |
|
} |
|
|
|
void |
|
calcoffsetsv(void) { |
|
unsigned int i; |
|
|
|
next = prev = curr; |
|
for(i = 0; i < lines && next; i++) |
|
next = next->right; |
|
mh = (dc.font.height + 2) * (i + 1); |
|
for(i = 0; i < lines && prev && prev->left; i++) |
|
prev = prev->left; |
|
} |
|
|
|
char * |
|
cistrstr(const char *s, const char *sub) { |
|
int c, csub; |
|
unsigned int len; |
|
|
|
if(!sub) |
|
return (char *)s; |
|
if((c = tolower(*sub++)) != '\0') { |
|
len = strlen(sub); |
|
do { |
|
do { |
|
if((csub = *s++) == '\0') |
|
return NULL; |
|
} |
|
while(tolower(csub) != c); |
|
} |
|
while(strncasecmp(s, sub, len) != 0); |
|
s--; |
|
} |
|
return (char *)s; |
|
} |
|
|
|
void |
|
cleanup(void) { |
|
Item *itm; |
|
|
|
while(allitems) { |
|
itm = allitems->next; |
|
free(allitems->text); |
|
free(allitems); |
|
allitems = itm; |
|
} |
|
cleanupdraw(&dc); |
|
XDestroyWindow(dpy, win); |
|
XUngrabKeyboard(dpy, CurrentTime); |
|
XCloseDisplay(dpy); |
|
} |
|
|
|
void |
|
drawitem(const char *s, unsigned long col[ColLast]) { |
|
const char *p; |
|
unsigned int w = textnw(&dc, text, strlen(text)); |
|
|
|
drawbox(&dc, col); |
|
drawtext(&dc, s, col); |
|
for(p = fstrstr(s, text); *text && (p = fstrstr(p, text)); p++) |
|
drawline(&dc, textnw(&dc, s, p-s) + dc.h/2 - 1, dc.h-2, w, 1, col); |
|
} |
|
|
|
void |
|
drawmenu(void) { |
|
dc.x = 0; |
|
dc.y = 0; |
|
dc.w = mw; |
|
dc.h = mh; |
|
drawbox(&dc, normcol); |
|
dc.h = dc.font.height + 2; |
|
dc.y = topbar ? 0 : mh - dc.h; |
|
/* print prompt? */ |
|
if(prompt) { |
|
dc.w = promptw; |
|
drawbox(&dc, selcol); |
|
drawtext(&dc, prompt, selcol); |
|
dc.x += dc.w; |
|
} |
|
dc.w = mw - dc.x; |
|
/* print command */ |
|
if(cmdw && item && lines == 0) |
|
dc.w = cmdw; |
|
drawtext(&dc, text, normcol); |
|
drawline(&dc, textnw(&dc, text, cur) + dc.h/2 - 2, 2, 1, dc.h-4, normcol); |
|
if(lines > 0) |
|
drawmenuv(); |
|
else if(curr) |
|
drawmenuh(); |
|
commitdraw(&dc, win); |
|
} |
|
|
|
void |
|
drawmenuh(void) { |
|
Item *i; |
|
|
|
dc.x += cmdw; |
|
dc.w = textw(&dc, "<"); |
|
drawtext(&dc, curr->left ? "<" : NULL, normcol); |
|
dc.x += dc.w; |
|
for(i = curr; i != next; i = i->right) { |
|
dc.w = MIN(textw(&dc, i->text), mw / 3); |
|
drawitem(i->text, (sel == i) ? selcol : normcol); |
|
dc.x += dc.w; |
|
} |
|
dc.w = textw(&dc, ">"); |
|
dc.x = mw - dc.w; |
|
drawtext(&dc, next ? ">" : NULL, normcol); |
|
} |
|
|
|
void |
|
drawmenuv(void) { |
|
Item *i; |
|
XWindowAttributes wa; |
|
|
|
dc.y = topbar ? dc.h : 0; |
|
dc.w = mw - dc.x; |
|
for(i = curr; i != next; i = i->right) { |
|
drawitem(i->text, (sel == i) ? selcol : normcol); |
|
dc.y += dc.h; |
|
} |
|
if(!XGetWindowAttributes(dpy, win, &wa)) |
|
eprint("cannot get window attributes"); |
|
XMoveResizeWindow(dpy, win, wa.x, wa.y + (topbar ? 0 : wa.height - mh), mw, mh); |
|
} |
|
|
|
void |
|
grabkeyboard(void) { |
|
unsigned int n; |
|
|
|
for(n = 0; n < 1000; n++) { |
|
if(!XGrabKeyboard(dpy, root, True, GrabModeAsync, GrabModeAsync, CurrentTime)) |
|
return; |
|
usleep(1000); |
|
} |
|
exit(EXIT_FAILURE); |
|
} |
|
|
|
void |
|
keypress(XKeyEvent *e) { |
|
char buf[sizeof text]; |
|
int num; |
|
unsigned int i, len; |
|
KeySym ksym; |
|
|
|
len = strlen(text); |
|
num = XLookupString(e, buf, sizeof buf, &ksym, NULL); |
|
if(ksym == XK_KP_Enter) |
|
ksym = XK_Return; |
|
else if(ksym >= XK_KP_0 && ksym <= XK_KP_9) |
|
ksym = (ksym - XK_KP_0) + XK_0; |
|
else if(IsFunctionKey(ksym) || IsKeypadKey(ksym) |
|
|| IsMiscFunctionKey(ksym) || IsPFKey(ksym) |
|
|| IsPrivateKeypadKey(ksym)) |
|
return; |
|
/* first check if a control mask is omitted */ |
|
if(e->state & ControlMask) { |
|
switch(tolower(ksym)) { |
|
default: |
|
return; |
|
case XK_a: |
|
ksym = XK_Home; |
|
break; |
|
case XK_b: |
|
ksym = XK_Left; |
|
break; |
|
case XK_c: |
|
ksym = XK_Escape; |
|
break; |
|
case XK_e: |
|
ksym = XK_End; |
|
break; |
|
case XK_f: |
|
ksym = XK_Right; |
|
break; |
|
case XK_h: |
|
ksym = XK_BackSpace; |
|
break; |
|
case XK_i: |
|
ksym = XK_Tab; |
|
break; |
|
case XK_j: |
|
case XK_m: |
|
ksym = XK_Return; |
|
break; |
|
case XK_k: |
|
text[cur] = '\0'; |
|
break; |
|
case XK_n: |
|
ksym = XK_Down; |
|
break; |
|
case XK_p: |
|
ksym = XK_Up; |
|
break; |
|
case XK_u: |
|
memmove(text, text + cur, sizeof text - cur + 1); |
|
cur = 0; |
|
match(); |
|
break; |
|
case XK_w: |
|
if(cur == 0) |
|
return; |
|
i = cur; |
|
while(i-- > 0 && text[i] == ' '); |
|
while(i-- > 0 && text[i] != ' '); |
|
memmove(text + i + 1, text + cur, sizeof text - cur + 1); |
|
cur = i + 1; |
|
match(); |
|
break; |
|
case XK_y: |
|
{ |
|
FILE *fp; |
|
char *s; |
|
if(!(fp = fopen("sselp", "r"))) |
|
eprint("cannot popen sselp\n"); |
|
s = fgets(buf, sizeof buf, fp); |
|
fclose(fp); |
|
if(!s) |
|
return; |
|
} |
|
num = strlen(buf); |
|
if(num && buf[num-1] == '\n') |
|
buf[--num] = '\0'; |
|
break; |
|
} |
|
} |
|
switch(ksym) { |
|
default: |
|
num = MIN(num, sizeof text); |
|
if(num && !iscntrl((int) buf[0])) { |
|
memmove(text + cur + num, text + cur, sizeof text - cur - num); |
|
memcpy(text + cur, buf, num); |
|
cur += num; |
|
match(); |
|
} |
|
break; |
|
case XK_BackSpace: |
|
if(cur == 0) |
|
return; |
|
for(i = 1; len - i > 0 && !IS_UTF8_1ST_CHAR(text[cur - i]); i++); |
|
memmove(text + cur - i, text + cur, sizeof text - cur + i); |
|
cur -= i; |
|
match(); |
|
break; |
|
case XK_Delete: |
|
if(cur == len) |
|
return; |
|
for(i = 1; cur + i < len && !IS_UTF8_1ST_CHAR(text[cur + i]); i++); |
|
memmove(text + cur, text + cur + i, sizeof text - cur); |
|
match(); |
|
break; |
|
case XK_End: |
|
if(cur < len) { |
|
cur = len; |
|
break; |
|
} |
|
while(next) { |
|
sel = curr = next; |
|
calcoffsets(); |
|
} |
|
while(sel && sel->right) |
|
sel = sel->right; |
|
break; |
|
case XK_Escape: |
|
exit(EXIT_FAILURE); |
|
case XK_Home: |
|
if(sel == item) { |
|
cur = 0; |
|
break; |
|
} |
|
sel = curr = item; |
|
calcoffsets(); |
|
break; |
|
case XK_Left: |
|
if(cur > 0 && (!sel || !sel->left || lines > 0)) { |
|
while(cur-- > 0 && !IS_UTF8_1ST_CHAR(text[cur])); |
|
break; |
|
} |
|
if(lines > 0) |
|
return; |
|
case XK_Up: |
|
if(!sel || !sel->left) |
|
return; |
|
sel = sel->left; |
|
if(sel->right == curr) { |
|
curr = prev; |
|
calcoffsets(); |
|
} |
|
break; |
|
case XK_Next: |
|
if(!next) |
|
return; |
|
sel = curr = next; |
|
calcoffsets(); |
|
break; |
|
case XK_Prior: |
|
if(!prev) |
|
return; |
|
sel = curr = prev; |
|
calcoffsets(); |
|
break; |
|
case XK_Return: |
|
fprintf(stdout, "%s", ((e->state & ShiftMask) || sel) ? sel->text : text); |
|
fflush(stdout); |
|
exit(EXIT_SUCCESS); |
|
case XK_Right: |
|
if(cur < len) { |
|
while(cur++ < len && !IS_UTF8_1ST_CHAR(text[cur])); |
|
break; |
|
} |
|
if(lines > 0) |
|
return; |
|
case XK_Down: |
|
if(!sel || !sel->right) |
|
return; |
|
sel = sel->right; |
|
if(sel == next) { |
|
curr = next; |
|
calcoffsets(); |
|
} |
|
break; |
|
case XK_Tab: |
|
if(!sel) |
|
return; |
|
strncpy(text, sel->text, sizeof text); |
|
cur = strlen(text); |
|
match(); |
|
break; |
|
} |
|
drawmenu(); |
|
} |
|
|
|
void |
|
match(void) { |
|
unsigned int len; |
|
Item *i, *itemend, *lexact, *lprefix, *lsubstr, *exactend, *prefixend, *substrend; |
|
|
|
len = strlen(text); |
|
item = lexact = lprefix = lsubstr = itemend = exactend = prefixend = substrend = NULL; |
|
for(i = allitems; i; i = i->next) |
|
if(!fstrncmp(text, i->text, len + 1)) |
|
appenditem(i, &lexact, &exactend); |
|
else if(!fstrncmp(text, i->text, len)) |
|
appenditem(i, &lprefix, &prefixend); |
|
else if(fstrstr(i->text, text)) |
|
appenditem(i, &lsubstr, &substrend); |
|
if(lexact) { |
|
item = lexact; |
|
itemend = exactend; |
|
} |
|
if(lprefix) { |
|
if(itemend) { |
|
itemend->right = lprefix; |
|
lprefix->left = itemend; |
|
} |
|
else |
|
item = lprefix; |
|
itemend = prefixend; |
|
} |
|
if(lsubstr) { |
|
if(itemend) { |
|
itemend->right = lsubstr; |
|
lsubstr->left = itemend; |
|
} |
|
else |
|
item = lsubstr; |
|
} |
|
curr = prev = next = sel = item; |
|
calcoffsets(); |
|
} |
|
|
|
void |
|
readstdin(void) { |
|
char *p, buf[sizeof text]; |
|
unsigned int len = 0, max = 0; |
|
Item *i, *new; |
|
|
|
i = NULL; |
|
while(fgets(buf, sizeof buf, stdin)) { |
|
len = strlen(buf); |
|
if(buf[len-1] == '\n') |
|
buf[--len] = '\0'; |
|
if(!(p = strdup(buf))) |
|
eprint("cannot strdup %u bytes\n", len); |
|
if((max = MAX(max, len)) == len) |
|
maxname = p; |
|
if(!(new = malloc(sizeof *new))) |
|
eprint("cannot malloc %u bytes\n", sizeof *new); |
|
new->next = new->left = new->right = NULL; |
|
new->text = p; |
|
if(!i) |
|
allitems = new; |
|
else |
|
i->next = new; |
|
i = new; |
|
} |
|
} |
|
|
|
void |
|
run(void) { |
|
XEvent ev; |
|
|
|
XSync(dpy, False); |
|
while(!XNextEvent(dpy, &ev)) |
|
switch(ev.type) { |
|
case Expose: |
|
if(ev.xexpose.count == 0) |
|
drawmenu(); |
|
break; |
|
case KeyPress: |
|
keypress(&ev.xkey); |
|
break; |
|
case VisibilityNotify: |
|
if(ev.xvisibility.state != VisibilityUnobscured) |
|
XRaiseWindow(dpy, win); |
|
break; |
|
} |
|
exit(EXIT_FAILURE); |
|
} |
|
|
|
void |
|
setup(void) { |
|
int i, j, x, y; |
|
#if XINERAMA |
|
int n; |
|
XineramaScreenInfo *info = NULL; |
|
#endif |
|
XModifierKeymap *modmap; |
|
XSetWindowAttributes wa; |
|
|
|
/* init modifier map */ |
|
modmap = XGetModifierMapping(dpy); |
|
for(i = 0; i < 8; i++) |
|
for(j = 0; j < modmap->max_keypermod; j++) { |
|
if(modmap->modifiermap[i * modmap->max_keypermod + j] |
|
== XKeysymToKeycode(dpy, XK_Num_Lock)) |
|
numlockmask = (1 << i); |
|
} |
|
XFreeModifiermap(modmap); |
|
|
|
dc.dpy = dpy; |
|
normcol[ColBG] = getcolor(&dc, normbgcolor); |
|
normcol[ColFG] = getcolor(&dc, normfgcolor); |
|
selcol[ColBG] = getcolor(&dc, selbgcolor); |
|
selcol[ColFG] = getcolor(&dc, selfgcolor); |
|
initfont(&dc, font); |
|
|
|
/* input window */ |
|
wa.override_redirect = True; |
|
wa.background_pixmap = ParentRelative; |
|
wa.event_mask = ExposureMask | KeyPressMask | VisibilityChangeMask; |
|
|
|
/* input window geometry */ |
|
mh = (dc.font.height + 2) * (lines + 1); |
|
#if XINERAMA |
|
if(XineramaIsActive(dpy) && (info = XineramaQueryScreens(dpy, &n))) { |
|
i = 0; |
|
if(n > 1) { |
|
int di; |
|
unsigned int dui; |
|
Window dummy; |
|
if(XQueryPointer(dpy, root, &dummy, &dummy, &x, &y, &di, &di, &dui)) |
|
for(i = 0; i < n; i++) |
|
if(INRECT(x, y, info[i].x_org, info[i].y_org, info[i].width, info[i].height)) |
|
break; |
|
} |
|
x = info[i].x_org; |
|
y = topbar ? info[i].y_org : info[i].y_org + info[i].height - mh; |
|
mw = info[i].width; |
|
XFree(info); |
|
} |
|
else |
|
#endif |
|
{ |
|
x = 0; |
|
y = topbar ? 0 : DisplayHeight(dpy, screen) - mh; |
|
mw = DisplayWidth(dpy, screen); |
|
} |
|
|
|
win = XCreateWindow(dpy, root, x, y, mw, mh, 0, |
|
DefaultDepth(dpy, screen), CopyFromParent, |
|
DefaultVisual(dpy, screen), |
|
CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa); |
|
|
|
setupdraw(&dc, win); |
|
if(prompt) |
|
promptw = MIN(textw(&dc, prompt), mw / 5); |
|
XMapRaised(dpy, win); |
|
} |
|
|
|
int |
|
main(int argc, char *argv[]) { |
|
unsigned int i; |
|
|
|
/* command line args */ |
|
progname = "dmenu"; |
|
for(i = 1; i < argc; i++) |
|
if(!strcmp(argv[i], "-i")) { |
|
fstrncmp = strncasecmp; |
|
fstrstr = cistrstr; |
|
} |
|
else if(!strcmp(argv[i], "-b")) |
|
topbar = False; |
|
else if(!strcmp(argv[i], "-l")) { |
|
if(++i < argc) lines = atoi(argv[i]); |
|
if(lines > 0) |
|
calcoffsets = calcoffsetsv; |
|
} |
|
else if(!strcmp(argv[i], "-fn")) { |
|
if(++i < argc) font = argv[i]; |
|
} |
|
else if(!strcmp(argv[i], "-nb")) { |
|
if(++i < argc) normbgcolor = argv[i]; |
|
} |
|
else if(!strcmp(argv[i], "-nf")) { |
|
if(++i < argc) normfgcolor = argv[i]; |
|
} |
|
else if(!strcmp(argv[i], "-p")) { |
|
if(++i < argc) prompt = argv[i]; |
|
} |
|
else if(!strcmp(argv[i], "-sb")) { |
|
if(++i < argc) selbgcolor = argv[i]; |
|
} |
|
else if(!strcmp(argv[i], "-sf")) { |
|
if(++i < argc) selfgcolor = argv[i]; |
|
} |
|
else if(!strcmp(argv[i], "-v")) { |
|
printf("dmenu-"VERSION", © 2006-2010 dmenu engineers, see LICENSE for details\n"); |
|
exit(EXIT_SUCCESS); |
|
} |
|
else { |
|
fputs("usage: dmenu [-i] [-b] [-l <lines>] [-fn <font>] [-nb <color>]\n" |
|
" [-nf <color>] [-p <prompt>] [-sb <color>] [-sf <color>] [-v]\n", stderr); |
|
exit(EXIT_FAILURE); |
|
} |
|
if(!setlocale(LC_CTYPE, "") || !XSupportsLocale()) |
|
fprintf(stderr, "dmenu: warning: no locale support\n"); |
|
if(!(dpy = XOpenDisplay(NULL))) |
|
eprint("cannot open display\n"); |
|
if(atexit(&cleanup) != 0) |
|
eprint("cannot register cleanup\n"); |
|
screen = DefaultScreen(dpy); |
|
root = RootWindow(dpy, screen); |
|
if(!(argp = malloc(sizeof *argp * (argc+2)))) |
|
eprint("cannot malloc %u bytes\n", sizeof *argp * (argc+2)); |
|
memcpy(argp + 2, argv + 1, sizeof *argp * argc); |
|
|
|
readstdin(); |
|
grabkeyboard(); |
|
setup(); |
|
if(maxname) |
|
cmdw = MIN(textw(&dc, maxname), mw / 3); |
|
match(); |
|
run(); |
|
return 0; |
|
}
|
|
|