#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>

const char *icon[] = {
    "\uf58d",                                         /* 0% */
    "\uf579", "\uf57a", "\uf57b", "\uf57c", "\uf57d", /* 50% */
    "\uf57e", "\uf57f", "\uf580", "\uf581", "\uf578", /* 100% */
};

typedef enum {
  CHARGED = 0,
  DISCHARGING = 1,
  CHARGING = 2,
  DISCONNECTED = 7,
  UNKNOWN = -1
} State;

#if defined(__linux__)
void battery_state(int *nbats, int bat[], State state[]) {
  FILE *fd;
  int i = 0;
  do {
    char filename[50], status[50];

    // get current capacity
    sprintf(filename, "/sys/class/power_supply/BAT%d/capacity", i);
    fd = fopen(filename, "rb");
    if (fd == NULL)
      break;
    fscanf(fd, "%d", &bat[i]);
    fclose(fd);
    // battery status (charging...)
    sprintf(filename, "/sys/class/power_supply/BAT%d/status", i);
    fd = fopen(filename, "rb");
    fgets(status, 50, fd);
    // puts(status);
    fclose(fd);
    if (!strcmp(status, "Not charging\n"))
      state[i] = CHARGED;
    else if (!strcmp(status, "Discharging\n"))
      state[i] = DISCHARGING;
    else if (!strcmp(status, "Charging\n"))
      state[i] = CHARGING;
    else
      state[i] = UNKNOWN;

    // next
    i++;
  } while (1);
  *nbats = i;
}
#elif defined(__FreeBSD__)
#include <sys/sysctl.h>
void battery_state(int *nbats, int bat[], State state[]) {
  size_t len = sizeof(int);
  int status, life = 0;

  *nbats = 1; // FIXME: deal with multiple batteries

  if (sysctlbyname("hw.acpi.battery.state", &status, &len, NULL, 0) ||
      sysctlbyname("hw.acpi.battery.life", &life, &len, NULL, 0)) {
    state[0] = UNKNOWN;
    bat[0] = 0;
  } else {
    bat[0] = life;
    state[0] = status;
  }
}
#endif

int main() {
  char *color;
  int life[2], nbats;
  State state[2];

  battery_state(&nbats, life, state);
  for (int i = 0; i < nbats; i++) {

    switch (state[i]) {
    case CHARGED: /* fully charged */
      printf("%s %d%%  ", icon[life[i] / 10], life[i]);
      break;
    case DISCHARGING: /* discharging */
      if (life[i] >= 50)
        color = "colour028"; /* green */
      else if (life[i] >= 25)
        color = "colour172"; /* yellow */
      else
        color = "colour088"; /* red */

      printf("#[fg=%s]%s#[fg=default] %d%%  ", color, icon[life[i] / 10],
             life[i]);
      break;
    case CHARGING: /* charging */
      printf("#[fg=yellow]\uf583#[fg=default] %d%%  ", life[i]);
      break;
    case DISCONNECTED: /* disconnected */
      printf("\uf492  ");
      break;
    default: /* unknown code */
      printf("\uf590  ");
    }
  }

  return EXIT_SUCCESS;
}