Merge cf83fc6e70 into sapling-pr-archive-poteto

2025-11-01 09:12:30 +00:00 · 2025-04-08 10:51:40 -04:00
parent 535e12b576 cf83fc6e70
commit 557cc951fc
1 changed files with 109 additions and 0 deletions
@@ -0,0 +1,109 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+# https://github.com/Chillee/ghstack_land_example/blob/main/.github/workflows/scripts/ghstack-perm-check.py
+
+import json
+import os
+import re
+import subprocess
+import sys
+
+import requests
+
+
+def main():
+    gh = requests.Session()
+    gh.headers.update(
+        {
+            "Authorization": f'Bearer {os.environ["GITHUB_TOKEN"]}',
+            "Accept": "application/vnd.github+json",
+            "X-GitHub-Api-Version": "2022-11-28",
+        }
+    )
+    NUMBER, head_ref = int(sys.argv[1]), sys.argv[2]
+    EV = json.loads(sys.stdin.read())
+    REPO = EV["repository"]
+
+    def must(cond, msg):
+        if not cond:
+            print(msg)
+            gh.post(
+                f"https://api.github.com/repos/{REPO}/issues/{NUMBER}/comments",
+                json={
+                    "body": f"ghstack bot failed: {msg}",
+                },
+            )
+            exit(1)
+
+    print(head_ref)
+    must(
+        head_ref and re.match(r"^gh/[A-Za-z0-9-]+/[0-9]+/head$", head_ref),
+        "Not a ghstack PR",
+    )
+    orig_ref = head_ref.replace("/head", "/orig")
+    print(":: Fetching newest main...")
+    must(os.system("git fetch origin main") == 0, "Can't fetch main")
+    print(":: Fetching orig branch...")
+    must(os.system(f"git fetch origin {orig_ref}") == 0, "Can't fetch orig branch")
+
+    proc = subprocess.Popen(
+        "git log FETCH_HEAD...$(git merge-base FETCH_HEAD origin/main)",
+        stdout=subprocess.PIPE,
+        shell=True,
+    )
+    out, _ = proc.communicate()
+    must(proc.wait() == 0, "`git log` command failed!")
+
+    pr_numbers = re.findall(
+        r"Pull Request resolved: https://github.com/.*?/pull/([0-9]+)",
+        out.decode("utf-8"),
+    )
+    pr_numbers = list(map(int, pr_numbers))
+    print(pr_numbers)
+    must(pr_numbers and pr_numbers[0] == NUMBER, "Extracted PR numbers not seems right!")
+
+    for n in pr_numbers:
+        print(f":: Checking PR status #{n}... ", end="")
+        resp = gh.get(f"https://api.github.com/repos/{REPO}/pulls/{n}")
+        must(resp.ok, "Error Getting PR Object!")
+        pr_obj = resp.json()
+
+        resp = gh.get(f"https://api.github.com/repos/{REPO}/pulls/{NUMBER}/reviews")
+        must(resp.ok, "Error Getting PR Reviews!")
+        reviews = resp.json()
+        idmap = {}
+        approved = True # TODO: REMOVE
+        for r in reviews:
+            s = r["state"]
+            if s not in ("COMMENTED",):
+                idmap[r["user"]["login"]] = r["state"]
+
+        for u, cc in idmap.items():
+            approved = approved or cc == "APPROVED"
+            must(
+                cc in ("APPROVED", "DISMISSED"),
+                f"@{u} has stamped PR #{n} `{cc}`, please resolve it first!",
+            )
+
+        must(approved, f"PR #{n} is not approved yet!")
+
+        resp = gh.get(f'https://api.github.com/repos/{REPO}/commits/{pr_obj["head"]["sha"]}/check-runs')
+        must(resp.ok, "Error getting check runs status!")
+        checkruns = resp.json()
+        for cr in checkruns["check_runs"]:
+            status = cr.get("conclusion", cr["status"])
+            name = cr["name"]
+            if name == "Copilot for PRs":
+                continue
+            must(
+                status in ("success", "neutral"),
+                f"PR #{n} check-run `{name}`'s status `{status}` is not success!",
+            )
+        print("SUCCESS!")
+
+    print(":: All PRs are ready to be landed!")
+
+
+if __name__ == "__main__":
+    main()