run-ci.py

import argparse
import os
import sys
from pathlib import Path
from typing import Optional

import tankerci
import tankerci.bump
import tankerci.conan
import tankerci.git
import tankerci.gitlab
from tankerci.conan import Profile, TankerSource


def prepare(
    tanker_source: TankerSource,
    host_profile: Profile,
    update: bool,
    tanker_ref: Optional[str],
) -> None:
    tanker_deployed_ref = tanker_ref
    if tanker_source == TankerSource.DEPLOYED and not tanker_deployed_ref:
        tanker_deployed_ref = "tanker/latest-stable@"
    tankerci.conan.install_tanker_source(
        tanker_source,
        output_path=Path.cwd() / "conan",
        host_profiles=[host_profile],
        build_profile=tankerci.conan.get_build_profile(),
        tanker_deployed_ref=tanker_deployed_ref,
    )
    tankerci.run("bundle", "install")
    tankerci.run("bundle", "exec", "rake", "tanker_libs")


def build(test: bool):
    tankerci.run("bundle", "install")
    if test:
        tankerci.run("bundle", "exec", "rake", "spec")


def lint() -> None:
    tankerci.run("bundle", "install")
    tankerci.run("bundle", "exec", "rake", "rubocop")


def set_credentials() -> None:
    gem_folder = Path.home() / ".gem"
    gem_folder.mkdir(exist_ok=True)
    api_key = os.environ["GEM_HOST_API_KEY"]
    credential_file = gem_folder / "credentials"
    credential_file.touch(mode=0o600)
    credential_file.write_text(f":rubygems_api_key: {api_key}")


def deploy(version: str) -> None:
    expected_libs = [
        "vendor/tanker/linux-x86_64/libctanker.so",
        "vendor/tanker/darwin-x86_64/libctanker.dylib",
        "vendor/tanker/darwin-aarch64/libctanker.dylib",
    ]
    for lib in expected_libs:
        expected_path = Path(lib)
        if not expected_path.exists():
            sys.exit(f"Error: {expected_path} does not exist!")

    tankerci.bump.bump_files(version)
    set_credentials()

    # NOTE: this commands also re-gerenates the lock as a side-effect since the
    # gemspec has changed - keep this before the git commands
    tankerci.run("bundle", "install")

    # NOTE: `bundle exec rake build` does not like dirty git repos, so make a
    # commit with the new changes first
    tankerci.git.run(Path.cwd(), "add", "--update", ".")
    tankerci.git.run(Path.cwd(), "commit", "--message", f"Bump to {version}")
    tankerci.run("bundle", "exec", "rake", "build")
    tankerci.run("bundle", "exec", "rake", "push")


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--isolate-conan-user-home",
        action="store_true",
        dest="home_isolation",
        default=False,
    )
    parser.add_argument("--remote", default="artifactory")

    subparsers = parser.add_subparsers(title="subcommands", dest="command")

    build_parser = subparsers.add_parser("build")
    build_parser.add_argument("--test", action="store_true")

    prepare_parser = subparsers.add_parser("prepare")
    prepare_parser.add_argument(
        "--use-tanker",
        type=TankerSource,
        default=TankerSource.EDITABLE,
        dest="tanker_source",
    )
    prepare_parser.add_argument(
        "--profile", dest="profiles", nargs="+", type=str, required=True
    )
    prepare_parser.add_argument("--tanker-ref")
    prepare_parser.add_argument(
        "--update",
        action="store_true",
        default=False,
        dest="update",
    )

    download_artifacts_parser = subparsers.add_parser("download-artifacts")
    download_artifacts_parser.add_argument("--project-id", required=True)
    download_artifacts_parser.add_argument("--pipeline-id", required=True)
    download_artifacts_parser.add_argument("--job-name", required=True)

    deploy_parser = subparsers.add_parser("deploy")
    deploy_parser.add_argument("--version", required=True)
    subparsers.add_parser("lint")

    args = parser.parse_args()
    command = args.command

    user_home = None

    if args.home_isolation:
        user_home = Path.cwd() / ".cache" / "conan" / args.remote
        if command == "prepare":
            # Because of GitLab issue https://gitlab.com/gitlab-org/gitlab/-/issues/254323
            # the downstream deploy jobs will be triggered even if upstream has failed
            # By removing the cache we ensure that we do not use a
            # previously built (and potentially broken) release candidate to deploy a binding
            tankerci.conan.run("remove", "tanker/*", "--force")

    ctx = tankerci.conan.ConanContextManager(
        [args.remote],
        conan_home=user_home,
    )
    if command == "build":
        with ctx:
            build(args.test)
    elif command == "prepare":
        profiles = Profile(args.profiles)
        with ctx:
            prepare(
                args.tanker_source,
                profiles,
                args.update,
                args.tanker_ref,
            )
    elif command == "deploy":
        with ctx:
            deploy(args.version)
    elif command == "lint":
        lint()
    elif command == "download-artifacts":
        tankerci.gitlab.download_artifacts(
            project_id=args.project_id,
            pipeline_id=args.pipeline_id,
            job_name=args.job_name,
        )
    else:
        parser.print_help()
        sys.exit(1)


if __name__ == "__main__":
    main()