1 files changed, 81 insertions, 0 deletions
diff --git a/files/nwg-panel/executors/arch_updates.py b/files/nwg-panel/executors/arch_updates.py
new file mode 100755
index 0000000..425bcce
--- /dev/null
+++ b/files/nwg-panel/executors/arch_updates.py
@@ -0,0 +1,81 @@
+#!/nix/store/lx8vhp4fxclp494svlfis3sb2g8z4l9h-python3-3.10.12/bin/python
+
+import os
+import time
+import subprocess
+
+
+# You may either use the full icon(s) path here, like e.g.:
+# "/home/piotr/.config/nwg-panel/icons_light/arch-linux.svg"
+# or just give the icon name, like below.
+
+# The icon name must either exist in your icon theme, or you may place `icon_name.svg`
+# custom files in '~/.config/nwg-panel/icons_light/' and '~/.config/nwg-panel/icons_dark/'.
+
+# This script needs the `au.sh` helper on path or in the same directory. See comments in `au.sh`.
+
+def main():
+    # Avoid checking on each panel restart: check if 15 minutes passed.
+    # Adjust the time (in seconds) to your liking.
+    # Make sure if the path below matches your temp directory.
+    file = "/tmp/arch-updates"
+    
+    if os.path.isfile(file):
+        if int(time.time() - os.stat(file).st_mtime) > 900:
+            arch, aur = check_updates()
+            save_string("{},{}".format(arch, aur), file)
+        else:
+            try:
+                vals = load_string(file).split(",")
+                arch, aur = int(vals[0]), int(vals[1])
+            except:
+                arch, aur = 0, 0
+    else:
+        arch, aur = check_updates()
+        save_string("{},{}".format(arch, aur), file)
+
+    if arch > 0 and aur > 0:
+        print("software-update-urgent")
+        print("{}/{}".format(arch, aur))
+    elif arch > 0:
+        print("software-update-available")
+        print("{}".format(arch))
+    elif aur > 0:
+        print("software-update-available")
+        print("{}".format(aur))
+
+
+def save_string(string, file):
+    try:
+        file = open(file, "wt")
+        file.write(string)
+        file.close()
+    except:
+        print("Error writing file '{}'".format(file))
+
+
+def load_string(path):
+    try:
+        with open(path, 'r') as file:
+            data = file.read()
+            return data
+    except:
+        return ""
+
+
+def check_updates():
+    arch, aur = 0, 0
+    try:
+        arch = len(subprocess.check_output(["checkupdates"]).decode("utf-8").splitlines())
+    except:
+        pass
+    try:
+        aur = len(subprocess.check_output(["trizen", "-Qqu", "-a"]).decode("utf-8").splitlines())
+    except:
+        pass
+
+    return arch, aur
+
+
+if __name__ == "__main__":
+    main()