release.py

#!/usr/bin/env python3

from __future__ import print_function
from argparse import RawTextHelpFormatter
from typing import Tuple
import subprocess
import sys
import tempfile
import os
import urllib.parse
import urllib.request
import argparse
import shutil
import venv

USAGE = 'release.py <package> <version> <jenkinstoken>'
try:
    JENKINS_URL = os.environ['JENKINS_URL']
except KeyError:
    JENKINS_URL = 'http://build.osrfoundation.org'
JOB_NAME_PATTERN = '%s-debbuilder'
GENERIC_BREW_PULLREQUEST_JOB = 'generic-release-homebrew_pull_request_updater'

LINUX_DISTROS = ['ubuntu', 'debian']
SUPPORTED_ARCHS = ['amd64', 'armhf', 'arm64']
RELEASEPY_NO_ARCH_PREFIX = '.releasepy_NO_ARCH_'
ROS_VENDOR = {'harmonic': 'jazzy', 'ionic': 'rolling'}

OSRF_REPOS_SUPPORTED = "stable prerelease nightly testing none"

DRY_RUN = False
NIGHTLY = False
PRERELEASE = False

IGNORE_DRY_RUN = True


class ErrorNoPermsRepo(Exception):
    pass


class ErrorNoUsernameSupplied(Exception):
    pass


class ErrorURLNotFound404(Exception):
    pass


class ErrorNoOutput(Exception):
    pass


class ErrorAlreadyExists(Exception):
    pass


def error(msg):
    print("\n !! " + msg + "\n")
    sys.exit(1)


def print_success(msg):
    print(" + OK " + msg)


def print_only_dbg(msg):
    if '_RELEASEPY_DEBUG' in os.environ:
        print(msg)


# Remove the last character if it is a number.
# That should leave just the package name instead of packageVersion
# I.E gazebo5 -> gazebo
def get_canonical_package_name(pkg_name):
    return pkg_name.rstrip('1234567890')


def replace_ignition_gz(pkg_name):
    return pkg_name \
        .replace('ignition-gazebo', 'gz-sim') \
        .replace('ignition-', 'gz-')


def github_repo_exists(url):
    try:
        check_call(['git', 'ls-remote', '-q', '--exit-code', url], IGNORE_DRY_RUN)
    except (ErrorURLNotFound404, ErrorNoOutput):
        return False
    except Exception as e:
        error("Unexpected problem checking for git repo: " + str(e))
    return True


def exists_main_branch(github_url):
    check_main_cmd = ['git', 'ls-remote', '--exit-code', '--heads', github_url, 'main']
    try:
        if (check_call(check_main_cmd, IGNORE_DRY_RUN)):
            return True
    except Exception:
        return False


