# vim: tabstop=4 shiftwidth=4 softtabstop=4
# Copyright 2011 Nicira Networks, Inc.
# 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: Somik Behera, Nicira Networks, Inc.
# @author: Brad Hall, Nicira Networks, Inc.
# @author: Dan Wendlandt, Nicira Networks, Inc.

import ConfigParser
import logging as LOG
import os
import sys
import unittest

from quantum.quantum_plugin_base import QuantumPluginBase
from optparse import OptionParser

import quantum.db.api as db
import ovs_db

CONF_FILE="ovs_quantum_plugin.ini"

LOG.basicConfig(level=LOG.WARN)
LOG.getLogger("ovs_quantum_plugin")

def find_config(basepath):
    for root, dirs, files in os.walk(basepath):
        if CONF_FILE in files:
            return os.path.join(root, CONF_FILE)
    return None

class VlanMap(object):
    vlans = {}
    def __init__(self):
        for x in xrange(2, 4094):
            self.vlans[x] = None
    def set(self, vlan_id, network_id):
        self.vlans[vlan_id] = network_id
    def acquire(self, network_id):
        for x in xrange(2, 4094):
            if self.vlans[x] == None:
                self.vlans[x] = network_id
                # LOG.debug("VlanMap::acquire %s -> %s" % (x, network_id))
                return x
        raise Exception("No free vlans..")
    def get(self, vlan_id):
        return self.vlans[vlan_id]
    def release(self, network_id):
        for x in self.vlans.keys():
            if self.vlans[x] == network_id:
                self.vlans[x] = None
                # LOG.debug("VlanMap::release %s" % (x))
                return
        LOG.error("No vlan found with network \"%s\"" % network_id)

