Merge pull request #395 from gitautoai/wes

Clean up all previous GitAuto comments when starting except its trigger comment

constants/messages.py

@@ -0,0 +1 @@
+CLICK_THE_CHECKBOX = "Click the checkbox below to generate a PR!"

services/check_run_handler.py

@@ -25,7 +25,7 @@
     Repository,
 )
 from services.github.pulls_manager import get_pull_request, get_pull_request_files
-from services.github.utilities import create_permission_url
+from services.github.github_utils import create_permission_url
 from services.openai.commit_changes import chat_with_agent
 from services.openai.chat import chat_with_ai
 from services.openai.instructions.identify_cause import IDENTIFY_CAUSE

services/gitauto_handler.py

@@ -13,6 +13,7 @@
     SUPABASE_SERVICE_ROLE_KEY,
     PR_BODY_STARTS_WITH,
 )
+from services.github.comment_manager import delete_my_comments
 from services.github.github_manager import (
     create_pull_request,
     create_remote_branch,
@@ -25,7 +26,7 @@
     add_reaction_to_issue,
 )
 from services.github.github_types import GitHubLabeledPayload
-from services.github.utilities import deconstruct_github_payload
+from services.github.github_utils import deconstruct_github_payload
 from services.jira.jira_manager import deconstruct_jira_payload
 from services.openai.commit_changes import chat_with_agent
 from services.openai.instructions.write_pr_body import WRITE_PR_BODY
@@ -85,6 +86,9 @@ async def handle_gitauto(
         )
     )
 
+    # Delete all comments made by GitAuto except the one with the checkbox to clean up the issue
+    delete_my_comments(base_args=base_args)
+
     # Notify the user if the request limit is reached and early return
     if requests_left <= 0 and IS_PRD and owner_name not in EXCEPTION_OWNERS:
         body = request_limit_reached(

services/github/comment_manager.py

@@ -0,0 +1,50 @@
+from requests import delete, get
+from config import GITHUB_API_URL, TIMEOUT
+from constants.messages import CLICK_THE_CHECKBOX
+from services.github.create_headers import create_headers
+from services.github.github_types import BaseArgs
+from utils.handle_exceptions import handle_exceptions
+
+
+@handle_exceptions(default_return_value=None, raise_on_error=False)
+def delete_a_comment(base_args: BaseArgs, comment_id: str):
+    """https://docs.github.com/en/rest/issues/comments?apiVersion=2022-11-28#delete-an-issue-comment"""
+    owner, repo, token = (base_args["owner"], base_args["repo"], base_args["token"])
+    url = f"{GITHUB_API_URL}/repos/{owner}/{repo}/issues/comments/{comment_id}"
+    headers: dict[str, str] = create_headers(token=token)
+    response = delete(url=url, headers=headers, timeout=TIMEOUT)
+    response.raise_for_status()
+
+
+@handle_exceptions(default_return_value=None, raise_on_error=False)
+def get_all_comments(base_args: BaseArgs):
+    """https://docs.github.com/en/rest/issues/comments?apiVersion=2022-11-28#list-issue-comments"""
+    owner, repo, issue_number, token = (
+        base_args["owner"],
+        base_args["repo"],
+        base_args["issue_number"],
+        base_args["token"],
+    )
+    url = f"{GITHUB_API_URL}/repos/{owner}/{repo}/issues/{issue_number}/comments"
+    headers: dict[str, str] = create_headers(token=token)
+    response = get(url=url, headers=headers, timeout=TIMEOUT)
+    response.raise_for_status()
+    comments: list[dict] = response.json()
+    # print(f"All comments: {dumps(obj=comments, indent=2)}")
+    return comments
+
+
+def filter_my_comments(comments: list[dict]):
+    """Filter out comments made by GitAuto except the one with the checkbox"""
+    return [
+        comment for comment in comments if CLICK_THE_CHECKBOX not in comment["body"]
+    ]
+
+
+def delete_my_comments(base_args: BaseArgs):
+    """Delete all comments made by GitAuto except the one with the checkbox"""
+    comments = get_all_comments(base_args)
+    my_comments = filter_my_comments(comments)
+    # print(f"My comments: {dumps(obj=my_comments, indent=2)}")
+    for comment in my_comments:
+        delete_a_comment(base_args=base_args, comment_id=comment["id"])

services/github/github_manager.py

@@ -38,6 +38,7 @@
     SUPABASE_SERVICE_ROLE_KEY,
     UTF8,
 )