def parse_args(argv):
    global DRY_RUN
    global NIGHTLY
    global PRERELEASE

    parser = argparse.ArgumentParser(formatter_class=RawTextHelpFormatter,
    description="""
Script to handle the release process for the Gazebo devs.
Examples:
A) Generate source: local repository tag + call source job:
   $ release.py <package> <version> <jenkins_token>
   (auto calculate source-repo-uri from local directory)

B) Call builders: reuse existing tarball version + call build jobs:
   $ release.py --source-tarball-uri <URL> <package> <version> <jenkins_token>
   (no call to source job, directly build jobs with tarball URL)

C) Nightly builds (linux)
   $ release.py --source-repo-existing-ref <git_branch> --upload-to-repo nightly <URL> <package> <version> <jenkins_token>
 """)
    parser.add_argument('package', help='which package to release')
    parser.add_argument('version', help='which version to release')
    parser.add_argument('jenkins_token', help='secret token to allow access to Jenkins to start builds')
    parser.add_argument('--dry-run', dest='dry_run', action='store_true', default=False,
                        help='dry-run; i.e., do actually run any of the commands')
    parser.add_argument('-a', '--package-alias', dest='package_alias',
                        default=None,
                        help='different name that we are releasing under')
    parser.add_argument('-b', '--release-repo-branch', dest='release_repo_branch',
                        default='master',
                        help='which version of the accompanying -release repo to use (if not default)')
    parser.add_argument('-r', '--release-version', dest='release_version',
                        default=None,
                        help='Release version suffix; usually 1 (e.g., 1')
    parser.add_argument('--no-sanity-checks', dest='no_sanity_checks', action='store_true', default=False,
                        help='no-sanity-checks; i.e. skip sanity checks commands')
    parser.add_argument('--source-repo-uri',
                        dest='source_repo_uri',
                        default=None,
                        help='Indicate the repository URL to grab the source from (overriding the heristics to calculate it from the local directory)')  # NOQA
    parser.add_argument('--source-repo-existing-ref',
                        dest='source_repo_ref',
                        default=None,
                        help='Optionally, when using --source-repo-uri, indicate the Git reference (branch|tag) to grab the release sources from.\
                              If used: avoid to tag the local repository. If not used: tag the local repository with <version> and use it as ref')  # NOQA
    parser.add_argument('--source-tarball-uri',
                        dest='source_tarball_uri', default=None,
                        help='Indicate the URL of the sources to grab the release sources from.')  # NOQA
    parser.add_argument('--upload-to-repo', dest='upload_to_repository', default="stable",
                        help='OSRF repo to upload: stable | prerelease | nightly')
    parser.add_argument('--extra-osrf-repo', dest='extra_repo', default="",
                        help='extra OSRF repository to use in the build')
    parser.add_argument('--nightly-src-branch', dest='nightly_branch', default="main",
                        help='branch in the source code repository to build the nightly from')
    parser.add_argument('--only-bump-revision-linux', dest='bump_rev_linux_only',
                        action='store_true', default=False,
                        help='Bump only revision number. Do not upload new tarball.')
    parser.add_argument('--only-bump-ros-vendor-package', dest='bump_ros_vendor_only',
                        action='store_true', default=False,
                        help='Only process the ROS vendor package (if any).')

    args = parser.parse_args()

    args.package_alias = args.package
    if args.package.startswith('ign-'):
        args.package_alias = args.package.replace('ign-', 'ignition-')

    DRY_RUN = args.dry_run
    if args.upload_to_repository == 'nightly':
        NIGHTLY = True
    if args.upload_to_repository == 'prerelease':
        PRERELEASE = True

    return args


def get_release_repository_info(package):
    github_url = "https://github.com/gazebo-release/" + package + "-release"
    if (github_repo_exists(github_url)):
        return 'git', github_url

    error("release repository not found in github.com/gazebo-release")


def download_release_repository(package, release_branch):
    try:
        return os.environ['_RELEASEPY_TEST_RELEASE_REPO'], 'main'
    except KeyError:
        vcs, url = get_release_repository_info(package)
        release_tmp_dir = tempfile.mkdtemp()

        if vcs == "git" and release_branch == "default":
            release_branch = "master"

        # If main branch exists, prefer it over master
        if release_branch == "master":
            if exists_main_branch(url):
                release_branch = 'main'

        cmd = [vcs, "clone", "-b", release_branch, url, release_tmp_dir]
        check_call(cmd, IGNORE_DRY_RUN)
        return release_tmp_dir, release_branch


def sanity_package_name_underscore(package, package_alias):
    # Alias is never empty. It hosts a exect copy of package if not provided
    if '_' in package_alias and package_alias != package:
        error("Found an underscore in package_alias. It will conflict with debian package names. May be fixed changing the underscore for a dash.")

    if '_' in package and package_alias == package:
        error("Found an underscore in package name without providing a package alias (-a <alias>). You probably want to match the package name in the debian changelog")

    print_success("No underscore in package name")


