Notify proxy maintainers

.github/workflows/notify_codeowners.py

@@ -0,0 +1,146 @@
+import github
+import click
+import urllib.request
+from pathlib import Path
+from typing import List, Tuple
+import re
+import fnmatch
+import glob
+import os
+
+
+def xor_strings(a, b):
+    result = int(a, 16) ^ int(b, 16)
+    return "{:x}".format(result)
+
+
+def get_github_client():
+    bot_token = "1353d990cdb8b8ceb1b73d301dce83cc0da3db29"
+    bot_token_key = "a1b2c3d47311f8e29e204f85a81b4df4a44e252c"
+
+    return github.Github(xor_strings(bot_token, bot_token_key))
+
+
+CLIENT = get_github_client()
+
+
+def check_user(user: str, line_idx: int):
+    if user[0] != "@":
+        raise ValueError(
+            f"User '{user}' at line {line_idx} of CODEOWNERS "
+            f"doesn't start with '@' "
+        )
+    user = user[1:]
+    if "/" not in user:
+        # it's a real user. Check for typos.
+        try:
+            CLIENT.get_user(user)
+        except github.UnknownObjectException:
+            raise KeyError(
+                f"User '{user}' line {line_idx} does not exist. "
+                f"Did you make a typo?"
+            )
+    return user
+
+
+def check_pattern(pattern: str, line_idx: int):
+    if pattern[0] == "/":
+        pattern = pattern[1:]
+
+    pattern = Pattern(pattern)
+
+    if not pattern.match_in_dir("."):
+        raise FileNotFoundError(
+            f"'{pattern.string}' present in CODEOWNERS line"
+            f" {line_idx} does not match any file in the repository. "
+            f"Did you make a typo?"
+        )
+
+    return pattern
+
+
+class Pattern:
+    def __init__(self, pattern: str):
+        self.string = pattern
+        if "*" in self.string:
+            self.regex = re.compile(fnmatch.translate(pattern))
+        else:
+            self.regex = None
+
+    def match(self, file):
+        if self.regex:
+            return re.match(self.regex, file) is not None
+        else:
+            return file.startswith(self.string)
+
+    def match_in_dir(self, directory):
+        list_files = glob.glob(f"{directory}/**/*", recursive=True)
+        list_files = [os.path.relpath(x, directory) for x in list_files]
+        matching_files = list(filter(self.match, list_files))
+        return bool(matching_files)
+
+
+CodeOwners = List[Tuple[Pattern, List[str]]]
+
+
+def parse_codeowners(text: str) -> CodeOwners:
+    result = []
+
+    for i, line in enumerate(text.splitlines()):
+        line = line.strip()
+        if line == "":
+            continue
+        if line[0] == "#":  # comment
+            continue
+        elements = list(filter(lambda x: x != "", line.split(" ")))
+
+        pattern = check_pattern(elements[0], i)
+        users = [check_user(user, i) for user in elements[1:]]
+        result.append((pattern, users))
+
+    return result
+
+
+nice_message = """
+
+You are owners of some files modified in this pull request.
+Would you kindly review the changes whenever you have the time to?
+Thank you very much.
+"""
+
+
+def craft_message(codeowners: CodeOwners, pull_request):
+
+    owners = set()
+    for file in pull_request.get_files():
+        for pattern, users in codeowners:
+            if not pattern.match(file.filename):
+                continue
+            owners.update(users)
+
+    owners = [f"@{owner}" for owner in owners]
+    return " ".join(owners + [nice_message])
+
+
+@click.command()
+@click.option("--pull-request-id")
+@click.option("--dry-run", is_flag=True)
+@click.argument("file")
+def notify_codeowners(pull_request_id, dry_run, file):
+    if file.startswith("http"):
+        text = urllib.request.urlopen(file).read().decode("utf-8")
+    else:
+        text = Path(file).read_text()
+    codeowners = parse_codeowners(text)
+
+    if pull_request_id:
+        pull_request_id = int(pull_request_id)
+        pull_request = CLIENT.get_repo("tensorflow/addons").get_pull(pull_request_id)
+        msg = craft_message(codeowners, pull_request)
+        print(msg)
+        if not dry_run:
+            pull_request.create_issue_comment(msg)
+
+
+if __name__ == "__main__":
+    notify_codeowners()

.github/workflows/notify_codeowners.yml

@@ -0,0 +1,28 @@
+name: Notify codeowners main
+
+on:
+  pull_request:
+    types: [opened]
+
+
+jobs:
+  notify-codeowners:
+    name: Notify codeowners
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+      - uses: actions/setup-python@v1
+        with:
+          python-version: 3.7
+      - run: pip install pygithub click
+      - uses: jwalton/gh-find-current-pr@v1
+          id: findPr
+          with:
+            github-token: ${{ secrets.GITHUB_TOKEN }}
+      - name: Drop a message for codeowners
+        env:
+          PR: ${{ steps.findPr.outputs.pr }}
+        run: |
+          python .github/workflows/notify_codeowners.py \
+              --pull-request-id=$PR \
+              https://raw.githubusercontent.com/tensorflow/addons/master/.github/CODEOWNERS

.github/workflows/valid_codeowners.yml

@@ -0,0 +1,25 @@
+name: Check codeowners file
+
+on:
+  push:
+    branches:
+      - master
+      - r*
+  pull_request:
+    branches:
+      - master
+      - r*
+
+
+jobs:
+  valid-codeowners:
+    name: Check that the CODEOWNERS is valid
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+      - uses: actions/setup-python@v1
+        with:
+          python-version: 3.7
+      - run: pip install pygithub click
+      - name: Check that the CODEOWNERS is valid
+        run: python .github/workflows/notify_codeowners.py .github/CODEOWNERS