rever.xsh

import os

from xonsh.tools import print_color

from rever.activity import activity


$PROJECT = $GITHUB_REPO = 'libcflib'
$ACTIVITIES = [
    'authors',
    'version_bump',
    'changelog',
    'tag',
    'push_tag',
    'ghrelease',
    'pypi',
    'conda_forge',
    'docker_build',
    'docker_push',
    'deploy_to_gcloud',
]

$VERSION_BUMP_PATTERNS = [
    ($PROJECT+'/__init__.py', r'__version__\s*=.*', "__version__ = '$VERSION'"),
    ('setup.py', r'VERSION\s*=.*', "VERSION = '$VERSION'")
    ]
$CHANGELOG_FILENAME = 'CHANGELOG.rst'
$CHANGELOG_TEMPLATE = 'TEMPLATE.rst'
$PUSH_TAG_REMOTE = 'git@github.com:regro/libcflib.git'

$GITHUB_ORG = 'regro'

# docker config
$DOCKERFILE = 'docker/Dockerfile'
$DOCKERFILE_TAGS = ('condaforge/libcflib:$VERSION',
                    'condaforge/libcflib:latest')

#
# Google Cloud
#
$GCLOUD_PROJECT_ID = 'libcflib'
$GCLOUD_ZONE = 'us-central1-a'
$GCLOUD_CLUSTER = 'libcflib-cluster00'


def _ensure_default_credentials():
    credfile = os.path.join($XDG_CONFIG_HOME, 'gcloud',
                            'application_default_credentials.json')
    if os.path.isfile(credfile):
        print_color('{YELLOW}Found ' + credfile + ' ...{NO_COLOR}')
    else:
        ![gcloud auth application-default login]
    $CLOUDSDK_CONTAINER_USE_APPLICATION_DEFAULT_CREDENTIALS = 'true'
    return credfile


def _ensure_account(n=0):
    if n > 3:
        raise RuntimeError('failed to log in to gcloud')
    account = $(gcloud config get-value account).strip()
    if account == '(unset)' or '@' not in account:
        n += 1
        print(f'gcloud account is {account}, login attempt {n}/3:')
        ![gcloud auth login]
        account = _ensure_account(n+1)
    return account


@activity
def deploy_to_gcloud():
    """Deploys the build docker containter to the google cloud"""
    # make sure we are logged in
    _ensure_default_credentials()
    account = _ensure_account()
    # get cluster credentials
    ![gcloud container clusters get-credentials --account @(account) \
      --zone=$GCLOUD_ZONE --project=$GCLOUD_PROJECT_ID $GCLOUD_CLUSTER]
    # set new image
    ![kubectl set image deployment/libcflib-app libcflib-app=docker.io/condaforge/libcflib:$VERSION]


# Ensure that we have the proper software to perform release
def _ensure_packages():
    clis = [
        ('gcloud', 'google-cloud-sdk'),
        ('kubectl', 'kubernetes'),
        ]
    bad = []
    for cli, package in clis:
        if not !(which @(cli)):
            bad.append((cli, package))
    if bad:
        s = ''
        for cli, package in bad:
            s += f'Could not find {cli}! Try installing:\n  $ conda install {package}'
        print(s)
        raise RuntimeError(s)


_ensure_packages()