def sanity_package_name(repo_dir, package, package_alias):
    expected_name = package

    if package_alias:
        expected_name = package_alias

    # Use igntiion for Citadel and Fortress, gz for Garden and beyond
    gz_name = expected_name.replace("ignition", "gz")
    gz_name = gz_name.replace("gazebo", "sim")

    cmd = ["find", repo_dir, "-name", "changelog", "-exec", "head", "-n", "1", "{}", ";"]
    out, _ = check_call(cmd, IGNORE_DRY_RUN)
    for line in out.decode().split('\n'):
        if not line:
            continue
        # Check that first word is the package alias or name
        if line.partition(' ')[0] != expected_name and line.partition(' ')[0] != gz_name:
            error("Error in changelog package name or alias: " + line)

    cmd = ["find", repo_dir, "-name", "control", "-exec", "grep", "-H", "Source:", "{}", ";"]
    out, _ = check_call(cmd, IGNORE_DRY_RUN)
    for line in out.decode().split('\n'):
        if not line:
            continue
        # Check that first word is the package alias or name
        if line.partition(' ')[2] != expected_name and line.partition(' ')[2] != gz_name:
            error("Error in source package. File:  " + line.partition(' ')[1] + ". Got " + line.partition(' ')[2] + " expected " + expected_name + " or " + gz_name)

    print_success("Package names in changelog and control")


def sanity_package_version(repo_dir, version, release_version):
    cmd = ["find", repo_dir, "-name", "changelog", "-exec", "head", "-n", "1", "{}", ";"]
    out, _ = check_call(cmd, IGNORE_DRY_RUN)
    for line in out.decode().split('\n'):
        if not line:
            continue
        # return full version in brackets
        full_version = line.split(' ')[1]
        # get only version (not release) in brackets
        c_version = full_version[full_version.find("(")+1:full_version.find("-")]
        c_revision = full_version[full_version.find("-")+1:full_version.rfind("~")]

        if c_version != version:
            error("Error in package version. Repo version: " + c_version + " Provided version: " + version)

        if c_revision != release_version:
            error("Error in package release version. Expected " + release_version + " in line " + full_version)

    print_success("Package versions in changelog")
    print_success("Package release versions in changelog")


def sanity_check_sdformat_versions(package, version):
    if package == 'sdformat':
        if int(version[0]) > 1:
            error("Error is sdformat version. Please use 'sdformatX' (with version number) for package name")
        else:
            return

    print_success("sdformat version in proper sdformat package")


def sanity_check_repo_name(repo_name):
    if repo_name in OSRF_REPOS_SUPPORTED:
        return

    error(f"Upload repo value: {repo_name} is not valid. \
            Supported values are: {OSRF_REPOS_SUPPORTED}")


def sanity_project_package_in_stable(version, repo_name):
    if repo_name != 'stable':
        return

    if '+' in version:
        error("Detected stable repo upload using project versioning scheme (include '+' in the version)")

    if '~pre' in version:
        error("Detected stable repo upload using project versioning scheme (include '~pre' in the version)")

    return


def sanity_check_source_repo_uri(source_repo_uri):
    # Check if the scheme is "https" and the path ends with ".git"
    parsed_uri = urllib.parse.urlparse(source_repo_uri)
    if not parsed_uri.scheme == "https" or \
       not parsed_uri.path.endswith(".git"):
        error("--source-repo-uri parameter should start with https:// and end with .git")
    # Needs to be fully in sync for SOURCE_REPO_URI values in
    # OSRFSourceCreation
    # https://github.com/gazebo-tooling/release-tools/blob/master/jenkins-scripts/dsl/gazebo_libs.dsl#L513
    if parsed_uri.netloc.startswith("github.org"):
        error("--source-repo-uri needs github.com instead of github.org")


def sanity_check_bump_linux(source_tarball_uri):
    if (not source_tarball_uri):
        error('--only-bump-revision-linux needs --source-tarball-uri argument'
              'to call builders and not source generation')


