initial commit

4 files changed, 279 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..c35e2cd
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+patches/
+dwmblocks
diff --git a/Makefile b/Makefile
new file mode 100644
index 0000000..25da945
--- /dev/null
+++ b/Makefile
@@ -0,0 +1,13 @@
+PREFIX ?= /usr/local
+CC ?= cc
+LDFLAGS = -lX11
+
+output: dwmblocks.c
+        ${CC}  dwmblocks.c $(LDFLAGS) -o dwmblocks
+clean:
+        rm -f *.o *.gch dwmblocks
+install: output
+        mkdir -p $(DESTDIR)$(PREFIX)/bin
+        install -m 0755 dwmblocks $(DESTDIR)$(PREFIX)/bin/dwmblocks
+uninstall:
+        rm -f $(DESTDIR)$(PREFIX)/bin/dwmblocks
diff --git a/blocks.h b/blocks.h
new file mode 100644
index 0000000..a9a79b3
--- /dev/null
+++ b/blocks.h
@@ -0,0 +1,17 @@
+//Modify this file to change what commands output to your statusbar, and recompile using the make command.
+//
+static const Block blocks[] = {
+        /*Icon*/        /*Command*/     /*Update Interval*/     /*Update Signal*/
+    { " ",              "~/.local/bin/statusbar/bar-cmus",              5,                                      12},
+    { " ",              "~/.local/bin/statusbar/bar-cpu",               2,                                      13},
+        { " ",          "~/.local/bin/statusbar/bar-gpu",               2,                                      14},
+        { " ",          "~/.local/bin/statusbar/bar-volume",    60,                                     10},
+        { " ",          "~/.local/bin/statusbar/bar-memory",    10,                                     16},
+        { " ",          "~/.local/bin/statusbar/bar-storage",   60,                                     17},
+        { " ",          "~/.local/bin/statusbar/bar-ipv4",              60,                                     18},
+        { " ",          "~/.local/bin/statusbar/bar-time",              1,                                      19},
+};
+
+//Sets delimiter between status commands. NULL character ('\0') means no delimiter.
+static char delim[] = " |";
+static unsigned int delimLen = 5;
diff --git a/dwmblocks.c b/dwmblocks.c
new file mode 100644
index 0000000..78fdeb5
--- /dev/null
+++ b/dwmblocks.c
@@ -0,0 +1,247 @@
+#include<stdlib.h>
+#include<stdio.h>
+#include<string.h>
+#include<unistd.h>
+#include<signal.h>
+#include<sys/wait.h>
+#ifndef NO_X
+#include<X11/Xlib.h>
+#endif
+#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 getcmds(int time);
+void getsigcmds(unsigned int signal);
+void setupsignals();
+void sighandler(int signum, siginfo_t *si, void *ucontext);
+int getstatus(char *str, char *last);
+void statusloop();
+void termhandler();
+void chldhandler();
+void pstdout();
+#ifndef NO_X
+void setroot();
+static void (*writestatus) () = setroot;
+static int setupX();
+static Display *dpy;
+static int screen;
+static Window root;
+#else
+static void (*writestatus) () = pstdout;
+#endif
+
+
+#include "blocks.h"
+
+static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0};
+static char statusstr[2][STATUSLENGTH];
+static int statusContinue = 1;
+static int returnStatus = 0;
+
+//opens process *cmd and stores output in *output
+void getcmd(const Block *block, char *output)
+{
+        if (block->signal)
+                *output++ = block->signal;
+        strcpy(output, block->icon);
+        FILE *cmdf = popen(block->command, "r");
+        if (!cmdf)
+                return;
+        int i = strlen(block->icon);
+        fgets(output+i, CMDLENGTH-i-delimLen, cmdf);
+        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 = { .sa_sigaction = sighandler, .sa_flags = SA_SIGINFO };
+#ifndef __OpenBSD__
+            /* initialize all real time signals with dummy handler */
+    for (int i = SIGRTMIN; i <= SIGRTMAX; i++) {
+        signal(i, dummysighandler);
+                sigaddset(&sa.sa_mask, i);
+        }
+#endif
+
+        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
+}
+
+#ifndef NO_X
+void setroot()
+{
+        if (!getstatus(statusstr[0], statusstr[1]))//Only set root if text has changed.
+                return;
+        XStoreName(dpy, root, statusstr[0]);
+        XFlush(dpy);
+}
+
+int setupX()
+{
+        dpy = XOpenDisplay(NULL);
+        if (!dpy) {
+                fprintf(stderr, "dwmblocks: Failed to open display\n");
+                return 0;
+        }
+        screen = DefaultScreen(dpy);
+        root = RootWindow(dpy, screen);
+        return 1;
+}
+#endif
+
+void pstdout()
+{
+        if (!getstatus(statusstr[0], statusstr[1]))//Only write out if text has changed.
+                return;
+        printf("%s\n",statusstr[0]);
+        fflush(stdout);
+}
+
+
+void statusloop()
+{
+        setupsignals();
+        int i = 0;
+        getcmds(-1);
+        while (1) {
+                getcmds(i++);
+                writestatus();
+                if (!statusContinue)
+                        break;
+                sleep(1.0);
+        }
+}
+
+#ifndef __OpenBSD__
+/* this signal handler should do nothing */
+void dummysighandler(int signum)
+{
+    return;
+}
+#endif
+
+void sighandler(int signum, siginfo_t *si, void *ucontext)
+{
+        if (si->si_value.sival_int) {
+                pid_t parent = getpid();
+                if (fork() == 0) {
+#ifndef NO_X
+                        if (dpy)
+                                close(ConnectionNumber(dpy));
+#endif
+                        int i;
+                        for (i = 0; i < LENGTH(blocks) && blocks[i].signal != signum-SIGRTMIN; i++);
+
+                        char shcmd[1024];
+                        sprintf(shcmd, "%s; kill -%d %d", blocks[i].command, SIGRTMIN+blocks[i].signal, parent);
+                        char *cmd[] = { "/bin/sh", "-c", shcmd, NULL };
+                        char button[2] = { '0' + si->si_value.sival_int, '\0' };
+                        setenv("BUTTON", button, 1);
+                        setsid();
+                        execvp(cmd[0], cmd);
+                        perror(cmd[0]);
+                        exit(EXIT_SUCCESS);
+                }
+        } else {
+                getsigcmds(signum-SIGPLUS);
+                writestatus();
+        }
+}
+
+void termhandler()
+{
+        statusContinue = 0;
+}
+
+void chldhandler()
+{
+        while (0 < waitpid(-1, NULL, WNOHANG));
+}
+
+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;
+        }
+#ifndef NO_X
+        if (!setupX())
+                return 1;
+#endif
+        delimLen = MIN(delimLen, strlen(delim));
+        delim[delimLen++] = '\0';
+        signal(SIGTERM, termhandler);
+        signal(SIGINT, termhandler);
+        signal(SIGCHLD, chldhandler);
+        statusloop();
+#ifndef NO_X
+        XCloseDisplay(dpy);
+#endif
+        return 0;
+}