# Copyright 2017 AT&T Intellectual Property.  All other rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Contains commands related to tasks against designs."""
import click
import json
import yaml

from drydock_provisioner.cli.task.actions import TaskList
from drydock_provisioner.cli.task.actions import TaskShow
from drydock_provisioner.cli.task.actions import TaskCreate
from drydock_provisioner.cli.task.actions import TaskBuildData


@click.group()
def task():
    """Drydock task commands."""


@task.command(name='create')
@click.option(
    '--design-ref', '-d', help='The design reference for this action')
@click.option('--action', '-a', help='The action to perform')
@click.option(
    '--node-names',
    '-n',
    help='The nodes targeted by this action, comma separated')
@click.option(
    '--rack-names',
    '-r',
    help='The racks targeted by this action, comma separated')
@click.option(
    '--node-tags',
    '-t',
    help='The nodes by tag name targeted by this action, comma separated')
@click.option(
    '--block/--no-block',
    '-b',
    help='The CLI will wait until the created completes before exitting',
    default=False)
@click.option(
    '--poll-interval',
    help='Polling interval to check task status in blocking mode.',
    default=15)
@click.pass_context
def task_create(ctx,
                design_ref=None,
                action=None,
                node_names=None,
                rack_names=None,
                node_tags=None,
                block=False,
                poll_interval=15):
    """Create a task."""
    if not design_ref:
        ctx.fail(
            'Error: Design reference must be specified using --design-ref')

    if not action:
        ctx.fail('Error: Action must be specified using --action')

    click.echo(
        json.dumps(
            TaskCreate(
                ctx.obj['CLIENT'],
                design_ref=design_ref,
                action_name=action,
                node_names=[x.strip() for x in node_names.split(',')]
                if node_names else [],
                rack_names=[x.strip() for x in rack_names.split(',')]
                if rack_names else [],
                node_tags=[x.strip()
                           for x in node_tags.split(',')] if node_tags else [],
                block=block,
                poll_interval=poll_interval).invoke()))


@task.command(name='list')
@click.pass_context
def task_list(ctx):
    """List tasks."""
    click.echo(json.dumps(TaskList(ctx.obj['CLIENT']).invoke()))


@task.command(name='show')
@click.option('--task-id', '-t', help='The required task id')
@click.option(
    '--block/--no-block',
    '-b',
    help='The CLI will wait until the created completes before exitting',
    default=False)
@click.pass_context
def task_show(ctx, task_id=None, block=False):
    """show a task's details."""
    if not task_id:
        ctx.fail('The task id must be specified by --task-id')

    click.echo(
        json.dumps(TaskShow(ctx.obj['CLIENT'], task_id=task_id).invoke()))


@task.command(name='builddata')
@click.option('--task-id', '-t', help='The required task id')
@click.option(
    '--output', '-o', help='The output format (yaml|json)', default='yaml')
@click.pass_context
def task_builddata(ctx, task_id=None, output='yaml'):
    """Show builddata assoicated with ``task_id``."""
    if not task_id:
        ctx.fail('The task id must be specified by --task-id')

    task_bd = TaskBuildData(ctx.obj['CLIENT'], task_id=task_id).invoke()

    if output == 'json':
        click.echo(json.dumps(task_bd))
    else:
        if output != 'yaml':
            click.echo(
                'Invalid output format {}, defaulting to YAML.'.format(output))
        click.echo(
            yaml.safe_dump(
                task_bd, allow_unicode=True, default_flow_style=False))