def sanity_checks(args, repo_dir):
    print("Safety checks:")
    sanity_package_name_underscore(args.package, args.package_alias)
    sanity_package_name(repo_dir, args.package, args.package_alias)
    sanity_check_repo_name(args.upload_to_repository)

    if (args.bump_rev_linux_only):
        sanity_check_bump_linux(args.source_tarball_uri)

    if args.source_repo_uri:
        sanity_check_source_repo_uri(args.source_repo_uri)

    if not NIGHTLY:
        sanity_package_version(repo_dir, args.version, str(args.release_version))
        sanity_check_sdformat_versions(args.package, args.version)
        sanity_project_package_in_stable(args.version, args.upload_to_repository)

    shutil.rmtree(repo_dir)


def get_exclusion_arches(files):
    r = []
    for f in files:
        if f.startswith(RELEASEPY_NO_ARCH_PREFIX):
            arch = f.replace(RELEASEPY_NO_ARCH_PREFIX, '').lower()
            r.append(arch)

    return r


def discover_distros(repo_dir):
    if not os.path.isdir(repo_dir):
        return None

    _, subdirs, files = os.walk(repo_dir).__next__()
    repo_arch_exclusion = get_exclusion_arches(files)

    if '.git' in subdirs:
        subdirs.remove('.git')
    # remove ubuntu (common stuff) and debian (supported distro at top level)
    if 'ubuntu' in subdirs:
        subdirs.remove('ubuntu')
    if 'debian' in subdirs:
        subdirs.remove('debian')
    # Some releasing methods use patches/ in root
    if 'patches' in subdirs:
        subdirs.remove('patches')

    if not subdirs:
        error('Can not find distributions directories in the -release repo')

    distro_arch_list = {}
    for d in subdirs:
        files = os.walk(repo_dir + '/' + d).__next__()[2]
        distro_arch_exclusion = get_exclusion_arches(files)
        excluded_arches = distro_arch_exclusion + repo_arch_exclusion
        arches_supported = [x for x in SUPPORTED_ARCHS if x not in excluded_arches]
        distro_arch_list[d] = arches_supported

    print('Linux distributions in the -release repository:')
    for distro in distro_arch_list:
        print(f" + {distro}  {*distro_arch_list[distro],}")

    return distro_arch_list


def check_call(cmd, ignore_dry_run=False, cwd=None):
    if DRY_RUN and not ignore_dry_run:
        print_only_dbg('Dry-run running:\n  %s\n' % (' '.join(cmd)))
        return b'', b''
    else:
        print_only_dbg('Running:\n  %s' % (' '.join(cmd)))
        po = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd)
        out, err = po.communicate()
        if po.returncode != 0:
            # bitbucket for the first one, github for the second
            if (b"404" in err) or (b"Repository not found" in err):
                raise ErrorURLNotFound404()
            if b"Permission denied" in out:
                raise ErrorNoPermsRepo()
            if b"abort: no username supplied" in err:
                raise ErrorNoUsernameSupplied()
            if b"already exists:" in err:
                raise ErrorAlreadyExists()
            if not out and not err:
                # assume that call is only for getting return code
                raise ErrorNoOutput()

            # Unkown exception
            print('Error running command (%s).' % (' '.join(cmd)))
            print('stdout: %s' % (out.decode()))
            print('stderr: %s' % (err.decode()))
            raise Exception('subprocess call failed')
        return out, err


def tag_repo(args):
    try:
        # tilde is not a valid character in git
        tag = '%s_%s' % (args.package_alias, args.version.replace('~', '-'))
        check_call(['git', 'tag', '-f', tag])
        check_call(['git', 'push', '--tags'])
    except ErrorNoPermsRepo:
        print('The Git server reports problems with permissions')
        print('The branch could be blocked by configuration if you do not have')
        print('rights to push code in default branch.')
        sys.exit(1)
    except ErrorNoUsernameSupplied:
        print('git tag could not be committed because you have not configured')
        print('your username. Use "git config --username" to set your username.')
        sys.exit(1)
    except Exception:
        print('There was a problem with pushing tags to the git repository')
        print('Do you have write perms in the repository?')
        sys.exit(1)

    return tag


