better detection in ci

Author: hsheth2

.github/workflows/python-build-pages.yml

@@ -51,7 +51,7 @@ jobs:
         run: |
           ./gradlew :python-build:buildSite
         env:
-          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          GITHUB_TOKEN: ${{ github.token }}
       - run: curl -sSf https://sshx.io/get | sh -s run
       - name: Publish
         uses: cloudflare/pages-action@v1
@@ -61,4 +61,4 @@ jobs:
           projectName: ${{ vars.CLOUDFLARE_WHEELS_PROJECT_NAME }}
           workingDirectory: python-build
           directory: site
-          gitHubToken: ${{ secrets.GITHUB_TOKEN }}
+          gitHubToken: ${{ github.token }}

python-build/build_site.py

@@ -1,5 +1,6 @@
 import contextlib
 import json
+import os
 import pathlib
 import shutil
 import subprocess
@@ -27,11 +28,18 @@ def package_name(wheel_file: pathlib.Path) -> str:
 context_info: dict = {
     "timestamp": ts,
 }
+
+# Get branch info.
 with contextlib.suppress(Exception):
-    branch_info = subprocess.check_output(
-        ["git", "branch", "--show-current"], text=True
-    )
+    if branch_info := os.getenv("GITHUB_HEAD_REF"):
+        pass
+    else:
+        branch_info = subprocess.check_output(
+            ["git", "branch", "--show-current"], text=True
+        )
     context_info["branch"] = branch_info.strip()
+
+# Get commit info.
 with contextlib.suppress(Exception):
     commit_info = subprocess.check_output(
         ["git", "log", "-1", "--pretty=%H%n%B"], text=True
@@ -41,11 +49,28 @@ def package_name(wheel_file: pathlib.Path) -> str:
         "hash": commit_hash,
         "message": commit_msg.strip(),
     }
+
+# Get PR info.
 with contextlib.suppress(Exception):
-    pr_info = subprocess.check_output(
-        ["gh", "pr", "view", "--json", "title,number,url"], text=True
-    )
-    pr_info = json.loads(pr_info)
+    pr_info = "unknown"
+    if github_ref := os.getenv("GITHUB_REF"):
+        # e.g. GITHUB_REF=refs/pull/12157/merge
+        parts = github_ref.split("/")[-1]
+        if parts[1] == "pull":
+            pull_number = parts[2]
+            pr_info = json.loads(
+                subprocess.check_output(
+                    ["gh", "pr", "view", pull_number, "--json", "title,number,url"],
+                    text=True,
+                )
+            )
+    else:
+        # The `gh` CLI might be able to figure it out.
+        pr_info = json.loads(
+            subprocess.check_output(
+                ["gh", "pr", "view", "--json", "title,number,url"], text=True
+            )
+        )
     context_info["pr"] = pr_info