# vim: tabstop=4 shiftwidth=4 softtabstop=4

# Copyright (c) 2014, Oracle and/or its affiliates. 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.

#

# @author: Girish Moodalbail, Oracle, Inc.

import rad.client as radcli

import rad.connect as radcon

import rad.bindings.com.oracle.solaris.rad.evscntl_1 as evsbind

from oslo.config import cfg

from quantum.api.rpc.agentnotifiers import dhcp_rpc_agent_api

from quantum.api.rpc.agentnotifiers import l3_rpc_agent_api

from quantum.api.v2 import attributes

from quantum.common import constants as l3_constants

from quantum.common import exceptions

from quantum.common import rpc as q_rpc

from quantum.common import topics

from quantum.db import dhcp_rpc_base

from quantum.db import l3_rpc_base

from quantum.extensions import l3

from quantum.extensions import providernet

from quantum.openstack.common import log as logging

from quantum.openstack.common import rpc

from quantum.plugins.evs.db import api as evs_db

from quantum.plugins.evs.db import l3nat as evs_l3nat

from quantum import quantum_plugin_base_v2

LOG = logging.getLogger(__name__)

evs_controller_opts = [

    cfg.StrOpt('evs_controller', default='ssh://evsuser@localhost',

               help=_("An URI that specifies an EVS controller"))

]

cfg.CONF.register_opts(evs_controller_opts, "EVS")

# Maps OpenStack network resource attributes to EVS properties

NETWORK_EVS_ATTRIBUTE_MAP = {

    'tenant_id': 'tenant',

    'network_id': 'evs',

    'id': 'evs',

    'name': 'evs',

    l3.EXTERNAL: 'OpenStack:' + l3.EXTERNAL,

}

# Maps OpenStack subnet resource attributes to EVS' IPnet properties

SUBNET_IPNET_ATTRIBUTE_MAP = {

    'tenant_id': 'tenant',

    'network_id': 'evs',

    'id': 'ipnet',

    'name': 'ipnet',

    'enable_dhcp': 'OpenStack:enable_dhcp',

    'dns_nameservers': 'OpenStack:dns_nameservers',

    'host_routes': 'OpenStack:host_routes',

}

# Maps OpenStack port resource attributes to EVS' VPort properties

PORT_VPORT_ATTRIBUTE_MAP = {

    'tenant_id': 'tenant',

    'network_id': 'evs',

    'id': 'vport',

    'name': 'vport',

    'device_id': 'OpenStack:device_id',

    'device_owner': 'OpenStack:device_owner',

}

class EVSControllerError(exceptions.QuantumException):

    message = _("EVS controller: %(errmsg)s")

    def __init__(self, evs_errmsg):

        super(EVSControllerError, self).__init__(errmsg=evs_errmsg)

class EVSOpNotSupported(exceptions.QuantumException):

    message = _("Operation not supported by EVS plugin: %(opname)s")

    def __init__(self, evs_errmsg):

        super(EVSOpNotSupported, self).__init__(opname=evs_errmsg)

class EVSRpcCallbacks(dhcp_rpc_base.DhcpRpcCallbackMixin,

                      l3_rpc_base.L3RpcCallbackMixin):

    RPC_API_VERSION = '1.0'

    def create_rpc_dispatcher(self):

        '''Get the rpc dispatcher for this manager.

        If a manager would like to set an rpc API version, or support more than

        one class as the target of rpc messages, override this method.

        '''

        return q_rpc.PluginRpcDispatcher([self])