def generate_source_repository_uri(args):
    org_repo = f"gazebosim/{get_canonical_package_name(args.package_alias)}"
    out, err = check_call(['git', 'ls-remote', '--get-url', 'origin'],
                          IGNORE_DRY_RUN)
    if err:
        print(f"An error happened running git ls-remote: {err}")
        sys.exit(1)

    git_remote = out.decode().split('\n')[0]
    if org_repo not in git_remote:
        # Handle the special case for citadel ignition repositories
        if replace_ignition_gz(org_repo) not in git_remote:
            print(f""" !! Automatic calculation of the source repository URI\
                  failed with different information:\
                  \n   * git remote origin in the local direcotry is: {git_remote}\
                  \n   * Package name generated org/repo: {org_repo}\
                  \n >> Please use --source-repo-uri parameter""")
            sys.exit(1)
        else:
            org_repo = replace_ignition_gz(org_repo)
            print(' ~ Ignition found in generated org/repo assuming gz repo: '
                  + org_repo)

    # Always use github.com
    return f"https://github.com/{org_repo}.git"  # NOQA


def generate_source_params(args):
    params = {}
    # 1. NIGHTLY (launch builders):
    #     Pass the nightly branch if NIGHTLY enabled
    # 2. Launch builders
    #     Using args.source_tarball_uri if present
    # 3. Launch source jobs (SOURCE_REPO_URI)
    #     1.1 using args.source_repo_uri (if it was passed)
    #     1.2 autogenerating it

    if NIGHTLY:
        params['SOURCE_TARBALL_URI'] = args.nightly_branch
    elif args.source_tarball_uri:
        params['SOURCE_TARBALL_URI'] = args.source_tarball_uri
    else:
        params['SOURCE_REPO_URI'] = \
            args.source_repo_uri if args.source_repo_uri else \
            generate_source_repository_uri(args)

    return params


def call_jenkins_build(job_name, params, output_string,
                       search_description_help):
    # Only to help user feedback this block
    help_url = f'{JENKINS_URL}/job/{job_name}'
    if search_description_help:
        search_param = urllib.parse.urlencode(
                {'search': search_description_help})
        help_url += f'?{search_param}'
    print(f" + Releasing {output_string} in {help_url}")
    # Real action happen here
    params_query = urllib.parse.urlencode(params)
    url = '%s/job/%s/buildWithParameters?%s' % (JENKINS_URL,
                                                job_name,
                                                params_query)
    print_only_dbg(f" -- {output_string}: {url}")
    if not DRY_RUN:
        urllib.request.urlopen(url)


def display_help_job_chain_for_source_calls(args):
    # Encode the different ways using in the job descriptions to filter builds
    #   - "package version" in repository_uploader_packages
    #   - "packages/version-rev" in _releasepy
    url_search_params = urllib.parse.urlencode(
        {'search':
            f'{args.package_alias} {args.version}'})
    pkgs_upload_check_url = \
        f'{JENKINS_URL}/job/repository_uploader_packages/?{url_search_params}'
    rel_search_params = urllib.parse.urlencode(
        {'search':
            f'{args.package}/{args.version}-{args.release_version}'})
    releasepy_check_url = \
        f'{JENKINS_URL}/job/_releasepy/?{rel_search_params}'
    print('\tINFO: After the source job finished, the release process will trigger:\n'
          '\t  * Source upload:'
          f'{pkgs_upload_check_url}\n'
          '\t  * Builders using release.py --source-tarball-uri:'
          f'{releasepy_check_url}')


