tools/kdetrayproxy/kdetrayproxy.c

   1 #include <X11/Xlib.h>
   2 #include <X11/Xatom.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <sys/select.h>
   7 #include <unistd.h>
   8
   9 typedef struct IList {
  10     Window win;
  11     int ignore_unmaps;
  12
  13     struct IList *next;
  14 } IList;
  15
  16 Display *display;
  17 Window root;
  18 Atom winhint;
  19 Atom roothint;
  20 int xfd;
  21 IList *list;
  22
  23 void init();
  24 void eventloop();
  25 void handleevent(XEvent *e);
  26 void addicon(Window win);
  27 void removeicon(Window win, int unmap);
  28 int issystray(Atom *a, int n);
  29 void updatehint();
  30 Window findclient(Window win);
  31 int ignore_errors(Display *d, XErrorEvent *e);
  32 void wait_time(unsigned int t);
  33
  34 int main()
  35 {
  36     init();
  37     updatehint();
  38     eventloop();
  39     return 0;
  40 }
  41
  42 void init()
  43 {
  44     display = XOpenDisplay(NULL);
  45     if (!display) {
  46         fprintf(stderr, "Could not open display\n");
  47         exit(EXIT_FAILURE);
  48     }
  49
  50     xfd = ConnectionNumber(display);
  51
  52     root = RootWindowOfScreen(DefaultScreenOfDisplay(display));
  53
  54     winhint = XInternAtom(display, "_KDE_NET_WM_SYSTEM_TRAY_WINDOW_FOR", 0);
  55     roothint = XInternAtom(display, "_KDE_NET_SYSTEM_TRAY_WINDOWS", 0);
  56
  57     XSelectInput(display, root, SubstructureNotifyMask);
  58 }
  59
  60 void eventloop()
  61 {
  62     XEvent e;
  63     fd_set set;
  64
  65     while (1) {
  66         int event = False;
  67         while (XPending(display)) {
  68             event = True;
  69             XNextEvent(display, &e);
  70             handleevent(&e);
  71         }
  72         if (!event) {
  73             FD_ZERO(&set);
  74             FD_SET(xfd, &set);
  75             select(xfd + 1, &set, NULL, NULL, NULL);
  76         }
  77     }
  78 }
  79
  80 void handleevent(XEvent *e)
  81 {
  82     switch (e->type) {
  83     case MapNotify:
  84     {
  85         Atom *a;
  86         int n;
  87         Window w;
  88
  89         w = findclient(e->xmap.window);
  90         if (w) {
  91             a = XListProperties(display, w, &n);
  92             if (issystray(a, n))
  93                 addicon(w);
  94             XFree(a);
  95         }
  96         break;
  97     }
  98     case UnmapNotify:
  99         removeicon(e->xunmap.window, True);
 100         break;
 101     case DestroyNotify:
 102         removeicon(e->xdestroywindow.window, False);
 103         break;
 104     }
 105 }
 106
 107 int ignore_errors(Display *d, XErrorEvent *e)
 108 {
 109     (void)d; (void)e;
 110     return 1;
 111 }
 112
 113 void addicon(Window win)
 114 {
 115     IList *it;
 116
 117     for (it = list; it; it = it->next)
 118         if (it->win == win) return; /* duplicate */
 119
 120     it = list;
 121     list = malloc(sizeof(IList));
 122     list->win = win;
 123     list->ignore_unmaps = 2;
 124     list->next = it;
 125
 126     XSelectInput(display, win, StructureNotifyMask);
 127     /* if i set the root hint too fast the dock app can fuck itself up */
 128     wait_time(1000000 / 8);
 129     updatehint();
 130 }
 131
 132 void removeicon(Window win, int unmap)
 133 {
 134     IList *it, *last = NULL;
 135     void *old;
 136
 137     for (it = list; it; last = it, it = it->next)
 138         if (it->win == win) {
 139             if (it->ignore_unmaps && unmap) {
 140                 it->ignore_unmaps--;
 141                 return;
 142             }
 143
 144             if (!last)
 145                 list = it->next;
 146             else
 147                 last->next = it->next;
 148
 149             XSync(display, False);
 150             old = XSetErrorHandler(ignore_errors);
 151             XSelectInput(display, win, NoEventMask);
 152             XSync(display, False);
 153             XSetErrorHandler(old);
 154             free(it);
 155
 156             updatehint();
 157         }
 158 }
 159
 160 int issystray(Atom *a, int n)
 161 {
 162     int i, r = False;
 163
 164     for (i = 0; i < n; ++i) {
 165         if (a[i] == winhint) {
 166             r = True;
 167             break;
 168         }
 169     }
 170     return r;
 171 }
 172
 173 void updatehint()
 174 {
 175     IList *it;
 176     int *wins, n, i;
 177
 178     for (it = list, n = 0; it; it = it->next, ++n) ;
 179     if (n) {
 180         wins = malloc(sizeof(int) * n);
 181         for (it = list, i = 0; it; it = it->next, ++i)
 182             wins[i] = it->win;
 183     }
 184     XChangeProperty(display, root, roothint, XA_WINDOW, 32, PropModeReplace,
 185                     (unsigned char*) wins, n);
 186 }
 187
 188 Window findclient(Window win)
 189 {
 190   Window r, *children;
 191   unsigned int n, i;
 192   Atom state = XInternAtom(display, "WM_STATE", True);
 193   Atom ret_type;
 194   int ret_format;
 195   unsigned long ret_items, ret_bytesleft;
 196   unsigned long *prop_return;
 197
 198   XQueryTree(display, win, &r, &r, &children, &n);
 199   for (i = 0; i < n; ++i) {
 200     Window w = findclient(children[i]);
 201     if (w) return w;
 202   }
 203
 204   /* try me */
 205   XGetWindowProperty(display, win, state, 0, 1,
 206                      False, state, &ret_type, &ret_format,
 207                      &ret_items, &ret_bytesleft,
 208                      (unsigned char**) &prop_return);
 209   if (ret_type == None || ret_items < 1)
 210     return None;
 211   return win; /* found it! */
 212 }
 213
 214 void wait_time(unsigned int t)
 215 {
 216     struct timeval time;
 217     time.tv_sec = 0;
 218     time.tv_usec = t;
 219     select(1, NULL, NULL, NULL, &time);
 220 }