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
  12 local notify_fg    = require("beautiful").fg_focus
  13 local naughty      = require("naughty")
  14 local wibox        = require("wibox")
  15
  16 local io           = io
  17 local tostring     = tostring
  18 local string       = { format = string.format,
  19                        gsub   = string.gsub }
  20
  21 local setmetatable = setmetatable
  22
  23 -- Network infos
  24 -- lain.widgets.net
  25 local net = {
  26     last_t = 0,
  27     last_r = 0
  28 }
  29
  30 function net.get_device()
  31     f = io.popen("ip link show | cut -d' ' -f2,9")
  32     ws = f:read("*all")
  33     f:close()
  34     ws = ws:match("%w+: UP")
  35     if ws ~= nil then
  36         return ws:gsub(": UP", "")
  37     else
  38         return ""
  39     end
  40 end
  41
  42 local function worker(args)
  43     local args = args or {}
  44     local timeout = args.timeout or 2
  45     local iface = args.iface or net.get_device()
  46     local units = args.units or 1024 --kb
  47     local settings = args.settings or function() end
  48
  49     net.widget = wibox.widget.textbox('')
  50
  51     helpers.set_map(iface, true)
  52
  53     function net.update()
  54         if iface == "" then iface = net.get_device() end
  55
  56         carrier = helpers.first_line('/sys/class/net/' .. iface ..
  57                                            '/carrier') or "0"
  58         state = helpers.first_line('/sys/class/net/' .. iface ..
  59                                            '/operstate') or "down"
  60         local now_t = helpers.first_line('/sys/class/net/' .. iface ..
  61                                            '/statistics/tx_bytes') or 0
  62         local now_r = helpers.first_line('/sys/class/net/' .. iface ..
  63                                            '/statistics/rx_bytes') or 0
  64
  65         sent = tostring((now_t - net.last_t) / timeout / units)
  66         sent = string.gsub(string.format('%.1f', sent), ",", ".")
  67
  68         received = tostring((now_r - net.last_r) / timeout / units)
  69         received = string.gsub(string.format('%.1f', received), ",", ".")
  70
  71         widget = net.widget
  72         settings()
  73
  74         net.last_t = now_t
  75         net.last_r = now_r
  76
  77         if carrier ~= "1"
  78         then
  79             if helpers.get_map(iface)
  80             then
  81                 n_title = iface
  82                 if n_title == "" then n_title = "network" end
  83                 naughty.notify({
  84                     title    = n_title,
  85                     text     = "no carrier",
  86                     timeout  = 7,
  87                     position = "top_left",
  88                     icon     = helpers.icons_dir .. "no_net.png",
  89                     fg       = notify_fg or "#FFFFFF"
  90                 })
  91                 helpers.set_map(iface, false)
  92             end
  93         else
  94             helpers.set_map(iface, true)
  95         end
  96     end
  97
  98     helpers.newtimer(iface, timeout, net.update)
  99
 100     return net.widget
 101 end
 102
 103 return setmetatable(net, { __call = function(_, ...) return worker(...) end })