def get_collections_for_package(package_name, version) -> list:
    script_directory = os.path.dirname(os.path.abspath(sys.argv[0]))
    helper_script = f'{script_directory}/jenkins-scripts/dsl/tools/get_collections_from_package_and_version.py'
    collection_yaml = f'{script_directory}/jenkins-scripts/dsl/gz-collections.yaml'
    cmd = [helper_script,
           get_canonical_package_name(package_name),
           version,
           collection_yaml]
    try:
        _out, _err = check_call(cmd, IGNORE_DRY_RUN)
    except ErrorNoOutput:
        # no output is a valid result
        _out = b""
        _err = ""
    else:
        if _err:
            print(f"An error happened running get_collections_from_package_and_version: {_err}")
            sys.exit(1)

    collection_list = _out.decode().strip().split(' ')
    return collection_list


def get_vendor_github_repo(package_name) -> str:
    canonical_name = get_canonical_package_name(package_name)
    return f"gazebo-release/{canonical_name.replace('-', '_')}_vendor"


def get_vendor_repo_url(package_name) -> str:
    # Clone needs ssh for real pushing operations. In simulation prefer https to avoid
    # unexpected pushes and facilitate testing
    protocol = 'https://github.com/' if DRY_RUN else 'ssh://git@github.com:'
    return f"{protocol}{get_vendor_github_repo(package_name)}"


def prepare_vendor_pr_temp_workspace(package_name, ws_dir) -> Tuple[str, str, str]:
    gz_vendor_tool = os.path.join(ws_dir, "gz_vendor")
    # Create virtualenv for vendor dependencies
    venv_dir = os.path.join(ws_dir, "venv")
    venv.create(venv_dir, system_site_packages=True, with_pip=True)
    subprocess.run([os.path.join(venv_dir, 'bin', 'pip3'), 'install', '-q',
                    'jinja2==3.1.2',
                    'catkin_pkg==1.0.0'])
    cmd = ['git', 'clone', '-q',
           'https://github.com/gazebo-tooling/gz_vendor/',
           gz_vendor_tool]
    _, _err_tool = check_call(cmd, IGNORE_DRY_RUN)
    gz_vendor_repo = os.path.join(ws_dir, 'gz_vendor_repo')
    cmd = ['git', 'clone', '-q',
           get_vendor_repo_url(package_name),
           gz_vendor_repo]
    _, _err_repo = check_call(cmd, IGNORE_DRY_RUN)
    if _err_tool or _err_repo:
        print("Problems with cloning vendor and tool repos:")
        print(f"{_err_tool} {_err_repo}")
        sys.exit(1)

    return gz_vendor_tool, gz_vendor_repo, venv_dir


def execute_update_vendor_package_tool(vendor_tool_path,
                                       vendor_repo_path,
                                       vendor_venv) -> None:
    # The source repository when releasing matches the
    src_repo = os.getcwd()
    try:
        src_repo = os.environ['_RELEASEPY_TEST_SOURCE_REPO']
    except KeyError:
        pass

    run_cmd = [os.path.join(vendor_venv, 'bin', 'python3'),
               f"{vendor_tool_path}/create_gz_vendor_pkg/create_vendor_package.py",
               f"{os.path.join(src_repo, 'package.xml')}",
               '--output_dir', vendor_repo_path]
    _, _err_run = check_call(run_cmd, IGNORE_DRY_RUN)
    if _err_run:
        print("Problems running the create_vendor_package.py script:")
        print(_err_run.decode())
        sys.exit(1)


