dwmst.c

#include <unistd.h>
#include <time.h>
#include <X11/Xlib.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <alsa/asoundlib.h>
#include <linux/wireless.h>

#include "dwmst.h"

static Display *dpy;

char *
smprintf(char *fmt, ...) {
	va_list fmtargs;
	char *ret;
	int len;

	va_start(fmtargs, fmt);
	len = vsnprintf(NULL, 0, fmt, fmtargs);
	va_end(fmtargs);

	ret = malloc(++len);
	if (ret == NULL) {
		perror("malloc");
		exit(1);
	}

	va_start(fmtargs, fmt);
	vsnprintf(ret, len, fmt, fmtargs);
	va_end(fmtargs);

	return ret;
}

int
is_up (char *device) {
	char devpath[32], state[5];
	FILE *infile;

	snprintf(devpath, sizeof(devpath), "/sys/class/net/%s/operstate", device);
	infile = fopen(devpath, "r");
	if(infile != NULL) {
		fscanf(infile, "%s", state);
		fclose(infile);
		if(strncmp(state, "up", 2) == 0)
			return 1;
	}
	return 0;
}

char *
get_skype(void) {
	if(access(SKYPE_LOCK, F_OK) == 0)
		return smprintf("%s", SKYPE_STR);
	else
		return NULL;
}

char *
get_net(struct iwreq wreq, int socket) {
	if(is_up(WIRELESS_DEVICE)) {
		char essid[30];

		memset(essid, 0, sizeof(essid));
		wreq.u.essid.pointer = essid;
		wreq.u.essid.length = sizeof(essid);
		ioctl(socket, SIOCGIWESSID, &wreq);
		return smprintf(WLAN_STR, essid);
	} else if(is_up(WIRED_DEVICE))
		return smprintf("%s", LAN_STR);
	else
		return smprintf("%s", NO_CON_STR);
}

char *
get_vol(snd_mixer_t *handle) {
	int mute = 0;
	long vol = 0, max = 0, min = 0;
	snd_mixer_elem_t *pcm_mixer, *max_mixer;
	snd_mixer_selem_id_t *vol_info, *mute_info;

	/*ToDo: maybe move all this to main?*/
	snd_mixer_handle_events(handle);
	snd_mixer_selem_id_malloc(&vol_info);
	snd_mixer_selem_id_malloc(&mute_info);
	snd_mixer_selem_id_set_name(vol_info, VOL_CH);
	snd_mixer_selem_id_set_name(mute_info, VOL_CH);
	pcm_mixer = snd_mixer_find_selem(handle, vol_info);
	max_mixer = snd_mixer_find_selem(handle, mute_info);
	snd_mixer_selem_get_playback_volume_range(pcm_mixer, &min, &max);
	snd_mixer_selem_get_playback_volume(pcm_mixer, 0, &vol);
	snd_mixer_selem_get_playback_switch(max_mixer, 0, &mute);
	snd_mixer_selem_id_free(vol_info);
	snd_mixer_selem_id_free(mute_info);

	if(mute == 0)
		return smprintf(VOL_MUTE_STR, (vol * 100) / max);
	return smprintf(VOL_STR, (vol * 100) / max);
}

char *
get_batt(void) {
	FILE *f;
	char state[11];

	if(access("/sys/class/power_supply/BAT1/", F_OK) != 0)
		return smprintf("%s", NO_BAT_STR);

	memset(state, 0, sizeof(state));
	f = fopen(BATT_STAT, "r");
	if(f == NULL)
		return NULL;
	fscanf(f, "%s\n", state);
	fclose(f);

	if(strncmp(state, "Full", 4) == 0)
		return smprintf("%s", BAT_FULL_STR);
	else if(strncmp(state, "Unknown", 7) == 0)
		return smprintf("%s", BAT_UNK_STR);

	float now, full, voltage, rate;
	unsigned int perc, minutes, hours;

	f = fopen (BATT_NOW, "r"); if(f == NULL) return NULL; fscanf (f, "%f\n", &now); fclose (f);
	f = fopen (BATT_FULL, "r"); if(f == NULL) return NULL; fscanf (f, "%f\n", &full); fclose (f);
	f = fopen (BATT_VOLT, "r"); if(f == NULL) return NULL; fscanf (f, "%f\n", &voltage); fclose (f);
	f = fopen (BATT_CNOW, "r"); if(f == NULL) return NULL; fscanf (f, "%f\n", &rate); fclose (f);

	now *= voltage;
	full *= voltage;
	rate *= voltage;
	perc = (now * 100) / full;

	if(perc > 100) perc = 100;

	if(strncmp(state, "Charging", 8) == 0) {
		minutes = 60 * ((full - now) / rate);
		hours = minutes / 60;
		if(hours > 24) hours = 24;
		minutes -= 60 * hours;
		return smprintf(BAT_CHRG_STR, perc, hours, minutes);
	} else {
		minutes = 60 * (now / rate);
		hours = minutes / 60;
		if(hours > 24) hours = 24;
		minutes -= 60 * hours;
		/*notify*/
		return smprintf(BAT_DIS_STR, perc, hours, minutes);
	}
}

char *
get_time(void) {
	char clock[38];
	time_t current;

	memset(clock, 0, sizeof(clock));
	time(&current);

	if(!strftime(clock, sizeof(clock) - 1, CLK_STR, localtime(&current)))
		return smprintf("%s", "strftime == 0");

	return smprintf("%s", clock);
}

void
setstatus(char *str) {
	XStoreName(dpy, DefaultRootWindow(dpy), str);
	XSync(dpy, False);
}

int
main(void) {
	struct iwreq wreq;
	snd_mixer_t *handle;
	int sockfd, loops = 60;
	char *status, *skype, *net, *vol, *batt, *clk;

	if(!(dpy = XOpenDisplay(NULL))) {
		fprintf(stderr, "dwmst: cannot open display.\n");
		return 1;
	}

	memset(&wreq, 0, sizeof(struct iwreq));
	snprintf(wreq.ifr_name, sizeof(WIRELESS_DEVICE), WIRELESS_DEVICE);
	sockfd = socket(AF_INET, SOCK_DGRAM, 0);

	snd_mixer_open(&handle, 0);
	snd_mixer_attach(handle, "default");
	snd_mixer_selem_register(handle, NULL, NULL);
	snd_mixer_load(handle);

	for(;;sleep(INTERVAL)) {
		if(++loops > 60) {
			loops = 0;
			if(sockfd > 0)
				net = get_net(wreq, sockfd);
			clk = get_time();
		}
		skype = get_skype();
		vol = get_vol(handle);
		batt = get_batt();
		status = smprintf("%s      %s      %s      %s      %s", skype, net, vol, batt, clk);
		setstatus(status);
		free(skype);
		free(vol);
		free(batt);
		free(status);
	}


	if(sockfd > 0)
		free(net);
	free(clk);
	XCloseDisplay(dpy);
	close(sockfd);
	snd_mixer_close(handle);
	return 0;
}