From dc2375e8c7a115fbc85f4667f9206e0ab2e5c838 Mon Sep 17 00:00:00 2001
From: Vladislav Yarmak <vladislav@vm-0.com>
Date: Mon, 20 May 2019 15:21:13 +0300
Subject: [PATCH] ndl: track cuda toolkit releases

---
 tools/nv-driver-locator/get_cuda_downloads.py | 71 +++++++++++++++++++
 .../nv-driver-locator.json.sample             |  5 ++
 tools/nv-driver-locator/nv-driver-locator.py  | 22 ++++++
 3 files changed, 98 insertions(+)
 create mode 100755 tools/nv-driver-locator/get_cuda_downloads.py

diff --git a/tools/nv-driver-locator/get_cuda_downloads.py b/tools/nv-driver-locator/get_cuda_downloads.py
new file mode 100755
index 0000000..4a212d5
--- /dev/null
+++ b/tools/nv-driver-locator/get_cuda_downloads.py
@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+
+import urllib.request
+import urllib.error
+import urllib.parse
+import codecs
+import enum
+import re
+from bs4 import BeautifulSoup
+
+USER_AGENT = 'Mozilla/5.0 (X11; Linux x86_64; rv:65.0) '\
+             'Gecko/20100101 Firefox/65.0'
+
+
+def parse_args():
+    import argparse
+
+    def check_positive_float(val):
+        val = float(val)
+        if val <= 0:
+            raise ValueError("Value %s is not valid positive float" %
+                             (repr(val),))
+        return val
+
+    parser = argparse.ArgumentParser(
+        description="Retrieves info about latest NVIDIA drivers from "
+        "downloads site",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser.add_argument("-T", "--timeout",
+                        type=check_positive_float,
+                        default=10.,
+                        help="timeout for network operations")
+    args = parser.parse_args()
+    return args
+
+
+def issue_request(timeout=10):
+    ENDPOINT = 'https://developer.nvidia.com/cuda-toolkit-archive'
+    http_req = urllib.request.Request(
+        ENDPOINT,
+        data=None,
+        headers={
+            'User-Agent': USER_AGENT
+        }
+    )
+    with urllib.request.urlopen(http_req, None, timeout) as resp:
+        coding = resp.headers.get_content_charset()
+        coding = coding if coding is not None else 'utf-8-sig'
+        decoder = codecs.getreader(coding)(resp)
+        res = decoder.read()
+    return res
+
+
+def get_latest_cuda_tk(*, timeout=10):
+    doc = issue_request(timeout)
+    soup = BeautifulSoup(doc, 'html.parser')
+
+    res = soup.find('strong', string=re.compile(r'^\s*latest\s+release\s*$', re.I)).\
+        parent.find('a', string=re.compile(r'^\s*cuda\s+toolkit\s+.*$', re.I)).\
+        string
+    return res
+
+
+def main():
+    import pprint
+    args = parse_args()
+    print(get_latest_cuda_tk(timeout=args.timeout))
+
+
+if __name__ == '__main__':
+    main()
diff --git a/tools/nv-driver-locator/nv-driver-locator.json.sample b/tools/nv-driver-locator/nv-driver-locator.json.sample
index de32460..8b53eed 100644
--- a/tools/nv-driver-locator/nv-driver-locator.json.sample
+++ b/tools/nv-driver-locator/nv-driver-locator.json.sample
@@ -175,6 +175,11 @@
                 "product": "Quadro",
                 "certlevel": "Certified"
             }
+        },
+        {
+            "type": "cuda_downloads",
+            "name": "cuda toolkit tracker",
+            "params": {}
         }
     ],
     "notifiers": [
diff --git a/tools/nv-driver-locator/nv-driver-locator.py b/tools/nv-driver-locator/nv-driver-locator.py
index ae3cdf9..6c167fc 100755
--- a/tools/nv-driver-locator/nv-driver-locator.py
+++ b/tools/nv-driver-locator/nv-driver-locator.py
@@ -197,6 +197,27 @@ class NvidiaDownloadsChannel(BaseChannel):
         }
 
 
+class CudaToolkitDownloadsChannel(BaseChannel):
+    def __init__(self, name, *,
+                 timeout=10):
+        self.name = name
+        gcd = importlib.import_module('get_cuda_downloads')
+        self._gcd = gcd
+        self._timeout = timeout
+
+    def get_latest_driver(self):
+        latest = self._gcd.get_latest_cuda_tk(timeout=self._timeout)
+        if not latest:
+            return None
+        return {
+            'DriverAttributes': {
+                'Version': '???',
+                'Name': latest,
+                'NameLocalized': latest,
+            }
+        }
+
+
 def parse_args():
     parser = argparse.ArgumentParser(
         description="Watches for GeForce experience driver updates for "
@@ -223,6 +244,7 @@ class DriverLocator:
         channel_types = {
             'gfe_client': GFEClientChannel,
             'nvidia_downloads': NvidiaDownloadsChannel,
+            'cuda_downloads': CudaToolkitDownloadsChannel,
         }
 
         channels = []