Merge pull request github#6123 from tamasvajk/feature/framework-coverage-pr

tamasvajk · web-flow · commit 1cddcdfcb112 · 2021-06-25T09:18:10.000+02:00
Add scheduled job to update framework coverage
diff --git a/.github/workflows/csv-coverage-update.yml b/.github/workflows/csv-coverage-update.yml
@@ -0,0 +1,44 @@
+name: Update framework coverage reports
+
+on:
+  workflow_dispatch:
+  schedule:
+    - cron: "0 0 * * *"
+
+jobs:
+  update:
+    name: Update framework coverage report
+    if: github.event.repository.fork == false
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Dump GitHub context
+      env:
+        GITHUB_CONTEXT: ${{ toJSON(github.event) }}
+      run: echo "$GITHUB_CONTEXT"
+    - name: Clone self (github/codeql)
+      uses: actions/checkout@v2
+      with:
+        path: ql
+        fetch-depth: 0
+    - name: Set up Python 3.8
+      uses: actions/setup-python@v2
+      with:
+        python-version: 3.8
+    - name: Download CodeQL CLI
+      env:
+         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+      run: |
+         gh release download --repo "github/codeql-cli-binaries" --pattern "codeql-linux64.zip"
+    - name: Unzip CodeQL CLI
+      run: unzip -d codeql-cli codeql-linux64.zip
+
+    - name: Generate coverage files
+      run: |
+        PATH="$PATH:codeql-cli/codeql" python ql/misc/scripts/library-coverage/generate-report.py ci ql ql
+
+    - name: Create pull request with changes
+      env:
+         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+      run: |
+        python ql/misc/scripts/library-coverage/create-pr.py ql "$GITHUB_REPOSITORY"
diff --git a/misc/scripts/library-coverage/create-pr.py b/misc/scripts/library-coverage/create-pr.py
@@ -0,0 +1,113 @@
+import sys
+import settings
+import utils
+import shutil
+import json
+import os
+
+"""
+This script compares the generated CSV coverage files with the ones in the codebase. And if they are different, it
+creates a PR with the changes. If a PR already exists, then it will be updated to reflect the latest changes.
+"""
+
+# Name of the branch that's used to push the updated coverage files, also the head of the PR
+branch_name = "workflow/coverage/update"
+# Name of the remote where the new commit is pushed
+remote = "origin"
+# Name of the branch that provides the base for the new branch, and the base of the PR
+main = "main"
+
+repo = sys.argv[2]
+owner = repo.split('/')[0]
+
+
+def overwrite_files():
+    languages = ['java']
+    for lang in languages:
+        repo_output_rst = settings.repo_output_rst.format(language=lang)
+        repo_output_csv = settings.repo_output_csv.format(language=lang)
+
+        generated_output_rst = settings.generated_output_rst.format(
+            language=lang)
+        generated_output_csv = settings.generated_output_csv.format(
+            language=lang)
+
+        exists = utils.check_file_exists(generated_output_rst)
+        if not exists:
+            print(
+                f"Generated RST file {generated_output_rst} is missing", file=sys.stderr)
+            sys.exit(1)
+
+        exists = utils.check_file_exists(generated_output_csv)
+        if not exists:
+            print(
+                f"Generated RST file {generated_output_csv} is missing", file=sys.stderr)
+            sys.exit(1)
+
+        shutil.move(generated_output_rst, repo_output_rst)
+        shutil.move(generated_output_csv, repo_output_csv)
+
+
+def get_pr_number(repo, owner, from_branch, to_branch):
+    ids = utils.subprocess_check_output(["gh", "api", "-X", "GET", f"repos/{repo}/pulls", "-f",
+                                        f"name={to_branch}", "-f", f"head={owner}:{from_branch}", "--jq", "[.[].number]"])
+
+    ids = json.loads(ids)
+
+    if len(ids) > 1:
+        print(
+            f"Found more than one PR that matches the branches. {ids}", file=sys.stderr)
+        sys.exit(1)
+
+    if len(ids) == 1:
+        print(f"Matching PR found: {ids[0]}")
+        return ids[0]
+
+    return 0
+
+
+def create_pr(repo, from_branch, to_branch):
+    print(f"Creating PR for {from_branch}")
+    utils.subprocess_check_output(["gh", "pr", "create", "-R", repo, "--base", to_branch,
+                                   "--head", from_branch, "--title", "Update CSV framework coverage reports",
+                                   "--body", "This PR changes the CSV framework coverage reports."])
+
+
+working_dir = ""
+if len(sys.argv) > 1:
+    working_dir = sys.argv[1]
+else:
+    print("Working directory is not specified")
+    exit(1)
+
+found_diff = False
+overwrite_files()
+
+os.chdir(working_dir)
+
+already_open_pr = get_pr_number(repo, owner, branch_name, main)
+try:
+    utils.subprocess_check_output(["git", "diff", "--exit-code"])
+    print("No differences found")
+    found_diff = False
+except:
+    print("Differences found")
+    found_diff = True
+
+if not found_diff:
+    if already_open_pr != 0:
+        # close the PR
+        utils.subprocess_run(
+            ["gh", "pr", "close", str(already_open_pr), "-R", repo])
+else:
+    utils.subprocess_run(["git", "config", "user.name",
+                         "github-actions[bot]"])
+    utils.subprocess_run(["git", "config", "user.email",
+                         "41898282+github-actions[bot]@users.noreply.github.com"])
+    utils.subprocess_run(["git", "add", "."])
+    utils.subprocess_run(
+        ["git", "commit", "-m", "Add changed framework coverage reports"])
+    utils.subprocess_run(["git", "branch", "-f", branch_name, main])
+    utils.subprocess_run(["git", "push", "-f", remote, branch_name])
+    if already_open_pr == 0:
+        create_pr(repo, branch_name, main)
