import argparse,sys
from clusterman.config import *
from  clusterman.commands import node, plugins, frontend

def main():
    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(dest="target", help='Target')

    ##### Node commands #####
    target_node = subparsers.add_parser("node")
    node_subparsers=target_node.add_subparsers(dest="command", help='Command')
    # Scan
    node_cmd_scan=node_subparsers.add_parser("scan")
    node_cmd_scan.add_argument("-t", "--timeout" ,help="Timeout", type=float)
    # Check
    node_cmd_scan=node_subparsers.add_parser("check")
    node_cmd_scan.add_argument("-t", "--timeout" ,help="Timeout", type=float)
    # List
    node_cmd_list=node_subparsers.add_parser("list")
    node_cmd_list.add_argument("-g", "--group" ,help="Group to list")
    # Exec
    node_cmd_list=node_subparsers.add_parser("exec")
    node_cmd_list.add_argument("-g", "--group" ,help="Group to run the command on")
    node_cmd_list.add_argument("-l", "--login" ,help="Use ssh username and password")
    node_cmd_list.add_argument("cmd",help="Command to run",nargs=argparse.REMAINDER)
    
    ##### Frontend commands #####
    target_frontend = subparsers.add_parser("frontend")
    frontend_subparsers=target_frontend.add_subparsers(dest="command", help='Command')
    # Info
    node_cmd_scan=frontend_subparsers.add_parser("info")
    
    ##### Plugins commands #####
    target_plugins = subparsers.add_parser("plugins")
    target_plugins.add_argument("name", help="Plugin's name")
    target_plugins.add_argument("parameters", help="Plugin's parameters",nargs=argparse.REMAINDER)
    
    # Check if command specified:
    if len(sys.argv)==1:
        parser.print_help(sys.stderr)
        sys.exit(1)
    # Parse arguments:
    args = parser.parse_args()

    # Run the proper handler
    if args.target == "node":
        if args.command == "scan":
            if args.timeout:
                node.scan(node_cmd_scan.timeout)
            else:
                node.scan(CONF["timeout"])
        elif args.command == "check":
            if args.timeout:
                node.check(node_cmd_scan.timeout)
            else:
                node.check(CONF["timeout"])
        elif args.command == "list":
            if args.group:
                node.ls(args.group)
            else:
                node.ls()
        elif args.command == "exec":
            use_key=not args.login
            if args.group:
                node.exec(args.cmd,args.group,use_key)
            else:
                node.exec(args.cmd, use_key=use_key)
        else:
            target_node.print_help(sys.stderr)
            sys.exit(1)  
    elif args.target == "frontend":
        if args.command == "info":
            frontend.info()
        else:
            target_frontend.print_help(sys.stderr)
            sys.exit(1)  
    elif args.target == "plugins":
        plugins.execute(args.name,args.parameters)