#!/bin/env python

### Add to ~/.config/waybar/config like so
# "custom/xbps-updates": {
#     "format": "{}  ",
#     "interval": 3600,
#     "return-type": "json",
#     "exec": "xbps-checkup-updates | jq --unbuffered --compact-output",
#     "tooltip": true,
# }

import json
import os
from typing import Any, NamedTuple

## updated with hourly cronjob that contains:
# xbps-install -S
# cd $HOME/void-packages && git pull
# xbps-checkvers -I > /tmp/xbps_updates_available
## by befault output is formatted like:
# pkg name  old_ver    new_ver  template name  repository (if known)
# minizip   1.2.13_1   1.3_1    minizip        ?
UPDATE_FILE = "/tmp/xbps_updates_available"


class Update(NamedTuple):
    package: str
    current_ver: str
    new_ver: str

    def __str__(self) -> str:
        return f"{self.package}: {self.current_ver} -> {self.new_ver}"


def main():
    total_packages = int(os.popen("xbps-query -l | wc -l").read())
    output = {
        "text": "",
        "alt": "",
        "tooltip": "",
        "percentage": Any,
    }
    updates = []
    # get list of updates from UPDATE_FILE
    with open(UPDATE_FILE) as file:
        for line in file:
            split_line = line.split(" ")
            updates.append(Update(split_line[0], split_line[1], split_line[2]))
    num_updates = len(updates)

    # format updates for waybar
    output["text"] = str(num_updates)
    if num_updates <= 0:
        output["tooltip"] = "Up to date!"
    else:
        output["tooltip"] = f"Updates available:\n{'\n'.join(str(x) for x in updates)}"
    output["alt"] = f"Updates: {num_updates} / {total_packages}"
    output["percentage"] = num_updates / total_packages

    print(json.dumps(output))


if __name__ == "__main__":
    main()