Copy github autolinks from one repo to all repos in an organization

copy_github_autolinks.py

#!/usr/bin/env python3

import argparse
import subprocess
import requests

API_URL = "https://api.github.com"


def get_github_token():
    """Fetch the GitHub token using 'gh auth token'."""
    try:
        token = subprocess.check_output(["gh", "auth", "token"], text=True).strip()
        if not token:
            raise Exception("No token returned by 'gh auth token'")
        return token
    except subprocess.CalledProcessError as e:
        print(
            "Error fetching GitHub token. Ensure you're logged in using 'gh auth login'."
        )
        raise e


def get_repos(org, headers):
    """Fetch all repositories in the organization."""
    url = f"{API_URL}/orgs/{org}/repos"
    repos = []
    params = {"per_page": 100}
    while url:
        response = requests.get(url, headers=headers, params=params)
        response.raise_for_status()
        repos.extend(response.json())
        url = response.links.get("next", {}).get("url")
    return [repo["full_name"] for repo in repos]


def get_autolinks(repo, headers):
    """Fetch all autolinks for a repository."""
    url = f"{API_URL}/repos/{repo}/autolinks"
    response = requests.get(url, headers=headers)
    response.raise_for_status()
    return response.json()


def add_autolink(repo, key_prefix, url_template, headers, dry_run):
    """Add an autolink to a repository."""
    if dry_run:
        print(
            f"[Dry Run] Would add autolink to {repo}: Prefix={key_prefix}, Template={url_template}"
        )
        return

    url = f"{API_URL}/repos/{repo}/autolinks"
    payload = {"key_prefix": key_prefix, "url_template": url_template}
    response = requests.post(url, headers=headers, json=payload)
    if response.status_code == 201:
        print(f"Successfully added autolink to {repo}: Prefix={key_prefix}")
    elif response.status_code == 422:
        print(f"Autolink already exists in {repo}: Prefix={key_prefix}")
    else:
        print(f"Failed to add autolink to {repo}: Prefix={key_prefix}")
        print(f"Error: {response.status_code} - {response.text}")


def main():
    parser = argparse.ArgumentParser(
        description="Copy autolinks from one repo to all repos in an organization."
    )
    parser.add_argument("--org", required=True, help="The GitHub organization name.")
    parser.add_argument(
        "--source-repo",
        required=True,
        help="The source repository containing autolinks (in the given org).",
    )
    parser.add_argument(
        "--dry-run",
        action="store_true",
        help="Simulate the process without making changes.",
    )
    args = parser.parse_args()

    # Fetch GitHub token
    token = get_github_token()
    headers = {
        "Authorization": f"Bearer {token}",
        "Accept": "application/vnd.github+json",
    }

    # Fetch repositories in the organization
    print(f"Fetching repositories for organization: {args.org}...")
    repos = get_repos(args.org, headers)

    # Fetch autolinks from the source repository
    source_repo = f"{args.org}/{args.source_repo}"
    print(f"Fetching autolinks from source repository: {source_repo}...")
    source_autolinks = get_autolinks(source_repo, headers)

    if not source_autolinks:
        print("No autolinks found in the source repository.")
        return

    # Copy autolinks to all repositories
    print("Copying autolinks to all repositories in the organization...")
    for repo in repos:
        print(f"Processing repository: {repo}")
        existing_autolinks = get_autolinks(repo, headers)
        existing_key_prefixes = {
            autolink["key_prefix"] for autolink in existing_autolinks
        }

        for autolink in source_autolinks:
            key_prefix = autolink["key_prefix"]
            url_template = autolink["url_template"]

            if key_prefix in existing_key_prefixes:
                print(
                    f"Autolink with prefix {key_prefix} already exists in {repo}. Skipping."
                )
            else:
                add_autolink(repo, key_prefix, url_template, headers, args.dry_run)

    print("Autolink copying completed.")


if __name__ == "__main__":
    main()