class OVSQuantumPlugin(QuantumPluginBase):
    def __init__(self, configfile=None):
        config = ConfigParser.ConfigParser()
        if configfile == None:
            if os.path.exists(CONF_FILE):
                configfile = CONF_FILE
            else:
                configfile = find_config(os.path.abspath(os.path.dirname(__file__)))
        if configfile == None:
            raise Exception("Configuration file \"%s\" doesn't exist" %
              (configfile))
        LOG.debug("Using configuration file: %s" % configfile)
        config.read(configfile)
        LOG.debug("Config: %s" % config)

        DB_NAME = config.get("DATABASE", "name")
        DB_USER = config.get("DATABASE", "user")
        DB_PASS = config.get("DATABASE", "pass")
        DB_HOST = config.get("DATABASE", "host")
        options = {"sql_connection": "mysql://%s:%s@%s/%s" % (DB_USER,
          DB_PASS, DB_HOST, DB_NAME)}
        db.configure_db(options)

        self.vmap = VlanMap()
        # Populate the map with anything that is already present in the
        # database
        vlans = ovs_db.get_vlans()
        for x in vlans:
            vlan_id, network_id = x
            # LOG.debug("Adding already populated vlan %s -> %s" % (vlan_id, network_id))
            self.vmap.set(vlan_id, network_id)

    def get_all_networks(self, tenant_id):
        nets = []
        for x in db.network_list(tenant_id):
            LOG.debug("Adding network: %s" % x.uuid)
            d = {}
            d["net-id"] = str(x.uuid)
            d["net-name"] = x.name
            nets.append(d)
        return nets

    def create_network(self, tenant_id, net_name):
        d = {}
        try:
             res = db.network_create(tenant_id, net_name)
             LOG.debug("Created newtork: %s" % res)
        except Exception, e:
            LOG.error("Error: %s" % str(e))
            return d
        d["net-id"] = str(res.uuid)
        d["net-name"] = res.name
        vlan_id = self.vmap.acquire(str(res.uuid))
        ovs_db.add_vlan_binding(vlan_id, str(res.uuid))
        return d

    def delete_network(self, tenant_id, net_id):
        net = db.network_destroy(net_id)
        d = {}
        d["net-id"] = str(net.uuid)
        ovs_db.remove_vlan_binding(net_id)
        self.vmap.release(net_id)
        return d

    def get_network_details(self, tenant_id, net_id):
        network = db.network_get(net_id)
        d = {}
        d["net-id"] = str(network.uuid)
        d["net-name"] = network.name
        d["net-ports"] = self.get_all_ports(tenant_id, net_id)
        return d

    def rename_network(self, tenant_id, net_id, new_name):
        try:
            net = db.network_rename(net_id, tenant_id, new_name)
        except Exception, e:
            raise Exception("Failed to rename network: %s" % str(e))
        d = {}
        d["net-id"] = str(net.uuid)
        d["net-name"] = net.name
        return d

    def get_all_ports(self, tenant_id, net_id):
        ids = []
        ports = db.port_list(net_id)
        for x in ports:
            LOG.debug("Appending port: %s" % x.uuid)
            d = {}
            d["port-id"] = str(x.uuid)
            ids.append(d)
        return ids

    def create_port(self, tenant_id, net_id, port_state=None):
        LOG.debug("Creating port with network_id: %s" % net_id)
        port = db.port_create(net_id)
        d = {}
        d["port-id"] = str(port.uuid)
        LOG.debug("-> %s" % (port.uuid))
        return d

    def delete_port(self, tenant_id, net_id, port_id):
        try:
            port = db.port_destroy(port_id)
        except Exception, e:
            raise Exception("Failed to delete port: %s" % str(e))
        d = {}
        d["port-id"] = str(port.uuid)
        return d

    def update_port(self, tenant_id, net_id, port_id, port_state):
        """
        Updates the state of a port on the specified Virtual Network.
        """
        LOG.debug("update_port() called\n")
        port = db.port_get(port_id)
        port['port-state'] = port_state
        return port

    def get_port_details(self, tenant_id, net_id, port_id):
        port = db.port_get(port_id)
        rv = {"port-id": port.uuid, "attachment": port.interface_id,
          "net-id": port.network_id, "port-state": "UP"}
        return rv

    def get_all_attached_interfaces(self, tenant_id, net_id):
        ports = db.port_list(net_id)
        ifaces = []
        for p in ports:
            ifaces.append(p.interface_id)
        return ifaces

    def plug_interface(self, tenant_id, net_id, port_id, remote_iface_id):
        db.port_set_attachment(port_id, remote_iface_id)
        ovs_db.update_network_binding(net_id, remote_iface_id)

    def unplug_interface(self, tenant_id, net_id, port_id):
        db.port_set_attachment(port_id, "")
        ovs_db.update_network_binding(net_id, None)

    def get_interface_details(self, tenant_id, net_id, port_id):
        res = db.port_get(port_id)
        return res.interface_id

class VlanMapTest(unittest.TestCase):
    def setUp(self):
        self.vmap = VlanMap()
    def tearDown(self):
        pass
    def testAddVlan(self):
        vlan_id = self.vmap.acquire("foobar")
        self.assertTrue(vlan_id == 2)
    def testReleaseVlan(self):
        vlan_id = self.vmap.acquire("foobar")
        self.vmap.release("foobar")
        self.assertTrue(self.vmap.get(vlan_id) == None)

