#!/usr/bin/env python3

import os
import glob

ICON = "\U000f0379"  # 󰍹


def main():
    files = glob.glob("/sys/class/drm/card*/device/gpu_busy_percent")
    if not files:
        return

    with open(files[0]) as f:
        load = int(f.readline())

    print(f" {ICON} {load}% \n")
    if load > 90:
        print(f"#{os.environ.get("BASE16_COLOR_00_HEX")}")
        print(f"#{os.environ.get("BASE16_COLOR_08_HEX")}")
    elif load > 60:
        print(f"#{os.environ.get("BASE16_COLOR_00_HEX")}")
        print(f"#{os.environ.get("BASE16_COLOR_0A_HEX")}")
    elif load > 30:
        print(f"#{os.environ.get("BASE16_COLOR_00_HEX")}")
        print(f"#{os.environ.get("BASE16_COLOR_0C_HEX")}")


if __name__ == "__main__":
    main()