X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/96df4b879033f447d494dfc9928d97cc34e5473b..84313cc876bce8b346ca83934de57b50488ab8cd:/widgets/net.lua diff --git a/widgets/net.lua b/widgets/net.lua index af97201..ed1cf4c 100644 --- a/widgets/net.lua +++ b/widgets/net.lua @@ -8,95 +8,113 @@ --]] 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 "network off" - 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 settings = args.settings or function() end + local net = { last_t = 0, last_r = 0 } - net.widget = wibox.widget.textbox('') - - helpers.set_map(iface, true) + 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 - function update() - net_now = {} + 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() - if iface == "" then iface = net.get_device() end + net.widget = wibox.widget.textbox('') - net_now.carrier = helpers.first_line('/sys/class/net/' .. iface .. - '/carrier') or "0" - net_now.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 + -- Compatibility with old API where iface was a string corresponding to 1 interface + if type(iface) == "string" then + iftable = {iface} + else + iftable = iface + end - net_now.sent = tostring((now_t - net.last_t) / timeout / units) - net_now.sent = string.gsub(string.format('%.1f', net_now.sent), ",", ".") + -- Mark all devices as initially online/active + for i, dev in ipairs(iftable) do + helpers.set_map(dev, true) + end - net_now.received = tostring((now_r - net.last_r) / timeout / units) - net_now.received = string.gsub(string.format('%.1f', net_now.received), ",", ".") + function update() + -- These are the totals over all specified interfaces + net_now = { + 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_now[dev] or net + + 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) + + if now_t ~= dev_before.last_t or now_r ~= dev_before.last_r then + dev_now.sent = (now_t - (dev_before.last_t or 0)) / timeout / units + net_now.sent = net_now.sent + dev_now.sent + dev_now.sent = string.gsub(string.format('%.1f', dev_now.sent), ',', '.') + dev_now.received = (now_r - (dev_before.last_r or 0)) / timeout / units + net_now.received = net_now.received + dev_now.received + dev_now.received = string.gsub(string.format('%.1f', dev_now.received), ',', '.') + end - widget = net.widget - settings() + total_t = total_t + now_t + total_r = total_r + now_r - net.last_t = now_t - net.last_r = now_r + net_now[dev] = dev_now - if net_now.carrier ~= "1" - then - if helpers.get_map(iface) - then + if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then naughty.notify({ - title = iface, + 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) + else + helpers.set_map(dev, true) end - else - helpers.set_map(iface, true) + end + + if total_t ~= net.last_t or total_r ~= net.last_r then + 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, 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 })