|
|
@@ -0,0 +1,565 @@ |
|
|
|
#include <err.h> |
|
|
|
#include <stdio.h> |
|
|
|
#include <stdlib.h> |
|
|
|
#include <string.h> |
|
|
|
#include <unistd.h> |
|
|
|
#include <X11/Xlib.h> |
|
|
|
#include <X11/Xutil.h> |
|
|
|
|
|
|
|
/* macros */ |
|
|
|
#define LEN(x) (sizeof (x) / sizeof (x[0])) |
|
|
|
|
|
|
|
/* color enum */ |
|
|
|
enum {ColorFG, ColorBG, ColorLast}; |
|
|
|
|
|
|
|
/* draw context structure */ |
|
|
|
struct DC { |
|
|
|
unsigned long unpressed[ColorLast]; |
|
|
|
unsigned long pressed[ColorLast]; |
|
|
|
unsigned long decoration[ColorLast]; |
|
|
|
|
|
|
|
Drawable d; |
|
|
|
GC gc; |
|
|
|
XFontStruct *font; |
|
|
|
int fonth; |
|
|
|
}; |
|
|
|
|
|
|
|
/* menu geometry structure */ |
|
|
|
struct Geometry { |
|
|
|
int itemb; /* item border */ |
|
|
|
int itemw; /* item width */ |
|
|
|
int itemh; /* item height */ |
|
|
|
int border; /* window border */ |
|
|
|
}; |
|
|
|
|
|
|
|
/* screen geometry structure */ |
|
|
|
struct ScreenGeometry { |
|
|
|
int cursx, cursy; /* cursor position */ |
|
|
|
int screenw, screenh; /* screen width and height */ |
|
|
|
}; |
|
|
|
|
|
|
|
/* menu item structure */ |
|
|
|
struct Item { |
|
|
|
char *label; |
|
|
|
char *output; |
|
|
|
int x, y; |
|
|
|
struct Item *next; |
|
|
|
struct Menu *submenu; |
|
|
|
}; |
|
|
|
|
|
|
|
/* menu structure */ |
|
|
|
struct Menu { |
|
|
|
struct Menu *parent; |
|
|
|
struct Item *list; |
|
|
|
struct Item *selected; |
|
|
|
int x, y, w, h; |
|
|
|
unsigned level; |
|
|
|
unsigned nitems; |
|
|
|
Window win; |
|
|
|
}; |
|
|
|
|
|
|
|
/* function declarations */ |
|
|
|
static unsigned long getcolor(const char *s); |
|
|
|
static void setupdc(void); |
|
|
|
static void setupgeom(void); |
|
|
|
static void setupgrab(void); |
|
|
|
static struct Item *allocitem(size_t count, const char *label, const char *output); |
|
|
|
static struct Menu *allocmenu(struct Menu *parent, struct Item *list, unsigned level); |
|
|
|
static void getmenuitem(Window win, int x, int y, |
|
|
|
struct Menu **menu_ret, struct Item **item_ret); |
|
|
|
static void drawmenu(void); |
|
|
|
static void calcscreengeom(void); |
|
|
|
static void calcmenu(struct Menu *menu); |
|
|
|
static void setcurrmenu(struct Menu *currmenu_new); |
|
|
|
static void parsestdin(void); |
|
|
|
static void run(void); |
|
|
|
static void cleanupexit(void); |
|
|
|
static void usage(void); |
|
|
|
|
|
|
|
/* X variables */ |
|
|
|
static Colormap colormap; |
|
|
|
static Display *dpy; |
|
|
|
static Window rootwin; |
|
|
|
static int screen; |
|
|
|
static struct DC dc; |
|
|
|
|
|
|
|
/* menu variables */ |
|
|
|
static struct Menu *rootmenu = NULL; |
|
|
|
static struct Menu *currmenu = NULL; |
|
|
|
|
|
|
|
/* geometry variables */ |
|
|
|
static struct Geometry geom; |
|
|
|
static struct ScreenGeometry sgeom; |
|
|
|
|
|
|
|
/* flag variables */ |
|
|
|
static Bool override_redirect = True; |
|
|
|
|
|
|
|
#include "config.h" |
|
|
|
|
|
|
|
int |
|
|
|
main(int argc, char *argv[]) |
|
|
|
{ |
|
|
|
int ch; |
|
|
|
|
|
|
|
while ((ch = getopt(argc, argv, "w")) != -1) { |
|
|
|
switch (ch) { |
|
|
|
case 'w': |
|
|
|
override_redirect = False; |
|
|
|
break; |
|
|
|
default: |
|
|
|
usage(); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
argc -= optind; |
|
|
|
argv += optind; |
|
|
|
|
|
|
|
/* open connection to server and set X variables */ |
|
|
|
if ((dpy = XOpenDisplay(NULL)) == NULL) |
|
|
|
errx(1, "cannot open display"); |
|
|
|
screen = DefaultScreen(dpy); |
|
|
|
rootwin = RootWindow(dpy, screen); |
|
|
|
colormap = DefaultColormap(dpy, screen); |
|
|
|
|
|
|
|
/* setup */ |
|
|
|
setupdc(); |
|
|
|
setupgeom(); |
|
|
|
setupgrab(); |
|
|
|
|
|
|
|
/* generate menus and recalculate them */ |
|
|
|
parsestdin(); |
|
|
|
if (rootmenu == NULL) |
|
|
|
errx(1, "no menu generated"); |
|
|
|
calcscreengeom(); |
|
|
|
calcmenu(rootmenu); |
|
|
|
|
|
|
|
/* run event loop */ |
|
|
|
run(); |
|
|
|
|
|
|
|
return 1; /* UNREACHABLE */ |
|
|
|
} |
|
|
|
|
|
|
|
/* get color from color string */ |
|
|
|
static unsigned long |
|
|
|
getcolor(const char *s) |
|
|
|
{ |
|
|
|
XColor color; |
|
|
|
|
|
|
|
if(!XAllocNamedColor(dpy, colormap, s, &color, &color)) |
|
|
|
errx(1, "cannot allocate color: %s", s); |
|
|
|
return color.pixel; |
|
|
|
} |
|
|
|
|
|
|
|
/* init draw context */ |
|
|
|
static void |
|
|
|
setupdc(void) |
|
|
|
{ |
|
|
|
/* get color pixels */ |
|
|
|
dc.unpressed[ColorBG] = getcolor(UNPRESSEDBG); |
|
|
|
dc.unpressed[ColorFG] = getcolor(UNPRESSEDFG); |
|
|
|
dc.pressed[ColorBG] = getcolor(PRESSEDBG); |
|
|
|
dc.pressed[ColorFG] = getcolor(PRESSEDFG); |
|
|
|
dc.decoration[ColorBG] = getcolor(DECORATIONBG); |
|
|
|
dc.decoration[ColorFG] = getcolor(DECORATIONFG); |
|
|
|
|
|
|
|
/* try to get font */ |
|
|
|
if ((dc.font = XLoadQueryFont(dpy, FONT)) == NULL) |
|
|
|
errx(1, "cannot load font"); |
|
|
|
dc.fonth = dc.font->ascent + dc.font->descent; |
|
|
|
|
|
|
|
/* create GC and set its font */ |
|
|
|
dc.gc = XCreateGC(dpy, rootwin, 0, NULL); |
|
|
|
XSetFont(dpy, dc.gc, dc.font->fid); |
|
|
|
} |
|
|
|
|
|
|
|
/* init menu geometry values */ |
|
|
|
static void |
|
|
|
setupgeom(void) |
|
|
|
{ |
|
|
|
geom.itemb = ITEMB; |
|
|
|
geom.itemh = dc.fonth + ITEMB * 2; |
|
|
|
geom.itemw = ITEMW; |
|
|
|
geom.border = BORDER; |
|
|
|
} |
|
|
|
|
|
|
|
/* grab pointer */ |
|
|
|
static void |
|
|
|
setupgrab(void) |
|
|
|
{ |
|
|
|
XGrabPointer(dpy, rootwin, True, ButtonPressMask | ButtonReleaseMask, |
|
|
|
GrabModeAsync, GrabModeAsync, None, None, CurrentTime); |
|
|
|
} |
|
|
|
|
|
|
|
/* allocate an item */ |
|
|
|
static struct Item * |
|
|
|
allocitem(size_t count, const char *label, const char *output) |
|
|
|
{ |
|
|
|
struct Item *item; |
|
|
|
|
|
|
|
if ((item = malloc(sizeof *item)) == NULL) |
|
|
|
err(1, "malloc"); |
|
|
|
if ((item->label = strdup(label)) == NULL) |
|
|
|
err(1, "strdup"); |
|
|
|
if ((item->output = strdup(output)) == NULL) |
|
|
|
err(1, "strdup"); |
|
|
|
item->x = 0; |
|
|
|
item->y = count * geom.itemh; |
|
|
|
item->next = NULL; |
|
|
|
item->submenu = NULL; |
|
|
|
|
|
|
|
return item; |
|
|
|
} |
|
|
|
|
|
|
|
/* allocate a menu */ |
|
|
|
static struct Menu * |
|
|
|
allocmenu(struct Menu *parent, struct Item *list, unsigned level) |
|
|
|
{ |
|
|
|
XSetWindowAttributes swa; |
|
|
|
struct Menu *menu; |
|
|
|
|
|
|
|
if ((menu = malloc(sizeof *menu)) == NULL) |
|
|
|
err(1, "malloc"); |
|
|
|
menu->parent = parent; |
|
|
|
menu->list = list; |
|
|
|
menu->selected = NULL; |
|
|
|
menu->x = 0; |
|
|
|
menu->y = 0; |
|
|
|
menu->w = geom.itemw; |
|
|
|
menu->h = geom.itemh; |
|
|
|
menu->level = level; |
|
|
|
menu->nitems = 0; |
|
|
|
|
|
|
|
swa.override_redirect = override_redirect; |
|
|
|
swa.background_pixel = dc.decoration[ColorBG]; |
|
|
|
swa.border_pixel = dc.decoration[ColorFG]; |
|
|
|
swa.event_mask = ExposureMask | KeyPressMask | ButtonPressMask | ButtonReleaseMask |
|
|
|
| PointerMotionMask; |
|
|
|
menu->win = XCreateWindow(dpy, rootwin, 0, 0, geom.itemw, geom.itemh, geom.border, |
|
|
|
CopyFromParent, CopyFromParent, CopyFromParent, |
|
|
|
CWOverrideRedirect | CWBackPixel | CWBorderPixel | CWEventMask, |
|
|
|
&swa); |
|
|
|
|
|
|
|
return menu; |
|
|
|
} |
|
|
|
|
|
|
|
/* create menus and items from the stdin */ |
|
|
|
static void |
|
|
|
parsestdin(void) |
|
|
|
{ |
|
|
|
char *s, buf[BUFSIZ]; |
|
|
|
char *label, *output; |
|
|
|
unsigned level = 0; |
|
|
|
unsigned i; |
|
|
|
struct Item *item, *p; |
|
|
|
struct Menu *menu; |
|
|
|
struct Menu *prevmenu = NULL; |
|
|
|
size_t count = 0; /* number of items in the current menu */ |
|
|
|
|
|
|
|
while (fgets(buf, BUFSIZ, stdin) != NULL) { |
|
|
|
level = 0; |
|
|
|
s = buf; |
|
|
|
|
|
|
|
while (*s == '\t') { |
|
|
|
level++; |
|
|
|
s++; |
|
|
|
} |
|
|
|
|
|
|
|
label = output = s; |
|
|
|
|
|
|
|
while (*s != '\0' && *s != '\t' && *s != '\n') |
|
|
|
s++; |
|
|
|
|
|
|
|
while (*s == '\t') |
|
|
|
*s++ = '\0'; |
|
|
|
|
|
|
|
if (*s != '\0' && *s != '\n') |
|
|
|
output = s; |
|
|
|
|
|
|
|
while (*s != '\0' && *s != '\n') |
|
|
|
s++; |
|
|
|
|
|
|
|
if (*s == '\n') |
|
|
|
*s = '\0'; |
|
|
|
|
|
|
|
item = allocitem(count, label, output); |
|
|
|
|
|
|
|
if (prevmenu == NULL) { /* there is no menu yet */ |
|
|
|
menu = allocmenu(NULL, item, level); |
|
|
|
rootmenu = menu; |
|
|
|
prevmenu = menu; |
|
|
|
count = 1; |
|
|
|
} else if (level < prevmenu->level) { /* item is continuation of a parent menu*/ |
|
|
|
for (menu = prevmenu, i = level; |
|
|
|
menu != NULL && i < prevmenu->level; |
|
|
|
menu = menu->parent, i++) |
|
|
|
; |
|
|
|
|
|
|
|
if (menu == NULL) |
|
|
|
errx(1, "reached NULL menu"); |
|
|
|
|
|
|
|
for (p = menu->list; p->next != NULL; p = p->next) |
|
|
|
; |
|
|
|
|
|
|
|
p->next = item; |
|
|
|
prevmenu = menu; |
|
|
|
} else if (level == prevmenu->level) { /* item is a continuation of current menu */ |
|
|
|
for (p = prevmenu->list; p->next != NULL; p = p->next) |
|
|
|
; |
|
|
|
p->next = item; |
|
|
|
} else if (level > prevmenu->level) { /* item begins a new menu */ |
|
|
|
menu = allocmenu(prevmenu, item, level); |
|
|
|
|
|
|
|
for (p = prevmenu->list; p->next != NULL; p = p->next) |
|
|
|
; |
|
|
|
|
|
|
|
p->submenu = menu; |
|
|
|
|
|
|
|
prevmenu = menu; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
/* calculate screen geometry */ |
|
|
|
static void |
|
|
|
calcscreengeom(void) |
|
|
|
{ |
|
|
|
Window w1, w2; /* unused variables */ |
|
|
|
int a, b; /* unused variables */ |
|
|
|
unsigned mask; /* unused variable */ |
|
|
|
|
|
|
|
XQueryPointer(dpy, rootwin, &w1, &w2, &sgeom.cursx, &sgeom.cursy, &a, &b, &mask); |
|
|
|
sgeom.screenw = DisplayWidth(dpy, screen); |
|
|
|
sgeom.screenh = DisplayHeight(dpy, screen); |
|
|
|
} |
|
|
|
|
|
|
|
/* recursivelly calculate height and position of the menus */ |
|
|
|
static void |
|
|
|
calcmenu(struct Menu *menu) |
|
|
|
{ |
|
|
|
XWindowChanges changes; |
|
|
|
struct Item *item, *p; |
|
|
|
size_t i; |
|
|
|
|
|
|
|
/* calculate number of items */ |
|
|
|
i = 0; |
|
|
|
for (item = menu->list; item != NULL; item = item->next) |
|
|
|
i++; |
|
|
|
menu->nitems = i; |
|
|
|
menu->h = geom.itemh * i; |
|
|
|
|
|
|
|
/* calculate menu's x and y positions */ |
|
|
|
if (menu->parent == NULL) { /* if root menu, calculate in respect to cursor */ |
|
|
|
if (sgeom.screenw - sgeom.cursx >= menu->w) |
|
|
|
menu->x = sgeom.cursx; |
|
|
|
else if (sgeom.cursx > menu->w) |
|
|
|
menu->x = sgeom.cursx - menu->w; |
|
|
|
|
|
|
|
if (sgeom.screenh - sgeom.cursy >= menu->h) |
|
|
|
menu->y = sgeom.cursy; |
|
|
|
else if (sgeom.screenh > menu->h) |
|
|
|
menu->y = sgeom.screenh - menu->h; |
|
|
|
} else { /* else, calculate in respect to parent menu */ |
|
|
|
|
|
|
|
/* search for the item in parent menu that generates this menu */ |
|
|
|
for (p = menu->parent->list; p->submenu != menu; p = p->next) |
|
|
|
; |
|
|
|
|
|
|
|
if (sgeom.screenw - (menu->parent->x + menu->parent->w) >= menu->w) |
|
|
|
menu->x = menu->parent->x + menu->parent->w; |
|
|
|
else if (menu->parent->x > menu->w) |
|
|
|
menu->x = menu->parent->x - menu->w; |
|
|
|
|
|
|
|
if (sgeom.screenh - p->y > menu->h) |
|
|
|
menu->y = p->y; |
|
|
|
else if (sgeom.screenh - menu->parent->y > menu->h) |
|
|
|
menu->y = menu->parent->y; |
|
|
|
else if (sgeom.screenh > menu->h) |
|
|
|
menu->y = sgeom.screenh - menu->h; |
|
|
|
} |
|
|
|
|
|
|
|
/* calculate position of each item in the menu */ |
|
|
|
for (i = 0, item = menu->list; item != NULL; item = item->next, i++) { |
|
|
|
item->x = menu->x; |
|
|
|
item->y = menu->y + i * geom.itemh; |
|
|
|
} |
|
|
|
|
|
|
|
/* update menu geometry */ |
|
|
|
changes.height = menu->h; |
|
|
|
changes.x = menu->x; |
|
|
|
changes.y = menu->y; |
|
|
|
XConfigureWindow(dpy, menu->win, CWHeight | CWX | CWY, &changes); |
|
|
|
|
|
|
|
for (item = menu->list; item != NULL; item = item->next) { |
|
|
|
if (item->submenu != NULL) |
|
|
|
calcmenu(item->submenu); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
/* get menu and item of given window and position */ |
|
|
|
static void |
|
|
|
getmenuitem(Window win, int x, int y, |
|
|
|
struct Menu **menu_ret, struct Item **item_ret) |
|
|
|
{ |
|
|
|
struct Menu *menu = NULL; |
|
|
|
struct Item *item = NULL; |
|
|
|
|
|
|
|
for (menu = currmenu; menu != NULL; menu = menu->parent) { |
|
|
|
if (menu->win == win) { |
|
|
|
for (item = menu->list; item != NULL; item = item->next) { |
|
|
|
if (x >= item->x && x <= item->x + geom.itemw && |
|
|
|
y >= item->y && y <= item->y + geom.itemh) { |
|
|
|
goto done; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
done: |
|
|
|
*menu_ret = menu; |
|
|
|
*item_ret = item; |
|
|
|
} |
|
|
|
|
|
|
|
/* set currentmenu to menu, umap previous menus and map current menu and its parents */ |
|
|
|
static void |
|
|
|
setcurrmenu(struct Menu *currmenu_new) |
|
|
|
{ |
|
|
|
struct Menu *menu; |
|
|
|
|
|
|
|
if (currmenu_new == currmenu) |
|
|
|
return; |
|
|
|
|
|
|
|
for (menu = currmenu; menu != NULL; menu = menu->parent) { |
|
|
|
XUnmapWindow(dpy, menu->win); |
|
|
|
} |
|
|
|
|
|
|
|
currmenu = currmenu_new; |
|
|
|
|
|
|
|
for (menu = currmenu; menu != NULL; menu = menu->parent) |
|
|
|
XMapWindow(dpy, menu->win); |
|
|
|
} |
|
|
|
|
|
|
|
/* draw items of the current menu and of its ancestors */ |
|
|
|
static void |
|
|
|
drawmenu(void) |
|
|
|
{ |
|
|
|
struct Menu *menu; |
|
|
|
struct Item *item; |
|
|
|
|
|
|
|
for (menu = currmenu; menu != NULL; menu = menu->parent) { |
|
|
|
size_t nitems; /* number of items before current item */ |
|
|
|
|
|
|
|
nitems = 0; |
|
|
|
for (item = menu->list; item != NULL; item = item->next) { |
|
|
|
unsigned long *color; |
|
|
|
size_t labellen; |
|
|
|
int labelx, labely; |
|
|
|
int y; |
|
|
|
|
|
|
|
/* determine item color */ |
|
|
|
if (item == menu->selected) |
|
|
|
color = dc.pressed; |
|
|
|
else |
|
|
|
color = dc.unpressed; |
|
|
|
|
|
|
|
/* calculate item's y position */ |
|
|
|
y = nitems * geom.itemh; |
|
|
|
|
|
|
|
/* draw item box */ |
|
|
|
XSetForeground(dpy, dc.gc, color[ColorBG]); |
|
|
|
XFillRectangle(dpy, menu->win, dc.gc, 0, y, |
|
|
|
geom.itemw, geom.itemh); |
|
|
|
|
|
|
|
/* draw item label */ |
|
|
|
labellen = strlen(item->label); |
|
|
|
labelx = 0 + dc.fonth; |
|
|
|
labely = y + dc.fonth + geom.itemb; |
|
|
|
XSetForeground(dpy, dc.gc, color[ColorFG]); |
|
|
|
XDrawString(dpy, menu->win, dc.gc, labelx, labely, item->label, labellen); |
|
|
|
|
|
|
|
/* draw triangle, if item contains a submenu */ |
|
|
|
if (item->submenu != NULL) { |
|
|
|
int trianglex = geom.itemw - (geom.itemb + dc.fonth); |
|
|
|
int triangley = y + geom.itemb; |
|
|
|
|
|
|
|
XPoint triangle[] = { |
|
|
|
{trianglex, triangley}, |
|
|
|
{trianglex + dc.fonth, triangley + dc.fonth/2}, |
|
|
|
{trianglex, triangley + dc.fonth}, |
|
|
|
{trianglex, triangley} |
|
|
|
}; |
|
|
|
|
|
|
|
XFillPolygon(dpy, menu->win, dc.gc, triangle, LEN(triangle), |
|
|
|
Convex, CoordModeOrigin); |
|
|
|
} |
|
|
|
|
|
|
|
nitems++; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
/* run event loop */ |
|
|
|
static void |
|
|
|
run(void) |
|
|
|
{ |
|
|
|
struct Menu *menu; |
|
|
|
struct Item *item; |
|
|
|
struct Item *previtem = NULL; |
|
|
|
XEvent ev; |
|
|
|
|
|
|
|
setcurrmenu(rootmenu); |
|
|
|
|
|
|
|
while (!XNextEvent(dpy, &ev)) { |
|
|
|
switch(ev.type) { |
|
|
|
case Expose: |
|
|
|
drawmenu(); |
|
|
|
break; |
|
|
|
case MotionNotify: |
|
|
|
getmenuitem(ev.xbutton.window, ev.xbutton.x_root, ev.xbutton.y_root, |
|
|
|
&menu, &item); |
|
|
|
if (menu != NULL && item != NULL) { |
|
|
|
if (previtem != item) { |
|
|
|
if (item->submenu != NULL) |
|
|
|
setcurrmenu(item->submenu); |
|
|
|
else |
|
|
|
setcurrmenu(menu); |
|
|
|
previtem = item; |
|
|
|
} else if (menu->selected != item) |
|
|
|
menu->selected = item; |
|
|
|
} |
|
|
|
drawmenu(); |
|
|
|
break; |
|
|
|
case ButtonRelease: |
|
|
|
getmenuitem(ev.xbutton.window, ev.xbutton.x_root, ev.xbutton.y_root, |
|
|
|
&menu, &item); |
|
|
|
if (menu != NULL && item != NULL) { |
|
|
|
if (item->submenu != NULL) { |
|
|
|
setcurrmenu(item->submenu); |
|
|
|
} else { |
|
|
|
printf("%s\n", item->output); |
|
|
|
cleanupexit(); |
|
|
|
} |
|
|
|
drawmenu(); |
|
|
|
} else { |
|
|
|
cleanupexit(); |
|
|
|
} |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
/* cleanup and exit */ |
|
|
|
static void |
|
|
|
cleanupexit(void) |
|
|
|
{ |
|
|
|
XCloseDisplay(dpy); |
|
|
|
exit(0); |
|
|
|
} |
|
|
|
|
|
|
|
/* show usage */ |
|
|
|
static void |
|
|
|
usage(void) |
|
|
|
{ |
|
|
|
(void)fprintf(stderr, "usage: xmenu [-w] menuname\n"); |
|
|
|
exit(1); |
|
|
|
} |