##############################################################################
# Copyright (c) 2017 Huawei Tech and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

import logging
import sys
import yaml
sys.path.insert(0, '/home/opnfv/bottlenecks/utils/infra_setup/passwordless_SSH/')
import ssh

logger = logging.getLogger(__name__)
with open('/tmp/pod.yaml') as f:
    dataMap = yaml.safe_load(f)
    for x in dataMap:
        for y in dataMap[x]:
            if (y['role']=='Controller') or (y['role']=='Compute'):
                ip = str(y['ip'])
                user = str(y['user'])
                pwd = str(y['password'])
                ssh_d = ssh.SSH(user, host= ip, password= pwd)
                status, stdout, stderr = ssh_d.execute("cd /etc && mkdir cadvisor-config")
                if status:
                    raise Exception("Command failed with non-zero status.")
                    logger.info(stdout.splitlines())
                with open("/home/opnfv/bottlenecks/monitor/cadvisor_install.sh") as stdin_file:
                    ssh_d.run("cat > /etc/cadvisor-config/install.sh", stdin=stdin_file)
                status, stdout, stderr = ssh_d.execute("sudo apt-get install docker.io")
                if status:
                    raise Exception("Command for installing docker failed.")
                    logger.info(stdout.splitlines())
                ssh_d.run("cd /etc/cadvisor-config/ && bash ./install.sh")