Change refresh to 30 seconds.

[dwm-status] / dwmstatus.c

/*
* Copy me if you can.
* by 20h
*/

#define _BSD_SOURCE
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>
#include <strings.h>
#include <sys/time.h>
#include <time.h>
#include <sys/types.h>
#include <sys/wait.h>

#include <X11/Xlib.h>

char *tzargentina = "America/Buenos_Aires";
char *tzutc = "UTC";
char *tzberlin = "Europe/Berlin";

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

void
settz(char *tzname)
{
       setenv("TZ", tzname, 1);
}

char *
mktimes(char *fmt, char *tzname)
{
       char buf[129];
       time_t tim;
       struct tm *timtm;

       settz(tzname);
       tim = time(NULL);
       timtm = localtime(&tim);
       if (timtm == NULL)
               return smprintf("");

       if (!strftime(buf, sizeof(buf)-1, fmt, timtm)) {
               fprintf(stderr, "strftime == 0\n");
               return smprintf("");
       }

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

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

char *
loadavg(void)
{
       double avgs[3];

       if (getloadavg(avgs, 3) < 0)
               return smprintf("");

       return smprintf("%.2f %.2f %.2f", avgs[0], avgs[1], avgs[2]);
}

char *
readfile(char *base, char *file)
{
       char *path, line[513];
       FILE *fd;

       memset(line, 0, sizeof(line));

       path = smprintf("%s/%s", base, file);
       fd = fopen(path, "r");
       free(path);
       if (fd == NULL)
               return NULL;

       if (fgets(line, sizeof(line)-1, fd) == NULL)
               return NULL;
       fclose(fd);

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

char *
getbattery(char *base)
{
       char *co, status;
       int descap, remcap;

       descap = -1;
       remcap = -1;

       co = readfile(base, "present");
       if (co == NULL)
               return smprintf("");
       if (co[0] != '1') {
               free(co);
               return smprintf("not present");
       }
       free(co);

       co = readfile(base, "charge_full_design");
       if (co == NULL) {
               co = readfile(base, "energy_full_design");
               if (co == NULL)
                       return smprintf("");
       }
       sscanf(co, "%d", &descap);
       free(co);

       co = readfile(base, "charge_now");
       if (co == NULL) {
               co = readfile(base, "energy_now");
               if (co == NULL)
                       return smprintf("");
       }
       sscanf(co, "%d", &remcap);
       free(co);

       co = readfile(base, "status");
       if (!strncmp(co, "Discharging", 11)) {
               status = '-';
       } else if(!strncmp(co, "Charging", 8)) {
               status = '+';
       } else {
               status = '?';
       }

       if (remcap < 0 || descap < 0)
               return smprintf("invalid");

       return smprintf("%.0f%%%c", ((float)remcap / (float)descap) * 100, status);
}

char *
gettemperature(char *base, char *sensor)
{
       char *co;

       co = readfile(base, sensor);
       if (co == NULL)
               return smprintf("");
       return smprintf("%02.0f°C", atof(co) / 1000);
}

char *
execscript(char *cmd)
{
       FILE *fp;
       char retval[1025], rv;

       memset(retval, 0, sizeof(retval));

       fp = popen(cmd, "r");
       if (fp == NULL)
               return smprintf("");

       rv = fgets(retval, sizeof(retval)-1, fp);
       pclose(fp);
       if (rv == NULL)
               return smprintf("");
       retval[strlen(retval)-1] = '\0';

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

int
main(void)
{
       char *status;
       char *avgs;
       char *bat;
       char *tmar;
       char *tmutc;
       char *tmbln;
       char *t0;
       char *t1;
       char *kbmap;

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

       for (;;sleep(30)) {
               avgs = loadavg();
               bat = getbattery("/sys/class/power_supply/BAT0");
               tmar = mktimes("%H:%M", tzargentina);
               tmutc = mktimes("%H:%M", tzutc);
               tmbln = mktimes("KW %W %a %d %b %H:%M %Z %Y", tzberlin);
               kbmap = execscript("setxkbmap -query | grep layout | cut -d':' -f 2- | tr -d ' '");
               t0 = gettemperature("/sys/devices/virtual/thermal/thermal_zone0", "temp");
               t1 = gettemperature("/sys/devices/virtual/thermal/thermal_zone1", "temp");

               status = smprintf("K:%s T:%s|%s L:%s B:%s A:%s U:%s %s",
                               kbmap, t0, t1, avgs, bat, tmar, tmutc,
                               tmbln);
               setstatus(status);

               free(kbmap);
               free(t0);
               free(t1);
               free(avgs);
               free(bat);
               free(tmar);
               free(tmutc);
               free(tmbln);
               free(status);
       }

       XCloseDisplay(dpy);

       return 0;
}