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 int getstatus(char *str, char *last);
  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[2][256];
  35 static int statusContinue = 1;
  36 static void (*writestatus) () = setroot;
  37
  38 void replace(char *str, char old, char new)
  39 {
  40         int N = strlen(str);
  41         for(int i = 0; i < N; i++)
  42                 if(str[i] == old)
  43                         str[i] = new;
  44 }
  45
  46 //opens process *cmd and stores output in *output
  47 void getcmd(const Block *block, char *output)
  48 {
  49         strcpy(output, block->icon);
  50         char *cmd = block->command;
  51         FILE *cmdf = popen(cmd,"r");
  52         if (!cmdf)
  53                 return;
  54         char c;
  55         int i = strlen(block->icon);
  56         while((c = fgetc(cmdf)) != EOF)
  57         {
  58                 output[i++] = c;
  59                 if(c == '\n')
  60                         break;
  61         }
  62         if (delim != '\0' && --i)
  63                 output[i++] = delim;
  64         output[i++] = '\0';
  65         pclose(cmdf);
  66 }
  67
  68 void getcmds(int time)
  69 {
  70         const Block* current;
  71         for(int i = 0; i < LENGTH(blocks); i++)
  72         {
  73                 current = blocks + i;
  74                 if ((current->interval != 0 && time % current->interval == 0) || time == -1)
  75                         getcmd(current,statusbar[i]);
  76         }
  77 }
  78
  79 void getsigcmds(int signal)
  80 {
  81         const Block *current;
  82         for (int i = 0; i < LENGTH(blocks); i++)
  83         {
  84                 current = blocks + i;
  85                 if (current->signal == signal)
  86                         getcmd(current,statusbar[i]);
  87         }
  88 }
  89
  90 void setupsignals()
  91 {
  92         for(int i = 0; i < LENGTH(blocks); i++)
  93         {
  94                 if (blocks[i].signal > 0)
  95                         signal(SIGRTMIN+blocks[i].signal, sighandler);
  96         }
  97
  98 }
  99
 100 int getstatus(char *str, char *last)
 101 {
 102         strcpy(last, str);
 103         str[0] = '\0';
 104         for(int i = 0; i < LENGTH(blocks); i++)
 105         {
 106                 strcat(str, statusbar[i]);
 107         }
 108         str[strlen(str)-1] = '\0';
 109         return strcmp(str, last);//0 if they are the same
 110 }
 111
 112 void setroot()
 113 {
 114         if (!getstatus(statusstr[0], statusstr[1]))//Only set root if text has changed.
 115                 return;
 116         Display *d = XOpenDisplay(NULL);
 117         if (d) {
 118                 dpy = d;
 119         }
 120         screen = DefaultScreen(dpy);
 121         root = RootWindow(dpy, screen);
 122         XStoreName(dpy, root, statusstr[0]);
 123         XCloseDisplay(dpy);
 124 }
 125
 126 void pstdout()
 127 {
 128         if (!getstatus(statusstr[0], statusstr[1]))//Only write out if text has changed.
 129                 return;
 130         printf("%s\n",statusstr[0]);
 131         fflush(stdout);
 132 }
 133
 134
 135 void statusloop()
 136 {
 137         setupsignals();
 138         int i = 0;
 139         getcmds(-1);
 140         while(statusContinue)
 141         {
 142                 getcmds(i);
 143                 writestatus();
 144                 sleep(1.0);
 145                 i++;
 146         }
 147 }
 148
 149 void statusinit()
 150 {
 151         statusloop();
 152 }
 153
 154
 155 void sighandler(int signum)
 156 {
 157         getsigcmds(signum-SIGRTMIN);
 158         writestatus();
 159 }
 160
 161 void termhandler(int signum)
 162 {
 163         statusContinue = 0;
 164         exit(0);
 165 }
 166
 167 int main(int argc, char** argv)
 168 {
 169         for(int i = 0; i < argc; i++)
 170         {
 171                 if (!strcmp("-d",argv[i]))
 172                         delim = argv[++i][0];
 173                 else if(!strcmp("-p",argv[i]))
 174                         writestatus = pstdout;
 175         }
 176         signal(SIGTERM, termhandler);
 177         signal(SIGINT, termhandler);
 178         statusinit();
 179 }