# TODO(bgh): Make the tests use a sqlite database instead of mysql
class OVSPluginTest(unittest.TestCase):
    def setUp(self):
        self.quantum = OVSQuantumPlugin()
        self.tenant_id = "testtenant"

    def testCreateNetwork(self):
        net1 = self.quantum.create_network(self.tenant_id, "plugin_test1")
        self.assertTrue(net1["net-name"] == "plugin_test1")

    def testGetNetworks(self):
        net1 = self.quantum.create_network(self.tenant_id, "plugin_test1")
        net2 = self.quantum.create_network(self.tenant_id, "plugin_test2")
        nets = self.quantum.get_all_networks(self.tenant_id)
        count = 0
        for x in nets:
            if "plugin_test" in x["net-name"]:
                count += 1
        self.assertTrue(count == 2)

    def testDeleteNetwork(self):
        net = self.quantum.create_network(self.tenant_id, "plugin_test1")
        self.quantum.delete_network(self.tenant_id, net["net-id"])
        nets = self.quantum.get_all_networks(self.tenant_id)
        count = 0
        for x in nets:
            if "plugin_test" in x["net-name"]:
                count += 1
        self.assertTrue(count == 0)

    def testRenameNetwork(self):
        net = self.quantum.create_network(self.tenant_id, "plugin_test1")
        net = self.quantum.rename_network(self.tenant_id, net["net-id"],
          "plugin_test_renamed")
        self.assertTrue(net["net-name"] == "plugin_test_renamed")

    def testCreatePort(self):
        net1 = self.quantum.create_network(self.tenant_id, "plugin_test1")
        port = self.quantum.create_port(self.tenant_id, net1["net-id"])
        ports = self.quantum.get_all_ports(self.tenant_id, net1["net-id"])
        count = 0
        for p in ports:
            count += 1
        self.assertTrue(count == 1)

    def testDeletePort(self):
        net1 = self.quantum.create_network(self.tenant_id, "plugin_test1")
        port = self.quantum.create_port(self.tenant_id, net1["net-id"])
        ports = self.quantum.get_all_ports(self.tenant_id, net1["net-id"])
        count = 0
        for p in ports:
            count += 1
        self.assertTrue(count == 1)
        for p in ports:
            self.quantum.delete_port(self.tenant_id, id, p["port-id"])
        ports = self.quantum.get_all_ports(self.tenant_id, net1["net-id"])
        count = 0
        for p in ports:
            count += 1
        self.assertTrue(count == 0)

    def testGetPorts(self):
        pass

    def testPlugInterface(self):
        net1 = self.quantum.create_network(self.tenant_id, "plugin_test1")
        port = self.quantum.create_port(self.tenant_id, net1["net-id"])
        self.quantum.plug_interface(self.tenant_id, net1["net-id"],
          port["port-id"], "vif1.1")
        port = self.quantum.get_port_details(self.tenant_id, net1["net-id"],
          port["port-id"])
        self.assertTrue(port["attachment"] == "vif1.1")

    def testUnPlugInterface(self):
        net1 = self.quantum.create_network(self.tenant_id, "plugin_test1")
        port = self.quantum.create_port(self.tenant_id, net1["net-id"])
        self.quantum.plug_interface(self.tenant_id, net1["net-id"],
          port["port-id"], "vif1.1")
        port = self.quantum.get_port_details(self.tenant_id, net1["net-id"],
          port["port-id"])
        self.assertTrue(port["attachment"] == "vif1.1")
        self.quantum.unplug_interface(self.tenant_id, net1["net-id"],
          port["port-id"])
        port = self.quantum.get_port_details(self.tenant_id, net1["net-id"],
          port["port-id"])
        self.assertTrue(port["attachment"] == "")

    def tearDown(self):
        networks = self.quantum.get_all_networks(self.tenant_id)
        # Clean up any test networks lying around
        for net in networks:
            id = net["net-id"]
            name = net["net-name"]
            if "plugin_test" in name:
                # Clean up any test ports lying around
                ports = self.quantum.get_all_ports(self.tenant_id, id)
                for p in ports:
                    self.quantum.delete_port(self.tenant_id, id, p["port-id"])
                self.quantum.delete_network(self.tenant_id, id)

if __name__ == "__main__":
    usagestr = "Usage: %prog [OPTIONS] <command> [args]"
    parser = OptionParser(usage=usagestr)
    parser.add_option("-v", "--verbose", dest="verbose",
      action="store_true", default=False, help="turn on verbose logging")

    options, args = parser.parse_args()

    if options.verbose:
        LOG.basicConfig(level=LOG.DEBUG)
    else:
        LOG.basicConfig(level=LOG.WARN)

    # Make sqlalchemy quieter
    LOG.getLogger('sqlalchemy.engine').setLevel(LOG.WARN)
    # Run the tests
    suite = unittest.TestLoader().loadTestsFromTestCase(OVSPluginTest)
    unittest.TextTestRunner(verbosity=2).run(suite)
    suite = unittest.TestLoader().loadTestsFromTestCase(VlanMapTest)
    unittest.TextTestRunner(verbosity=2).run(suite)