widgets/net.lua

   1
   2 --[[
   3
   4      Licensed under GNU General Public License v2
   5       * (c) 2013,      Luke Bonham
   6       * (c) 2010-2012, Peter Hofmann
   7
   8 --]]
   9
  10 local helpers      = require("lain.helpers")
  11 local naughty      = require("naughty")
  12 local wibox        = require("wibox")
  13
  14 local string       = { format = string.format,
  15                        gsub   = string.gsub,
  16                        match  = string.match }
  17
  18 local setmetatable = setmetatable
  19
  20 -- Network infos
  21 -- lain.widgets.net
  22
  23 local function worker(args)
  24     local net = helpers.make_widget_textbox()
  25     net.last_t = 0
  26     net.last_r = 0
  27     net.devices = {}
  28
  29     function net.get_first_device()
  30         local ws = helpers.read_pipe("ip link show | cut -d' ' -f2,9")
  31         ws = ws:match("%w+: UP") or ws:match("ppp%w+: UNKNOWN")
  32         if ws then return { ws:match("(%w+):") }
  33         else return {} end
  34     end
  35
  36     local args     = args or {}
  37     local timeout  = args.timeout or 2
  38     local units    = args.units or 1024 --kb
  39     local notify   = args.notify or "on"
  40     local screen   = args.screen or 1
  41     local settings = args.settings or function() end
  42     local iface    = args.iface or net.get_first_device()
  43
  44     -- Compatibility with old API where iface was a string corresponding to 1 interface
  45     if type(iface) == "string" then
  46         iftable = {iface}
  47     else
  48         iftable = iface
  49     end
  50
  51     -- Mark all devices as initially online/active
  52     for i, dev in ipairs(iftable) do
  53         helpers.set_map(dev, true)
  54     end
  55
  56     function update()
  57         -- This check is required to ensure we keep looking for one device if
  58         -- none is found by net.get_first_device() at startup (i.e. iftable = {})
  59         if next(iftable) == nil then
  60             iftable = net.get_first_device()
  61         end
  62
  63         -- These are the totals over all specified interfaces
  64         net_now = {
  65             -- New api - Current state of requested devices
  66             devices  = {},
  67             -- Bytes since last iteration
  68             sent     = 0,
  69             received = 0
  70         }
  71
  72         -- Total bytes transfered
  73         local total_t = 0
  74         local total_r = 0
  75
  76         for i, dev in ipairs(iftable) do
  77             local dev_now = {}
  78             local dev_before = net.devices[dev] or { last_t = 0, last_r = 0 }
  79
  80             dev_now.carrier  = helpers.first_line(string.format('/sys/class/net/%s/carrier', dev)) or '0'
  81             dev_now.state    = helpers.first_line(string.format('/sys/class/net/%s/operstate', dev)) or 'down'
  82
  83             local now_t      = tonumber(helpers.first_line(string.format('/sys/class/net/%s/statistics/tx_bytes', dev)) or 0)
  84             local now_r      = tonumber(helpers.first_line(string.format('/sys/class/net/%s/statistics/rx_bytes', dev)) or 0)
  85
  86             dev_now.sent     = (now_t - dev_before.last_t) / timeout / units
  87             dev_now.received = (now_r - dev_before.last_r) / timeout / units
  88
  89             net_now.sent     = net_now.sent     + dev_now.sent
  90             net_now.received = net_now.received + dev_now.received
  91
  92             dev_now.sent     = string.gsub(string.format('%.1f', dev_now.sent), ',', '.')
  93             dev_now.received = string.gsub(string.format('%.1f', dev_now.received), ',', '.')
  94
  95             dev_now.last_t   = now_t
  96             dev_now.last_r   = now_r
  97
  98             -- This will become dev_before in the next update/iteration
  99             net.devices[dev] = dev_now
 100
 101             total_t  = total_t + now_t
 102             total_r  = total_r + now_r
 103
 104             -- Notify only once when connection is loss
 105             if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then
 106                 naughty.notify({
 107                     title    = dev,
 108                     text     = "no carrier",
 109                     icon     = helpers.icons_dir .. "no_net.png",
 110                     screen   = screen
 111                 })
 112                 helpers.set_map(dev, false)
 113             elseif string.match(dev_now.carrier, "1") then
 114                 helpers.set_map(dev, true)
 115             end
 116
 117             -- Old api compatibility
 118             net_now.carrier      = dev_now.carrier
 119             net_now.state        = dev_now.state
 120             -- And new api
 121             net_now.devices[dev] = dev_now
 122             -- With the new api new_now.sent and net_now.received will be the
 123             -- totals across all specified devices
 124
 125         end
 126
 127         if total_t ~= net.last_t or total_r ~= net.last_r then
 128             -- Convert to a string to round the digits after the float point
 129             net_now.sent     = string.gsub(string.format('%.1f', net_now.sent), ',', '.')
 130             net_now.received = string.gsub(string.format('%.1f', net_now.received), ',', '.')
 131
 132             net.last_t = total_t
 133             net.last_r = total_r
 134         end
 135
 136         widget = net.widget
 137         settings()
 138     end
 139
 140     helpers.newtimer(iface, timeout, update)
 141
 142     return net
 143 end
 144
 145 return setmetatable({}, { __call = function(_, ...) return worker(...) end })