amancevice/dip

"""
Define options for dip commands.
"""
import os
import re

import click


def validate_env(ctx, param, value):
    """ Validate --env option. """
    # pylint: disable=unused-argument
    environment = {}
    for val in value:

        # Export from ENV
        if re.match(r'^[A-Z_]+$', val):
            env = val
            val = os.getenv(val)

        # Export from CLI
        elif re.match(r'^[A-Z_]+=.*$', val):
            env, val = re.split(r'([A-Z_]+)=(.*)', val)[1:-1]

        # Raise Bad usage
        else:
            raise click.BadParameter("Use 'ENV=VALUE', or 'ENV'")
        environment[env] = val
    return environment


def validate_secret(ctx, param, value):
    """ Validate --secret option. """
    # pylint: disable=unused-argument
    for val in value:
        if not re.match(r'^[A-Z_]+$', val):
            raise click.BadParameter("Use 'CAPITALS_AND_UNDERSCORES'")
    return value


def ensure_remote(ctx, param, value):
    """ Ensure --remote is used. """
    # pylint: disable=unused-argument
    if value and ctx.params.get('remote') == (None, None):
        raise click.BadParameter('Invalid without "-r" / "--remote" option')
    return value


def expand_home(ctx, param, value):
    """ Expand home argument to absolute path. """
    # pylint: disable=unused-argument
    if value is not None:
        return os.path.abspath(os.path.expanduser(value))
    return value


def split_remote(ctx, param, value):
    """ Split remote/branch into tuple. """
    # pylint: disable=unused-argument
    if value:
        try:
            remote, branch = value.split('/')
        except ValueError:
            remote, branch = value, None
        return remote, branch
    return value, value


class NameVal(click.types.StringParamType):
    """ Override of the StringParamType. """
    name = 'NAME=VALUE'


class Name(click.types.StringParamType):
    """ Override of the StringParamType. """
    name = 'NAME'


ARGS = click.argument('ARGS', nargs=-1)
HOME = click.argument('HOME', callback=expand_home)
KEYS = click.argument('KEYS', is_eager=True, nargs=-1)
NAME = click.argument('NAME', type=Name())
NAMES = click.argument('NAMES', nargs=-1, type=Name())
AUTO_UPGRADE = click.option('-a', '--auto-upgrade',
                            callback=ensure_remote,
                            help='Auto-upgrade out-of-date remotes',
                            is_flag=True)
DOTENV = click.option('-d', '--dotenv',
                      help='Load ENV variables from file in HOME')
EDIT = click.option('-e', '--edit',
                    help='Edit settings (requires EDITOR to be set in ENV)',
                    is_flag=True)
ENV = click.option('-e', '--env',
                   callback=validate_env,
                   help='Runtime ENV variable',
                   multiple=True,
                   type=NameVal())
FORCE = click.option('-f', '--force',
                     help='Do not prompt',
                     is_flag=True,
                     prompt='Are you sure?')
NO_EXE = click.option('-o', '--no-exe',
                      help='Install without executable',
                      is_flag=True)
QUICK = click.option('-q', '--quick',
                     help='Do not check remote before running',
                     is_flag=True)
QUIET = click.option('-q', '--quiet',
                     help='Do not show output',
                     is_flag=True)
SECRET = click.option('-x', '--secret',
                      callback=validate_secret,
                      help='Set secret ENV',
                      multiple=True,
                      type=Name())
SLEEP = click.option('-s', '--sleep',
                     callback=ensure_remote,
                     help='Number of seconds to sleep when remote differs',
                     type=click.INT)
PATH = click.option('-p', '--path',
                    callback=expand_home,
                    help='Path to write executable')
REMOTE = click.option('-r', '--remote',
                      callback=split_remote,
                      help='Optional git remote/branch',
                      is_eager=True)