dwmblocks.c

   1 #include<stdlib.h>
   2 #include<stdio.h>
   3 #include<string.h>
   4 #include<unistd.h>
   5 #include<signal.h>
   6 #include<X11/Xlib.h>
   7 #define LENGTH(X)               (sizeof(X) / sizeof (X[0]))
   8
   9 typedef struct {
  10         char* icon;
  11         char* command;
  12         unsigned int interval;
  13         unsigned int signal;
  14 } Block;
  15 void sighandler(int num);
  16 void replace(char *str, char old, char new);
  17 void getcmds(int time);
  18 void getsigcmds(int signal);
  19 void setupsignals();
  20 void getstatus(char *str);
  21 void setroot();
  22 void statusloop();
  23 void statusinit();
  24 void sighandler(int signum);
  25 void termhandler(int signum);
  26
  27
  28 #include "blocks.h"
  29
  30 static Display *dpy;
  31 static int screen;
  32 static Window root;
  33 static char statusbar[LENGTH(blocks)][50] = {0};
  34 static char statusstr[256];
  35 static char *statuscat;
  36 static const char *volupcmd[]  = { "volup", NULL };
  37 static const char *voldowncmd[]  = { "voldown", NULL };
  38 static const char *volmutecmd[]  = { "volmute", NULL };
  39 static int statusContinue = 1,volmuted = 0;
  40
  41 void replace(char *str, char old, char new)
  42 {
  43         int N = strlen(str);
  44         for(int i = 0; i < N; i++)
  45                 if(str[i] == old)
  46                         str[i] = new;
  47 }
  48
  49 //opens process *cmd and stores output in *output
  50 void getcmd(const Block *block, char *output)
  51 {
  52         strcpy(output, block->icon);
  53         char *cmd = block->command;
  54         FILE *cmdf = popen(cmd,"r");
  55         if (!cmdf)
  56                 return;
  57         //int N = strlen(output);
  58         char c;
  59         int sl;
  60         int i = sl = strlen(block->icon);
  61         while((c = fgetc(cmdf)) != EOF)
  62                 output[i++] = c;
  63         pclose(cmdf);
  64         if (i == sl) { //return empty string if command has no output
  65                 output[0] = '\0';
  66                 return;
  67         }
  68         if (delim[0] != '\0' && --i)
  69                 for (int j = 0; delim[j]; j++)
  70                         output[i++] = delim[j];
  71         else
  72                 output[--i] = '\0';
  73 }
  74
  75 void getcmds(int time)
  76 {
  77         const Block* current;
  78         for(int i = 0; i < LENGTH(blocks); i++)
  79         {
  80                 current = blocks + i;
  81                 if ((current->interval != 0 && time % current->interval == 0) || time == -1)
  82                         getcmd(current,statusbar[i]);
  83         }
  84 }
  85
  86 void getsigcmds(int signal)
  87 {
  88         const Block *current;
  89         for (int i = 0; i < LENGTH(blocks); i++)
  90         {
  91                 current = blocks + i;
  92                 if (current->signal == signal)
  93                         getcmd(current,statusbar[i]);
  94         }
  95 }
  96
  97 void setupsignals()
  98 {
  99         for(int i = 0; i < LENGTH(blocks); i++)
 100         {
 101                 if (blocks[i].signal > 0)
 102                         signal(SIGRTMIN+blocks[i].signal, sighandler);
 103         }
 104
 105 }
 106
 107 void getstatus(char *str)
 108 {
 109         int j = 0;
 110         for(int i = 0; i < LENGTH(blocks); j+=strlen(statusbar[i++]))
 111         {
 112                 strcpy(str + j, statusbar[i]);
 113         }
 114         for (int i = 0; delim[i]; i++)
 115                 str[--j] = '\0';
 116
 117 }
 118
 119 void setroot()
 120 {
 121         Display *d = XOpenDisplay(NULL);
 122         if (d) {
 123                 dpy = d;
 124         }
 125         screen = DefaultScreen(dpy);
 126         root = RootWindow(dpy, screen);
 127         getstatus(statusstr);
 128         XStoreName(dpy, root, statusstr);
 129         XCloseDisplay(dpy);
 130 }
 131
 132
 133 void statusloop()
 134 {
 135         setupsignals();
 136         int i = 0;
 137         getcmds(-1);
 138         while(statusContinue)
 139         {
 140                 getcmds(i);
 141                 setroot();
 142                 sleep(1.0);
 143                 i++;
 144         }
 145 }
 146
 147 void statusinit()
 148 {
 149         statusloop();
 150 }
 151
 152
 153 void sighandler(int signum)
 154 {
 155         getsigcmds(signum-SIGRTMIN);
 156         setroot();
 157 }
 158
 159 void termhandler(int signum)
 160 {
 161         statusContinue = 0;
 162         exit(0);
 163 }
 164
 165 int main(int argc, char** argv)
 166 {
 167         for(int i = 0; i < argc; i++)
 168         {
 169                 if (!strcmp("-d",argv[i]))
 170                         delim = argv[++i];
 171         }
 172         signal(SIGTERM, termhandler);
 173         signal(SIGINT, termhandler);
 174         statusinit();
 175 }