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 = { last_t = 0, last_r = 0 }
  25
  26     function net.get_first_device()
  27         local ws = helpers.read_pipe("ip link show | cut -d' ' -f2,9")
  28         ws = ws:match("%w+: UP") or ws:match("ppp%w+: UNKNOWN")
  29         if ws then return { ws:match("(%w+):") }
  30         else return {} end
  31     end
  32
  33     local args     = args or {}
  34     local timeout  = args.timeout or 2
  35     local units    = args.units or 1024 --kb
  36     local notify   = args.notify or "on"
  37     local screen   = args.screen or 1
  38     local settings = args.settings or function() end
  39     local iface    = args.iface or net.get_first_device()
  40
  41     net.widget = wibox.widget.textbox('')
  42
  43     -- Compatibility with old API where iface was a string corresponding to 1 interface
  44     if type(iface) == "string" then
  45         iftable = {iface}
  46     else
  47         iftable = iface
  48     end
  49
  50     -- Mark all devices as initially online/active
  51     for i, dev in ipairs(iftable) do
  52         helpers.set_map(dev, true)
  53     end
  54
  55     function update()
  56         -- These are the totals over all specified interfaces
  57         net_now = {
  58             sent     = 0,
  59             received = 0
  60         }
  61
  62         -- Total bytes transfered
  63         local total_t = 0
  64         local total_r = 0
  65
  66         for i, dev in ipairs(iftable) do
  67             local dev_now = {}
  68             local dev_before = net_now[dev] or net
  69
  70             dev_now.carrier  = helpers.first_line(string.format('/sys/class/net/%s/carrier', dev)) or '0'
  71             dev_now.state    = helpers.first_line(string.format('/sys/class/net/%s/operstate', dev)) or 'down'
  72
  73             local now_t      = tonumber(helpers.first_line(string.format('/sys/class/net/%s/statistics/tx_bytes', dev)) or 0)
  74             local now_r      = tonumber(helpers.first_line(string.format('/sys/class/net/%s/statistics/rx_bytes', dev)) or 0)
  75
  76             if now_t ~= dev_before.last_t or now_r ~= dev_before.last_r then
  77                 dev_now.sent     = (now_t - (dev_before.last_t or 0)) / timeout / units
  78                 net_now.sent     = net_now.sent + dev_now.sent
  79                 dev_now.sent     = string.gsub(string.format('%.1f', dev_now.sent), ',', '.')
  80                 dev_now.received = (now_r - (dev_before.last_r or 0)) / timeout / units
  81                 net_now.received = net_now.received + dev_now.received
  82                 dev_now.received = string.gsub(string.format('%.1f', dev_now.received), ',', '.')
  83             end
  84
  85             total_t  = total_t + now_t
  86             total_r  = total_r + now_r
  87
  88             net_now[dev] = dev_now
  89
  90             if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then
  91                 naughty.notify({
  92                     title    = dev,
  93                     text     = "no carrier",
  94                     icon     = helpers.icons_dir .. "no_net.png",
  95                     screen   = screen
  96                 })
  97                 helpers.set_map(dev, false)
  98             else
  99                 helpers.set_map(dev, true)
 100             end
 101         end
 102
 103         if total_t ~= net.last_t or total_r ~= net.last_r then
 104             net_now.sent     = string.gsub(string.format('%.1f', net_now.sent), ',', '.')
 105             net_now.received = string.gsub(string.format('%.1f', net_now.received), ',', '.')
 106
 107             widget = net.widget
 108             settings()
 109
 110             net.last_t = total_t
 111             net.last_r = total_r
 112         end
 113     end
 114
 115     helpers.newtimer(iface, timeout, update)
 116
 117     return setmetatable(net, { __index = net.widget })
 118 end
 119
 120 return setmetatable({}, { __call = function(_, ...) return worker(...) end })