From 95c3eb377bab413ac2d9238d7726898dedf33ee2 Mon Sep 17 00:00:00 2001
From: alesapin <alesapin@gmail.com>
Date: Wed, 29 Sep 2021 10:47:15 +0300
Subject: [PATCH] Add finish check

---
 .github/workflows/main.yml | 12 ++++++++++-
 tests/ci/finish_check.py   | 41 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 52 insertions(+), 1 deletion(-)
 create mode 100644 tests/ci/finish_check.py

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 05ed78d8c07..3931bc1538d 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -37,7 +37,7 @@ jobs:
         with:
           name: changed_images
           path: ${{ runner.temp }}/docker_images_check/changed_images.json
-  Style-Check:
+  StyleCheck:
     needs: DockerHubPush
     runs-on: [self-hosted]
     steps:
@@ -76,3 +76,13 @@ jobs:
 #          TEMP_PATH: ${{runner.temp}}/pvs_check
 #          REPO_COPY: ${{runner.temp}}/pvs_check/ClickHouse
 #        run: mkdir -p ${{runner.temp}}/pvs_check && cp -r $GITHUB_WORKSPACE $TEMP_PATH && cd $REPO_COPY/tests/ci && python3 pvs_check.py
+  Finish-Check:
+    needs: [Style-Check, DockerHubPush, CheckLabels]
+    runs-on: [self-hosted]
+    steps:
+      - name: Check out repository code
+        uses: actions/checkout@v2
+      - name: Finish label
+        run: cd $GITHUB_WORKSPACE/tests/ci && python3 finish_check.py
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
diff --git a/tests/ci/finish_check.py b/tests/ci/finish_check.py
new file mode 100644
index 00000000000..b481c5b658c
--- /dev/null
+++ b/tests/ci/finish_check.py
@@ -0,0 +1,41 @@
+#!/usr/bin/env python3
+import logging
+from github import Github
+import os
+
+NAME = 'Run Check (actions)'
+
+def filter_statuses(statuses):
+    """
+    Squash statuses to latest state
+    1. context="first", state="success", update_time=1
+    2. context="second", state="success", update_time=2
+    3. context="first", stat="failure", update_time=3
+    =========>
+    1. context="second", state="success"
+    2. context="first", stat="failure"
+    """
+    filt = {}
+    for status in sorted(statuses, key=lambda x: x.updated_at):
+        filt[status.context] = status
+    return filt
+
+
+def get_commit(gh, commit_sha):
+    repo = gh.get_repo(os.getenv("GITHUB_REPOSITORY", "ClickHouse/ClickHouse"))
+    commit = repo.get_commit(commit_sha)
+    return commit
+
+if __name__ == "__main__":
+    logging.basicConfig(level=logging.INFO)
+    with open(os.getenv('GITHUB_EVENT_PATH'), 'r') as event_file:
+        event = json.load(event_file)
+
+    pr_info = PRInfo(event, need_orgs=True)
+    gh = Github(os.getenv("GITHUB_TOKEN"))
+    commit = get_commit(gh, pr_info.sha)
+
+    url = f"https://github.com/ClickHouse/ClickHouse/actions/runs/{os.getenv('GITHUB_RUN_ID')}"
+    statuses = filter_statuses(list(commit.get_statuses()))
+    if NAME in statuses and statuses[NAME].state == "pending":
+        commit.create_status(context=NAME, description="All checks finished", state="success", target_url=url)