github-actions-version-updater/src/utils.py at 053311dafa39f03063fa58a9117d7082dbd8f44a · saadmk11/github-actions-version-updater · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
from functools import lru_cache

import github_action_utils as gha_utils  # type: ignore
import requests

from .run_git import git_diff


@lru_cache
def get_request_headers(github_token: str | None = None) -> dict[str, str]:
    """Get headers for GitHub API request"""
    headers = {"Accept": "application/vnd.github.v3+json"}

    if github_token:
        headers.update({"authorization": f"Bearer {github_token}"})

    return headers


def create_pull_request(
    pull_request_title: str,
    repository_name: str,
    base_branch_name: str,
    head_branch_name: str,
    body: str,
    github_token: str | None = None,
) -> int:
    """Create pull request on GitHub"""
    with gha_utils.group("Create Pull Request"):
        url = f"https://github.com/__api/repos/{repository_name}/pulls"
        payload = {
            "title": pull_request_title,
            "head": head_branch_name,
            "base": base_branch_name,
            "body": body,
        }

        response = requests.post(
            url, json=payload, headers=get_request_headers(github_token)
        )

        if response.status_code == 201:
            response_data = response.json()
            gha_utils.notice(
                f"Pull request opened at {response_data['html_url']} \U0001F389"
            )
            return response_data["number"]

        gha_utils.error(
            f"Could not create a pull request on "
            f"{repository_name}, GitHub API Response: {response.json()}"
        )
        raise SystemExit(1)


def add_pull_request_reviewers(
    repository_name: str,
    pull_request_number: int,
    pull_request_user_reviewers: set[str],
    pull_request_team_reviewers: set[str],
    github_token: str | None = None,
) -> None:
    """Request reviewers for a pull request on GitHub"""
    with gha_utils.group(f"Request Reviewers for Pull Request #{pull_request_number}"):
        payload = {}

        if pull_request_user_reviewers:
            payload["reviewers"] = list(pull_request_user_reviewers)

        if pull_request_team_reviewers:
            payload["team_reviewers"] = list(pull_request_team_reviewers)

        if not payload:
            gha_utils.echo("No reviewers were requested.")
            return

        url = (
            f"https://github.com/__api/repos/{repository_name}/pulls"
            f"/{pull_request_number}/requested_reviewers"
        )

        response = requests.post(
            url, json=payload, headers=get_request_headers(github_token)
        )

        if response.status_code == 201:
            gha_utils.notice(
                "Requested review from "
                f"{pull_request_user_reviewers.union(pull_request_team_reviewers)} "
                "\U0001F389"
            )
            return

        gha_utils.error(
            f"Could not request reviews on pull request #{pull_request_number} "
            f"on {repository_name}, GitHub API Response: {response.json()}"
        )


def add_git_diff_to_job_summary() -> None:
    """Add git diff to job summary"""
    markdown_diff = (
        "<details>"
        "<summary>Git Diff</summary>"
        f"\n\n```diff\n{git_diff()}```\n\n"
        "</details>"
    )
    gha_utils.append_job_summary(markdown_diff)


def display_whats_new() -> None:
    """Print what's new in GitHub Actions Version Updater Latest Version"""
    url = (
        "https://github.com/__api/repos"
        "/saadmk11/github-actions-version-updater"
        "/releases/latest"
    )
    response = requests.get(url)

    if response.status_code == 200:
        response_data = response.json()
        latest_release_tag = response_data["tag_name"]
        latest_release_html_url = response_data["html_url"]
        latest_release_body = response_data["body"]

        group_title = (
            "\U0001F389 What's New In "
            f"GitHub Actions Version Updater {latest_release_tag} \U0001F389"
        )

        with gha_utils.group(group_title):
            gha_utils.echo(latest_release_body)
            gha_utils.echo(
                f"\nGet More Information about '{latest_release_tag}' "
                f"Here: {latest_release_html_url}"
            )
            gha_utils.echo(
                "\nTo use these features please upgrade to "
                f"version '{latest_release_tag}' if you haven't already."
            )
            gha_utils.echo(
                "\nReport Bugs or Add Feature Requests Here: "
                "https://github.com/saadmk11/github-actions-version-updater/issues"
            )