bundles/sshmon: rewrite check_github_for_new_release

bundles/sshmon/files/check_github_for_new_release

@@ -1,33 +1,66 @@
 #!/usr/bin/env python3
 
-from requests import get
+
 from sys import argv, exit
 
-if not len(argv) == 3:
-    print('Usage: check_github_for_new_release <repo> <version>')
-    exit(3)
+from requests import get
+
+try:
+    # Arch Linux (possibly future Ubuntu releases)
+    from packaging.version import parse
+except ModuleNotFoundError:
+    # Debian
+    from setuptools._vendor.packaging.version import parse
+
+
+assert (
+    len(argv) >= 3
+), "Usage: check_github_for_new_release <repo> <version> [tag prefix]"
 
 repo = argv[1]
 current_version = argv[2]
+if len(argv) >= 4:
+    tag_prefix = argv[3]
+else:
+    tag_prefix = ""
 
 try:
-    r = get('https://api.github.com/repos/{}/releases/latest'.format(repo), headers={'Accept': 'application/json'}).json()
-except Exception as e:
-    print(repr(e))
-    exit(3)
+    r = get(
+        "https://api.github.com/repos/{}/releases".format(repo),
+        headers={"Accept": "application/json"},
+    )
+    r.raise_for_status()
 
-newest_release = r['tag_name']
+    releases = r.json()
 
-if newest_release != current_version:
-    print('There is a newer version available: {} (currently installed: {})'.format(newest_release, current_version))
+    newest_release = None
 
-    if r['prerelease'] or r['draft']:
-        print('Please note this is only a prerelease/draft.')
+    for i in releases:
+        if i["tag_name"].startswith(tag_prefix):
+            if newest_release is None or parse(i["tag_name"]) > parse(
+                newest_release["tag_name"]
+            ):
+                newest_release = i
+
+    assert newest_release is not None, "Could not determine latest release"
+
+    if parse(newest_release["tag_name"]) > parse(current_version):
+        print(
+            "There is a newer version available: {} (currently installed: {})".format(
+                newest_release["tag_name"], current_version
+            )
+        )
+
+        if newest_release["prerelease"] or newest_release["draft"]:
+            print("Please note this is only a prerelease/draft.")
 
             # Don't alert people for prereleases.
             exit(0)
         else:
             exit(2)
-else:
-    print('Currently installed version matches newest release on github')
+    else:
+        print("Currently installed version matches newest release on github")
         exit(0)
+except Exception as e:
+    print(repr(e))
+    exit(3)

bundles/sshmon/metadata.py

@@ -7,6 +7,7 @@ defaults = {
             'libwww-perl': {},  # needed by check_nginx_status
             'monitoring-plugins': {},
             'python3-requests': {},
+            'python3-setuptools': {},  # needed by check_github_for_new_release
             'sysstat': {},  # needed by check_cpu_stats
         },
     },