A mirror of phillbush's xmenu.
 
 
 
 
 

915 lines
23 KiB

  1. #include <err.h>
  2. #include <stdio.h>
  3. #include <stdlib.h>
  4. #include <string.h>
  5. #include <unistd.h>
  6. #include <X11/Xlib.h>
  7. #include <X11/Xutil.h>
  8. #include <X11/Xresource.h>
  9. #include <X11/XKBlib.h>
  10. #include <X11/Xft/Xft.h>
  11. #define PROGNAME "xmenu"
  12. #define ITEMPREV 0
  13. #define ITEMNEXT 1
  14. /* macros */
  15. #define LEN(x) (sizeof (x) / sizeof (x[0]))
  16. #define MAX(x,y) ((x)>(y)?(x):(y))
  17. #define MIN(x,y) ((x)<(y)?(x):(y))
  18. /* color enum */
  19. enum {ColorFG, ColorBG, ColorLast};
  20. /* draw context structure */
  21. struct DC {
  22. XftColor normal[ColorLast];
  23. XftColor selected[ColorLast];
  24. XftColor decoration[ColorLast];
  25. Drawable d;
  26. GC gc;
  27. XftFont *font;
  28. };
  29. /* menu geometry structure */
  30. struct Geometry {
  31. int itemb; /* item border */
  32. int itemw; /* item width */
  33. int itemh; /* item height */
  34. int border; /* window border width */
  35. int separator; /* menu separator width */
  36. };
  37. /* screen geometry structure */
  38. struct ScreenGeometry {
  39. int cursx, cursy; /* cursor position */
  40. int screenw, screenh; /* screen width and height */
  41. };
  42. /* menu item structure */
  43. struct Item {
  44. char *label; /* string to be drawed on menu */
  45. char *output; /* string to be outputed when item is clicked */
  46. int y; /* item y position relative to menu */
  47. int h; /* item height */
  48. size_t labellen; /* strlen(label) */
  49. struct Item *prev; /* previous item */
  50. struct Item *next; /* next item */
  51. struct Menu *submenu; /* submenu spawned by clicking on item */
  52. };
  53. /* menu structure */
  54. struct Menu {
  55. struct Menu *parent; /* parent menu */
  56. struct Item *caller; /* item that spawned the menu */
  57. struct Item *list; /* list of items contained by the menu */
  58. struct Item *selected; /* item currently selected in the menu */
  59. int x, y, w, h; /* menu geometry */
  60. unsigned level; /* menu level relative to root */
  61. Drawable pixmap; /* pixmap to draw the menu on */
  62. XftDraw *draw;
  63. Window win; /* menu window to map on the screen */
  64. };
  65. /* function declarations */
  66. static int menuexist(void);
  67. static void getcolor(const char *s, XftColor *color);
  68. static void getresources(void);
  69. static void setupdc(void);
  70. static void setupgeom(void);
  71. static struct Item *allocitem(const char *label, const char *output);
  72. static struct Menu *allocmenu(struct Menu *parent, struct Item *list, unsigned level);
  73. static void getmenuitem(Window win, int y, struct Menu **menu_ret, struct Item **item_ret);
  74. static void drawmenu(void);
  75. static void calcscreengeom(void);
  76. static void calcmenu(struct Menu *menu);
  77. static void recalcmenu(struct Menu *menu);
  78. static void grabpointer(void);
  79. static void grabkeyboard(void);
  80. static void setcurrmenu(struct Menu *currmenu_new);
  81. static void parsestdin(void);
  82. static void run(void);
  83. static void freewindow(struct Menu *menu);
  84. static void cleanup(void);
  85. static void usage(void);
  86. /* X variables */
  87. static Colormap colormap;
  88. static Display *dpy;
  89. static Visual *visual;
  90. static Window rootwin;
  91. static int screen;
  92. static struct DC dc;
  93. static Atom wmdelete;
  94. /* menu variables */
  95. static struct Menu *rootmenu = NULL;
  96. static struct Menu *currmenu = NULL;
  97. static char **menutitle;
  98. static int menutitlecount;
  99. /* geometry variables */
  100. static struct Geometry geom;
  101. static struct ScreenGeometry screengeom;
  102. /* flag variables */
  103. static int wflag = 0;
  104. #include "config.h"
  105. int
  106. main(int argc, char *argv[])
  107. {
  108. int ch;
  109. while ((ch = getopt(argc, argv, "w")) != -1) {
  110. switch (ch) {
  111. case 'w':
  112. wflag = 1;
  113. break;
  114. default:
  115. usage();
  116. break;
  117. }
  118. }
  119. argc -= optind;
  120. argv += optind;
  121. menutitle = argv;
  122. menutitlecount = argc;
  123. /* open connection to server and set X variables */
  124. if ((dpy = XOpenDisplay(NULL)) == NULL)
  125. errx(1, "cannot open display");
  126. screen = DefaultScreen(dpy);
  127. visual = DefaultVisual(dpy, screen);
  128. rootwin = RootWindow(dpy, screen);
  129. colormap = DefaultColormap(dpy, screen);
  130. wmdelete=XInternAtom(dpy, "WM_DELETE_WINDOW", True);
  131. /* exit if another menu exists */
  132. if (menuexist()) {
  133. XCloseDisplay(dpy);
  134. return 1;
  135. }
  136. /* setup */
  137. getresources();
  138. setupdc();
  139. setupgeom();
  140. /* generate menus and recalculate them */
  141. parsestdin();
  142. if (rootmenu == NULL)
  143. errx(1, "no menu generated");
  144. calcscreengeom();
  145. calcmenu(rootmenu);
  146. /* grab mouse and keyboard */
  147. if (!wflag) {
  148. grabpointer();
  149. grabkeyboard();
  150. }
  151. /* map root menu */
  152. currmenu = rootmenu;
  153. XMapWindow(dpy, rootmenu->win);
  154. /* run event loop */
  155. run();
  156. cleanup();
  157. return 0;
  158. }
  159. /* check whether another menu exists */
  160. static int
  161. menuexist(void)
  162. {
  163. Window wina, winb; /* unused variables */
  164. Window *children;
  165. unsigned nchildren;
  166. XClassHint classh;
  167. if (XQueryTree(dpy, rootwin, &wina, &winb, &children, &nchildren) == 0)
  168. errx(1, "could not query tree");
  169. while (nchildren-- > 0) {
  170. if (XGetClassHint(dpy, *children, &classh) != 0)
  171. if (strcmp(classh.res_class, PROGNAME) == 0)
  172. return 1;
  173. children++;
  174. }
  175. return 0;
  176. }
  177. /* read xrdb for configuration options */
  178. static void
  179. getresources(void)
  180. {
  181. char *xrm;
  182. long n;
  183. XrmInitialize();
  184. if ((xrm = XResourceManagerString(dpy))) {
  185. char *type;
  186. XrmDatabase xdb;
  187. XrmValue xval;
  188. xdb = XrmGetStringDatabase(xrm);
  189. if (XrmGetResource(xdb, "xmenu.menuborder", "*", &type, &xval) == True)
  190. if ((n = strtol(xval.addr, NULL, 10)) > 0)
  191. menuborder = n;
  192. if (XrmGetResource(xdb, "xmenu.separatorsize", "*", &type, &xval) == True)
  193. if ((n = strtol(xval.addr, NULL, 10)) > 0)
  194. separatorsize = n;
  195. if (XrmGetResource(xdb, "xmenu.itemborder", "*", &type, &xval) == True)
  196. if ((n = strtol(xval.addr, NULL, 10)) > 0)
  197. itemborder = n;
  198. if (XrmGetResource(xdb, "xmenu.width", "*", &type, &xval) == True)
  199. if ((n = strtol(xval.addr, NULL, 10)) > 0)
  200. width = n;
  201. if (XrmGetResource(xdb, "xmenu.background", "*", &type, &xval) == True)
  202. background = strdup(xval.addr);
  203. if (XrmGetResource(xdb, "xmenu.foreground", "*", &type, &xval) == True)
  204. foreground = strdup(xval.addr);
  205. if (XrmGetResource(xdb, "xmenu.selbackground", "*", &type, &xval) == True)
  206. selbackground = strdup(xval.addr);
  207. if (XrmGetResource(xdb, "xmenu.selforeground", "*", &type, &xval) == True)
  208. selforeground = strdup(xval.addr);
  209. if (XrmGetResource(xdb, "xmenu.separator", "*", &type, &xval) == True)
  210. separator = strdup(xval.addr);
  211. if (XrmGetResource(xdb, "xmenu.border", "*", &type, &xval) == True)
  212. border = strdup(xval.addr);
  213. if (XrmGetResource(xdb, "xmenu.font", "*", &type, &xval) == True)
  214. font = strdup(xval.addr);
  215. XrmDestroyDatabase(xdb);
  216. }
  217. }
  218. /* get color from color string */
  219. static void
  220. getcolor(const char *s, XftColor *color)
  221. {
  222. if(!XftColorAllocName(dpy, visual, colormap, s, color))
  223. errx(1, "cannot allocate color: %s", s);
  224. }
  225. /* init draw context */
  226. static void
  227. setupdc(void)
  228. {
  229. /* get color pixels */
  230. getcolor(background, &dc.normal[ColorBG]);
  231. getcolor(foreground, &dc.normal[ColorFG]);
  232. getcolor(selbackground, &dc.selected[ColorBG]);
  233. getcolor(selforeground, &dc.selected[ColorFG]);
  234. getcolor(separator, &dc.decoration[ColorBG]);
  235. getcolor(border, &dc.decoration[ColorFG]);
  236. /* try to get font */
  237. if ((dc.font = XftFontOpenName(dpy, screen, font)) == NULL)
  238. errx(1, "cannot load font");
  239. /* create GC */
  240. dc.gc = XCreateGC(dpy, rootwin, 0, NULL);
  241. }
  242. /* init menu geometry values */
  243. static void
  244. setupgeom(void)
  245. {
  246. geom.itemb = itemborder;
  247. geom.itemh = dc.font->height + itemborder * 2;
  248. geom.itemw = width;
  249. geom.border = menuborder;
  250. geom.separator = separatorsize;
  251. }
  252. /* allocate an item */
  253. static struct Item *
  254. allocitem(const char *label, const char *output)
  255. {
  256. struct Item *item;
  257. if ((item = malloc(sizeof *item)) == NULL)
  258. err(1, "malloc");
  259. if (*label == '\0') {
  260. item->label = NULL;
  261. item->output = NULL;
  262. } else {
  263. if ((item->label = strdup(label)) == NULL)
  264. err(1, "strdup");
  265. if ((item->output = strdup(output)) == NULL)
  266. err(1, "strdup");
  267. }
  268. item->y = 0;
  269. item->h = item->label ? geom.itemh : geom.separator;
  270. if (item->label == NULL)
  271. item->labellen = 0;
  272. else
  273. item->labellen = strlen(item->label);
  274. item->next = NULL;
  275. item->submenu = NULL;
  276. return item;
  277. }
  278. /* allocate a menu */
  279. static struct Menu *
  280. allocmenu(struct Menu *parent, struct Item *list, unsigned level)
  281. {
  282. XSetWindowAttributes swa;
  283. struct Menu *menu;
  284. if ((menu = malloc(sizeof *menu)) == NULL)
  285. err(1, "malloc");
  286. menu->parent = parent;
  287. menu->list = list;
  288. menu->caller = NULL;
  289. menu->selected = NULL;
  290. menu->w = geom.itemw;
  291. menu->h = 0; /* calculated by calcmenu() */
  292. menu->x = 0; /* calculated by calcmenu() */
  293. menu->y = 0; /* calculated by calcmenu() */
  294. menu->level = level;
  295. swa.override_redirect = (wflag) ? False : True;
  296. swa.background_pixel = dc.decoration[ColorBG].pixel;
  297. swa.border_pixel = dc.decoration[ColorFG].pixel;
  298. swa.event_mask = ExposureMask | KeyPressMask | ButtonPressMask | ButtonReleaseMask
  299. | PointerMotionMask | LeaveWindowMask;
  300. menu->win = XCreateWindow(dpy, rootwin, 0, 0, geom.itemw, geom.itemh, geom.border,
  301. CopyFromParent, CopyFromParent, CopyFromParent,
  302. CWOverrideRedirect | CWBackPixel | CWBorderPixel | CWEventMask,
  303. &swa);
  304. XSetWMProtocols(dpy, menu->win, &wmdelete, 1);
  305. return menu;
  306. }
  307. /* create menus and items from the stdin */
  308. static void
  309. parsestdin(void)
  310. {
  311. char *s, buf[BUFSIZ];
  312. char *label, *output;
  313. unsigned level = 0;
  314. unsigned i;
  315. struct Item *curritem = NULL; /* item currently being read */
  316. struct Menu *prevmenu = NULL; /* menu the previous item was added to */
  317. struct Item *item; /* dummy item for for loops */
  318. struct Menu *menu; /* dummy menu for for loops */
  319. size_t count = 0; /* number of items in the current menu */
  320. while (fgets(buf, BUFSIZ, stdin) != NULL) {
  321. level = 0;
  322. s = buf;
  323. while (*s == '\t') {
  324. level++;
  325. s++;
  326. }
  327. label = output = s;
  328. while (*s != '\0' && *s != '\t' && *s != '\n')
  329. s++;
  330. while (*s == '\t')
  331. *s++ = '\0';
  332. if (*s != '\0' && *s != '\n')
  333. output = s;
  334. while (*s != '\0' && *s != '\n')
  335. s++;
  336. if (*s == '\n')
  337. *s = '\0';
  338. curritem = allocitem(label, output);
  339. if (prevmenu == NULL) { /* there is no menu yet */
  340. menu = allocmenu(NULL, curritem, level);
  341. rootmenu = menu;
  342. prevmenu = menu;
  343. count = 1;
  344. curritem->prev = NULL;
  345. curritem->next = NULL;
  346. } else if (level < prevmenu->level) { /* item is continuation of a parent menu*/
  347. for (menu = prevmenu, i = level;
  348. menu != NULL && i < prevmenu->level;
  349. menu = menu->parent, i++)
  350. ;
  351. if (menu == NULL)
  352. errx(1, "reached NULL menu");
  353. for (item = menu->list; item->next != NULL; item = item->next)
  354. ;
  355. item->next = curritem;
  356. curritem->prev = item;
  357. curritem->next = NULL;
  358. prevmenu = menu;
  359. } else if (level == prevmenu->level) { /* item is a continuation of current menu */
  360. for (item = prevmenu->list; item->next != NULL; item = item->next)
  361. ;
  362. item->next = curritem;
  363. curritem->prev = item;
  364. curritem->next = NULL;
  365. } else if (level > prevmenu->level) { /* item begins a new menu */
  366. menu = allocmenu(prevmenu, curritem, level);
  367. for (item = prevmenu->list; item->next != NULL; item = item->next)
  368. ;
  369. item->submenu = menu;
  370. menu->caller = item;
  371. curritem->prev = NULL;
  372. curritem->next = NULL;
  373. prevmenu = menu;
  374. }
  375. count++;
  376. }
  377. }
  378. /* calculate screen geometry */
  379. static void
  380. calcscreengeom(void)
  381. {
  382. Window w1, w2; /* unused variables */
  383. int a, b; /* unused variables */
  384. unsigned mask; /* unused variable */
  385. XQueryPointer(dpy, rootwin, &w1, &w2, &screengeom.cursx, &screengeom.cursy, &a, &b, &mask);
  386. screengeom.screenw = DisplayWidth(dpy, screen);
  387. screengeom.screenh = DisplayHeight(dpy, screen);
  388. }
  389. /* recursivelly calculate menu geometry and set window hints */
  390. static void
  391. calcmenu(struct Menu *menu)
  392. {
  393. static XClassHint classh = {PROGNAME, PROGNAME};
  394. XWindowChanges changes;
  395. XTextProperty textprop;
  396. XSizeHints sizeh;
  397. XGlyphInfo ext;
  398. struct Item *item;
  399. int labelwidth;
  400. /* calculate items positions and menu width and height */
  401. menu->w = geom.itemw;
  402. for (item = menu->list; item != NULL; item = item->next) {
  403. item->y = menu->h;
  404. if (item->label == NULL) /* height for separator item */
  405. menu->h += geom.separator;
  406. else
  407. menu->h += geom.itemh;
  408. XftTextExtentsUtf8(dpy, dc.font, (XftChar8 *)item->label,
  409. item->labellen, &ext);
  410. labelwidth = ext.xOff + dc.font->height * 2;
  411. menu->w = MAX(menu->w, labelwidth);
  412. }
  413. /* calculate menu's x and y positions */
  414. if (menu->parent == NULL) { /* if root menu, calculate in respect to cursor */
  415. if (screengeom.screenw - screengeom.cursx >= menu->w)
  416. menu->x = screengeom.cursx;
  417. else if (screengeom.cursx > menu->w)
  418. menu->x = screengeom.cursx - menu->w;
  419. if (screengeom.screenh - screengeom.cursy >= menu->h)
  420. menu->y = screengeom.cursy;
  421. else if (screengeom.screenh > menu->h)
  422. menu->y = screengeom.screenh - menu->h;
  423. XStringListToTextProperty(menutitle, menutitlecount, &textprop);
  424. } else { /* else, calculate in respect to parent menu */
  425. if (screengeom.screenw - (menu->parent->x + menu->parent->w + geom.border) >= menu->w)
  426. menu->x = menu->parent->x + menu->parent->w + geom.border;
  427. else if (menu->parent->x > menu->w + geom.border)
  428. menu->x = menu->parent->x - menu->w - geom.border;
  429. if (screengeom.screenh - (menu->caller->y + menu->parent->y) > menu->h)
  430. menu->y = menu->caller->y + menu->parent->y;
  431. else if (screengeom.screenh - menu->parent->y > menu->h)
  432. menu->y = menu->parent->y;
  433. else if (screengeom.screenh > menu->h)
  434. menu->y = screengeom.screenh - menu->h;
  435. XStringListToTextProperty(&(menu->caller->output), 1, &textprop);
  436. }
  437. /* update menu geometry */
  438. changes.height = menu->h;
  439. changes.width = menu->w;
  440. changes.x = menu->x;
  441. changes.y = menu->y;
  442. XConfigureWindow(dpy, menu->win, CWWidth | CWHeight | CWX | CWY, &changes);
  443. /* set window manager hints */
  444. sizeh.flags = PMaxSize | PMinSize;
  445. sizeh.min_width = sizeh.max_width = menu->w;
  446. sizeh.min_height = sizeh.max_height = menu->h;
  447. XSetWMProperties(dpy, menu->win, &textprop, NULL, NULL, 0, &sizeh,
  448. NULL, &classh);
  449. /* create pixmap and XftDraw */
  450. menu->pixmap = XCreatePixmap(dpy, menu->win, menu->w, menu->h,
  451. DefaultDepth(dpy, screen));
  452. menu->draw = XftDrawCreate(dpy, menu->pixmap, visual, colormap);
  453. /* calculate positions of submenus */
  454. for (item = menu->list; item != NULL; item = item->next) {
  455. if (item->submenu != NULL)
  456. calcmenu(item->submenu);
  457. }
  458. }
  459. /* recalculate menu position in respect to its parent */
  460. static void
  461. recalcmenu(struct Menu *menu)
  462. {
  463. XWindowAttributes parentwin;
  464. if (menu->parent == NULL)
  465. return;
  466. XGetWindowAttributes(dpy, menu->parent->win, &parentwin);
  467. if (screengeom.screenw - (parentwin.x + menu->parent->w + geom.border) >= menu->w)
  468. menu->x = parentwin.x + menu->parent->w + geom.border;
  469. else if (parentwin.x > menu->w + geom.border)
  470. menu->x = parentwin.x - menu->w - geom.border;
  471. if (screengeom.screenh - (menu->caller->y + parentwin.y) > menu->h)
  472. menu->y = menu->caller->y + parentwin.y;
  473. else if (screengeom.screenh - parentwin.y > menu->h)
  474. menu->y = parentwin.y;
  475. else if (screengeom.screenh > menu->h)
  476. menu->y = screengeom.screenh - menu->h;
  477. XMoveWindow(dpy, menu->win, menu->x, menu->y);
  478. }
  479. /* try to grab pointer, we may have to wait for another process to ungrab */
  480. static void
  481. grabpointer(void)
  482. {
  483. struct timespec ts = { .tv_sec = 0, .tv_nsec = 1000000 };
  484. int i;
  485. for (i = 0; i < 1000; i++) {
  486. if (XGrabPointer(dpy, rootwin, True, ButtonPressMask,
  487. GrabModeAsync, GrabModeAsync, None,
  488. None, CurrentTime) == GrabSuccess)
  489. return;
  490. nanosleep(&ts, NULL);
  491. }
  492. errx(1, "cannot grab keyboard");
  493. }
  494. /* try to grab keyboard, we may have to wait for another process to ungrab */
  495. static void
  496. grabkeyboard(void)
  497. {
  498. struct timespec ts = { .tv_sec = 0, .tv_nsec = 1000000 };
  499. int i;
  500. for (i = 0; i < 1000; i++) {
  501. if (XGrabKeyboard(dpy, rootwin, True, GrabModeAsync,
  502. GrabModeAsync, CurrentTime) == GrabSuccess)
  503. return;
  504. nanosleep(&ts, NULL);
  505. }
  506. errx(1, "cannot grab keyboard");
  507. }
  508. /* get menu and item of given window and position */
  509. static void
  510. getmenuitem(Window win, int y,
  511. struct Menu **menu_ret, struct Item **item_ret)
  512. {
  513. struct Menu *menu = NULL;
  514. struct Item *item = NULL;
  515. for (menu = currmenu; menu != NULL; menu = menu->parent) {
  516. if (menu->win == win) {
  517. for (item = menu->list; item != NULL; item = item->next) {
  518. if (y >= item->y && y <= item->y + item->h) {
  519. goto done;
  520. }
  521. }
  522. }
  523. }
  524. done:
  525. *menu_ret = menu;
  526. *item_ret = item;
  527. }
  528. /* set currentmenu to menu, umap previous menus and map current menu and its parents */
  529. static void
  530. setcurrmenu(struct Menu *currmenu_new)
  531. {
  532. struct Menu *menu, *menu_;
  533. struct Item *item;
  534. struct Menu *lcamenu; /* lowest common ancestor menu */
  535. unsigned minlevel; /* level of the closest to root menu */
  536. unsigned maxlevel; /* level of the closest to root menu */
  537. if (currmenu_new == currmenu)
  538. return;
  539. /* find lowest common ancestor menu */
  540. lcamenu = rootmenu;
  541. if (currmenu != NULL) {
  542. minlevel = MIN(currmenu_new->level, currmenu->level);
  543. maxlevel = MAX(currmenu_new->level, currmenu->level);
  544. if (currmenu_new->level == maxlevel) {
  545. menu = currmenu_new;
  546. menu_ = currmenu;
  547. } else {
  548. menu = currmenu;
  549. menu_ = currmenu_new;
  550. }
  551. while (menu->level > minlevel)
  552. menu = menu->parent;
  553. while (menu != menu_) {
  554. menu = menu->parent;
  555. menu_ = menu_->parent;
  556. }
  557. lcamenu = menu;
  558. }
  559. /* unmap menus from currmenu (inclusive) until lcamenu (exclusive) */
  560. for (menu = currmenu; menu != lcamenu; menu = menu->parent) {
  561. menu->selected = NULL;
  562. XUnmapWindow(dpy, menu->win);
  563. }
  564. currmenu = currmenu_new;
  565. /* map menus from currmenu (inclusive) until lcamenu (exclusive) */
  566. item = NULL;
  567. for (menu = currmenu; menu != lcamenu; menu = menu->parent) {
  568. if (wflag)
  569. recalcmenu(menu);
  570. XMapWindow(dpy, menu->win);
  571. if (item != NULL)
  572. menu->selected = item;
  573. item = menu->caller;
  574. }
  575. }
  576. /* draw items of the current menu and of its ancestors */
  577. static void
  578. drawmenu(void)
  579. {
  580. struct Menu *menu;
  581. struct Item *item;
  582. for (menu = currmenu; menu != NULL; menu = menu->parent) {
  583. for (item = menu->list; item != NULL; item = item->next) {
  584. XftColor *color;
  585. int labelx, labely;
  586. /* determine item color */
  587. if (item == menu->selected)
  588. color = dc.selected;
  589. else
  590. color = dc.normal;
  591. /* continue if item is a separator */
  592. if (item->label == NULL)
  593. continue;
  594. /* draw item box */
  595. XSetForeground(dpy, dc.gc, color[ColorBG].pixel);
  596. XDrawRectangle(dpy, menu->pixmap, dc.gc, 0, item->y,
  597. menu->w, item->h);
  598. XFillRectangle(dpy, menu->pixmap, dc.gc, 0, item->y,
  599. menu->w, item->h);
  600. /* draw item label */
  601. labelx = 0 + dc.font->height;
  602. labely = item->y + dc.font->height + geom.itemb / 2;
  603. XSetForeground(dpy, dc.gc, color[ColorFG].pixel);
  604. XftDrawStringUtf8(menu->draw, &color[ColorFG], dc.font,
  605. labelx, labely, item->label,
  606. item->labellen);
  607. /* draw triangle, if item contains a submenu */
  608. if (item->submenu != NULL) {
  609. int trianglex = menu->w - dc.font->height + geom.itemb - 1;
  610. int triangley = item->y + (3 * item->h)/8 -1;
  611. XPoint triangle[] = {
  612. {trianglex, triangley},
  613. {trianglex + item->h/8 + 1, item->y + item->h/2},
  614. {trianglex, triangley + item->h/4 + 2},
  615. {trianglex, triangley}
  616. };
  617. XFillPolygon(dpy, menu->pixmap, dc.gc, triangle, LEN(triangle),
  618. Convex, CoordModeOrigin);
  619. }
  620. XCopyArea(dpy, menu->pixmap, menu->win, dc.gc, 0, item->y,
  621. menu->w, item->h, 0, item->y);
  622. }
  623. }
  624. }
  625. /* cycle through the items; non-zero direction is next, zero is prev */
  626. static struct Item *
  627. itemcycle(int direction)
  628. {
  629. struct Item *item;
  630. struct Item *lastitem;
  631. item = NULL;
  632. if (direction == ITEMNEXT) {
  633. if (currmenu->selected == NULL)
  634. item = currmenu->list;
  635. else if (currmenu->selected->next != NULL)
  636. item = currmenu->selected->next;
  637. while (item != NULL && item->label == NULL)
  638. item = item->next;
  639. if (item == NULL)
  640. item = currmenu->list;
  641. } else {
  642. for (lastitem = currmenu->list;
  643. lastitem != NULL && lastitem->next != NULL;
  644. lastitem = lastitem->next)
  645. ;
  646. if (currmenu->selected == NULL)
  647. item = lastitem;
  648. else if (currmenu->selected->prev != NULL)
  649. item = currmenu->selected->prev;
  650. while (item != NULL && item->label == NULL)
  651. item = item->prev;
  652. if (item == NULL)
  653. item = lastitem;
  654. }
  655. return item;
  656. }
  657. /* run event loop */
  658. static void
  659. run(void)
  660. {
  661. struct Menu *menu;
  662. struct Item *item;
  663. struct Item *previtem = NULL;
  664. KeySym ksym;
  665. XEvent ev;
  666. while (!XNextEvent(dpy, &ev)) {
  667. switch(ev.type) {
  668. case Expose:
  669. if (ev.xexpose.count == 0)
  670. drawmenu();
  671. break;
  672. case MotionNotify:
  673. getmenuitem(ev.xbutton.window, ev.xbutton.y, &menu, &item);
  674. if (menu != NULL && item != NULL) {
  675. if (previtem != item) {
  676. if (item->submenu != NULL)
  677. setcurrmenu(item->submenu);
  678. else
  679. setcurrmenu(menu);
  680. previtem = item;
  681. drawmenu();
  682. } else if (menu->selected != item) {
  683. menu->selected = item;
  684. drawmenu();
  685. }
  686. }
  687. break;
  688. case ButtonRelease:
  689. getmenuitem(ev.xbutton.window, ev.xbutton.y, &menu, &item);
  690. if (menu != NULL && item != NULL) {
  691. selectitem:
  692. if (item->label == NULL)
  693. break; /* ignore separators */
  694. if (item->submenu != NULL) {
  695. setcurrmenu(item->submenu);
  696. } else {
  697. printf("%s\n", item->output);
  698. return;
  699. }
  700. currmenu->selected = currmenu->list;
  701. drawmenu();
  702. break;
  703. } else {
  704. return;
  705. }
  706. case ButtonPress:
  707. getmenuitem(ev.xbutton.window, ev.xbutton.y, &menu, &item);
  708. if (menu == NULL || item == NULL)
  709. return;
  710. break;
  711. case KeyPress:
  712. ksym = XkbKeycodeToKeysym(dpy, ev.xkey.keycode, 0, 0);
  713. if (ksym == XK_Escape && currmenu == rootmenu)
  714. return;
  715. /* Shift-Tab = ISO_Left_Tab */
  716. if (ksym == XK_Tab && (ev.xkey.state & ShiftMask))
  717. ksym = XK_ISO_Left_Tab;
  718. /* cycle through menu */
  719. item = NULL;
  720. if (ksym == XK_ISO_Left_Tab || ksym == XK_Up) {
  721. item = itemcycle(ITEMPREV);
  722. } else if (ksym == XK_Tab || ksym == XK_Down) {
  723. item = itemcycle(ITEMNEXT);
  724. } else if ((ksym == XK_Return || ksym == XK_Right) &&
  725. currmenu->selected != NULL) {
  726. item = currmenu->selected;
  727. goto selectitem;
  728. } else if ((ksym == XK_Escape || ksym == XK_Left) &&
  729. currmenu->parent != NULL) {
  730. item = currmenu->parent->selected;
  731. setcurrmenu(currmenu->parent);
  732. } else
  733. break;
  734. currmenu->selected = item;
  735. drawmenu();
  736. break;
  737. case LeaveNotify:
  738. currmenu->selected = NULL;
  739. drawmenu();
  740. break;
  741. case ClientMessage: /* user closed a window */
  742. return;
  743. }
  744. }
  745. }
  746. /* recursivelly free a pixmap */
  747. static void
  748. freewindow(struct Menu *menu)
  749. {
  750. struct Item *item;
  751. for (item = menu->list; item != NULL; item = item->next)
  752. if (item->submenu != NULL)
  753. freewindow(item->submenu);
  754. XFreePixmap(dpy, menu->pixmap);
  755. XftDrawDestroy(menu->draw);
  756. XDestroyWindow(dpy, menu->win);
  757. }
  758. /* cleanup and exit */
  759. static void
  760. cleanup(void)
  761. {
  762. freewindow(rootmenu);
  763. XftColorFree(dpy, visual, colormap, &dc.normal[ColorBG]);
  764. XftColorFree(dpy, visual, colormap, &dc.normal[ColorFG]);
  765. XftColorFree(dpy, visual, colormap, &dc.selected[ColorBG]);
  766. XftColorFree(dpy, visual, colormap, &dc.selected[ColorFG]);
  767. XftColorFree(dpy, visual, colormap, &dc.decoration[ColorBG]);
  768. XftColorFree(dpy, visual, colormap, &dc.decoration[ColorFG]);
  769. XFreeGC(dpy, dc.gc);
  770. XCloseDisplay(dpy);
  771. }
  772. /* show usage */
  773. static void
  774. usage(void)
  775. {
  776. (void)fprintf(stderr, "usage: xmenu [-w] title...\n");
  777. exit(1);
  778. }