+from constants.messages import CLICK_THE_CHECKBOX
 from services.github.create_headers import create_headers
 from services.github.github_types import (
     BaseArgs,
@@ -269,7 +270,7 @@ def create_comment_on_issue_with_gitauto_button(payload: GitHubLabeledPayload) -
         )
     )
 
-    body = "Click the checkbox below to generate a PR!\n- [ ] Generate PR"
+    body = f"{CLICK_THE_CHECKBOX}\n- [ ] Generate PR"
     if PRODUCT_ID != "gitauto":
         body += " - " + PRODUCT_ID
 

services/github/utilities.py → services/github/github_utils.py

@@ -14,7 +14,7 @@
     get_installation_access_token,
     get_user_public_email,
 )
-from utils.extract_urls import extract_urls
+from utils.extract_urls import extract_image_urls, extract_urls
 from utils.handle_exceptions import handle_exceptions
 
 
@@ -65,6 +65,10 @@ def deconstruct_github_payload(payload: GitHubLabeledPayload):
 
     # Extract other information
     github_urls, other_urls = extract_urls(text=issue_body)
+    # print(f"github_urls: {github_urls}")
+    # print(f"other_urls: {other_urls}")
+    image_urls = extract_image_urls(text=issue_body)
+    # print(f"image_urls: {image_urls}")
     installation_id: int = payload["installation"]["id"]
     token: str = get_installation_access_token(installation_id=installation_id)
     sender_email: str = get_user_public_email(username=sender_name, token=token)
@@ -91,6 +95,7 @@ def deconstruct_github_payload(payload: GitHubLabeledPayload):
         "is_automation": is_automation,
         "reviewers": reviewers,
         "github_urls": github_urls,
+        "image_urls": image_urls,
         "other_urls": other_urls,
     }
 

utils/extract_urls.py

@@ -1,7 +1,19 @@
-import re
+from re import findall, match
+from utils.handle_exceptions import handle_exceptions
 
 
-def extract_urls(text) -> tuple[list[str], list[str]]:
+@handle_exceptions(default_return_value=[], raise_on_error=False)
+def extract_image_urls(text: str) -> list[str]:
+    """Extract URLs from img tags in the given text.
+
+    Example 1: ['https://github.com/user-attachments/assets/123']
+    """
+    pattern = r'<img[^>]*src="([^"]*)"[^>]*>'
+    urls: list[str] = findall(pattern, text)
+    return urls
+
+
+def extract_urls(text: str) -> tuple[list[str], list[str]]:
     """
     ?: Matches 0 or 1 occurrence of the preceding expression.
     +: Matches 1 or more occurrences of the preceding expression.
@@ -24,8 +36,8 @@ def extract_urls(text) -> tuple[list[str], list[str]]:
     )
     all_url_pattern = r"https?://[^\s)]+"
 
-    all_urls = re.findall(all_url_pattern, text)
-    github_urls = [url for url in all_urls if re.match(github_pattern, url)]
-    other_urls = [url for url in all_urls if url not in github_urls]
+    all_urls: list[str] = findall(all_url_pattern, text)
+    github_urls: list[str] = [url for url in all_urls if match(github_pattern, url)]
+    other_urls: list[str] = [url for url in all_urls if url not in github_urls]
 
     return github_urls, other_urls