ccm

#!/usr/bin/env python

import os
import sys
import warnings

import pkg_resources
from six import print_

from ccmlib import common
from ccmlib.cmds import cluster_cmds, command, node_cmds
from ccmlib.remote import (PARAMIKO_IS_AVAILABLE, execute_ccm_remotely,
                           get_remote_options, get_remote_usage)


def get_command(kind, cmd):
    cmd_name = kind.lower().capitalize() + cmd.lower().capitalize() + "Cmd"
    try:
        klass = (cluster_cmds if kind.lower() == 'cluster' else node_cmds).__dict__[cmd_name]
    except KeyError:
        return None
    if not issubclass(klass, command.Cmd):
        return None
    return klass()


def print_subcommand_usage(kind):
    for cmd_name in (cluster_cmds if kind.lower() == 'cluster' else node_cmds).commands():
        cmd = get_command(kind, cmd_name)
        if not cmd:
            print_("Internal error, unknown command {0}".format(cmd_name))
            exit(1)
        print_("  {0:14} {1}".format(cmd_name, cmd.description()))


def print_global_usage():
    print_("Usage:")
    remote_options = ""
    if PARAMIKO_IS_AVAILABLE:
        remote_options = " [remote_options]"
    print_("  ccm" + remote_options + " <cluster_cmd> [options]")
    print_("  ccm" + remote_options + " <node_name> <node_cmd> [options]")
    if PARAMIKO_IS_AVAILABLE:
        print_("")
        print_(get_remote_usage())
    print_("")
    print_("Where <cluster_cmd> is one of")
    print_subcommand_usage('cluster')
    print_("")
    print_("or <node_name> is the name of a node of the current cluster and <node_cmd> is one of")
    print_subcommand_usage('node')
    exit(1)


for entry_point in pkg_resources.iter_entry_points(group='ccm_extension'):
    entry_point.load()()

common.check_win_requirements()

if len(sys.argv) <= 1:
    print_("Missing arguments")
    print_global_usage()

arg1 = sys.argv[1].lower()

# show-*-cmds are undocumented commands that emit a list of
# the appropriate type of subcommand. This is used by the bash completion script.
if arg1 == 'show-cluster-cmds':
    for cmd_name in cluster_cmds.commands():
        print_(cmd_name)
    exit(1)

if arg1 == 'show-node-cmds':
    for cmd_name in node_cmds.commands():
        print_(cmd_name)
    exit(1)

# Attempt a remote execution (only occurs if remote options are valid and ssh_host is set)
# NOTE: An execption may be thrown if there are errors
if PARAMIKO_IS_AVAILABLE:
    remote_options, ccm_args = get_remote_options();
    if not remote_options is None and not remote_options.ssh_host is None:
        ouput, exit_status = execute_ccm_remotely(remote_options, ccm_args)
        sys.exit(exit_status)

if arg1 in cluster_cmds.commands():
    kind = 'cluster'
    cmd = arg1
    cmd_args = sys.argv[2:]
else:
    if len(sys.argv) <= 2:
        print_("Missing arguments")
        print_global_usage()
    kind = 'node'
    node = arg1
    cmd = sys.argv[2]
    cmd_args = [node] + sys.argv[3:]

if os.getenv('CASSANDRA_HOME') is not None:
    warnings.warn("'CASSANDRA_HOME' is set to be {}".format(os.getenv('CASSANDRA_HOME')))

cmd = get_command(kind, cmd)
if not cmd:
    print_("Unknown node or command: {0}".format(arg1))
    exit(1)

parser = cmd.get_parser()

(options, args) = parser.parse_args(cmd_args)
cmd.validate(parser, options, args)

cmd.run()