X-Git-Url: https://git.xinqibao.xyz/dmenu.git/blobdiff_plain/595e7976601fc77acf12015d3f5f6843e2cdd706..6cc0b0dc086feaf944b166d0b459ac407192ea5e:/dmenu.c

diff --git a/dmenu.c b/dmenu.c
index 0d89474..40b220c 100644
--- a/dmenu.c
+++ b/dmenu.c
@@ -1,235 +1,241 @@
 /* See LICENSE file for copyright and license details. */
 #include <ctype.h>
-#include <locale.h>
-#include <stdarg.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <strings.h>
 #include <unistd.h>
-#include <X11/keysym.h>
 #include <X11/Xlib.h>
+#include <X11/Xatom.h>
 #include <X11/Xutil.h>
 #ifdef XINERAMA
 #include <X11/extensions/Xinerama.h>
 #endif
+#include "draw.h"
 
-/* macros */
-#define CLEANMASK(mask)         (mask & ~(numlockmask | LockMask))
-#define INRECT(X,Y,RX,RY,RW,RH) ((X) >= (RX) && (X) < (RX) + (RW) && (Y) >= (RY) && (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))
+#define INRECT(x,y,rx,ry,rw,rh) ((x) >= (rx) && (x) < (rx)+(rw) && (y) >= (ry) && (y) < (ry)+(rh))
+#define MIN(a,b)                ((a) < (b) ? (a) : (b))
+#define MAX(a,b)                ((a) > (b) ? (a) : (b))
 
 typedef struct Item Item;
 struct Item {
 	char *text;
-	Item *next;         /* traverses all items */
-	Item *left, *right; /* traverses items matching current search pattern */
+	Item *left, *right;
 };
 
-/* forward declarations */
-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 appenditem(Item *item, Item **list, Item **last);
+static void calcoffsets(void);
 static void drawmenu(void);
-static void drawmenuh(void);
-static void drawmenuv(void);
-static void eprint(const char *errstr, ...);
-static Bool grabkeyboard(void);
-static void kpress(XKeyEvent * e);
-static void match(char *pattern);
+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(Bool topbar);
-
-#include "config.h"
-#include "draw.h"
+static void setup(void);
+
+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 long normcol[ColLast];
+static unsigned long selcol[ColLast];
+static Atom utf8;
+static Bool topbar = True;
+static DC *dc;
+static Item *items = NULL;
+static Item *matches, *matchend;
+static Item *prev, *curr, *next, *sel;
+static Window root, win;
 
-/* variables */
-static char *maxname = NULL;
-static char *prompt = NULL;
-static char text[4096];
-static int cmdw = 0;
-static int promptw = 0;
-static int ret = 0;
-static unsigned int numlockmask = 0;
-static Bool running = True;
-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;
 static int (*fstrncmp)(const char *, const char *, size_t) = strncmp;
-static char *(*fstrstr)(const char *, const char *) = strstr;
-static unsigned int lines = 0;
-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;
-}
+int
+main(int argc, char *argv[]) {
+	Bool fast = False;
+	int i;
 
-void
-calcoffsetsh(void) {
-	unsigned int w;
+	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], "-f"))
+			fast = True;
+		else if(!strcmp(argv[i], "-i"))
+			fstrncmp = strncasecmp;
+		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;
 
-	if(!curr)
-		return;
-	w = promptw + cmdw + 2 * spaceitem;
-	for(next = curr; next && w < mw; next=next->right)
-		w += MIN(textw(next->text), mw / 3);
-	w = promptw + cmdw + 2 * spaceitem;
-	for(prev = curr; prev && prev->left && w < mw; prev=prev->left)
-		w += MIN(textw(prev->left->text), mw / 3);
-}
+	dc = initdc();
+	initfont(dc, font);
 
