Merge pull request #5 from mattalbr/main

Support repo-agnostic commenting from github bot.

Author: patrick91

app/github.py

@@ -31,44 +31,56 @@ def has_signature(comment: GithubComment, slug: str) -> bool:
     )
 
 
-def get_comments_link(pr_number: int) -> str:
-    url = f"/repos/strawberry-graphql/strawberry/issues/{pr_number}/comments"
+def get_comments_link(pr_number: int, repo: typing.Optional[str] = None) -> str:
+    url = (
+        f"/repos/strawberry-graphql/{repo or 'strawberry'}/issues/{pr_number}/comments"
+    )
 
     return API_BASE + url
 
 
-def get_labels_link(pr_number: int) -> str:
-    url = f"/repos/strawberry-graphql/strawberry/issues/{pr_number}/labels"
+def get_labels_link(pr_number: int, repo: typing.Optional[str] = None) -> str:
+    url = f"/repos/strawberry-graphql/{repo or 'strawberry'}/issues/{pr_number}/labels"
 
     return API_BASE + url
 
 
-def get_comments(pr_number: int) -> typing.List[GithubComment]:
-    comments_link = get_comments_link(pr_number)
+def get_comments(
+    pr_number: int, repo: typing.Optional[str] = None
+) -> typing.List[GithubComment]:
+    comments_link = get_comments_link(pr_number, repo=repo)
 
     response = httpx.get(comments_link)
 
     return response.json()
 
 
-def get_labels(pr_number) -> typing.List[GithubLabel]:
-    labels_link = get_labels_link(pr_number)
+def get_labels(
+    pr_number: int, repo: typing.Optional[str] = None
+) -> typing.List[GithubLabel]:
+    labels_link = get_labels_link(pr_number, repo=repo)
 
     response = httpx.get(labels_link)
 
     return response.json()
 
 
-def add_or_edit_comment(pr_number: int, comment_template: str, slug: str):
-    current_comments = get_comments(pr_number)
+def add_or_edit_comment(
+    pr_number: int, comment_template: str, slug: str, repo: typing.Optional[str] = None
+):
+    current_comments = get_comments(pr_number, repo=repo)
 
     previous_comment = next(
         (comment for comment in current_comments if has_signature(comment, slug)),
         None,
     )
 
     method = httpx.patch if previous_comment else httpx.post
-    url = previous_comment["url"] if previous_comment else get_comments_link(pr_number)
+    url = (
+        previous_comment["url"]
+        if previous_comment
+        else get_comments_link(pr_number, repo=repo)
+    )
 
     response = method(
         url,
@@ -79,7 +91,9 @@ def add_or_edit_comment(pr_number: int, comment_template: str, slug: str):
     response.raise_for_status()
 
 
-def update_labels(pr_number: int, release_info: typing.Optional[ReleaseInfo]):
+def update_labels(
+    pr_number: int, release_info: typing.Optional[ReleaseInfo], repo: str | None = None
+):
     labels_to_add = {"bot:has-release-file"}
     labels_to_remove: typing.Set[str] = set()
 
@@ -92,10 +106,10 @@ def update_labels(pr_number: int, release_info: typing.Optional[ReleaseInfo]):
         new_release_label = f"bot:release-type-{release_info.change_type.value}"
         labels_to_add.add(new_release_label)
 
-    labels_url = get_labels_link(pr_number)
+    labels_url = get_labels_link(pr_number, repo=repo)
 
     current_labels_url_by_name = {
-        label["name"]: label["url"] for label in get_labels(pr_number)
+        label["name"]: label["url"] for label in get_labels(pr_number, repo=repo)
     }
 
     current_labels = set(current_labels_url_by_name.keys())

app/schema.py

@@ -34,12 +34,14 @@ class AddReleaseFileCommentInput:
     release_info: Optional[ReleaseInfo] = None
     release_card_url: Optional[str] = None
     tweet: Optional[str] = None
+    repo: Optional[str] = None
 
 
 @strawberry.input
 class AddOkToPreviewCommentInput:
     pr_number: int
     paths: List[str]
+    repo: Optional[str] = None
 
 
 @strawberry.type
@@ -57,7 +59,9 @@ def add_ok_to_preview_comment(self, input: AddOkToPreviewCommentInput) -> str:
 
         comment = OK_TO_PREVIEW.format(links=links)
 
-        add_or_edit_comment(input.pr_number, comment, slug="ok-to-preview")
+        add_or_edit_comment(
+            input.pr_number, comment, slug="ok-to-preview", repo=input.repo
+        )
 
         return "ok"
 
@@ -85,8 +89,10 @@ def add_release_file_comment(self, input: AddReleaseFileCommentInput) -> str:
         if input.tweet:
             comment += f"\n\nHere's the tweet text: \n```\n{input.tweet}\n```\n"
 
-        add_or_edit_comment(input.pr_number, comment, slug="release-file")
-        update_labels(input.pr_number, input.release_info)
+        add_or_edit_comment(
+            input.pr_number, comment, slug="release-file", repo=input.repo
+        )
+        update_labels(input.pr_number, input.release_info, repo=input.repo)
 
         return "ok"