# Copyright (c) 2016 Orange.
# All Rights Reserved.
#
#    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

from networking_bagpipe.tests.common import json_fixtures

from neutron.common import utils
from neutron.tests.fullstack.resources import config as neutron_cfg


ROOTWRAP_DAEMON_CMD_DFLT = ("sudo /usr/local/bin/oslo-rootwrap-daemon "
                            "/etc/bagpipe-bgp/rootwrap.conf")


def bagpipe_agent_config_fixture_init_common(self, env_desc, host_desc,
                                             local_ip):
    agent_config = self.config.get('agent')
    if agent_config is None:
        self.config.update({'agent': {}})
        agent_config = self.config.get('agent')

    agent_exts = agent_config.get('extensions', '').split(',')

    if env_desc.bagpipe_ml2:
        agent_exts.append('bagpipe')
    if env_desc.bgpvpn:
        agent_exts.append('bagpipe_bgpvpn')

    agent_config.update({
        'extensions': ','.join(filter(None, agent_exts))
    })

    self.config.update({
        'bagpipe': {
            'bagpipe_bgp_ip': local_ip
        }
    })


class OVSConfigFixture(neutron_cfg.OVSConfigFixture):

    def __init__(self, env_desc, host_desc, temp_dir, local_ip, mpls_bridge):
        super().__init__(env_desc, host_desc,
                         temp_dir, local_ip)
        bagpipe_agent_config_fixture_init_common(
            self, env_desc, host_desc, local_ip)

        self.config.bagpipe.update({
            'mpls_bridge': mpls_bridge,
            'tun_to_mpls_peer_patch_port':
                utils.get_rand_device_name(prefix='to-mpls'),
            'mpls_to_tun_peer_patch_port':
                utils.get_rand_device_name(prefix='to-tun'),
            'mpls_to_int_peer_patch_port':
                utils.get_rand_device_name(prefix='mpls-to-int'),
            'int_to_mpls_peer_patch_port':
                utils.get_rand_device_name(prefix='int-to-mpls'),
        })


class JsonFixture(neutron_cfg.ConfigFixture):
    """A fixture that holds a JSON configuration."""
    def _setUp(self):
        cfg_fixture = json_fixtures.JsonFileFixture(
            self.base_filename, self.config, self.temp_dir)
        self.useFixture(cfg_fixture)
        self.filename = cfg_fixture.filename


class BagpipeBGPConfigFixture(neutron_cfg.ConfigFixture):

    def __init__(self, env_desc, host_desc, temp_dir, local_ip, bgp_peer,
                 bgp_port, mpls_bridge, mpls_interface):
        super().__init__(
            env_desc, host_desc, temp_dir,
            base_filename='bgp.conf')

        self.config.update({
            'DEFAULT': {
                'debug': True,
            },
            'COMMON': {
                'root_helper_daemon': os.environ.get('OS_ROOTWRAP_DAEMON_CMD',
                                                     ROOTWRAP_DAEMON_CMD_DFLT)
            },
            'BGP': {
                'local_address': local_ip,
                'peers': bgp_peer,
                'my_as': '64512',
                'bgp_port': bgp_port,
            },
            'API': {
                'host': local_ip,
                'port': '8082'
            },
            'DATAPLANE_DRIVER_IPVPN': {
                'dataplane_driver': self.env_desc.ipvpn_driver,
                'ovs_bridge': mpls_bridge,
                'proxy_arp': 'False'
            },
            'DATAPLANE_DRIVER_EVPN': {
                'dataplane_driver': self.env_desc.evpn_driver
            }
        })

        if self.env_desc.ipvpn_driver != 'dummy':
            if self.env_desc.ipvpn_encap == 'vxlan':
                self.config['DATAPLANE_DRIVER_IPVPN'].update({
                    'vxlan_encap': 'True',
                    'mpls_interface': ''
                })
            if 'mpls-gre' in self.env_desc.ipvpn_encap:
                self.config['DATAPLANE_DRIVER_IPVPN'].update({
                    'mpls_interface': '*gre*',
                    'gre_tunnel': self._generate_gre_tunnel()
                })
            if self.env_desc.ipvpn_encap == 'mpls-gre-l3':
                self.config['DATAPLANE_DRIVER_IPVPN'].update({
                    'gre_tunnel_options': "packet_type=legacy_l3"
                })
            if self.env_desc.ipvpn_encap == 'bare-mpls':
                self.config['DATAPLANE_DRIVER_IPVPN'].update({
                    'mpls_interface': mpls_interface
                })

    def _generate_gre_tunnel(self):
        return utils.get_rand_device_name(prefix='mpls-gre')


class GoBGPConfigFixture(JsonFixture):

    def __init__(self, env_desc, host_desc, temp_dir,
                 bgp_peer, bgp_port, host_ips):
        super().__init__(
            env_desc, host_desc, temp_dir,
            base_filename='gobgp.conf')

        self.config.update({
            'global': {
                'config': {
                    'as': '64512',
                    'router-id': bgp_peer,
                    'port': bgp_port
                }
            }
        })

        neighbors = list()
        for host_ip in host_ips:
            neighbor = {
                'config': {
                    'neighbor-address': host_ip,
                    'peer-as': '64512'
                },
                'transport': {
                    'config': {
                        'passive-mode': 'true'
                    }
                },
                'route-reflector': {
                    'config': {
                        'route-reflector-client': 'true',
                        'route-reflector-cluster-id': '1.2.3.4'
                    }
                }
            }

            afi_safis = list()
            for afi_safi in ('rtc', 'l2vpn-evpn', 'l3vpn-ipv4-unicast',
                             'l3vpn-ipv4-flowspec'):
                afi_safis.append({
                    'config': {
                        'afi-safi-name': afi_safi
                    }
                })

            neighbor.update({'afi-safis': afi_safis})
            neighbors.append(neighbor)

        self.config.update({'neighbors': neighbors})
