from github import Github
from subprocess import check_output
import shlex
from tqdm import tqdm
import click


@click.command()
@click.option(
    "--access-token",
    required=True,
    help="""
Github Access token that has repo:public_repo and user:read:user permission.

Create them at https://github.com/settings/tokens/new
""",
)
@click.option(
    "--ray-path", required=True, help="File system path to the ray directory")
@click.option(
    "--prev-branch",
    required=True,
    help="Previous version branch like ray-0.7.1")
@click.option(
    "--curr-branch",
    required=True,
    help="Current version branch like ray-0.7.2")
def run(access_token, ray_path, prev_branch, curr_branch):
    # Generate command
    cmd = []
    cmd.append(f"cd {ray_path}")
    cmd.append(f'git log {prev_branch}..{curr_branch} --pretty=format:"%s" '
               ' | grep -Eo "#(\d+)"')
    joined = " && ".join(cmd)
    cmd = f"bash -c '{joined}'"
    cmd = shlex.split(cmd)
    print("Executing", cmd)

    # Sort the PR numbers
    pr_numbers = [
        int(l.lstrip("#")) for l in check_output(cmd).decode().split()
    ]
    print("PR numbers", pr_numbers)

    # Use Github API to fetch the
    g = Github(access_token)
    ray_repo = g.get_repo("ray-project/ray")
    logins = set()
    for num in tqdm(pr_numbers):
        try:
            logins.add(ray_repo.get_pull(num).user.login)
        except Exception as e:
            print(e)

    print()
    print("Here's the list of contributors")
    print("=" * 10)
    print()
    print("@" + ", @".join(logins))
    print()
    print("=" * 10)


if __name__ == "__main__":
    run()