bar and blocks

1 files changed, 215 insertions, 0 deletions

diff --git a/wm/someblocks/someblocks.c b/wm/someblocks/someblocks.c
new file mode 100644
index 0000000..f149896
--- /dev/null
+++ b/wm/someblocks/someblocks.c
@@ -0,0 +1,215 @@
+#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 sighandler(int signum);
+int getstatus(char *str, char *last);
+void statusloop();
+void termhandler();
+void pstdout();
+void psomebar();
+static void (*writestatus) () = psomebar;
+
+#include "blocks.h"
+
+static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0};
+static char statusstr[2][STATUSLENGTH];
+static int statusContinue = 1;
+static int returnStatus = 0;
+static char somebarPath[128];
+static int somebarFd = -1;
+
+//opens process *cmd and stores output in *output
+void getcmd(const Block *block, char *output)
+{
+	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 = {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()
+{
+	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)
+{
+	getsigcmds(signum-SIGPLUS);
+	writestatus();
+}
+
+void termhandler()
+{
+	statusContinue = 0;
+}
+
+void sigpipehandler()
+{
+	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;
+}