X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/8d3868d7ce528cfc075e246c8935bae2724e0868..f76d33e8cd5bc84f44758e8047294227deb3f29d:/widget/net.lua

diff --git a/widget/net.lua b/widget/net.lua
index c6ebaca..987f047 100644
--- a/widget/net.lua
+++ b/widget/net.lua
@@ -1,32 +1,27 @@
-
 --[[
-                                                  
-     Licensed under GNU General Public License v2 
-      * (c) 2013,      Luke Bonham                
-      * (c) 2010-2012, Peter Hofmann              
-                                                  
+
+     Licensed under GNU General Public License v2
+      * (c) 2013,      Luca CPZ
+      * (c) 2010-2012, Peter Hofmann
+
 --]]
 
-local helpers      = require("lain.helpers")
-local naughty      = require("naughty")
-local wibox        = require("wibox")
-local string       = { format = string.format,
-                       match  = string.match }
-local setmetatable = setmetatable
+local helpers = require("lain.helpers")
+local naughty = require("naughty")
+local wibox   = require("wibox")
+local string  = { format = string.format, match = string.match }
 
 -- Network infos
 -- lain.widget.net
 
-local function worker(args)
-    local net = { widget = wibox.widget.textbox() }
-    net.last_t = 0
-    net.last_r = 0
-    net.devices = {}
-
+local function factory(args)
+    local net        = { widget = wibox.widget.textbox(), devices = {} }
     local args       = args or {}
     local timeout    = args.timeout or 2
-    local units      = args.units or 1024 --kb
+    local units      = args.units or 1024 -- KB
     local notify     = args.notify or "on"
+    local wifi_state = args.wifi_state or "off"
+    local eth_state  = args.eth_state or "off"
     local screen     = args.screen or 1
     local settings   = args.settings or function() end
 
@@ -35,29 +30,23 @@ local function worker(args)
                 (type(args.iface) == "table" and args.iface)) or {}
 
     function net.get_device()
-        helpers.async(string.format("ip link show", device_cmd), function(ws)
-            ws = ws:match("(%w+): <BROADCAST,MULTICAST,.-UP,LOWER_UP>")
-            net.iface = ws and { ws } or {}
+        helpers.line_callback("ip link", function(line)
+            net.iface[#net.iface + 1] = not string.match(line, "LOOPBACK") and string.match(line, "(%w+): <") or nil
         end)
     end
 
     if #net.iface == 0 then net.get_device() end
 
-    function update()
+    function net.update()
         -- These are the totals over all specified interfaces
         net_now = {
-            -- New api - Current state of requested devices
             devices  = {},
             -- Bytes since last iteration
             sent     = 0,
             received = 0
         }
 
-        -- Total bytes transfered
-        local total_t = 0
-        local total_r = 0
-
-        for i, dev in ipairs(net.iface) do
+        for _, dev in ipairs(net.iface) do
             local dev_now    = {}
             local dev_before = net.devices[dev] or { last_t = 0, last_r = 0 }
             local now_t      = tonumber(helpers.first_line(string.format("/sys/class/net/%s/statistics/tx_bytes", dev)) or 0)
@@ -69,55 +58,56 @@ local function worker(args)
             dev_now.sent     = (now_t - dev_before.last_t) / timeout / units
             dev_now.received = (now_r - dev_before.last_r) / timeout / units
 
-            net_now.sent     = net_now.sent     + dev_now.sent
+            net_now.sent     = net_now.sent + dev_now.sent
             net_now.received = net_now.received + dev_now.received
 
-            dev_now.sent     = string.format('%.1f', dev_now.sent)
-            dev_now.received = string.format('%.1f', dev_now.received)
+            dev_now.sent     = string.format("%.1f", dev_now.sent)
+            dev_now.received = string.format("%.1f", dev_now.received)
 
             dev_now.last_t   = now_t
             dev_now.last_r   = now_r
 
-            -- This will become dev_before in the next update/iteration
-            net.devices[dev] = dev_now
+            if wifi_state == "on" and helpers.first_line(string.format("/sys/class/net/%s/uevent", dev)) == "DEVTYPE=wlan" and string.match(dev_now.carrier, "1") then
+                dev_now.wifi   = true
+                dev_now.signal = tonumber(string.match(helpers.lines_from("/proc/net/wireless")[3], "(%-%d+%.)")) or nil
+            end
 
-            total_t  = total_t + now_t
-            total_r  = total_r + now_r
+            if eth_state == "on" and helpers.first_line(string.format("/sys/class/net/%s/uevent", dev)) ~= "DEVTYPE=wlan" and string.match(dev_now.carrier, "1") then
+                dev_now.ethernet = true
+            end
 
-            -- Notify only once when connection is loss
+            net.devices[dev] = dev_now
+
+            -- Notify only once when connection is lost
             if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then
-                naughty.notify({
+                naughty.notify {
                     title    = dev,
-                    text     = "no carrier",
+                    text     = "No carrier",
                     icon     = helpers.icons_dir .. "no_net.png",
                     screen   = screen
-                })
+                }
                 helpers.set_map(dev, false)
             elseif string.match(dev_now.carrier, "1") then
                 helpers.set_map(dev, true)
             end
 
-            net_now.carrier      = dev_now.carrier
-            net_now.state        = dev_now.state
+            net_now.carrier = dev_now.carrier
+            net_now.state = dev_now.state
             net_now.devices[dev] = dev_now
-            -- new_now.sent and net_now.received will be the
-            -- totals across all specified devices
+            -- net_now.sent and net_now.received will be
+            -- the totals across all specified devices
         end
 
-        if total_t ~= net.last_t or total_r ~= net.last_r then
-            net_now.sent     = string.format('%.1f', net_now.sent)
-            net_now.received = string.format('%.1f', net_now.received)
-            net.last_t       = total_t
-            net.last_r       = total_r
-        end
+        net_now.sent = string.format("%.1f", net_now.sent)
+        net_now.received = string.format("%.1f", net_now.received)
 
         widget = net.widget
         settings()
     end
 
-    helpers.newtimer("network", timeout, update)
+    helpers.newtimer("network", timeout, net.update)
 
     return net
 end
 
-return setmetatable({}, { __call = function(_, ...) return worker(...) end })
+return factory