-void
-calcoffsetsv(void) {
-	unsigned int h;
+	if(fast) {
+		setup();
+		readstdin();
+	}
+	else {
+		readstdin();
+		setup();
+	}
+	match();
+	run();
+	return EXIT_FAILURE;
 
-	if(!curr)
-		return;
-	h = (dc.font.height + 2) * lines;
-	for(next = curr; next && h > 0; next = next->right)
-		h -= dc.font.height + 2;
-	h = (dc.font.height + 2) * lines;
-	for(prev = curr; prev && prev->left && h > 0; prev = prev->left)
-		h -= dc.font.height + 2;
+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;
 }
 
-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
+appenditem(Item *item, Item **list, Item **last) {
+	if(!*last)
+		*list = item;
+	else
+		(*last)->right = item;
+	item->left = *last;
+	item->right = NULL;
+	*last = item;
 }
 
 void
-cleanup(void) {
-	drawcleanup();
-	XDestroyWindow(dpy, win);
-	XUngrabKeyboard(dpy, CurrentTime);
+calcoffsets(void) {
+	unsigned int i, n;
+
+	if(lines > 0)
+		n = lines * bh;
+	else
+		n = mw - (promptw + inputw + textw(dc, "<") + textw(dc, ">"));
+
+	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
 drawmenu(void) {
-	dc.x = 0;
-	dc.y = 0;
-	dc.w = mw;
-	dc.h = mh;
-	drawtext(NULL, dc.norm);
-	/* print prompt? */
+	int curpos;
+	Item *item;
+
+	dc->x = 0;
+	dc->y = 0;
+	dc->h = bh;
+	drawrect(dc, 0, 0, mw, mh, True, BG(dc, normcol));
+
 	if(prompt) {
-		dc.w = promptw;
-		drawtext(prompt, dc.sel);
-		dc.x += dc.w;
+		dc->w = promptw;
+		drawtext(dc, prompt, selcol);
+		dc->x = dc->w;
 	}
-	dc.w = mw - dc.x;
-	/* print command */
-	if(cmdw && item && lines == 0)
-		dc.w = cmdw;
-	drawtext(*text ? text : NULL, dc.norm);
-	if(curr) {
-		if(lines > 0)
-			drawmenuv();
-		else
-			drawmenuh();
+	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, True, FG(dc, normcol));
+
+	if(lines > 0) {
+		dc->w = mw - dc->x;
+		for(item = curr; item != next; item = item->right) {
+			dc->y += dc->h;
+			drawtext(dc, item->text, (item == sel) ? selcol : normcol);
+		}
 	}
-	XCopyArea(dpy, dc.drawable, win, dc.gc, 0, 0, mw, mh, 0, 0);
-	XFlush(dpy);
-}
-
-void
-drawmenuh(void) {
-	Item *i;
-
-	dc.x += cmdw;
-	dc.w = spaceitem;
-	drawtext(curr->left ? "<" : NULL, dc.norm);
-	dc.x += dc.w;
-	for(i = curr; i != next; i=i->right) {
-		dc.w = MIN(textw(i->text), mw / 3);
-		drawtext(i->text, (sel == i) ? dc.sel : dc.norm);
-		dc.x += dc.w;
+	else if(matches) {
+		dc->x += inputw;
+		dc->w = textw(dc, "<");
+		if(curr->left)
+			drawtext(dc, "<", normcol);
+		for(item = curr; item != next; item = item->right) {
+			dc->x += dc->w;
+			dc->w = MIN(textw(dc, item->text), mw - dc->x - textw(dc, ">"));
+			drawtext(dc, item->text, (item == sel) ? selcol : normcol);
+		}
+		dc->w = textw(dc, ">");
+		dc->x = mw - dc->w;
+		if(next)
+			drawtext(dc, ">", normcol);
 	}
-	dc.w = spaceitem;
-	dc.x = mw - dc.w;
-	drawtext(next ? ">" : NULL, dc.norm);
+	mapdc(dc, win, mw, mh);
 }
 
-void
-drawmenuv(void) {
-	Item *i;
-
-	dc.w = mw - dc.x;
-	dc.h = dc.font.height + 2;
-	dc.y = dc.h;
-	for(i = curr; i != next; i=i->right) {
-		drawtext(i->text, (sel == i) ? dc.sel : dc.norm);
-		dc.y += dc.h;
-	}
-	dc.h = mh - dc.y;
-	drawtext(NULL, dc.norm);
+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;
 }
 
-Bool
+void
 grabkeyboard(void) {
-	unsigned int len;
+	int i;
 
-	for(len = 1000; len; len--) {
-		if(XGrabKeyboard(dpy, parent, True, GrabModeAsync, GrabModeAsync, CurrentTime)
-		== GrabSuccess)
-			break;
+	for(i = 0; i < 1000; i++) {
+		if(!XGrabKeyboard(dc->dpy, root, True, GrabModeAsync, GrabModeAsync, CurrentTime))
+			return;
 		usleep(1000);
 	}
-	return len > 0;
+	eprintf("cannot grab keyboard\n");
 }
 
 void
-kpress(XKeyEvent * e) {
-	char buf[sizeof text];
-	int num;
-	unsigned int i, len;
+insert(const char *s, ssize_t 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;
+	match();
+}
+
+void
+keypress(XKeyEvent *ev) {
+	char buf[32];
+	size_t 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) {
+	XLookupString(ev, buf, sizeof buf, &ksym, NULL);
+	if(ev->state & ControlMask)
 		switch(tolower(ksym)) {
 		default:
 			return;
@@ -242,6 +248,9 @@ kpress(XKeyEvent * e) {
 		case XK_c:
 			ksym = XK_Escape;
 			break;
+		case XK_d:
+			ksym = XK_Delete;
+			break;
 		case XK_e:
 			ksym = XK_End;
 			break;
@@ -257,70 +266,76 @@ kpress(XKeyEvent * e) {
 		case XK_j:
 			ksym = XK_Return;
 			break;
+		case XK_k:  /* delete right */
+			text[cursor] = '\0';
+			match();
+			break;
 		case XK_n:
 			ksym = XK_Down;
 			break;
 		case XK_p:
 			ksym = XK_Up;
 			break;
-		case XK_u:
-			text[0] = '\0';
-			match(text);
+		case XK_u:  /* delete left */
+			insert(NULL, 0 - cursor);
 			break;
-		case XK_w:
-			if(len == 0)
-				return;
-			i = len;
-			while(i-- > 0 && text[i] == ' ');
-			while(i-- > 0 && text[i] != ' ');
-			text[++i] = '\0';
-			match(text);
-			break;
-		case XK_x:
-			execlp("dinput", "dinput", text, NULL); /* todo: argv */
-			eprint("dmenu: cannot exec dinput:");
+		case XK_w:  /* delete word */
+			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, utf8, win, CurrentTime);
+			return;
 		}
-	}
 	switch(ksym) {
 	default:
-		num = MIN(num, sizeof text);
-		if(num && !iscntrl((int) buf[0])) {
-			memcpy(text + len, buf, num + 1);
-			len += num;
-			match(text);
-		}
+		if(!iscntrl(*buf))
+			insert(buf, strlen(buf));
 		break;
-	case XK_BackSpace:
-		if(len == 0)
+	case XK_Delete:
+		if(cursor == len)
 			return;
-		for(i = 1; len - i > 0 && !IS_UTF8_1ST_CHAR(text[len - i]); i++);
-		len -= i;
-		text[len] = '\0';
-		match(text);
+		cursor = nextrune(+1);
+	case XK_BackSpace:
+		if(cursor > 0)
+			insert(NULL, nextrune(-1) - cursor);
 		break;
 	case XK_End:
-		while(next) {
-			sel = curr = next;
+		if(cursor < len) {
+			cursor = len;
+			break;
+		}
+		if(next) {
+			curr = matchend;
 			calcoffsets();
+			curr = prev;
+			calcoffsets();
+			while(next && (curr = curr->right))
+				calcoffsets();
 		}
-		while(sel && sel->right)
-			sel = sel->right;
+		sel = matchend;
 		break;
 	case XK_Escape:
-		ret = 1;
-		running = False;
-		return;
+		exit(EXIT_FAILURE);
 	case XK_Home:
-		sel = curr = item;
+		if(sel == matches) {
+			cursor = 0;
+			break;
+		}
+		sel = curr = matches;
 		calcoffsets();
 		break;
 	case XK_Left:
-	case XK_Up:
-		if(!sel || !sel->left)
+		if(cursor > 0 && (!sel || !sel->left || lines > 0)) {
+			cursor = nextrune(-1);
+			break;
+		}
+		else if(lines > 0)
 			return;
-		sel = sel->left;
-		if(sel->right == curr) {
+	case XK_Up:
+		if(sel && sel->left && (sel = sel->left)->right == curr) {
 			curr = prev;
 			calcoffsets();
 		}
@@ -338,19 +353,18 @@ kpress(XKeyEvent * e) {
 		calcoffsets();
 		break;
 	case XK_Return:
-		if((e->state & ShiftMask) || !sel)
-			fprintf(stdout, "%s", text);
-		else
-			fprintf(stdout, "%s", sel->text);
-		fflush(stdout);
-		running = False;
-		return;
+	case XK_KP_Enter:
+		fputs((sel && !(ev->state & ShiftMask)) ? sel->text : text, stdout);
+		exit(EXIT_SUCCESS);
 	case XK_Right:
-	case XK_Down:
-		if(!sel || !sel->right)
+		if(cursor < len) {
+			cursor = nextrune(+1);
+			break;
+		}
+		else if(lines > 0)
 			return;
-		sel = sel->right;
-		if(sel == next) {
+	case XK_Down:
+		if(sel && sel->right && (sel = sel->right) == next) {
 			curr = next;
 			calcoffsets();
 		}
@@ -359,77 +373,89 @@ kpress(XKeyEvent * e) {
 		if(!sel)
 			return;
 		strncpy(text, sel->text, sizeof text);
-		match(text);
+		cursor = strlen(text);
+		match();
 		break;
 	}
 	drawmenu();
 }
 
 void
-match(char *pattern) {
-	unsigned int plen;
-	Item *i, *itemend, *lexact, *lprefix, *lsubstr, *exactend, *prefixend, *substrend;
+match(void) {
+	size_t len = strlen(text);
+	Item *item, *lexact, *lprefix, *lsubstr, *exactend, *prefixend, *substrend;
+
+	matches = lexact = lprefix = lsubstr = matchend = exactend = prefixend = substrend = NULL;
+	for(item = items; item && item->text; item++)
+		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(!pattern)
-		return;
-	plen = strlen(pattern);
-	item = lexact = lprefix = lsubstr = itemend = exactend = prefixend = substrend = NULL;
-	for(i = allitems; i; i = i->next)
-		if(!fstrncmp(pattern, i->text, plen + 1))
-			appenditem(i, &lexact, &exactend);
-		else if(!fstrncmp(pattern, i->text, plen))
-			appenditem(i, &lprefix, &prefixend);
-		else if(fstrstr(i->text, pattern))
-			appenditem(i, &lsubstr, &substrend);
 	if(lexact) {
-		item = lexact;
-		itemend = exactend;
+		matches = lexact;
+		matchend = exactend;
 	}
 	if(lprefix) {
-		if(itemend) {
-			itemend->right = lprefix;
-			lprefix->left = itemend;
+		if(matchend) {
+			matchend->right = lprefix;
+			lprefix->left = matchend;
 		}
 		else
-			item = lprefix;
-		itemend = prefixend;
+			matches = lprefix;
+		matchend = prefixend;
 	}
 	if(lsubstr) {
-		if(itemend) {
-			itemend->right = lsubstr;
-			lsubstr->left = itemend;
+		if(matchend) {
+			matchend->right = lsubstr;
+			lsubstr->left = matchend;
 		}
 		else
-			item = lsubstr;
+			matches = lsubstr;
+		matchend = substrend;
 	}
-	curr = prev = next = sel = item;
+	curr = sel = matches;
 	calcoffsets();
 }
 
+size_t
+nextrune(int incr) {
+	size_t n, 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;
+	int di;
+	unsigned long dl;
+	Atom da;
+
+	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);
+	drawmenu();
+}
+
 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("dmenu: cannot strdup %u bytes\n", len);
-		if((max = MAX(max, len)) == len)
-			maxname = p;
-		if(!(new = malloc(sizeof *new)))
-			eprint("dmenu: 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;
+	char buf[sizeof text], *p;
+	size_t i, size = 0;
+
+	for(i = 0; fgets(buf, sizeof buf, stdin); items[++i].text = NULL) {
+		if(i+1 == size / sizeof *items || !items)
+			if(!(items = realloc(items, (size += BUFSIZ))))
+				eprintf("cannot realloc %u bytes:", size);
+		if((p = strchr(buf, '\n')))
+			*p = '\0';
+		if(!(items[i].text = strdup(buf)))
+			eprintf("cannot strdup %u bytes:", strlen(buf)+1);
+		inputw = MAX(inputw, textw(dc, items[i].text));
 	}
 }
 
@@ -437,154 +463,83 @@ void
 run(void) {
 	XEvent ev;
 
-	/* main event loop */
-	while(running && !XNextEvent(dpy, &ev))
-		switch (ev.type) {
-		case KeyPress:
-			kpress(&ev.xkey);
-			break;
+	while(!XNextEvent(dc->dpy, &ev))
+		switch(ev.type) {
 		case Expose:
 			if(ev.xexpose.count == 0)
 				drawmenu();
 			break;
+		case KeyPress:
+			keypress(&ev.xkey);
+			break;
+		case SelectionNotify:
+			if(ev.xselection.property == utf8)
+				paste();
+			break;
 		case VisibilityNotify:
-			if (ev.xvisibility.state != VisibilityUnobscured)
-				XRaiseWindow(dpy, win);
+			if(ev.xvisibility.state != VisibilityUnobscured)
+				XRaiseWindow(dc->dpy, win);
 			break;
 		}
 }
 
 void
-setup(Bool topbar) {
-	int i, j, x, y;
-#if XINERAMA
+setup(void) {
+	int x, y, screen;
+	XSetWindowAttributes wa;
+#ifdef XINERAMA
 	int n;
-	XineramaScreenInfo *info = NULL;
+	XineramaScreenInfo *info;
 #endif
-	XModifierKeymap *modmap;
-	XSetWindowAttributes wa;
-	XWindowAttributes pwa;
-
-	/* 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);
 
-	initfont(font);
+	screen = DefaultScreen(dc->dpy);
+	root = RootWindow(dc->dpy, screen);
+	utf8 = XInternAtom(dc->dpy, "UTF8_STRING", False);
 
-	/* menu window */
-	wa.override_redirect = True;
-	wa.background_pixmap = ParentRelative;
-	wa.event_mask = ExposureMask | ButtonPressMask | KeyPressMask | VisibilityChangeMask;
-
-	/* menu window geometry */
-	mh = (dc.font.height + 2) * (lines + 1);
-#if XINERAMA
-	if(parent == RootWindow(dpy, screen) && XineramaIsActive(dpy) && (info = XineramaQueryScreens(dpy, &n))) {
-		i = 0;
-		if(n > 1) {
-			int di;
-			unsigned int dui;
-			Window dummy;
-			if(XQueryPointer(dpy, parent, &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;
-		}
+	normcol[ColBG] = getcolor(dc, normbgcolor);
+	normcol[ColFG] = getcolor(dc, normfgcolor);
+	selcol[ColBG] = getcolor(dc, selbgcolor);
+	selcol[ColFG] = getcolor(dc, selfgcolor);
+
+	/* menu geometry */
+	bh = dc->font.height + 2;
+	lines = MAX(lines, 0);
+	mh = (lines + 1) * bh;
+#ifdef XINERAMA
+	if((info = XineramaQueryScreens(dc->dpy, &n))) {
+		int i, di;
+		unsigned int du;
+		Window dw;
+
+		XQueryPointer(dc->dpy, root, &dw, &dw, &x, &y, &di, &di, &du);
+		for(i = 0; i < n-1; i++)
+			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 = topbar ? info[i].y_org : info[i].y_org + info[i].height - mh;
+		y = info[i].y_org + (topbar ? 0 : info[i].height - mh);
 		mw = info[i].width;
 		XFree(info);
 	}
 	else
 #endif
 	{
-		XGetWindowAttributes(dpy, parent, &pwa);
 		x = 0;
-		y = topbar ? 0 : pwa.height - mh;
-		mw = pwa.width;
+		y = topbar ? 0 : DisplayHeight(dc->dpy, screen) - mh;
+		mw = DisplayWidth(dc->dpy, screen);
 	}
-
-	win = XCreateWindow(dpy, parent, x, y, mw, mh, 0,
-			DefaultDepth(dpy, screen), CopyFromParent,
-			DefaultVisual(dpy, screen),
-			CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
-
-	drawsetup();
-	if(maxname)
-		cmdw = MIN(textw(maxname), mw / 3);
-	if(prompt)
-		promptw = MIN(textw(prompt), mw / 5);
-	text[0] = '\0';
-	match(text);
-	XMapRaised(dpy, win);
-}
-
-int
-main(int argc, char *argv[]) {
-	unsigned int i;
-	Bool topbar = True;
-
-	/* command line args */
-	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], "-e")) {
-			if(++i < argc) parent = atoi(argv[i]);
-		}
-		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"))
-			eprint("dmenu-"VERSION", © 2006-2010 dmenu engineers, see LICENSE for details\n");
-		else
-			eprint("usage: dmenu [-i] [-b] [-e <xid>] [-l <lines>] [-fn <font>] [-nb <color>]\n"
-			       "             [-nf <color>] [-p <prompt>] [-sb <color>] [-sf <color>] [-v]\n");
-	if(!setlocale(LC_CTYPE, "") || !XSupportsLocale())
-		fprintf(stderr, "dmenu: warning: no locale support\n");
-	if(!(dpy = XOpenDisplay(NULL)))
-		eprint("dmenu: cannot open display\n");
-	screen = DefaultScreen(dpy);
-	if(!parent)
-		parent = RootWindow(dpy, screen);
-
-	readstdin();
-	running = grabkeyboard();
-
-	setup(topbar);
-	drawmenu();
-	XSync(dpy, False);
-	run();
-	cleanup();
-	XCloseDisplay(dpy);
-	return ret;
+	/* menu window */
+	wa.override_redirect = True;
+	wa.background_pixmap = ParentRelative;
+	wa.event_mask = ExposureMask | KeyPressMask | VisibilityChangeMask;
+	win = XCreateWindow(dc->dpy, root, x, y, mw, mh, 0,
+	                    DefaultDepth(dc->dpy, screen), CopyFromParent,
+	                    DefaultVisual(dc->dpy, screen),
+	                    CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
+
+	grabkeyboard();
+	resizedc(dc, mw, mh);
+	inputw = MIN(inputw, mw/3);
+	promptw = prompt ? textw(dc, prompt) : 0;
+	XMapRaised(dc->dpy, win);
 }