#    Copyright 2015 Mirantis, Inc.
#
#    Licensed 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 time
import subprocess

from fuelweb_test import logger
from fuelweb_test.helpers import checkers
from fuelweb_test.settings import CONTRAIL_PLUGIN_PATH
from proboscis.asserts import assert_true
from settings import VSRX_TEMPLATE_PATH
import openstack


BOND_CONFIG = [
    {
        'mac': None,
        'mode': 'balance-rr',
        'name': 'lnx-bond0',
        'slaves': [
            {'name': 'eth3'},
            {'name': 'eth4'},
        ],
        'state': None,
        'type': 'bond',
        'assigned_networks': []
    }
]

INTERFACES = {
    'eth0': ['fuelweb_admin', 'storage'],
    'eth1': ['public'],
    'eth2': ['management'],
    'lnx-bond0': ['private']
}


def upload_contrail_packages(obj):
    node_ssh = obj.env.d_env.get_admin_remote()
    if os.path.splitext(obj.pack_path)[1] == ".deb":
        pkg_name = os.path.basename(obj.pack_path)
        logger.debug("Uploading package {0} "
                     "to master node".format(pkg_name))
        node_ssh.upload(obj.pack_path, obj.pack_copy_path)
    else:
        raise Exception('Failed to upload file to the master node')


def install_packages(obj, remote):
    command = "cd " + obj.pack_copy_path + " && ./install.sh"
    logger.info('The command is %s', command)
    remote.execute_async(command)
    time.sleep(50)
    os.path.isfile(obj.add_package)


def prepare_contrail_plugin(obj, slaves=None, options={}):
    """Copy necessary packages to the master node and install them"""

    obj.env.revert_snapshot("ready_with_%d_slaves" % slaves)

    # copy plugin to the master node
    checkers.upload_tarball(
        obj.env.d_env.get_admin_remote(),
        CONTRAIL_PLUGIN_PATH, '/var')

    # install plugin
    checkers.install_plugin_check_code(
        obj.env.d_env.get_admin_remote(),
        plugin=os.path.basename(CONTRAIL_PLUGIN_PATH))
    # FIXME: when opencontrail v3.0 is available
    # FIXME: remove the following line from 'or True'
    if obj.CONTRAIL_DISTRIBUTION == 'juniper' or True:
        # copy additional packages to the master node
        upload_contrail_packages(obj)

        # install packages
        install_packages(obj, obj.env.d_env.get_admin_remote())

    # prepare fuel
    openstack.assign_net_provider(obj, **options)


def activate_plugin(obj):
    """Enable plugin in contrail settings"""
    plugin_name = 'contrail'
    msg = "Plugin couldn't be enabled. Check plugin version. Test aborted"
    assert_true(
        obj.fuel_web.check_plugin_exists(obj.cluster_id, plugin_name),
        msg)
    logger.debug('we have contrail element')

    # FIXME: uncomment next block when opencontrail v3.0 is available
    """
    if obj.CONTRAIL_DISTRIBUTION == 'juniper':
        option = {'metadata/enabled': True,
                  'contrail_distribution/value': 'juniper', }
    else:
        option = {'metadata/enabled': True, }
    """

    # FIXME: remove next line when opencontrail v3.0 is available
    option = {'metadata/enabled': True, }

    obj.fuel_web.update_plugin_data(obj.cluster_id, plugin_name, option)


def activate_vsrx():
    """Activate vSRX1 image"""

    logger.info('Configure iptables and route...')
    command = 'sudo /sbin/iptables -F'
    subprocess.call(command, shell=True)
    command = 'sudo /sbin/iptables -t nat -F'
    subprocess.call(command, shell=True)
    command = 'sudo /sbin/iptables -t nat -A POSTROUTING -o eth0 -j MASQUERADE'
    subprocess.call(command, shell=True)

    if not VSRX_TEMPLATE_PATH:
        logger.info("#" * 10 + 'VSRX_TEMPLATE_PATH is not defined, '
                    'OSTF will not be running' + "#" * 10)
        return False

    logger.info("#" * 10 + 'Delete previous installation of vSRX...')
    subprocess.call('virsh destroy vSRX1', shell=True)
    subprocess.call('virsh undefine vSRX1', shell=True)
    command = 'sed -r "s/ENV_NAME/$ENV_NAME/g" {0} > logs/vSRX1.xml'.\
        format(VSRX_TEMPLATE_PATH)
    subprocess.call(command, shell=True)
    command = 'virsh create logs/vSRX1.xml'
    logger.info("#" * 10 + 'Create vSRX...')
    if subprocess.call(command, shell=True):
        logger.info("#" * 10 + 'VSRX could not be established, '
                    'OSTF will not be running' + "#" * 10)
        return False

    command = 'sudo ip route add 10.100.1.0/24 via 10.109.1.250'
    subprocess.call(command, shell=True)
    return True


def net_group_preparation(obj):
    """Prepare network group for network template """

    node_ssh = obj.env.d_env.get_admin_remote()

    # preparing commands for gateway setting and  private interface updating
    commands = [
        'fuel network-group --delete --network 5',
        'fuel network-group --create --name fuel network-group'
        '--create --name private --cidr 10.109.3.0/24 '
        '--gateway 10.109.3.1 --nodegroup 1',
        'fuel network-group --set --network 6 --meta "'
        '{"name": "private", "notation": "cidr", "render_type": null,'
        '"map_priority": 2, "configurable": true, "use_gateway": true,'
        ' "render_addr_mask": "internal", "vlan_start": null, '
        '"cidr": "10.109.3.0/24"}"']

    for i in commands:
        node_ssh.execute_async(i)
        time.sleep(40)