# -*- coding: utf-8 -*-
##----------------------------------------------------------------------
## NAG.SNR.get_lldp_neighbors
##----------------------------------------------------------------------
## Copyright (C) 2007-2012 The NOC Project
## See LICENSE for details
##----------------------------------------------------------------------

## NOC modules
from noc.sa.script import Script as NOCScript
from noc.sa.interfaces import IGetLLDPNeighbors
from noc.sa.interfaces.base import MACAddressParameter
from noc.lib.validators import is_int, is_ipv4


class Script(NOCScript):
    name = "DLink.DxS_Smart.get_lldp_neighbors"
    implements = [IGetLLDPNeighbors]

    def execute(self):
        r = []
        # Try SNMP first
        if self.snmp and self.access_profile.snmp_ro:
            try:

# lldpRemLocalPortNum
# lldpRemChassisIdSubtype lldpRemChassisId
# lldpRemPortIdSubtype lldpRemPortId
# lldpRemSysName lldpRemSysCapEnabled
                for v in self.snmp.get_tables(
                    ["1.3.6.1.2.1.2.2.1.1",
                    "1.0.8802.1.1.2.1.4.1.1.4", "1.0.8802.1.1.2.1.4.1.1.5",
                    "1.0.8802.1.1.2.1.4.1.1.6", "1.0.8802.1.1.2.1.4.1.1.7",
                    "1.0.8802.1.1.2.1.4.1.1.9", "1.0.8802.1.1.2.1.4.1.1.12"
                    ], bulk=True):
		    if v[6]:
			local_interface = int(v[0].split('.')[1])
			remote_chassis_id_subtype = v[2]
            	        remotechassisid = ":".join(["%02x" % ord(c) for c in v[3]])
            	        remote_port_subtype = v[4]
            		remote_system_name = v[6]
            		remote_port = v[5]
            		remote_capabilities = (["%02x" % ord(x) for x in v[7]])[0][0:1]
                	i = {"local_interface": local_interface, "neighbors": []}
            		n = {
                	    "remote_chassis_id_subtype": remote_chassis_id_subtype,
                	    "remote_chassis_id": remotechassisid,
                	    "remote_port_subtype": remote_port_subtype,
                	    "remote_port": remote_port,
                	    "remote_capabilities": remote_capabilities,
                	}
            		if remote_system_name:
            	    	    n["remote_system_name"] = remote_system_name
    			i["neighbors"].append(n)
            		r.append(i)
		if r:
		    return r

            except self.snmp.TimeOutError:
                pass
	return r