#!/usr/bin/env python
import json
import os
import sys

__all__ = ["_mem", "_backlight", "_hwtemp"]

_BITS = []


def _bit(idx):
    def wrapper(func):
        _BITS.append([idx, func.__name__.replace("_", ""), func])
        return func

    return wrapper


@_bit(6)
def _mem():
    with open("/proc/meminfo") as meminfo_fp:
        attrs = {}
        for line in meminfo_fp.readlines():
            parts = line.split(":", 2)
            attrs[parts[0].strip()] = parts[1].strip()
        avail = int(attrs["MemAvailable"].split()[0])
        total = int(attrs["MemTotal"].split()[0])
        used = int(((total - avail) / total) * 100)
        color = None
        if used > 75:
            color = "#ff0000"
        if used < 25:
            color = "#00ff00"
        return "M:{}%".format(used), color


@_bit(0)
def _backlight(dev_dir="/sys/class/backlight/intel_backlight"):
    with open("{}/brightness".format(dev_dir)) as br_fp:
        with open("{}/max_brightness".format(dev_dir)) as mb_fp:
            value = int(br_fp.read().strip())
            max_value = int(mb_fp.read().strip())
            pct = int(float(float(value) / float(max_value)) * 100)
            color = None
            if pct >= 75:
                color = "#ffaa00"
            if pct <= 25:
                color = "#5599ff"
            return "☼:{}%".format(pct), color


THINKFAN_CONF = "/etc/thinkfan.conf"
ACPI_IBM_FAN = "/proc/acpi/ibm/fan"


@_bit(7)
def _hwtemp(conf=THINKFAN_CONF, fan=ACPI_IBM_FAN):
    if not os.path.exists(conf):
        return None, None

    temps = []

    with open("/etc/thinkfan.conf") as tfc_fp:
        for line in tfc_fp.readlines():
            if not line.startswith("hwmon "):
                continue
            try:
                with open(line.split(" ")[1].strip()) as temp_fp:
                    temps.append(float(temp_fp.read()))
            except (OSError, IOError) as exc:
                sys.stderr.write(str(exc) + "\n")

    avg_temp = float(sum(temps)) / len(temps) / 1000.0
    color = None

    fan_level = "unset"
    try:
        with open(fan) as fan_fp:
            for line in fan_fp.readlines():
                if not line.startswith("level:\t\t"):
                    continue
                fan_level = int(line.replace("level:\t\t", ""))
    except (OSError, IOError) as exc:
        sys.stderr.write(str(exc) + "\n")

    if avg_temp > 75:
        color = "#ff0000"
    if avg_temp >= 50:
        color = "#ffaa00"
    if avg_temp <= 25:
        color = "#5599ff"
    return "T:{:.1f}°C L:{}".format(avg_temp, fan_level), color


def _print_line(message):
    sys.stdout.write(message + "\n")
    sys.stdout.flush()


def _read_line():
    try:
        line = sys.stdin.readline().strip()
        if not line:
            sys.exit(3)
        return line
    except KeyboardInterrupt:
        sys.exit()


def main(bits=_BITS):
    _print_line(_read_line())
    _print_line(_read_line())

    while True:
        line, prefix = _read_line(), ""
        if line.startswith(","):
            line, prefix = line[1:], ","

        loaded = json.loads(line)
        for idx, name, func in bits:
            try:
                value, color = func()
                if value is None:
                    continue

                record = dict(full_text=str(value), name=name)
                if color is not None:
                    record.update(dict(color=color))

                loaded.insert(idx, record)
            except Exception as exc:
                sys.stderr.write(str(exc) + "\n")
                sys.stderr.flush()

        _print_line(prefix + json.dumps(loaded))


if __name__ == "__main__":
    main()