dwmblocks/dwmblocks.c

189 lines
3.5 KiB
C
Raw Normal View History

2019-03-24 22:31:05 +00:00
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<unistd.h>
#include<signal.h>
#include<X11/Xlib.h>
#define LENGTH(X) (sizeof(X) / sizeof (X[0]))
#define CMDLENGTH 50
2019-03-24 22:31:05 +00:00
typedef struct {
2019-03-30 03:10:46 +00:00
char* icon;
2019-03-24 22:31:05 +00:00
char* command;
unsigned int interval;
unsigned int signal;
} Block;
void dummysighandler(int num);
2019-03-24 22:31:05 +00:00
void sighandler(int num);
void replace(char *str, char old, char new);
void getcmds(int time);
#ifndef __OpenBSD__
void getsigcmds(int signal);
void setupsignals();
void sighandler(int signum);
#endif
int getstatus(char *str, char *last);
void setroot();
void statusloop();
void termhandler(int signum);
2019-03-24 22:31:05 +00:00
#include "blocks.h"
static Display *dpy;
static int screen;
static Window root;
static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0};
static char statusstr[2][256];
static int statusContinue = 1;
static void (*writestatus) () = setroot;
2019-03-24 22:31:05 +00:00
void replace(char *str, char old, char new)
{
int N = strlen(str);
for(int i = 0; i < N; i++)
if(str[i] == old)
str[i] = new;
}
//opens process *cmd and stores output in *output
2019-03-30 03:10:46 +00:00
void getcmd(const Block *block, char *output)
2019-03-24 22:31:05 +00:00
{
2019-03-30 03:10:46 +00:00
strcpy(output, block->icon);
char *cmd = block->command;
2019-03-24 22:31:05 +00:00
FILE *cmdf = popen(cmd,"r");
if (!cmdf)
return;
char c;
int i = strlen(block->icon);
fgets(output+i, CMDLENGTH-i, cmdf);
i = strlen(output);
if (delim != '\0' && --i)
output[i++] = delim;
output[i++] = '\0';
2019-03-24 22:31:05 +00:00
pclose(cmdf);
}
void getcmds(int time)
{
const Block* current;
for(int i = 0; i < LENGTH(blocks); i++)
{
2019-03-24 22:31:05 +00:00
current = blocks + i;
if ((current->interval != 0 && time % current->interval == 0) || time == -1)
2019-03-30 03:10:46 +00:00
getcmd(current,statusbar[i]);
2019-03-24 22:31:05 +00:00
}
}
#ifndef __OpenBSD__
2019-03-24 22:31:05 +00:00
void getsigcmds(int signal)
{
const Block *current;
for (int i = 0; i < LENGTH(blocks); i++)
{
current = blocks + i;
if (current->signal == signal)
2019-03-30 03:10:46 +00:00
getcmd(current,statusbar[i]);
2019-03-24 22:31:05 +00:00
}
}
void setupsignals()
{
/* initialize all real time signals with dummy handler */
for(int i = SIGRTMIN; i <= SIGRTMAX; i++)
signal(i, dummysighandler);
2019-03-24 22:31:05 +00:00
for(int i = 0; i < LENGTH(blocks); i++)
{
2019-03-24 22:31:05 +00:00
if (blocks[i].signal > 0)
signal(SIGRTMIN+blocks[i].signal, sighandler);
}
}
#endif
2019-03-24 22:31:05 +00:00
int getstatus(char *str, char *last)
2019-03-24 22:31:05 +00:00
{
strcpy(last, str);
str[0] = '\0';
for(int i = 0; i < LENGTH(blocks); i++)
strcat(str, statusbar[i]);
str[strlen(str)-1] = '\0';
return strcmp(str, last);//0 if they are the same
2019-03-24 22:31:05 +00:00
}
void setroot()
{
if (!getstatus(statusstr[0], statusstr[1]))//Only set root if text has changed.
return;
2019-03-24 22:31:05 +00:00
Display *d = XOpenDisplay(NULL);
if (d) {
dpy = d;
}
screen = DefaultScreen(dpy);
root = RootWindow(dpy, screen);
XStoreName(dpy, root, statusstr[0]);
2019-03-24 22:31:05 +00:00
XCloseDisplay(dpy);
}
void pstdout()
{
if (!getstatus(statusstr[0], statusstr[1]))//Only write out if text has changed.
return;
printf("%s\n",statusstr[0]);
fflush(stdout);
}
2019-03-24 22:31:05 +00:00
void statusloop()
2019-03-24 22:31:05 +00:00
{
#ifndef __OpenBSD__
2019-03-24 22:31:05 +00:00
setupsignals();
#endif
2019-03-24 22:31:05 +00:00
int i = 0;
getcmds(-1);
while(statusContinue)
{
getcmds(i);
writestatus();
2019-03-24 22:31:05 +00:00
sleep(1.0);
i++;
}
}
#ifndef __OpenBSD__
/* this signal handler should do nothing */
void dummysighandler(int signum)
{
return;
}
#endif
#ifndef __OpenBSD__
2019-03-24 22:31:05 +00:00
void sighandler(int signum)
{
getsigcmds(signum-SIGRTMIN);
writestatus();
2019-03-24 22:31:05 +00:00
}
#endif
2019-03-24 22:31:05 +00:00
void termhandler(int signum)
{
statusContinue = 0;
exit(0);
}
2019-03-30 03:10:46 +00:00
int main(int argc, char** argv)
2019-03-24 22:31:05 +00:00
{
2019-03-30 03:10:46 +00:00
for(int i = 0; i < argc; i++)
{
2019-03-30 03:10:46 +00:00
if (!strcmp("-d",argv[i]))
delim = argv[++i][0];
else if(!strcmp("-p",argv[i]))
writestatus = pstdout;
2019-03-30 03:10:46 +00:00
}
2019-03-24 22:31:05 +00:00
signal(SIGTERM, termhandler);
signal(SIGINT, termhandler);
statusloop();
2019-03-24 22:31:05 +00:00
}