class EVSQuantumPluginV2(quantum_plugin_base_v2.QuantumPluginBaseV2,

                         evs_l3nat.EVS_L3_NAT_db_mixin):

    """Implements v2 Neutron Plug-in API specification.

    All the quantum API calls to create/delete/retrieve Network/Subnet/Port

    are forwarded to EVS controller through Solaris RAD. The RAD connection

    to EVS Controller is over SSH. In order that this plugin can communicate

    with EVS Controller non-interactively and securely, one should setup SSH

    authentication with pre-shared keys between the host running neutron-server

    and the host running EVS controller.

    The following table maps OpenStack Neutron resources and attributes to

    Solaris Elastic Virtual Switch resources and attributes

    |---------------------+------------------+------------------------------|

    | OpenStack Neutron   | Solaris EVS      | Comments                     |

    |---------------------+------------------+------------------------------|

    | Network             | EVS              | Represents an isolated L2    |

    | -- name             | -- name          | segment; implemented either  |

    | -- id               | -- uuid          | through VLANs or VXLANs      |

    | -- tenant_id        | -- tenant        |                              |

    | -- shared           | Always False     |                              |

    | -- admin_state_up   | Always True      |                              |

    | -- status           | Always ACTIVE    |                              |

    | -- provider:        |                  |                              |

    |    network_type     |  -- l2-type      | (either VLAN or VXLAN)       |

    | -- provider:        |                  |                              |

    |    segmentation_id  |  -- vlanid/vni   |                              |

    |                     |                  |                              |

    |                     |                  |                              |

    | Subnet              | IPnet            | An IP network represents     |

    | -- name             | -- name          | a block of either IPv4       |

    | -- id               | -- uuid          | or IPv6 addresses (subnet)   |

    | -- network_id       | -- evs           | along with a default router  |

    | -- tenant_id        | -- tenant        | for the block                |

    | -- cidr             | -- subnet        |                              |

    | -- gateway_ip       | -- defrouter     |                              |

    | -- allocation_pools | -- start/stop    |                              |

    | -- dns_nameservers  | -- OpenStack:\   |                              |

    |                     | dns_nameservers  |                              |

    | -- host_routes      | -- OpenStack:\   |                              |

    |                     | host_routes      |                              |

    | -- enable_dhcp      | -- OpenStack:\   |                              |

    |                     | enable_dhcp      |                              |

    | -- shared           | Always False     |                              |

    |                     |                  |                              |

    | Port                | VPort            | A VPort represents the point |

    | -- name             | -- name          | of attachment between the    |

    | -- id               | -- uuid          | VNIC and an EVS. It          |

    | -- network_id       | -- evs           | encapsulates various network |

    | -- tenant_id        | -- tenant        | configuration parameters (   |

    | -- status           | -- status        | MAC addresses, IP addresses, |

    | -- mac_address      | -- macaddr       | and SLAs)                    |

    | -- fixed_ips        | -- ipaddr        |                              |

    | -- device_id        | -- OpenStack:\   |                              |

    |                     |    device_id     |                              |

    | -- device_owner     | -- OpenStack:\   |                              |

    |                     |    device_owner  |                              |

    | -- security_groups  | -- Not Supported |                              |

    | -- admin_state_up   | Always UP        |                              |

    |---------------------+------------------+------------------------------|

    """

    _supported_extension_aliases = ["provider", "router", "quotas"]

    def __init__(self):

        # Since EVS Framework does not support router and floatingip

        # resources, the plugin itself will maintain a DB for these

        # two resources

        evs_db.configure_db()

        # Since there is no connect_uri() yet, we need to do this ourselves

        # parse ssh://user@hostname

        suh = cfg.CONF.EVS.evs_controller.split('://')

        if len(suh) != 2 or suh[0] != 'ssh' or not suh[1].strip():

            raise SystemExit(_("Specified evs_controller is invalid"))

        uh = suh[1].split('@')

        if len(uh) != 2 or not uh[0].strip() or not uh[1].strip():

            raise SystemExit(_("'user' and 'hostname' need to be specified "

                               "for evs_controller"))

        # TODO(gmoodalb): - try few times before you give up

        self._rc = radcon.connect_ssh(uh[1], user=uh[0])

        self._evsc = self._rc.get_object(evsbind.EVSController())

        self._setup_rpc()

    def _setup_rpc(self):

        # RPC support

        self.topic = topics.PLUGIN

        self.conn = rpc.create_connection(new=True)

        self.callbacks = EVSRpcCallbacks()

        self.dispatcher = self.callbacks.create_rpc_dispatcher()

        self.conn.create_consumer(self.topic, self.dispatcher, fanout=False)

        # Consume from all consumers in a thread

        self.conn.consume_in_thread()

        self.dhcp_agent_notifier = dhcp_rpc_agent_api.DhcpAgentNotifyAPI()

    @property

    def supported_extension_aliases(self):

        return self._supported_extension_aliases

    def _fields(self, resource, fields):

        if fields:

            return dict(((key, item) for key, item in resource.iteritems()

                         if key in fields))

        return resource

    def _convert_evs_to_network(self, evs):

        """Converts an EVS structure into Neutron Network structure."""

        networkdict = dict()

        networkdict['name'] = evs.name

        networkdict['id'] = evs.uuid

        networkdict['subnets'] = ([ipnet.uuid for ipnet in evs.ipnets]

                                  if evs.ipnets else [])

        networkdict['tenant_id'] = evs.tenantname

        networkdict[l3.EXTERNAL] = False

        for prop in evs.props:

            if prop.name == 'l2-type':

                networkdict[providernet.NETWORK_TYPE] = prop.value

            if prop.name == 'vlanid' or prop.name == 'vni':

                networkdict[providernet.SEGMENTATION_ID] = int(prop.value)

            if prop.name == NETWORK_EVS_ATTRIBUTE_MAP[l3.EXTERNAL]:

                networkdict[l3.EXTERNAL] = \

                    (True if prop.value == 'True' else False)

        # fixed values as EVS framework doesn't support this

        networkdict['admin_state_up'] = True

        networkdict['status'] = 'ACTIVE'

        networkdict['shared'] = False

        return networkdict

    def _convert_ipnet_to_subnet(self, ipnet):

        """Converts an EVS IPnet structure into Neutron Subnet structure."""

        subnetdict = dict()

        subnetdict['name'] = ipnet.name

        subnetdict['network_id'] = ipnet.evsuuid

        subnetdict['id'] = ipnet.uuid

        subnetdict['tenant_id'] = ipnet.tenantname

        subnetdict['ip_version'] = \

            (4 if ipnet.ipvers == evsbind.IPVersion.IPV4 else 6)

        # assign default values to some subnet attributes

        subnetdict['dns_nameservers'] = []

        subnetdict['host_routes'] = []

        subnetdict['enable_dhcp'] = False

        for prop in ipnet.props:

            if prop.name == 'defrouter':

                subnetdict['gateway_ip'] = prop.value

            if prop.name == 'subnet':

                subnetdict['cidr'] = prop.value

            if prop.name == SUBNET_IPNET_ATTRIBUTE_MAP['enable_dhcp']:

                subnetdict['enable_dhcp'] = \

                    (True if prop.value == 'True' else False)

            if prop.name == SUBNET_IPNET_ATTRIBUTE_MAP['dns_nameservers']:

                subnetdict['dns_nameservers'] = prop.value.split(',')

            if prop.name == SUBNET_IPNET_ATTRIBUTE_MAP['host_routes']:

                hrlist = []

                vlist = prop.value.split(',')

                for i in range(0, len(vlist), 2):

                    hrlist.append({vlist[i]: vlist[i + 1]})

                subnetdict['host_routes'] = hrlist

        subnetdict['shared'] = False

        return subnetdict

    def _convert_vport_to_port(self, context, vport):

        """Converts an EVS VPort structure into Neutron port structure."""

        portdict = dict()

        portdict['admin_state_up'] = True

        portdict['id'] = vport.uuid

        portdict['name'] = vport.name

        portdict['network_id'] = vport.evsuuid

        #TODO(gmoodalb): set to host/zonename/vnicname?

        portdict['device_id'] = ''

        portdict['device_owner'] = ''

        for prop in vport.props:

            if (prop.name == 'macaddr'):

                portdict['mac_address'] = prop.value

            if (prop.name == 'ipaddr'):

                evs = self.get_network(context, vport.evsuuid)

                portdict['fixed_ips'] = \

                    [{

                        'ip_address': prop.value.split('/')[0],

                        'subnet_id': evs['subnets'][0],

                    }]

            if (prop.name == 'OpenStack:device_id'):

                portdict['device_id'] = prop.value

            if (prop.name == 'OpenStack:device_owner'):

                portdict['device_owner'] = prop.value

        portdict['security_groups'] = []

        portdict['status'] = 'ACTIVE'

        portdict['tenant_id'] = vport.tenantname

        return portdict

    def _apply_rsrc_props_filter(self, rsrclist, filters):

        # if all of the filter values are None, then return

        if all([value is None for value in filters.values()]):

            return

        rsrc_to_remove = []

        for rsrc in rsrclist:

            propdict = dict((prop.name, prop.value) for prop in rsrc.props)

            for key, value in filters.iteritems():

                if value is None:

                    continue

                if key not in propdict:

                    rsrc_to_remove.append(rsrc)

                    break

                elif isinstance(value, list):

                    strlist = [str(v) for v in value]

                    if propdict[key] not in strlist:

                        rsrc_to_remove.append(rsrc)

                        break

                # TODO(gmoodalb): - check if it's an instance of basestring?

                elif propdict[key] != str(value):

                    rsrc_to_remove.append(rsrc)

                    break

        for rsrc in rsrc_to_remove:

            rsrclist.remove(rsrc)

    def create_subnet(self, context, subnet):

        """Creates a subnet(IPnet) for a given network(EVS).

         An IP network represents a block of either IPv4 or IPv6 addresses

         (i.e., subnet) along with a default router for the block. Only one

         IPnet can be associated with an EVS. All the zones/VNICs that

         connect to the EVS, through a VPort, will get an IP address from the

         IPnet associated with the EVS.

        """

        ipnetname = subnet['subnet']['name']

        if not ipnetname:

            ipnetname = None

        proplist = ['subnet=%s' % (subnet['subnet']['cidr'])]

        # obtain the optional default router

        defrouter = subnet['subnet']['gateway_ip']

        if defrouter is None:

            # user specified --no-gateway, we don't support it

            raise EVSOpNotSupported(_("cannot use --no-gateway"))

        if defrouter is not attributes.ATTR_NOT_SPECIFIED:

            proplist.append('defrouter=%s' % (defrouter))

        # obtain the optional DNS nameservers

        nameservers = subnet['subnet']['dns_nameservers']

        if attributes.is_attr_set(nameservers):

            proplist.append('%s=%s' %

                            (SUBNET_IPNET_ATTRIBUTE_MAP['dns_nameservers'],

                            ','.join(nameservers)))

        # obtain the host routes

        hostroutes = subnet['subnet']['host_routes']

        if attributes.is_attr_set(hostroutes):

            hrlist = ['%s,%s' % (destination, nexthop)

                      for destination, nexthop in hostroutes]

            proplist.append('%s=%s' %

                            (SUBNET_IPNET_ATTRIBUTE_MAP['host_routes'],

                             ",".join(hrlist)))

        enable_dhcp = subnet['subnet']['enable_dhcp']

        proplist.append('%s=%s' %

                        (SUBNET_IPNET_ATTRIBUTE_MAP['enable_dhcp'],

                         enable_dhcp))

        propstr = None

        if proplist:

            propstr = ",".join(proplist)

        # TODO(gmoodalb): extract the tenant id if an admin is creating for

        # someone else

        try:

            evsuuid = subnet['subnet']['network_id']

            tenantname = subnet['subnet']['tenant_id']

            pat = radcli.ADRGlobPattern(

                {'name': evsuuid, 'tenant': tenantname})

            evs = self._rc.get_object(evsbind.EVS(), pat)

            ipnet = evs.addIPnet(propstr, ipnetname)

        except radcli.ObjectError as oe:

            raise EVSControllerError(oe.get_payload().errmsg)

        retval = self._convert_ipnet_to_subnet(ipnet)

        # notify dhcp agent of subnet creation

        self.dhcp_agent_notifier.notify(context, {'subnet': retval},

                                        'subnet.create.end')

        return retval

    def update_subnet(self, context, id, subnet):

        evs_rpccall_sync = subnet.pop('evs_rpccall_sync', False)

        if not (set(subnet['subnet'].keys()) == set(('enable_dhcp',))):

                raise EVSOpNotSupported(_("only enable_dhcp can be updated"))

        propstr = "%s=%s" % (SUBNET_IPNET_ATTRIBUTE_MAP['enable_dhcp'],

                             subnet['subnet']['enable_dhcp'])

        try:

            pat = radcli.ADRGlobPattern({'uuid': id})

            ipnetlist = self._rc.list_objects(evsbind.IPnet(), pat)

            assert len(ipnetlist) == 1

            ipnet = self._rc.get_object(ipnetlist[0])

            ipnet.setProperty(propstr)

        except radcli.ObjectError as oe:

            raise EVSControllerError(oe.get_payload().errmsg)

        retval = self.get_subnet(context, id)

        # notify dhcp agent of subnet update

        methodname = 'subnet.update.end'

        payload = {'subnet': retval}

        if not evs_rpccall_sync:

            self.dhcp_agent_notifier.notify(context, payload, methodname)

        else:

            msg = self.dhcp_agent_notifier.make_msg(

                methodname.replace(".", "_"), payload=payload)

            self.dhcp_agent_notifier.call(context, msg,

                                          topic=topics.DHCP_AGENT)

        return retval

    def get_subnet(self, context, id, fields=None):

        try:

            ipnetlist = self._evsc.getIPnetInfo('ipnet=%s' % (id))

        except radcli.ObjectError as oe:

            raise EVSControllerError(oe.get_payload().errmsg)

        if ipnetlist:

            subnetdict = self._convert_ipnet_to_subnet(ipnetlist[0])

            return self._fields(subnetdict, fields)

        return {}

    def get_subnets(self, context, filters=None, fields=None,

                    sorts=None, limit=None, marker=None, page_reverse=False):

        filterstr = None

        # EVS desn't support filtering of resource based on

        # properties, so we will have to filter ourselves

        ipnet_props = {'OpenStack:enable_dhcp': None}

        if filters is not None:

            filterlist = []

            for key, value in filters.items():

                if key == 'shared':

                    if any(value):

                        return []

                    continue

                if key == 'verbose':

                    continue

                if key == 'enable_dhcp':

                    ipnet_props[SUBNET_IPNET_ATTRIBUTE_MAP[key]] = value

                    continue

                key = SUBNET_IPNET_ATTRIBUTE_MAP.get(key, key)

                if isinstance(value, list):

                    value = ",".join([str(val) for val in value])

                filterlist.append("%s=%s" % (key, value))

            if filterlist:

                filterstr = ",".join(filterlist)

        LOG.debug(_("calling ListIPnet from get_subnets() filterstr: '%s'")

                  % (filterstr))

        try:

            ipnetlist = self._evsc.getIPnetInfo(filterstr)

        except radcli.ObjectError as oe:

            raise EVSControllerError(oe.get_payload().errmsg)

        self._apply_rsrc_props_filter(ipnetlist, ipnet_props)

        retme = []

        for ipnet in ipnetlist:

            subnetdict = self._convert_ipnet_to_subnet(ipnet)

            retme.append(self._fields(subnetdict, fields))

        return retme

    def get_subnets_count(self, context, filters=None):

        return len(self.get_ipnets(context, filters))

    def _release_subnet_dhcp_port(self, context, subnet, delete_network):

        """Release any dhcp port associated with the subnet"""

        filters = dict(evs=subnet['network_id'])

        portlist = self.get_ports(context, filters)

        if delete_network:

            # One can delete a network if there is only one port that has a

            # VNIC attached to it and that port happens to be a DHCP port.

            ports_with_deviceid = [port for port in portlist

                                   if port['device_id'] != '']

            update_subnet = len(ports_with_deviceid) == 1

        else:

            # One can delete a subnet if there is only one port and that

            # port happens to be a DHCP port.

            update_subnet = len(portlist) == 1

        if update_subnet: