Initial commit

[xmenu] / xmenu.c

#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);
}