import pyroute2
import netifaces
import bumblebee.module

class Module(bumblebee.module.Module):
    def __init__(self, args):
        super(Module, self).__init__(args)
        self._exclude = ( "lo", "virbr" )
        self._interfaces = [ i for i in netifaces.interfaces() if not i.startswith(self._exclude) ]
        self._index = 0
        self._intf = self._interfaces[0] if len(self._interfaces) > 0 else None
        self._cache = {}
        self._state = "down"

    def data(self):
        if len(self._interfaces) <= self._index:
            return "n/a"
        self._intf = self._interfaces[self._index]
        self._state = "down"
        addr = []

        if netifaces.AF_INET in netifaces.ifaddresses(self._intf):
            for ip in netifaces.ifaddresses(self._intf)[netifaces.AF_INET]:
                if "addr" in ip and ip["addr"] != "":
                    addr.append(ip["addr"])
                    self._state = "up"

        return "{} {} {}".format(self._intf, self._state, ", ".join(addr))

    def next(self):
        self._index += 1
        if self._index < len(self._interfaces):
            return True
        self._index = 0
        # reload to support hotplug
        self._interfaces = [ i for i in netifaces.interfaces() if not i.startswith(self._exclude) ]
        return False

    def _iswlan(self, intf):
        if not "wlan{}".format(intf) in self._cache:
            iw = pyroute2.IW()
            ip = pyroute2.IPRoute()
            idx = ip.link_lookup(ifname=intf)[0]
            try:
                iw.get_interface_by_ifindex(idx)
                self._cache["wlan{}".format(intf)] = True
            except exception as e:
                self._cache["wlan{}".format(intf)] = False
        return self._cache

    def state(self):
        t = "wireless" if self._iswlan(self._intf) else "wired"
        return "{}-{}".format(t, self._state)

    def warning(self):
        return self._state != "up"

    def critical(self):
        return self._state == "down"

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