from cinderclient import client as cc
from heatclient import client as hc
from keystoneclient.v2_0 import client as kc
from novaclient.v1_1 import client as nc
import os
import re
import sys
import time

CONF = dict()
keys = ["os_username", "os_password", "os_auth_url", "os_tenant_name",
        "os_image_endpoint"]


def load_conf():
    for item in keys:
        CONF[item] = os.environ[item]


def get_nova_client():
    return nc.Client(username=CONF["os_username"],
                     api_key=CONF["os_password"],
                     auth_url=CONF["os_auth_url"],
                     project_id=CONF["os_tenant_name"])


def get_auth_token():
    keystone = kc.Client(username=CONF["os_username"],
                         password=CONF["os_password"],
                         tenant_name=CONF["os_tenant_name"],
                         auth_url=CONF["os_auth_url"])
    return keystone.auth_token


def get_heat_client():
    return hc.Client('1',
                     endpoint=CONF["os_image_endpoint"],
                     token=get_auth_token())


def get_cinder_client():
    return cc.Client('1',
                     CONF["os_username"],
                     CONF["os_password"],
                     CONF["os_tenant_name"],
                     CONF["os_auth_url"])


def cleanup_heat():
    current_name = sys.argv[2]
    client = get_heat_client()
    stacks = client.stacks.list(show_hidden=True)
    name_regex = re.compile(current_name)
    deleted_stacks = []

    for stack in stacks:
        if name_regex.match(stack.stack_name):
            deleted_stacks.append(stack.stack_name)
            print(stack.stack_name)
            client.stacks.delete(stack.stack_name)
    if not deleted_stacks:
        return
    else:
        # Let Heat delete stacks
        time.sleep(60)

    stacks = client.stacks.list(show_hidden=True)
    for stack in stacks:
        if stack.stack_name in deleted_stacks:
            # Resource cleanup is required
            print("At least one stack wasn't deleted!")
            print("Performing resources cleanup...")
            cleanup()


def cleanup():
    client = get_nova_client()
    cinder_client = get_cinder_client()
    servers = client.servers.list()
    volumes = cinder_client.volumes.list()
    secgroups = client.security_groups.list()
    current_name = sys.argv[2]
    name_regex = re.compile(current_name)

    for server in servers:
        if name_regex.match(server.name):
            print(server.name)
            fl_ips = client.floating_ips.findall(instance_id=server.id)
            for fl_ip in fl_ips:
                    client.floating_ips.delete(fl_ip.id)
            client.servers.delete(server.id)

    time.sleep(20)
    for volume in volumes:
        if name_regex.match(volume.display_name):
            print(volume.display_name)
            volume.delete()

    for group in secgroups:
        if name_regex.match(group.name):
            print(group.name)
            group.delete()


def main(argv):
    load_conf()

    if "cleanup" in argv:
        cleanup()

    if "cleanup-heat" in argv:
        cleanup_heat()


if __name__ == "__main__":
    main(sys.argv[1:])