Pass along GITHUB_TOKEN is availabel to avoid GitHub rate limits

Signed-off-by: Christoph Obexer <cobexer@gradle.com>

Author: cobexer

toolbox.sh

@@ -8,6 +8,7 @@ fi
 
 exec docker run --rm -ti \
 	-v "$(pwd):/workspace" \
+	-e GITHUB_TOKEN \
 	-w /workspace \
 	gradle-dockerhub-toolbox \
 	python3 "$@"

update.py

@@ -4,6 +4,16 @@
 import os
 import sys
 import hashlib
+from urllib.parse import urlparse
+
+def github_headers(url):
+    token = os.environ.get("GITHUB_TOKEN")
+    if not token:
+        return {}
+    host = urlparse(url).hostname or ""
+    if host == "api.github.com" or host == "github.com" or host.endswith(".github.com") or host == "objects.githubusercontent.com":
+        return {"Authorization": f"Bearer {token}"}
+    return {}
 
 def get_gradle_version(base_version):
     response = requests.get(f"https://services.gradle.org/versions/{base_version}", timeout=10)
@@ -25,15 +35,15 @@ def get_gradle_version(base_version):
 
 def calculate_sha256(url):
     sha256_hash = hashlib.sha256()
-    with requests.get(url, stream=True, timeout=300) as r:
+    with requests.get(url, stream=True, timeout=300, headers=github_headers(url)) as r:
         r.raise_for_status()
         for chunk in r.iter_content(chunk_size=8192):
             sha256_hash.update(chunk)
     return sha256_hash.hexdigest()
 
 def get_sha256(url):
     try:
-        response = requests.get(url, timeout=10)
+        response = requests.get(url, timeout=10, headers=github_headers(url))
         response.raise_for_status()
         return response.text.strip()
     except requests.exceptions.HTTPError as e:
@@ -44,7 +54,8 @@ def get_sha256(url):
         raise
 
 def get_graalvm_info(jdk_version):
-    response = requests.get("https://api.github.com/repos/graalvm/graalvm-ce-builds/releases?per_page=20&page=1", timeout=10)
+    url = "https://api.github.com/repos/graalvm/graalvm-ce-builds/releases?per_page=20&page=1"
+    response = requests.get(url, timeout=10, headers=github_headers(url))
     response.raise_for_status()
     releases = response.json()