[modules] Add NIC module

Add a module that displays the status of all NICs (interface name, list
of IPs and state).

In its status, it also exposes whether it's a WiFi or a wired NIC.

For this functionality, additional code was implemented to allow a
module to add multiple elements to the bar at once. The framework calls
the module until its "next()" method return False.

bumblebee/module.py

@@ -15,4 +15,7 @@ class Module(object):
     def state(self):
         return "default"
 
+    def next(self):
+        return False
+
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

bumblebee/modules/nic.py

@@ -0,0 +1,61 @@
+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

bumblebee/outputs/i3.py

@@ -15,6 +15,7 @@ class i3bar(bumblebee.output.Output):
     def add(self, obj):
         theme = self.theme()
 
+        while True:
             data = {
                 u"full_text": "{}{}{}".format(theme.prefix(obj), obj.data(), theme.suffix(obj)),
                 "color": theme.color(obj),
@@ -37,6 +38,9 @@ class i3bar(bumblebee.output.Output):
                     })
 
             self._data.append(data)
+            if obj.next() == False:
+                break
+            theme.next()
 
     def get(self):
         data = json.dumps(self._data)

bumblebee/themes/solarized-powerline.json

@@ -32,6 +32,22 @@
 	"cpu": {
 		"prefix": "  "
 	},
+	"nic": {
+		"states": {
+			"wireless-up": {
+				"prefix": "  "
+			},
+			"wireless-down": {
+				"prefix": "  "
+			},
+			"wired-up": {
+				"prefix": "  "
+			},
+			"wired-down": {
+				"prefix": "  "
+			}
+		}
+	},
 	"battery": {
 		"states": {
 			"charged": {