Support closing mend issues by repository

src/cli.py

@@ -469,6 +469,98 @@ def issues_create(
     click.echo(issue)
 
 
+@issues_cli.command("list")
+@click.option(
+    "-c",
+    "--creator",
+    default="mend-for-github-com[bot]",
+    prompt="Creator username",
+)
+@click.option(
+    "-r",
+    "--repository",
+    prompt="Repository name",
+)
+@click.option(
+    "-t",
+    "--token",
+    prompt=False,
+    type=str,
+    default=None,
+    hide_input=True,
+    confirmation_prompt=False,
+    show_envvar=True,
+)
+@click.option("-o", "--organization", prompt="Organization name", type=str)
+def issues_list(
+    creator: str,
+    repository: str,
+    organization: str,
+    token: str,
+) -> None:
+    """Get issues created by an user on a repository"""
+
+    issues_res = issues.search(
+        creator=creator,
+        repository=repository,
+        organization=organization,
+        token=token,
+    )
+
+    click.echo(issues_res)
+
+
+@issues_cli.command("close_mend")
+@click.option(
+    "-c",
+    "--creator",
+    default="mend-for-github-com[bot]",
+    prompt="Creator username",
+)
+@click.option(
+    "-r",
+    "--repository",
+    prompt="Repository name",
+)
+@click.option(
+    "-t",
+    "--token",
+    prompt=False,
+    type=str,
+    default=None,
+    hide_input=True,
+    confirmation_prompt=False,
+    show_envvar=True,
+)
+@click.option("-o", "--organization", prompt="Organization name", type=str)
+def issues_close_mend(
+    creator: str,
+    repository: str,
+    organization: str,
+    token: str,
+) -> None:
+    """Close all issues created by a specific user on a repository"""
+
+    issues_res = issues.search(
+        creator=creator,
+        repository=repository,
+        organization=organization,
+        token=token,
+    )
+
+    if not issues_res:
+        return
+
+    res = issues.close_issues(
+        issue_numbers=issues_res,
+        repository=repository,
+        organization=organization,
+        token=token,
+    )
+
+    click.echo(f"Closed {res} issues from {creator} on {organization}/{repository}.")
+
+
 ###########
 # Secrets #
 ###########
@@ -601,6 +693,13 @@ def mass_cli() -> None:
     type=click.BOOL,
     prompt="Deploy the Dependency Reviewer?",
 )
+@click.option(
+    "-m",
+    "--mend",
+    type=click.BOOL,
+    default=False,
+    prompt="Close Mend issues?",
+)
 @click.argument("input_repos_list", type=click.File("r"))
 @click.argument("output_csv", type=click.File("a", lazy=True))
 @click.option(
@@ -621,6 +720,7 @@ def mass_deploy(
     dependabot: bool,
     codeql: bool,
     reviewer: bool,
+    mend: bool,
     input_repos_list: Any,
     output_csv: Any,
     organization: str,
@@ -716,6 +816,24 @@ def mass_deploy(
                 organization, token, repo
             )
 
+        if mend:
+            issues_res = issues.search(
+                creator="mend-for-github-com[bot]",
+                repository=repo,
+                organization=organization,
+                token=token,
+            )
+
+            if not issues_res:
+                continue
+
+            issues.close_issues(
+                issue_numbers=issues_res,
+                repository=repo,
+                organization=organization,
+                token=token,
+            )
+
         print(
             f"Done: {actions_res},{secretscanner_res}, {pushprotection_res}, {dependabot_res}, {codeql_res}, {reviewer_res}, {issue_secretscanner_res}, {issue_pushprotection_res}, {issue_dependabot_res}, {issue_codeql_res}"
         )

src/ghas_cli/utils/issues.py

@@ -45,3 +45,79 @@ def create(
         return False
     else:
         return issue.json()["html_url"]
+
+
+def search(
+    creator: str,
+    repository: str,
+    organization: str,
+    token: str,
+) -> List:
+    """List issues of a repository"""
+
+    headers = network.get_github_headers(token)
+
+    params = {"state": "open", "creator": creator, "per_page": 100}
+
+    # Retry if rate-limited
+    i = 0
+    while i < network.RETRIES:
+        issue = requests.get(
+            url=f"https://api.github.com/repos/{organization}/{repository}/issues",
+            params=params,
+            headers=headers,
+        )
+        if issue.status_code == 200:
+            break
+
+        if network.check_rate_limit(issue):
+            time.sleep(network.SLEEP_1_MINUTE)
+
+        i += 1
+
+    if issue.status_code != 200:
+        return False
+
+    issue_list = []
+    for i in issue.json():
+        issue_list.append(i["number"])
+
+    return issue_list
+
+
+def close_issues(
+    issue_numbers: List,
+    repository: str,
+    organization: str,
+    token: str,
+) -> None:
+    """Close a list of issues on a repository"""
+
+    headers = network.get_github_headers(token)
+
+    payload = {"state": "closed", "state_reason": "not_planned"}
+
+    success_count = 0
+    for issue_number in issue_numbers:
+        # Retry if rate-limited
+        i = 0
+        while i < network.RETRIES:
+            issue = requests.patch(
+                url=f"https://api.github.com/repos/{organization}/{repository}/issues/{issue_number}",
+                json=payload,
+                headers=headers,
+            )
+
+            if issue.status_code == 200:
+                success_count += 1
+                break
+
+            if network.check_rate_limit(issue):
+                time.sleep(network.SLEEP_1_MINUTE)
+
+            i += 1
+
+    if issue.status_code != 200:
+        return False
+
+    return success_count