pegasust
/
dotfiles
1
0
Fork 0
Code Issues 6 Pull Requests Projects Releases Wiki Activity

nixGL/fetch: cleanup

std
Hung 2023-04-02 01:35:36 -07:00
parent 335efeb2d9
commit 17937c2146
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
out-of-tree/nixGL/fetch.py
View File

@ -1,5 +1,6 @@
import http.client import http.client
import json import json
import sys
NV_FREE_X86_URL = "download.nvidia.com" NV_FREE_X86_URL = "download.nvidia.com"
POTENTIAL_SHA256_EXTS = [".sha256sum", ".sha256"] POTENTIAL_SHA256_EXTS = [".sha256sum", ".sha256"]
@ -10,7 +11,7 @@ dir_html = response.read()
assert (response.status < 400), "Error occurred fetching for source from" assert (response.status < 400), "Error occurred fetching for source from"
def scrape_driver_versions(dir_html: bytes) -> list[str]: def scrape_driver_versions(dir_html: bytes):
# The idea is to recursively follows all interesting `src` from `<a href={src}>` # The idea is to recursively follows all interesting `src` from `<a href={src}>`
def _rec(dir_html: bytes, href_url_start: int = 0, so_far: list[str] = []) -> list[str]: def _rec(dir_html: bytes, href_url_start: int = 0, so_far: list[str] = []) -> list[str]:
MATCH_START = b"<span class=\'dir\'><a href=\'" MATCH_START = b"<span class=\'dir\'><a href=\'"
@ -28,8 +29,15 @@ def scrape_driver_versions(dir_html: bytes) -> list[str]:
assert version_end != -1, "Should have end-signaling /" assert version_end != -1, "Should have end-signaling /"
so_far.append(dir_html[potential_version_start:version_end].decode()) so_far.append(dir_html[potential_version_start:version_end].decode())
return _rec(dir_html, version_end, so_far) return _rec(dir_html, version_end, so_far)
return _rec(dir_html, 0, [])
versions = _rec(dir_html, 0, [])
num_versions = len(versions)
right_pad = " " * 50
for i, version in enumerate(versions):
print(f"[{i+1}/{num_versions}] Processing version {version}{right_pad}", end="\r", file=sys.stderr)
yield version
print()
versions = scrape_driver_versions(dir_html) versions = scrape_driver_versions(dir_html)
download_urls_of = lambda ver: [f"/XFree86/Linux-x86_64/{ver}/NVIDIA-Linux-x86_64-{ver}.run"] download_urls_of = lambda ver: [f"/XFree86/Linux-x86_64/{ver}/NVIDIA-Linux-x86_64-{ver}.run"]
@ -58,4 +66,3 @@ print(json.dumps({
"sha256": sha256 "sha256": sha256
} for (version, dl_url, sha256) in fetch_data if sha256 is not None}, indent=4)) } for (version, dl_url, sha256) in fetch_data if sha256 is not None}, indent=4))
# execution: fetch.py >nvidia_versions.json # execution: fetch.py >nvidia_versions.json

1
out-of-tree/nixGL/flake.nix
View File

@ -28,7 +28,6 @@
nixVulkanNvidia = pkgs.auto.nixVulkanNvidia; nixVulkanNvidia = pkgs.auto.nixVulkanNvidia;
nixVulkanIntel = pkgs.nixVulkanIntel; nixVulkanIntel = pkgs.nixVulkanIntel;
nvida-fetch
}; };
# deprecated attributes for retro compatibility # deprecated attributes for retro compatibility