harness-deployment

#!/usr/bin/env python
import sys
from pathlib import Path

from cloudharness_utils.constants import DEPLOYMENT_PATH

from ch_cli_tools.api import deploy, hosts_info, perform_migration
from ch_cli_tools.model import CHDeployConfig, CHProject
from ch_cli_tools.utils import generation_argv

HERE = Path(__file__).resolve().parent
ROOT = HERE.parent

if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser(
        description="Walk filesystem inside ./applications and define build and deploy scripts."
    )

    parser.add_argument(
        "paths",
        metavar="paths",
        default=[ROOT],
        type=Path,
        nargs="*",
        help="Base paths to start looking for applications. The paths will be processed "
        "in the given order so determining the build and overriding precedence.",
    )

    parser.add_argument(
        "-i",
        "--include",
        dest="include",
        action="append",
        default=[],
        help="Specify the applications to include and exclude the rest. "
        "Omit to build images for all Docker files.",
    )

    parser.add_argument(
        "-t",
        "--tag",
        dest="tag",
        action="store",
        default=None,
        help="Add a tag with the current version (default no value, content hash is used)",
    )

    parser.add_argument(
        "-n",
        "--namespace",
        dest="namespace",
        action="store",
        default=None,
        help="Specify the namespace of the deployment (default taken from values.yaml)",
    )
    parser.add_argument(
        "--name",
        dest="name",
        action="store",
        default=None,
        help="Specify the helm chart name (default: namespace when provided, otherwise chart default)",
    )
    parser.add_argument(
        "--chart-version",
        dest="chart_version",
        action="store",
        default=None,
        help="Specify the helm chart version",
    )
    parser.add_argument(
        "--app-version",
        dest="app_version",
        action="store",
        default=None,
        help="Specify the helm chart appVersion",
    )

    parser.add_argument(
        "-r",
        "--registry",
        dest="registry",
        action="store",
        default="",
        help="Specify image registry prefix",
    )
    parser.add_argument(
        "-rsn",
        "-rs",
        "--registry-secret-name",
        dest="registry_secret_name",
        action="store",
        default="",
        help="Specify image registry secret (-rs is deprecated, use -rsn)",
    )
    parser.add_argument(
        "-o",
        "--output",
        dest="output_path",
        action="store",
        default="./deployment",
        help="Specify helm chart base path (default `./deployment)`",
    )
    parser.add_argument(
        "-d",
        "--domain",
        dest="domain",
        action="store",
        default="cloudharness.metacell.us",
        help="Specify a domain different from cloudharness.metacell.us",
    )
    parser.add_argument(
        "-l",
        "--local",
        dest="local",
        action="store_true",
        help="Specify for local deployments info and setup",
    )
    parser.add_argument(
        "-u",
        "--disable-security",
        dest="unsecured",
        action="store_true",
        help="Disable secured gatekeepers access",
    )
    parser.add_argument(
        "-ex",
        "--exclude",
        dest="exclude",
        action="append",
        default=[],
        help="Specify application to exclude from the deployment",
    )
    parser.add_argument(
        "-e",
        "--env",
        dest="env",
        action="store",
        default="",
        help="Specify the name of the environment(s) (will load values-[ENV].yaml files). Specify multiple envs with dashes (e.g env1-env2)",
    )

    parser.add_argument(
        "-dtls",
        "--disable-tls",
        dest="no_tls",
        action="store_true",
        help="Disable tls (serve on http)",
    )
    parser.add_argument(
        "--deploy", dest="deploy", action="store_true", help="Deploy helm chart"
    )
    parser.add_argument(
        "-N",
        "--no-cd",
        dest="no_cd_gen",
        action="store_const",
        default=None,
        const=True,
        help=f"Do not generate ci/cd files",
    )
    parser.add_argument(
        "-we",
        "--write-env",
        dest="write_env",
        action="store_const",
        default=None,
        const=True,
        help=f"Write build env to .env file in {DEPLOYMENT_PATH}",
    )
    parser.add_argument(
        "--migrate",
        dest="migration",
        action="store_true",
        help="Perform a migration of the deployment to the latest supported version.",
    )
    parser.add_argument(
        "--migrate-accept-all",
        dest="migration_accept_all",
        action="store_true",
        help="Perform a migration of the deployment to the latest supported version without asking for confirmation. Use with caution.",
    )
    parser.add_argument(
        "--docker-compose",
        dest="docker_compose",
        action="store_true",
        help="Generate docker-compose.yaml and dedicated Skaffold configuration",
    )
    parser.add_argument(
        "-cu",
        "--cache-url",
        dest="image_cache_url",
        action="store",
        default="",
        help="Specify a custom cache URL for the docker images. This is used to speed up builds by caching layers. Example: https://cache.cloudharness.io?repository=[REGISTRY]/[IMAGE_NAME]&tag=[TAG]",
    )

    args, unknown = parser.parse_known_args(sys.argv[1:])

    root_paths = [Path.cwd() / path for path in args.paths]

    envs = args.env.split("-") if args.env else ()

    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(1)

    if unknown:
        print(
            "There are unknown args. Make sure to call the script with the accepted args. Try --help"
        )
        print(f"unknown: {unknown}")
        sys.exit(2)

    if args.migration_accept_all:
        perform_migration(root_paths, accept_all=True)
    elif args.migration:
        perform_migration(root_paths, accept_all=False)

    generation_argv.set(sys.argv)

    config = CHDeployConfig(
        includes=args.include,
        excludes=args.exclude,
        registry=args.registry,
        tag=args.tag,
        local=args.local,
        namespace=args.namespace,
        registry_secret_name=args.registry_secret_name,
        output_path=args.output_path,
        tls=not args.no_tls,
        secured_gatekeepers=not args.unsecured,
        domain=args.domain,
        env=envs or None,
        chart_name=args.name,
        chart_version=args.chart_version,
        app_version=args.app_version,
    )

    if args.docker_compose:
        config.backend = "compose"

    project = CHProject.init(root_paths, config=config)

    if args.docker_compose:
        project.docker_compose.generate()
    else:
        project.helm_chart.generate()
    helm_values = project.build_final_helm_values()

    if not args.no_cd_gen and envs and not args.docker_compose:
        project.codefresh.generate()

    if args.write_env:
        project.codefresh.write_env_file(
            root_paths[-1] / DEPLOYMENT_PATH / ".env",
            args.image_cache_url,
        )

    project.skaffold.generate()
    project.tilt.generate()

    if not args.docker_compose:
        project.vscode_debug.generate()
        hosts_info(helm_values)

    if args.deploy:
        deploy(args.namespace, args.output_path)