rinri
/
dmenu-rinri
1
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
My dmenu build
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.
392 Commits
1 Branch
1.2 MiB
 
 
 
 
 
 
Tree: 86019c5855
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '86019c5855'
${ noResults }
dmenu-rinri/dmenu.c

545 lines
12 KiB
Raw Blame History 

  1. /* See LICENSE file for copyright and license details. */

  2. #include <ctype.h>

  3. #include <stdio.h>

  4. #include <stdlib.h>

  5. #include <string.h>

  6. #include <unistd.h>

  7. #include <X11/Xlib.h>

  8. #include <X11/Xatom.h>

  9. #include <X11/Xutil.h>

  10. #ifdef XINERAMA

  11. #include <X11/extensions/Xinerama.h>

  12. #endif

  13. #include "draw.h"

  14. 

  15. #define INRECT(x,y,rx,ry,rw,rh) ((x) >= (rx) && (x) < (rx)+(rw) && (y) >= (ry) && (y) < (ry)+(rh))

  16. #define MIN(a,b)                ((a) < (b) ? (a) : (b))

  17. #define MAX(a,b)                ((a) > (b) ? (a) : (b))

  18. 

  19. typedef struct Item Item;

  20. struct Item {

  21. 	char *text;

  22. 	Item *next;          /* traverses all items */

  23. 	Item *left, *right;  /* traverses matching items */

  24. };

  25. 

  26. static void appenditem(Item *item, Item **list, Item **last);

  27. static void calcoffsets(void);

  28. static void drawmenu(void);

  29. static char *fstrstr(const char *s, const char *sub);

  30. static void grabkeyboard(void);

  31. static void insert(const char *s, ssize_t n);

  32. static void keypress(XKeyEvent *ev);

  33. static void match(void);

  34. static size_t nextrune(int incr);

  35. static void paste(void);

  36. static void readstdin(void);

  37. static void run(void);

  38. static void setup(void);

  39. 

  40. static char text[BUFSIZ] = "";

  41. static int bh, mw, mh;

  42. static int inputw = 0;

  43. static int lines = 0;

  44. static int monitor = -1;

  45. static int promptw;

  46. static size_t cursor = 0;

  47. static const char *font = NULL;

  48. static const char *prompt = NULL;

  49. static const char *normbgcolor = "#cccccc";

  50. static const char *normfgcolor = "#000000";

  51. static const char *selbgcolor  = "#0066ff";

  52. static const char *selfgcolor  = "#ffffff";

  53. static unsigned long normcol[ColLast];

  54. static unsigned long selcol[ColLast];

  55. static Atom utf8;

  56. static Bool topbar = True;

  57. static DC *dc;

  58. static Item *items = NULL;

  59. static Item *matches, *sel;

  60. static Item *prev, *curr, *next;

  61. static Window root, win;

  62. 

  63. static int (*fstrncmp)(const char *, const char *, size_t) = strncmp;

  64. 

  65. int

  66. main(int argc, char *argv[]) {

  67. 	Bool fast = False;

  68. 	int i;

  69. 

  70. 	progname = "dmenu";

  71. 	for(i = 1; i < argc; i++)

  72. 		/* single flags */

  73. 		if(!strcmp(argv[i], "-v")) {

  74. 			fputs("dmenu-"VERSION", © 2006-2011 dmenu engineers, see LICENSE for details\n", stdout);

  75. 			exit(EXIT_SUCCESS);

  76. 		}

  77. 		else if(!strcmp(argv[i], "-b"))

  78. 			topbar = False;

  79. 		else if(!strcmp(argv[i], "-f"))

  80. 			fast = True;

  81. 		else if(!strcmp(argv[i], "-i"))

  82. 			fstrncmp = strncasecmp;

  83. 		else if(i == argc-1)

  84. 			goto usage;

  85. 		/* double flags */

  86. 		else if(!strcmp(argv[i], "-l"))

  87. 			lines = atoi(argv[++i]);

  88. 		else if(!strcmp(argv[i], "-m"))

  89. 			monitor = atoi(argv[++i]);

  90. 		else if(!strcmp(argv[i], "-p"))

  91. 			prompt = argv[++i];

  92. 		else if(!strcmp(argv[i], "-fn"))

  93. 			font = argv[++i];

  94. 		else if(!strcmp(argv[i], "-nb"))

  95. 			normbgcolor = argv[++i];

  96. 		else if(!strcmp(argv[i], "-nf"))

  97. 			normfgcolor = argv[++i];

  98. 		else if(!strcmp(argv[i], "-sb"))

  99. 			selbgcolor = argv[++i];

  100. 		else if(!strcmp(argv[i], "-sf"))

  101. 			selfgcolor = argv[++i];

  102. 		else

  103. 			goto usage;

  104. 

  105. 	dc = initdc();

  106. 	initfont(dc, font);

  107. 

  108. 	if(fast) {

  109. 		setup();

  110. 		readstdin();

  111. 	}

  112. 	else {

  113. 		readstdin();

  114. 		setup();

  115. 	}

  116. 	match();

  117. 	run();

  118. 	return EXIT_FAILURE;

  119. 

  120. usage:

  121. 	fputs("usage: dmenu [-b] [-f] [-i] [-l lines] [-m monitor] [-p prompt] [-fn font]\n"

  122. 	      "             [-nb color] [-nf color] [-sb color] [-sf color] [-v]\n", stderr);

  123. 	return EXIT_FAILURE;

  124. }

  125. 

  126. void

  127. appenditem(Item *item, Item **list, Item **last) {

  128. 	if(!*last)

  129. 		*list = item;

  130. 	else

  131. 		(*last)->right = item;

  132. 	item->left = *last;

  133. 	item->right = NULL;

  134. 	*last = item;

  135. }

  136. 

  137. void

  138. calcoffsets(void) {

  139. 	unsigned int i, n;

  140. 

  141. 	if(lines > 0)

  142. 		n = lines * bh;

  143. 	else

  144. 		n = mw - (promptw + inputw + textw(dc, "<") + textw(dc, ">"));

  145. 

  146. 	for(i = 0, next = curr; next; next = next->right)

  147. 		if((i += (lines > 0) ? bh : MIN(textw(dc, next->text), n)) > n)

  148. 			break;

  149. 	for(i = 0, prev = curr; prev && prev->left; prev = prev->left)

  150. 		if((i += (lines > 0) ? bh : MIN(textw(dc, prev->left->text), n)) > n)

  151. 			break;

  152. }

  153. 

  154. void

  155. drawmenu(void) {

  156. 	int curpos;

  157. 	Item *item;

  158. 

  159. 	dc->x = 0;

  160. 	dc->y = 0;

  161. 	dc->h = bh;

  162. 	drawrect(dc, 0, 0, mw, mh, True, BG(dc, normcol));

  163. 

  164. 	if(prompt) {

  165. 		dc->w = promptw;

  166. 		drawtext(dc, prompt, selcol);

  167. 		dc->x = dc->w;

  168. 	}

  169. 	dc->w = (lines > 0 || !matches) ? mw - dc->x : inputw;

  170. 	drawtext(dc, text, normcol);

  171. 	if((curpos = textnw(dc, text, cursor) + dc->h/2 - 2) < dc->w)

  172. 		drawrect(dc, curpos, 2, 1, dc->h - 4, True, FG(dc, normcol));

  173. 

  174. 	if(lines > 0) {

  175. 		dc->w = mw - dc->x;

  176. 		for(item = curr; item != next; item = item->right) {

  177. 			dc->y += dc->h;

  178. 			drawtext(dc, item->text, (item == sel) ? selcol : normcol);

  179. 		}

  180. 	}

  181. 	else if(matches) {

  182. 		dc->x += inputw;

  183. 		dc->w = textw(dc, "<");

  184. 		if(curr->left)

  185. 			drawtext(dc, "<", normcol);

  186. 		for(item = curr; item != next; item = item->right) {

  187. 			dc->x += dc->w;

  188. 			dc->w = MIN(textw(dc, item->text), mw - dc->x - textw(dc, ">"));

  189. 			drawtext(dc, item->text, (item == sel) ? selcol : normcol);

  190. 		}

  191. 		dc->w = textw(dc, ">");

  192. 		dc->x = mw - dc->w;

  193. 		if(next)

  194. 			drawtext(dc, ">", normcol);

  195. 	}

  196. 	mapdc(dc, win, mw, mh);

  197. }

  198. 

  199. char *

  200. fstrstr(const char *s, const char *sub) {

  201. 	size_t len;

  202. 

  203. 	for(len = strlen(sub); *s; s++)

  204. 		if(!fstrncmp(s, sub, len))

  205. 			return (char *)s;

  206. 	return NULL;

  207. }

  208. 

  209. void

  210. grabkeyboard(void) {

  211. 	int i;

  212. 

  213. 	for(i = 0; i < 1000; i++) {

  214. 		if(!XGrabKeyboard(dc->dpy, root, True, GrabModeAsync, GrabModeAsync, CurrentTime))

  215. 			return;

  216. 		usleep(1000);

  217. 	}

  218. 	eprintf("cannot grab keyboard\n");

  219. }

  220. 

  221. void

  222. insert(const char *s, ssize_t n) {

  223. 	if(strlen(text) + n > sizeof text - 1)

  224. 		return;

  225. 	memmove(text + cursor + n, text + cursor, sizeof text - cursor - MAX(n, 0));

  226. 	if(n > 0)

  227. 		memcpy(text + cursor, s, n);

  228. 	cursor += n;

  229. 	match();

  230. }

  231. 

  232. void

  233. keypress(XKeyEvent *ev) {

  234. 	char buf[32];

  235. 	size_t len;

  236. 	KeySym ksym;

  237. 

  238. 	len = strlen(text);

  239. 	XLookupString(ev, buf, sizeof buf, &ksym, NULL);

  240. 	if(ev->state & ControlMask)

  241. 		switch(tolower(ksym)) {

  242. 		default:

  243. 			return;

  244. 		case XK_a:

  245. 			ksym = XK_Home;

  246. 			break;

  247. 		case XK_b:

  248. 			ksym = XK_Left;

  249. 			break;

  250. 		case XK_c:

  251. 			ksym = XK_Escape;

  252. 			break;

  253. 		case XK_d:

  254. 			ksym = XK_Delete;

  255. 			break;

  256. 		case XK_e:

  257. 			ksym = XK_End;

  258. 			break;

  259. 		case XK_f:

  260. 			ksym = XK_Right;

  261. 			break;

  262. 		case XK_h:

  263. 			ksym = XK_BackSpace;

  264. 			break;

  265. 		case XK_i:

  266. 			ksym = XK_Tab;

  267. 			break;

  268. 		case XK_j:

  269. 			ksym = XK_Return;

  270. 			break;

  271. 		case XK_k:  /* delete right */

  272. 			text[cursor] = '\0';

  273. 			match();

  274. 			break;

  275. 		case XK_n:

  276. 			ksym = XK_Down;

  277. 			break;

  278. 		case XK_p:

  279. 			ksym = XK_Up;

  280. 			break;

  281. 		case XK_u:  /* delete left */

  282. 			insert(NULL, 0 - cursor);

  283. 			break;

  284. 		case XK_w:  /* delete word */

  285. 			while(cursor > 0 && text[nextrune(-1)] == ' ')

  286. 				insert(NULL, nextrune(-1) - cursor);

  287. 			while(cursor > 0 && text[nextrune(-1)] != ' ')

  288. 				insert(NULL, nextrune(-1) - cursor);

  289. 			break;

  290. 		case XK_y:  /* paste selection */

  291. 			XConvertSelection(dc->dpy, XA_PRIMARY, utf8, utf8, win, CurrentTime);

  292. 			return;

  293. 		}

  294. 	switch(ksym) {

  295. 	default:

  296. 		if(!iscntrl(*buf))

  297. 			insert(buf, strlen(buf));

  298. 		break;

  299. 	case XK_Delete:

  300. 		if(cursor == len)

  301. 			return;

  302. 		cursor = nextrune(+1);

  303. 	case XK_BackSpace:

  304. 		if(cursor > 0)

  305. 			insert(NULL, nextrune(-1) - cursor);

  306. 		break;

  307. 	case XK_End:

  308. 		if(cursor < len) {

  309. 			cursor = len;

  310. 			break;

  311. 		}

  312. 		while(next) {

  313. 			sel = curr = next;

  314. 			calcoffsets();

  315. 		}

  316. 		while(sel && sel->right)

  317. 			sel = sel->right;

  318. 		break;

  319. 	case XK_Escape:

  320. 		exit(EXIT_FAILURE);

  321. 	case XK_Home:

  322. 		if(sel == matches) {

  323. 			cursor = 0;

  324. 			break;

  325. 		}

  326. 		sel = curr = matches;

  327. 		calcoffsets();

  328. 		break;

  329. 	case XK_Left:

  330. 		if(cursor > 0 && (!sel || !sel->left || lines > 0)) {

  331. 			cursor = nextrune(-1);

  332. 			break;

  333. 		}

  334. 		else if(lines > 0)

  335. 			return;

  336. 	case XK_Up:

  337. 		if(sel && sel->left && (sel = sel->left)->right == curr) {

  338. 			curr = prev;

  339. 			calcoffsets();

  340. 		}

  341. 		break;

  342. 	case XK_Next:

  343. 		if(!next)

  344. 			return;

  345. 		sel = curr = next;

  346. 		calcoffsets();

  347. 		break;

  348. 	case XK_Prior:

  349. 		if(!prev)

  350. 			return;

  351. 		sel = curr = prev;

  352. 		calcoffsets();

  353. 		break;

  354. 	case XK_Return:

  355. 	case XK_KP_Enter:

  356. 		fputs((sel && !(ev->state & ShiftMask)) ? sel->text : text, stdout);

  357. 		exit(EXIT_SUCCESS);

  358. 	case XK_Right:

  359. 		if(cursor < len) {

  360. 			cursor = nextrune(+1);

  361. 			break;

  362. 		}

  363. 		else if(lines > 0)

  364. 			return;

  365. 	case XK_Down:

  366. 		if(sel && sel->right && (sel = sel->right) == next) {

  367. 			curr = next;

  368. 			calcoffsets();

  369. 		}

  370. 		break;

  371. 	case XK_Tab:

  372. 		if(!sel)

  373. 			return;

  374. 		strncpy(text, sel->text, sizeof text);

  375. 		cursor = strlen(text);

  376. 		match();

  377. 		break;

  378. 	}

  379. 	drawmenu();

  380. }

  381. 

  382. void

  383. match(void) {

  384. 	size_t len;

  385. 	Item *item, *itemend, *lexact, *lprefix, *lsubstr, *exactend, *prefixend, *substrend;

  386. 

  387. 	len = strlen(text);

  388. 	matches = lexact = lprefix = lsubstr = itemend = exactend = prefixend = substrend = NULL;

  389. 	for(item = items; item; item = item->next)

  390. 		if(!fstrncmp(text, item->text, len + 1))

  391. 			appenditem(item, &lexact, &exactend);

  392. 		else if(!fstrncmp(text, item->text, len))

  393. 			appenditem(item, &lprefix, &prefixend);

  394. 		else if(fstrstr(item->text, text))

  395. 			appenditem(item, &lsubstr, &substrend);

  396. 

  397. 	if(lexact) {

  398. 		matches = lexact;

  399. 		itemend = exactend;

  400. 	}

  401. 	if(lprefix) {

  402. 		if(itemend) {

  403. 			itemend->right = lprefix;

  404. 			lprefix->left = itemend;

  405. 		}

  406. 		else

  407. 			matches = lprefix;

  408. 		itemend = prefixend;

  409. 	}

  410. 	if(lsubstr) {

  411. 		if(itemend) {

  412. 			itemend->right = lsubstr;

  413. 			lsubstr->left = itemend;

  414. 		}

  415. 		else

  416. 			matches = lsubstr;

  417. 	}

  418. 	curr = sel = matches;

  419. 	calcoffsets();

  420. }

  421. 

  422. size_t

  423. nextrune(int incr) {

  424. 	size_t n, len;

  425. 

  426. 	len = strlen(text);

  427. 	for(n = cursor + incr; n >= 0 && n < len && (text[n] & 0xc0) == 0x80; n += incr);

  428. 	return n;

  429. }

  430. 

  431. void

  432. paste(void) {

  433. 	char *p, *q;

  434. 	int di;

  435. 	unsigned long dl;

  436. 	Atom da;

  437. 

  438. 	XGetWindowProperty(dc->dpy, win, utf8, 0, (sizeof text / 4) + 1, False,

  439. 	                   utf8, &da, &di, &dl, &dl, (unsigned char **)&p);

  440. 	insert(p, (q = strchr(p, '\n')) ? q-p : strlen(p));

  441. 	XFree(p);

  442. 	drawmenu();

  443. }

  444. 

  445. void

  446. readstdin(void) {

  447. 	char buf[sizeof text], *p;

  448. 	Item *item, **end;

  449. 

  450. 	for(end = &items; fgets(buf, sizeof buf, stdin); *end = item, end = &item->next) {

  451. 		if((p = strchr(buf, '\n')))

  452. 			*p = '\0';

  453. 		if(!(item = calloc(1, sizeof *item)))

  454. 			eprintf("cannot malloc %u bytes:", sizeof *item);

  455. 		if(!(item->text = strdup(buf)))

  456. 			eprintf("cannot strdup %u bytes:", strlen(buf)+1);

  457. 		inputw = MAX(inputw, textw(dc, item->text));

  458. 	}

  459. }

  460. 

  461. void

  462. run(void) {

  463. 	XEvent ev;

  464. 

  465. 	while(!XNextEvent(dc->dpy, &ev))

  466. 		switch(ev.type) {

  467. 		case Expose:

  468. 			if(ev.xexpose.count == 0)

  469. 				drawmenu();

  470. 			break;

  471. 		case KeyPress:

  472. 			keypress(&ev.xkey);

  473. 			break;

  474. 		case SelectionNotify:

  475. 			if(ev.xselection.property == utf8)

  476. 				paste();

  477. 			break;

  478. 		case VisibilityNotify:

  479. 			if(ev.xvisibility.state != VisibilityUnobscured)

  480. 				XRaiseWindow(dc->dpy, win);

  481. 			break;

  482. 		}

  483. }

  484. 

  485. void

  486. setup(void) {

  487. 	int x, y, screen;

  488. 	XSetWindowAttributes wa;

  489. #ifdef XINERAMA

  490. 	int n;

  491. 	XineramaScreenInfo *info;

  492. #endif

  493. 

  494. 	screen = DefaultScreen(dc->dpy);

  495. 	root = RootWindow(dc->dpy, screen);

  496. 	utf8 = XInternAtom(dc->dpy, "UTF8_STRING", False);

  497. 

  498. 	normcol[ColBG] = getcolor(dc, normbgcolor);

  499. 	normcol[ColFG] = getcolor(dc, normfgcolor);

  500. 	selcol[ColBG] = getcolor(dc, selbgcolor);

  501. 	selcol[ColFG] = getcolor(dc, selfgcolor);

  502. 

  503. 	/* menu geometry */

  504. 	bh = dc->font.height + 2;

  505. 	lines = MAX(lines, 0);

  506. 	mh = (lines + 1) * bh;

  507. #ifdef XINERAMA

  508. 	if((info = XineramaQueryScreens(dc->dpy, &n))) {

  509. 		int i, di;

  510. 		unsigned int du;

  511. 		Window dw;

  512. 

  513. 		XQueryPointer(dc->dpy, root, &dw, &dw, &x, &y, &di, &di, &du);

  514. 		for(i = 0; i < n-1; i++)

  515. 			if((monitor == info[i].screen_number)

  516. 			|| (monitor < 0 && INRECT(x, y, info[i].x_org, info[i].y_org, info[i].width, info[i].height)))

  517. 				break;

  518. 		x = info[i].x_org;

  519. 		y = info[i].y_org + (topbar ? 0 : info[i].height - mh);

  520. 		mw = info[i].width;

  521. 		XFree(info);

  522. 	}

  523. 	else

  524. #endif

  525. 	{

  526. 		x = 0;

  527. 		y = topbar ? 0 : DisplayHeight(dc->dpy, screen) - mh;

  528. 		mw = DisplayWidth(dc->dpy, screen);

  529. 	}

  530. 	/* menu window */

  531. 	wa.override_redirect = True;

  532. 	wa.background_pixmap = ParentRelative;

  533. 	wa.event_mask = ExposureMask | KeyPressMask | VisibilityChangeMask;

  534. 	win = XCreateWindow(dc->dpy, root, x, y, mw, mh, 0,

  535. 	                    DefaultDepth(dc->dpy, screen), CopyFromParent,

  536. 	                    DefaultVisual(dc->dpy, screen),

  537. 	                    CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);

  538. 

  539. 	grabkeyboard();

  540. 	resizedc(dc, mw, mh);

  541. 	inputw = MIN(inputw, mw/3);

  542. 	promptw = prompt ? textw(dc, prompt) : 0;

  543. 	XMapRaised(dc->dpy, win);

  544. }