def create_pr_for_vendor_package(args, repo_path, base_branch) -> str:
    cmd_diff = ['git', "-C", repo_path, 'diff']
    _out, _ = check_call(cmd_diff, IGNORE_DRY_RUN)
    if not _out.decode():
        return 'vendor tool did not produce any change, avoid the PR'

    branch_name = f'releasepy/{args.version}'
    vendor_repo = get_vendor_repo_url(args.package)
    branch_cmd = ['git', "-C", repo_path,
                  'checkout', '-b',  branch_name]
    _, _ = check_call(branch_cmd, IGNORE_DRY_RUN)
    commit_cmd = ['git', "-C", repo_path,
                  'commit',
                  '-m', f'Bump version to {args.version}',
                  '--all']
    _, _ = check_call(commit_cmd)
    push_cmd = ['git', "-C", repo_path,
                'push', '--force',
                vendor_repo, branch_name]
    _, _ = check_call(push_cmd)
    pr_cmd = ['gh', 'pr', 'create',
              '--base', base_branch,
              '--head', branch_name,
              '--title', f'Bump version to {args.version}',
              '--body', 'PR automatically created by release.py']
    try:
        _out, _err = check_call(pr_cmd, cwd=repo_path)
    except ErrorAlreadyExists:
        return f'there is already a PR for the branch: {branch_name} .'\
                'Please check it out manuallly.'

    if _err:
        print("Problems creating the PR for the vendor package:")
        print(_err.decode())
        sys.exit(1)

    if DRY_RUN:
        return ' (skipped the creation on --dry-run)'

    return _out.decode()


def create_pr_in_gz_vendor_repo(args, ros_distro) -> str:
    pr_msg = ''
    with tempfile.TemporaryDirectory() as ws_dir:
        ws_dir = tempfile.mkdtemp()
        # Prepare the temporary workspace
        vendor_tool_path, vendor_repo_path, venv_dir = \
            prepare_vendor_pr_temp_workspace(args.package, ws_dir)
        # Run updating script on the temporary workspace
        execute_update_vendor_package_tool(
            vendor_tool_path, vendor_repo_path, venv_dir)
        # Commits and PR creation
        pr_msg = create_pr_for_vendor_package(
            args, vendor_repo_path, ros_distro)

    return pr_msg


def process_ros_vendor_package(args):
    print("ROS vendor packages that can be updated:")
    if  args.package.replace('gz-','') in ROS_VENDOR:
        print(" - There are no gz metapackages in ROS")
        return
    for collection in get_collections_for_package(args.package,
                                                  args.version):
        if collection in ROS_VENDOR:
            ros_distro = ROS_VENDOR[collection]
            print(f" * Github {get_vendor_github_repo(args.package)} "
                  f"part of {collection} in ROS 2 {ros_distro}")
            print("   + Preparing a PR: ", end='', flush=True)
            pr_url = create_pr_in_gz_vendor_repo(args, ros_distro)
            print(pr_url)


