solar/cli.py

#!/usr/bin/env python
import click
import json
#import matplotlib
#matplotlib.use('Agg')  # don't show windows
#import matplotlib.pyplot as plt
import networkx as nx
import os
import subprocess

from solar.core import actions as xa
from solar.core import resource as xr
from solar.core import signals as xs
from solar import operations
from solar import state


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


def init_cli_resource():
    @click.group()
    def resource():
        pass

    cli.add_command(resource)

    @click.command()
    @click.argument('resource_path')
    @click.argument('action_name')
    def action(action_name, resource_path):
        print 'action', resource_path, action_name
        r = xr.load(resource_path)
        xa.resource_action(r, action_name)

    resource.add_command(action)

    @click.command()
    @click.argument('name')
    @click.argument('base_path')
    @click.argument('dest_path')
    @click.argument('args')
    def create(args, dest_path, base_path, name):
        print 'create', name, base_path, dest_path, args
        args = json.loads(args)
        xr.create(name, base_path, dest_path, args)

    resource.add_command(create)

    @click.command()
    @click.argument('resource_path')
    @click.argument('tag_name')
    @click.option('--add/--delete', default=True)
    def tag(add, tag_name, resource_path):
        print 'Tag', resource_path, tag_name, add
        r = xr.load(resource_path)
        if add:
            r.add_tag(tag_name)
        else:
            r.remove_tag(tag_name)
        r.save()

    resource.add_command(tag)

    @click.command()
    @click.argument('path')
    @click.option('--all/--one', default=False)
    @click.option('--tag', default=None)
    @click.option('--use-json/--no-use-json', default=False)
    def show(use_json, tag, all, path):
        import json
        import six

        printer = lambda r: six.print_(r)
        if use_json:
            printer = lambda r: six.print_(json.dumps(r.to_dict()))

        if all or tag:
            for name, resource in xr.load_all(path).items():
                show = True
                if tag:
                    if tag not in resource.tags:
                        show = False

                if show:
                    printer(resource)
        else:
            printer(xr.load(path))

    resource.add_command(show)

    @click.command()
    @click.argument('name')
    @click.argument('args')
    def update(name, args):
        args = json.loads(args)
        all = xr.load_all()
        r = all[name]
        r.update(args)
    resource.add_command(update)


def init_cli_connect():
    @click.command()
    @click.argument('emitter')
    @click.argument('receiver')
    @click.option('--mapping', default=None)
    def connect(mapping, receiver, emitter):
        print 'Connect', emitter, receiver
        emitter = xr.load(emitter)
        receiver = xr.load(receiver)
        print emitter
        print receiver
        if mapping is not None:
            mapping = json.loads(mapping)
        xs.connect(emitter, receiver, mapping=mapping)

    cli.add_command(connect)

    @click.command()
    @click.argument('emitter')
    @click.argument('receiver')
    def disconnect(receiver, emitter):
        print 'Disconnect', emitter, receiver
        emitter = xr.load(emitter)
        receiver = xr.load(receiver)
        print emitter
        print receiver
        xs.disconnect(emitter, receiver)

    cli.add_command(disconnect)


def init_changes():
    @click.group()
    def changes():
        pass

    cli.add_command(changes)

    @click.command()
    def stage():
        log = operations.stage_changes()
        print log.show()

    changes.add_command(stage)

    @click.command()
    @click.option('--one', is_flag=True, default=False)
    def commit(one):
        if one:
            operations.commit_one()
        else:
            operations.commit_changes()

    changes.add_command(commit)

    @click.command()
    @click.option('--limit', default=5)
    def history(limit):
        print state.CL().show()

    changes.add_command(history)

    @click.command()
    @click.option('--last', is_flag=True, default=False)
    @click.option('--all', is_flag=True, default=False)
    @click.option('--uid', default=None)
    def rollback(last, all, uid):
        if last:
            print operations.rollback_last()
        elif all:
            print operations.rollback_all()
        elif uid:
            print operations.rollback_uid(uid)

    changes.add_command(rollback)


def init_cli_connections():
    @click.group()
    def connections():
        pass

    cli.add_command(connections)

    @click.command()
    def show():
        print json.dumps(xs.CLIENTS, indent=2)

    connections.add_command(show)

    # TODO: this requires graphing libraries
    @click.command()
    def graph():
        #g = xs.connection_graph()
        g = xs.detailed_connection_graph()

        nx.write_dot(g, 'graph.dot')
        subprocess.call(['dot', '-Tpng', 'graph.dot', '-o', 'graph.png'])

        # Matplotlib
        #pos = nx.spring_layout(g)
        #nx.draw_networkx_nodes(g, pos)
        #nx.draw_networkx_edges(g, pos, arrows=True)
        #nx.draw_networkx_labels(g, pos)
        #plt.axis('off')
        #plt.savefig('graph.png')

    connections.add_command(graph)


def init_cli_deployment_config():
    @click.command()
    @click.argument('filepath')
    def deploy(filepath):
        print 'Deploying from file {}'.format(filepath)
        xd.deploy(filepath)

    cli.add_command(deploy)


if __name__ == '__main__':
    init_cli_resource()
    init_cli_connect()
    init_cli_connections()
    init_cli_deployment_config()
    init_changes()

    cli()