--]]
local helpers = require("lain.helpers")
-
-local notify_fg = require("beautiful").fg_focus
local naughty = require("naughty")
local wibox = require("wibox")
-local io = io
-local tostring = tostring
local string = { format = string.format,
- gsub = string.gsub }
+ gsub = string.gsub,
+ match = string.match }
local setmetatable = setmetatable
-- Network infos
-- lain.widgets.net
-local net = {
- last_t = 0,
- last_r = 0
-}
-
-function net.get_device()
- f = io.popen("ip link show | cut -d' ' -f2,9")
- ws = f:read("*all")
- f:close()
- ws = ws:match("%w+: UP")
- if ws ~= nil then
- return ws:gsub(": UP", "")
- else
- return ""
- end
-end
local function worker(args)
- local args = args or {}
- local timeout = args.timeout or 2
- local iface = args.iface or net.get_device()
- local units = args.units or 1024 --kb
+ local net = { last_t = 0, last_r = 0, devices = {} }
+
+ function net.get_first_device()
+ local ws = helpers.read_pipe("ip link show | cut -d' ' -f2,9")
+ ws = ws:match("%w+: UP") or ws:match("ppp%w+: UNKNOWN")
+ if ws then return { ws:match("(%w+):") }
+ else return {} end
+ end
+
+ local args = args or {}
+ local timeout = args.timeout or 2
+ local units = args.units or 1024 --kb
+ local notify = args.notify or "on"
+ local screen = args.screen or 1
local settings = args.settings or function() end
+ local iface = args.iface or net.get_first_device()
net.widget = wibox.widget.textbox('')
- helpers.set_map(iface, true)
+ -- Compatibility with old API where iface was a string corresponding to 1 interface
+ if type(iface) == "string" then
+ iftable = {iface}
+ else
+ iftable = iface
+ end
+
+ -- Mark all devices as initially online/active
+ for i, dev in ipairs(iftable) do
+ helpers.set_map(dev, true)
+ end
+
+ function update()
+ -- This check is required to ensure we keep looking for one device if
+ -- none is found by net.get_first_device() at startup (i.e. iftable = {})
+ if next(iftable) == nil then
+ iftable = net.get_first_device()
+ end
+
+ -- 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(iftable) do
+ local dev_now = {}
+ local dev_before = net.devices[dev] or { last_t = 0, last_r = 0 }
+
+ dev_now.carrier = helpers.first_line(string.format('/sys/class/net/%s/carrier', dev)) or '0'
+ dev_now.state = helpers.first_line(string.format('/sys/class/net/%s/operstate', dev)) or 'down'
+
+ local now_t = tonumber(helpers.first_line(string.format('/sys/class/net/%s/statistics/tx_bytes', dev)) or 0)
+ local now_r = tonumber(helpers.first_line(string.format('/sys/class/net/%s/statistics/rx_bytes', dev)) or 0)
- function net.update()
- if iface == "" then iface = net.get_device() end
+ dev_now.sent = (now_t - dev_before.last_t) / timeout / units
+ dev_now.received = (now_r - dev_before.last_r) / timeout / units
- carrier = helpers.first_line('/sys/class/net/' .. iface ..
- '/carrier') or "0"
- state = helpers.first_line('/sys/class/net/' .. iface ..
- '/operstate') or "down"
- local now_t = helpers.first_line('/sys/class/net/' .. iface ..
- '/statistics/tx_bytes') or 0
- local now_r = helpers.first_line('/sys/class/net/' .. iface ..
- '/statistics/rx_bytes') or 0
+ net_now.sent = net_now.sent + dev_now.sent
+ net_now.received = net_now.received + dev_now.received
- sent = tostring((now_t - net.last_t) / timeout / units)
- sent = string.gsub(string.format('%.1f', sent), ",", ".")
+ dev_now.sent = string.gsub(string.format('%.1f', dev_now.sent), ',', '.')
+ dev_now.received = string.gsub(string.format('%.1f', dev_now.received), ',', '.')
- received = tostring((now_r - net.last_r) / timeout / units)
- received = string.gsub(string.format('%.1f', received), ",", ".")
+ dev_now.last_t = now_t
+ dev_now.last_r = now_r
- widget = net.widget
- settings()
+ -- This will become dev_before in the next update/iteration
+ net.devices[dev] = dev_now
- net.last_t = now_t
- net.last_r = now_r
+ total_t = total_t + now_t
+ total_r = total_r + now_r
- if carrier ~= "1"
- then
- if helpers.get_map(iface)
- then
- n_title = iface
- if n_title == "" then n_title = "network" end
+ -- Notify only once when connection is loss
+ if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then
naughty.notify({
- title = n_title,
+ title = dev,
text = "no carrier",
- timeout = 7,
- position = "top_left",
icon = helpers.icons_dir .. "no_net.png",
- fg = notify_fg or "#FFFFFF"
+ screen = screen
})
- helpers.set_map(iface, false)
+ helpers.set_map(dev, false)
+ elseif string.match(dev_now.carrier, "1") then
+ helpers.set_map(dev, true)
end
- else
- helpers.set_map(iface, true)
+
+ -- Old api compatibility
+ net_now.carrier = dev_now.carrier
+ net_now.state = dev_now.state
+ -- And new api
+ net_now.devices[dev] = dev_now
+ -- With the new api new_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
+ -- Convert to a string to round the digits after the float point
+ net_now.sent = string.gsub(string.format('%.1f', net_now.sent), ',', '.')
+ net_now.received = string.gsub(string.format('%.1f', net_now.received), ',', '.')
+
+ widget = net.widget
+ settings()
+
+ net.last_t = total_t
+ net.last_r = total_r
end
end
- helpers.newtimer(iface, timeout, net.update)
+ helpers.newtimer(iface, timeout, update)
- return net.widget
+ return setmetatable(net, { __index = net.widget })
end
-return setmetatable(net, { __call = function(_, ...) return worker(...) end })
+return setmetatable({}, { __call = function(_, ...) return worker(...) end })