#!/usr/bin/python3 -W ignore::DeprecationWarning
# -*- coding: utf-8 -*-
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.


import os
import sys
import glob
from random import choice
import string
from optparse import OptionParser
import mysql.connector
import paramiko
from threading import Thread

# ---- This snippet of code adds the sources path and the waf configured PYTHONDIR to the Python path ----
# ---- We do this so cloud_utils can be looked up in the following order:
# ---- 1) Sources directory
# ---- 2) waf configured PYTHONDIR
# ---- 3) System Python path
for pythonpath in (
    "@PYTHONDIR@",
    os.path.join(
        os.path.dirname(__file__), os.path.pardir, os.path.pardir, "python", "lib"
    ),
):
    if os.path.isdir(pythonpath):
        sys.path.insert(0, pythonpath)
# ---- End snippet of code ----
from cloud_utils import check_call, CalledProcessError, read_properties

cfg = "@MSCONF@/db.properties"

# ---------------------- option parsing and command line checks ------------------------


usage = """%prog <license file> <-a | host names / IP addresses...>

This command deploys the license file specified in the command line into a specific XenServer host or all XenServer hosts known to the management server."""

parser = OptionParser(usage=usage)
parser.add_option(
    "-a",
    "--all",
    action="store_true",
    dest="all",
    default=False,
    help="deploy to all known hosts rather that a single host",
)

# ------------------ functions --------------------


def e(msg):
    parser.error(msg)


def getknownhosts(host, username, password):
    conn = mysql.connector.connect(host=host, user=username, password=password)
    cur = conn.cursor()
    cur.execute(
        "SELECT h.private_ip_address,d.value FROM cloud.host h inner join cloud.host_details d on (h.id = d.host_id) where d.name = 'username' and setup = 1"
    )
    usernames = dict(cur.fetchall())
    cur.execute(
        "SELECT h.private_ip_address,d.value FROM cloud.host h inner join cloud.host_details d on (h.id = d.host_id) where d.name = 'password' and setup = 1"
    )
    passwords = dict(cur.fetchall())
    creds = dict([[x, (usernames[x], passwords[x])] for x in list(usernames.keys())])
    cur.close()
    conn.close()
    return creds


def splitlast(string, splitter):
    splitted = string.split(splitter)
    first, last = splitter.join(splitted[:-1]), splitted[-1]
    return first, last


def parseuserpwfromhosts(hosts):
    creds = {}
    for host in hosts:
        user = "root"
        password = ""
        if "@" in host:
            user, host = splitlast(host, "@")
            if ":" in user:
                user, password = splitlast(user, ":")
        creds[host] = (user, password)
    return creds


class XenServerConfigurator(Thread):

    def __init__(self, host, user, password, keyfiledata):
        Thread.__init__(self)
        self.host = host
        self.user = user
        self.password = password
        self.keyfiledata = keyfiledata
        self.retval = None  # means all's good
        self.stdout = ""
        self.stderr = ""
        self.state = "initialized"

    def run(self):
        try:
            self.state = "running"
            c = paramiko.SSHClient()
            c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            c.connect(self.host, username=self.user, password=self.password)
            sftp = c.open_sftp()
            sftp.chdir("/tmp")
            f = sftp.open("xen-license", "w")
            f.write(self.keyfiledata)
            f.close()
            sftp.close()
            stdin, stdout, stderr = c.exec_command(
                "xe host-license-add license-file=/tmp/xen-license"
            )
            c.exec_command("false")
            self.stdout = stdout.read(-1)
            self.stderr = stderr.read(-1)
            self.retval = stdin.channel.recv_exit_status()
            c.close()
            if self.retval != 0:
                self.state = "failed"
            else:
                self.state = "finished"

        except Exception as e:
            self.state = "failed"
            self.retval = e
            # raise

    def __str__(self):
        if self.state == "failed":
            return "<%s XenServerConfigurator on %s@%s: %s>" % (
                self.state,
                self.user,
                self.host,
                str(self.retval),
            )
        else:
            return "<%s XenServerConfigurator on %s@%s>" % (
                self.state,
                self.user,
                self.host,
            )


# ------------- actual code --------------------

(options, args) = parser.parse_args()

try:
    licensefile, args = args[0], args[1:]
except IndexError:
    e("The first argument must be the license file to use")

if options.all:
    if len(args) != 0:
        e("IP addresses cannot be specified if -a is specified")
    config = read_properties(cfg)
    creds = getknownhosts(
        config["db.cloud.host"],
        config["db.cloud.username"],
        config["db.cloud.password"],
    )
    hosts = list(creds.keys())
else:
    if not args:
        e("You must specify at least one IP address, or -a")
    hosts = args
    creds = parseuserpwfromhosts(hosts)

try:
    keyfiledata = file(licensefile).read(-1)
except OSError as e:
    sys.stderr.write("The file %s cannot be opened" % licensefile)
    sys.exit(1)

configurators = []
for host, (user, password) in list(creds.items()):
    configurators.append(XenServerConfigurator(host, user, password, keyfiledata))


for c in configurators:
    c.start()

for c in configurators:
    print(c.host + "...", end=" ")
    c.join()
    if c.state == "failed":
        if c.retval:
            msg = "failed with return code %s: %s%s" % (c.retval, c.stdout, c.stderr)
            msg = msg.strip()
            print(msg)
        else:
            print("failed: %s" % c.retval)
    else:
        print("done")

successes = len([a for a in configurators if not a.state == "failed"])
failures = len([a for a in configurators if a.state == "failed"])

print("%3s successes" % successes)
print("%3s failures" % failures)