#!/usr/bin/python

import click

from orch import graph
from orch import tasks


@click.group()
def orchestration():
    pass


@click.command()
@click.argument('plan', type=click.File('rb'))
def create(plan):
    click.echo(graph.create_plan(plan.read()))


@click.command()
@click.argument('uid')
def report(uid):
    colors = {
        'PENDING': 'white',
        'ERROR': 'red',
        'SUCCESS': 'green',
        'INPROGRESS': 'yellow'}

    report = graph.report_topo(uid)
    for item in report:
        click.echo(
            click.style('{} -> {}'.format(item[0], item[1]), fg=colors[item[1]]))


@click.command()
@click.argument('uid')
def execute(uid):
    tasks.schedule_start.apply_async(args=[uid], queue='master')


@click.command()
@click.argument('uid')
@click.option('--reset', default=False, is_flag=True)
def restart(uid, reset):
    if reset:
        graph.reset(uid)
    tasks.schedule_start.apply_async(args=[uid], queue='master')


@click.command()
@click.argument('uid')
def reset(uid):
    graph.reset(uid)


orchestration.add_command(create)
orchestration.add_command(report)
orchestration.add_command(execute)
orchestration.add_command(restart)
orchestration.add_command(reset)


if __name__ == '__main__':
    orchestration()