#!/usr/bin/env python
import argparse
import os
import easyfab


def init_command(args):
    directory = os.getcwd()
    easyfab.init(directory)


def add_deployment(args):
    directory = os.getcwd()
    try:
        easyfab.add_deployment(directory, args.name)
    except:
        raise


if __name__ == '__main__':

    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(help='Available commands')

    # init easyfab command
    init_parser = subparsers.add_parser('init',
                                        help='Inits easyfab fabric file and deployments directory')
    init_parser.set_defaults(func=init_command)

    # add new deployment
    add_parser = subparsers.add_parser(
        'add', help='Creates blank new deployment')
    add_parser.add_argument('name')
    add_parser.set_defaults(func=add_deployment)

    args = parser.parse_args()
    args.func(args)
