# -*- coding: utf-8 -*-
import os
import sys
import yaml

from devops.models import Environment


def create_config():
    env = os.environ

    conf_path = env['CONF_PATH']
    with open(conf_path) as c:
        conf = yaml.load(c.read())

    env_name = env['ENV_NAME']
    image_path = env['IMAGE_PATH']
    slaves_count = int(env['SLAVES_COUNT'])

    conf['env_name'] = env_name
    node_params = conf['rack-01-node-params']
    node_params['volumes'][0]['source_image'] = image_path

    group = conf['groups'][0]
    for i in range(slaves_count):
        group['nodes'].append({'name': 'slave-{}'.format(i),
                               'role': 'slave'})
    for node in group['nodes']:
        node['params'] = node_params
    return {'template': {'devops_settings': conf}}

def get_env():
    env = os.environ
    env_name = env['ENV_NAME']
    return Environment.get(name=env_name)

def get_master_ip(env):
    admin=env.get_node(role='master')
    return admin.get_ip_address_by_network_name('public')

def get_slave_ips(env):
    slaves = env.get_nodes(role='slave')
    ips = []
    for slave in slaves:
        ips.append(slave.get_ip_address_by_network_name('public'))
    return ips

def define_from_config(conf):
    env = Environment.create_environment(conf)
    env.define()
    env.start()


if __name__ == '__main__':
    if len(sys.argv) != 2:
        sys.exit(2)
    cmd = sys.argv[1]
    if cmd == 'create_env':
        config = create_config()
        define_from_config(config)
    elif cmd == 'get_admin_ip':
        print get_master_ip(get_env())
    elif cmd == 'get_slaves_ips':
        print get_slave_ips(get_env())