219 lines
4.5 KiB
C
219 lines
4.5 KiB
C
#define _POSIX_C_SOURCE 200809L
|
|
#include<stdlib.h>
|
|
#include<stdio.h>
|
|
#include<string.h>
|
|
#include<unistd.h>
|
|
#include<fcntl.h>
|
|
#include<errno.h>
|
|
#include<signal.h>
|
|
#ifdef __OpenBSD__
|
|
#define SIGPLUS SIGUSR1+1
|
|
#define SIGMINUS SIGUSR1-1
|
|
#else
|
|
#define SIGPLUS SIGRTMIN
|
|
#define SIGMINUS SIGRTMIN
|
|
#endif
|
|
#define LENGTH(X) (sizeof(X) / sizeof (X[0]))
|
|
#define CMDLENGTH 50
|
|
#define MIN( a, b ) ( ( a < b) ? a : b )
|
|
#define STATUSLENGTH (LENGTH(blocks) * CMDLENGTH + 1)
|
|
|
|
typedef struct {
|
|
char* icon;
|
|
char* command;
|
|
unsigned int interval;
|
|
unsigned int signal;
|
|
} Block;
|
|
#ifndef __OpenBSD__
|
|
void dummysighandler(int num);
|
|
#endif
|
|
void sighandler(int num);
|
|
void getcmds(int time);
|
|
void getsigcmds(unsigned int signal);
|
|
void setupsignals(void);
|
|
void sighandler(int signum);
|
|
int getstatus(char *str, char *last);
|
|
void statusloop(void);
|
|
void termhandler(int sig);
|
|
void pstdout(void);
|
|
void psomebar(void);
|
|
static void (*writestatus) (void) = psomebar;
|
|
|
|
#include "blocks.h"
|
|
|
|
static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0};
|
|
static char statusstr[2][STATUSLENGTH];
|
|
static int statusContinue = 1;
|
|
static char somebarPath[128];
|
|
static int somebarFd = -1;
|
|
|
|
//opens process *cmd and stores output in *output
|
|
void getcmd(const Block *block, char *output)
|
|
{
|
|
FILE *cmdf;
|
|
int i;
|
|
strcpy(output, block->icon);
|
|
cmdf = popen(block->command, "r");
|
|
if (!cmdf)
|
|
return;
|
|
i = strlen(block->icon);
|
|
if (fgets(output+i, CMDLENGTH-i-delimLen, cmdf) != (output+i)) {
|
|
exit(1);
|
|
}
|
|
i = strlen(output);
|
|
if (i == 0) {
|
|
//return if block and command output are both empty
|
|
pclose(cmdf);
|
|
return;
|
|
}
|
|
if (delim[0] != '\0') {
|
|
//only chop off newline if one is present at the end
|
|
i = output[i-1] == '\n' ? i-1 : i;
|
|
strncpy(output+i, delim, delimLen);
|
|
}
|
|
else
|
|
output[i++] = '\0';
|
|
pclose(cmdf);
|
|
}
|
|
|
|
void getcmds(int time)
|
|
{
|
|
const Block* current;
|
|
for (unsigned int i = 0; i < LENGTH(blocks); i++) {
|
|
current = blocks + i;
|
|
if ((current->interval != 0 && time % current->interval == 0) || time == -1)
|
|
getcmd(current,statusbar[i]);
|
|
}
|
|
}
|
|
|
|
void getsigcmds(unsigned int signal)
|
|
{
|
|
const Block *current;
|
|
for (unsigned int i = 0; i < LENGTH(blocks); i++) {
|
|
current = blocks + i;
|
|
if (current->signal == signal)
|
|
getcmd(current,statusbar[i]);
|
|
}
|
|
}
|
|
|
|
void setupsignals()
|
|
{
|
|
struct sigaction sa = {0};
|
|
#ifndef __OpenBSD__
|
|
/* initialize all real time signals with dummy handler */
|
|
sa.sa_handler = dummysighandler;
|
|
for (int i = SIGRTMIN; i <= SIGRTMAX; i++)
|
|
sigaction(i, &sa, NULL);
|
|
#endif
|
|
|
|
sa.sa_handler = sighandler;
|
|
for (unsigned int i = 0; i < LENGTH(blocks); i++) {
|
|
if (blocks[i].signal > 0)
|
|
sigaction(SIGMINUS+blocks[i].signal, &sa, NULL);
|
|
}
|
|
|
|
}
|
|
|
|
int getstatus(char *str, char *last)
|
|
{
|
|
strcpy(last, str);
|
|
str[0] = '\0';
|
|
for (unsigned int i = 0; i < LENGTH(blocks); i++)
|
|
strcat(str, statusbar[i]);
|
|
str[strlen(str)-strlen(delim)] = '\0';
|
|
return strcmp(str, last);//0 if they are the same
|
|
}
|
|
|
|
void pstdout()
|
|
{
|
|
if (!getstatus(statusstr[0], statusstr[1]))//Only write out if text has changed.
|
|
return;
|
|
printf("%s\n",statusstr[0]);
|
|
fflush(stdout);
|
|
}
|
|
|
|
|
|
void psomebar()
|
|
{
|
|
if (!getstatus(statusstr[0], statusstr[1]))//Only write out if text has changed.
|
|
return;
|
|
if (somebarFd < 0) {
|
|
somebarFd = open(somebarPath, O_WRONLY|O_CLOEXEC);
|
|
if (somebarFd < 0 && errno == ENOENT) {
|
|
// assume somebar is not ready yet
|
|
sleep(1);
|
|
somebarFd = open(somebarPath, O_WRONLY|O_CLOEXEC);
|
|
}
|
|
if (somebarFd < 0) {
|
|
perror("open");
|
|
return;
|
|
}
|
|
}
|
|
dprintf(somebarFd, "status %s\n", statusstr[0]);
|
|
}
|
|
|
|
|
|
void statusloop()
|
|
{
|
|
int i;
|
|
setupsignals();
|
|
i = 0;
|
|
getcmds(-1);
|
|
while (1) {
|
|
getcmds(i++);
|
|
writestatus();
|
|
if (!statusContinue)
|
|
break;
|
|
sleep(1);
|
|
}
|
|
}
|
|
|
|
#ifndef __OpenBSD__
|
|
/* this signal handler should do nothing */
|
|
void dummysighandler(int signum)
|
|
{
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
void sighandler(int signum)
|
|
{
|
|
getsigcmds(signum-SIGPLUS);
|
|
writestatus();
|
|
}
|
|
|
|
void termhandler(int sig)
|
|
{
|
|
statusContinue = 0;
|
|
}
|
|
|
|
void sigpipehandler(int sig)
|
|
{
|
|
close(somebarFd);
|
|
somebarFd = -1;
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
for (int i = 0; i < argc; i++) {//Handle command line arguments
|
|
if (!strcmp("-d",argv[i]))
|
|
strncpy(delim, argv[++i], delimLen);
|
|
else if (!strcmp("-p",argv[i]))
|
|
writestatus = pstdout;
|
|
else if (!strcmp("-s",argv[i]))
|
|
strcpy(somebarPath, argv[++i]);
|
|
}
|
|
|
|
if (!strlen(somebarPath)) {
|
|
strcpy(somebarPath, getenv("XDG_RUNTIME_DIR"));
|
|
strcat(somebarPath, "/somebar-0");
|
|
}
|
|
|
|
delimLen = MIN(delimLen, strlen(delim));
|
|
delim[delimLen++] = '\0';
|
|
signal(SIGTERM, termhandler);
|
|
signal(SIGINT, termhandler);
|
|
signal(SIGPIPE, sigpipehandler);
|
|
statusloop();
|
|
return 0;
|
|
}
|