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 #define CMDLENGTH               50
   9
  10 typedef struct {
  11         char* icon;
  12         char* command;
  13         unsigned int interval;
  14         unsigned int signal;
  15 } Block;
  16 void dummysighandler(int num);
  17 void sighandler(int num);
  18 void replace(char *str, char old, char new);
  19 void getcmds(int time);
  20 #ifndef __OpenBSD__
  21 void getsigcmds(int signal);
  22 void setupsignals();
  23 void sighandler(int signum);
  24 #endif
  25 int getstatus(char *str, char *last);
  26 void setroot();
  27 void statusloop();
  28 void termhandler(int signum);
  29
  30
  31 #include "blocks.h"
  32
  33 static Display *dpy;
  34 static int screen;
  35 static Window root;
  36 static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0};
  37 static char statusstr[2][256];
  38 static int statusContinue = 1;
  39 static void (*writestatus) () = setroot;
  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         char c;
  58         int i = strlen(block->icon);
  59         fgets(output+i, CMDLENGTH-i, cmdf);
  60         i = strlen(output);
  61         if (delim != '\0' && --i)
  62                 output[i++] = delim;
  63         output[i++] = '\0';
  64         pclose(cmdf);
  65 }
  66
  67 void getcmds(int time)
  68 {
  69         const Block* current;
  70         for(int i = 0; i < LENGTH(blocks); i++)
  71         {
  72                 current = blocks + i;
  73                 if ((current->interval != 0 && time % current->interval == 0) || time == -1)
  74                         getcmd(current,statusbar[i]);
  75         }
  76 }
  77
  78 #ifndef __OpenBSD__
  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     /* initialize all real time signals with dummy handler */
  93     for(int i = SIGRTMIN; i <= SIGRTMAX; i++)
  94         signal(i, dummysighandler);
  95
  96         for(int i = 0; i < LENGTH(blocks); i++)
  97         {
  98                 if (blocks[i].signal > 0)
  99                         signal(SIGRTMIN+blocks[i].signal, sighandler);
 100         }
 101
 102 }
 103 #endif
 104
 105 int getstatus(char *str, char *last)
 106 {
 107         strcpy(last, str);
 108         str[0] = '\0';
 109         for(int i = 0; i < LENGTH(blocks); i++)
 110                 strcat(str, statusbar[i]);
 111         str[strlen(str)-1] = '\0';
 112         return strcmp(str, last);//0 if they are the same
 113 }
 114
 115 void setroot()
 116 {
 117         if (!getstatus(statusstr[0], statusstr[1]))//Only set root if text has changed.
 118                 return;
 119         Display *d = XOpenDisplay(NULL);
 120         if (d) {
 121                 dpy = d;
 122         }
 123         screen = DefaultScreen(dpy);
 124         root = RootWindow(dpy, screen);
 125         XStoreName(dpy, root, statusstr[0]);
 126         XCloseDisplay(dpy);
 127 }
 128
 129 void pstdout()
 130 {
 131         if (!getstatus(statusstr[0], statusstr[1]))//Only write out if text has changed.
 132                 return;
 133         printf("%s\n",statusstr[0]);
 134         fflush(stdout);
 135 }
 136
 137
 138 void statusloop()
 139 {
 140 #ifndef __OpenBSD__
 141         setupsignals();
 142 #endif
 143         int i = 0;
 144         getcmds(-1);
 145         while(statusContinue)
 146         {
 147                 getcmds(i);
 148                 writestatus();
 149                 sleep(1.0);
 150                 i++;
 151         }
 152 }
 153
 154 #ifndef __OpenBSD__
 155 /* this signal handler should do nothing */
 156 void dummysighandler(int signum)
 157 {
 158     return;
 159 }
 160 #endif
 161
 162 #ifndef __OpenBSD__
 163 void sighandler(int signum)
 164 {
 165         getsigcmds(signum-SIGRTMIN);
 166         writestatus();
 167 }
 168 #endif
 169
 170 void termhandler(int signum)
 171 {
 172         statusContinue = 0;
 173         exit(0);
 174 }
 175
 176 int main(int argc, char** argv)
 177 {
 178         for(int i = 0; i < argc; i++)
 179         {
 180                 if (!strcmp("-d",argv[i]))
 181                         delim = argv[++i][0];
 182                 else if(!strcmp("-p",argv[i]))
 183                         writestatus = pstdout;
 184         }
 185         signal(SIGTERM, termhandler);
 186         signal(SIGINT, termhandler);
 187         statusloop();
 188 }