Merge pull request #11239 from id/0710-ci-rerun-apps-version-check-on-release

ci: rerun apps version check on release

.github/workflows/release.yaml

@@ -101,3 +101,21 @@ jobs:
           push "el/8" "packages/$PROFILE-$VERSION-el8-arm64.rpm"
           push "el/9" "packages/$PROFILE-$VERSION-el9-amd64.rpm"
           push "el/9" "packages/$PROFILE-$VERSION-el9-arm64.rpm"
+
+  rerun-apps-version-check:
+    runs-on: ubuntu-22.04
+    if: github.repository_owner == 'emqx' && github.event_name == 'release'
+    needs:
+      - upload
+    permissions:
+      pull-requests: read
+      checks: read
+      actions: write
+    steps:
+      - uses: actions/checkout@v3
+      - name: trigger re-run of app versions check on open PRs
+        shell: bash
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          python3 scripts/rerun-apps-version-check.py

scripts/rerun-apps-version-check.py

@@ -0,0 +1,122 @@
+#!/usr/bin/env python3
+# Usage: python3 rerun-apps-version-check.py -t <github token> -r <repo>
+#
+# Default repo is emqx/emqx
+#
+import requests
+import http.client
+import json
+import os
+import sys
+import time
+import math
+import inspect
+from optparse import OptionParser
+from urllib.parse import urlparse, parse_qs
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
+
+user_agent = sys.argv[0].split('/')[-1]
+
+def query(owner, repo):
+    return """
+query {
+  repository(owner: "%s", name: "%s") {
+    pullRequests(last: 25, states: OPEN) {
+      nodes {
+        url
+        commits(last: 1) {
+          nodes {
+            commit {
+              checkSuites(first: 17) {
+                nodes {
+                  url
+                  checkRuns(first: 1, filterBy: {checkName: "check_apps_version"}) {
+                    nodes {
+                      url
+                    }
+                  }
+                }
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+}
+    """ % (owner, repo)
+
+
+def get_headers(token: str):
+    return {'Accept': 'application/vnd.github+json',
+            'Authorization': f'Bearer {token}',
+            'X-GitHub-Api-Version': '2022-11-28',
+            'User-Agent': f'{user_agent}'
+            }
+
+def get_session():
+    session = requests.Session()
+
+    retries = Retry(total=10,
+                    backoff_factor=1,  # 1s
+                    allowed_methods=None,
+                    status_forcelist=[ 429, 500, 502, 503, 504 ])  # Retry on these status codes
+
+    session.mount('https://', HTTPAdapter(max_retries=retries))
+
+    return session
+
+def get_check_suite_ids(token: str, repo: str):
+    session = get_session()
+    url = f'https://api.github.com/graphql'
+    [repo_owner, repo_repo] = repo.split('/')
+    r = session.post(url, headers=get_headers(token), json={'query': query(repo_owner, repo_repo)})
+    if r.status_code == 200:
+        resp = r.json()
+        if not 'data' in resp:
+            print(f'Failed to fetch check runs: {r.status_code}\n{r.json()}')
+            sys.exit(1)
+        ids = []
+        for pr in resp['data']['repository']['pullRequests']['nodes']:
+            if not pr['commits']['nodes']:
+                continue
+            if not pr['commits']['nodes'][0]['commit']['checkSuites']['nodes']:
+                continue
+            for node in pr['commits']['nodes'][0]['commit']['checkSuites']['nodes']:
+                if node['checkRuns']['nodes']:
+                    id = node['checkRuns']['nodes'][0]['url'].rsplit('/', 1)[-1]
+                    url_parsed = urlparse(node['url'])
+                    params = parse_qs(url_parsed.query)
+                    check_suite_id = params['check_suite_id'][0] 
+                    ids.extend([check_suite_id])
+        return ids
+    else:
+        print(f'Failed to fetch check runs: {r.status_code}\n{r.text}')
+        sys.exit(1)
+
+def rerequest_check_suite(token: str, repo: str, check_suite_id: str):
+    session = get_session()
+    url = f'https://api.github.com/repos/{repo}/check-suites/{check_suite_id}/rerequest'
+    r = session.post(url, headers=get_headers(token))
+    if r.status_code == 201:
+        print(f'Successfully triggered rerequest for check suite {check_suite_id}')
+    else:
+        print(f'Failed to trigger rerequest for check suite {check_suite_id}: {r.status_code}\n{r.text}')
+
+def main():
+    parser = OptionParser()
+    parser.add_option("-r", "--repo", dest="repo",
+                      help="github repo", default="emqx/emqx")
+    parser.add_option("-t", "--token", dest="gh_token",
+                      help="github API token")
+    (options, args) = parser.parse_args()
+
+    # Get github token from env var if provided, else use the one from command line.
+    # The token must be exported in the env from ${{ secrets.GITHUB_TOKEN }} in the workflow.
+    token = os.environ['GITHUB_TOKEN'] if 'GITHUB_TOKEN' in os.environ else options.gh_token
+    for id in get_check_suite_ids(token, options.repo):
+        rerequest_check_suite(token, options.repo, id)
+
+if __name__ == '__main__':
+    main()