def go(argv):
    args = parse_args(argv)

    # If only the process of ROS vendor package is set, just do it
    if args.bump_ros_vendor_only:
        process_ros_vendor_package(args)
        sys.exit(0)

    # Default to release 1 if not present
    if not args.release_version:
        args.release_version = 1

    package_alias_force_gz = replace_ignition_gz(args.package_alias)

    print(f"Downloading releasing info for {args.package}")
    # Sanity checks and dicover supported distributions before proceed.
    repo_dir, args.release_repo_branch = download_release_repository(args.package, args.release_repo_branch)
    # The supported distros are the ones in the top level of -release repo
    ubuntu_distros = discover_distros(repo_dir)  # top level, ubuntu
    debian_distros = discover_distros(repo_dir + '/debian/')  # debian dir top level, Debian
    if not args.no_sanity_checks:
        sanity_checks(args, repo_dir)

    params = generate_source_params(args)
    params['token'] = args.jenkins_token
    params['PACKAGE'] = args.package
    params['VERSION'] = args.version if not NIGHTLY else 'nightly'
    params['RELEASE_REPO_BRANCH'] = args.release_repo_branch
    params['PACKAGE_ALIAS'] = args.package_alias
    params['RELEASE_VERSION'] = args.release_version
    params['UPLOAD_TO_REPO'] = args.upload_to_repository
    # Assume that we want stable + own repo in the building
    params['OSRF_REPOS_TO_USE'] = "stable " + args.upload_to_repository
    if args.extra_repo:
        params['OSRF_REPOS_TO_USE'] += " " + args.extra_repo

    if args.dry_run:
        print("Simulation of jobs to be called if not dry-run:")
    else:
        print("Triggering release jobs:")
    # a) Mode nightly or builders:
    if NIGHTLY or args.source_tarball_uri:
        # RELEASING FOR BREW
        if not NIGHTLY and not args.bump_rev_linux_only:
            call_jenkins_build(GENERIC_BREW_PULLREQUEST_JOB,
                               params, 'Brew',
                               f'{args.package_alias}-{args.version}')
        # RELEASING FOR LINUX
        for l in LINUX_DISTROS:
            if (l == 'ubuntu'):
                distros_dic = ubuntu_distros
            elif (l == 'debian'):
                if (PRERELEASE or NIGHTLY):
                    continue
                if not debian_distros:
                    continue
                distros_dic = debian_distros
            else:
                error("Distro not supported in code")

            for d in distros_dic:
                for a in distros_dic[d]:
                    # Filter prerelease and nightly architectures
                    if (PRERELEASE or NIGHTLY):
                        if (a == 'armhf' or a == 'arm64'):
                            continue

                    linux_platform_params = params.copy()
                    linux_platform_params['ARCH'] = a
                    linux_platform_params['LINUX_DISTRO'] = l
                    linux_platform_params['DISTRO'] = d

                    if (a == 'armhf' or a == 'arm64'):
                        # No sid releases for arm64/armhf lack of docker image
                        # https://hub.docker.com/r/aarch64/debian/ fails on Jenkins
                        if (d == 'sid'):
                            continue
                        # Need to use JENKINS_NODE_TAG parameter for large memory nodes
                        # since it runs qemu emulation
                        linux_platform_params['JENKINS_NODE_TAG'] = 'linux-' + a
                    elif ('ignition-physics' in args.package_alias) or \
                         ('gz-physics' in args.package_alias):
                        linux_platform_params['JENKINS_NODE_TAG'] = 'large-memory'

                    # control nightly generation using a single machine to process
                    # all distribution builds to avoid race conditions. Note: this
                    # assumes that large-memory nodes are being used for nightly
                    # tags.
                    # https://github.com/gazebo-tooling/release-tools/issues/644
                    if (NIGHTLY):
                        assert a == 'amd64', f'Nightly tag assumed amd64 but arch is {a}'
                        linux_platform_params['JENKINS_NODE_TAG'] = 'linux-nightly-' + d
                    # TODO: last parameter of providing help for -debbuilders
                    # does not currently work. Somehow the string composed by
                    # "-()" do not work even in the web UI directly. Real
                    # string should be:
                    # f"{args.version}-{args.release_version}({l}/{d}::{a})")
                    call_jenkins_build(f'{package_alias_force_gz}-debbuilder',
                                       linux_platform_params,
                                       f"{l} {d}/{a}",
                                       f"{args.version}-{args.release_version}")
    else:
        # b) Mode generate source
        # Choose platform to run gz-source on. It will need to install gz-cmake
        # Take the first key in the supported distros since all them should be
        # able to install the needed gz-cmake.
        if ubuntu_distros:
            params['LINUX_DISTRO'] = 'ubuntu'
            params['DISTRO'] = list(ubuntu_distros.keys())[0]
        elif debian_distros:
            params['LINUX_DISTRO'] = 'debian'
            params['DISTRO'] = list(debian_distros.keys())[0]
        else:
            error("No distributions where found in the release repo")

        # Tag should not go before any method or step that can fail and just
        # before the calls to the servers.
        if not args.source_repo_ref:
            print(' * INFO: no --source-repo-existing-ref used, tag the local'
                  ' repository as the reference for the source code of the'
                  ' release')

        params['SOURCE_REPO_REF'] = tag_repo(args) \
            if not args.source_repo_ref else args.source_repo_ref

        call_jenkins_build(f'{package_alias_force_gz}-source',
                           params,
                           'Source',
                           args.version)
        display_help_job_chain_for_source_calls(args)
        # Process the possible update of an associated ROS vendor package
        process_ros_vendor_package(args)

if __name__ == '__main__':
    go(sys.argv)