#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;
}