net.lua: support wifi/ethernet connection indicators

This commit is contained in:
cool-cool-sweat 2018-01-22 22:39:48 -05:00 committed by Luca CPZ
parent 35856d49a2
commit 5e2d2977cf
1 changed files with 23 additions and 13 deletions

View File

@ -20,6 +20,8 @@ local function factory(args)
local timeout = args.timeout or 2 local timeout = args.timeout or 2
local units = args.units or 1024 -- KB local units = args.units or 1024 -- KB
local notify = args.notify or "on" local notify = args.notify or "on"
local wifi_state = args.wifi_state or "off"
local eth_state = args.eth_state or "off"
local screen = args.screen or 1 local screen = args.screen or 1
local settings = args.settings or function() end local settings = args.settings or function() end
@ -28,9 +30,8 @@ local function factory(args)
(type(args.iface) == "table" and args.iface)) or {} (type(args.iface) == "table" and args.iface)) or {}
function net.get_device() function net.get_device()
helpers.async(string.format("ip link show", device_cmd), function(ws) helpers.line_callback("ip link", function(line)
ws = ws:match("(%w+): <BROADCAST,MULTICAST,.-UP,LOWER_UP>") net.iface[#net.iface + 1] = not string.match(line, "LOOPBACK") and string.match(line, "(%w+): <") or nil
net.iface = ws and { ws } or {}
end) end)
end end
@ -45,7 +46,7 @@ local function factory(args)
received = 0 received = 0
} }
for i, dev in ipairs(net.iface) do for _, dev in ipairs(net.iface) do
local dev_now = {} local dev_now = {}
local dev_before = net.devices[dev] or { last_t = 0, last_r = 0 } local dev_before = net.devices[dev] or { last_t = 0, last_r = 0 }
local now_t = tonumber(helpers.first_line(string.format("/sys/class/net/%s/statistics/tx_bytes", dev)) or 0) local now_t = tonumber(helpers.first_line(string.format("/sys/class/net/%s/statistics/tx_bytes", dev)) or 0)
@ -66,9 +67,18 @@ local function factory(args)
dev_now.last_t = now_t dev_now.last_t = now_t
dev_now.last_r = now_r dev_now.last_r = now_r
if wifi_state == "on" and helpers.first_line(string.format("/sys/class/net/%s/uevent", dev)) == "DEVTYPE=wlan" and string.match(dev_now.carrier, "1") then
dev_now.wifi = true
dev_now.signal = tonumber(string.match(helpers.lines_from("/proc/net/wireless")[3], "(%-%d+%.)")) or nil
end
if eth_state == "on" and helpers.first_line(string.format("/sys/class/net/%s/uevent", dev)) ~= "DEVTYPE=wlan" and string.match(dev_now.carrier, "1") then
dev_now.ethernet = true
end
net.devices[dev] = dev_now net.devices[dev] = dev_now
-- Notify only once when connection is loss -- Notify only once when connection is lost
if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then
naughty.notify { naughty.notify {
title = dev, title = dev,
@ -84,7 +94,7 @@ local function factory(args)
net_now.carrier = dev_now.carrier net_now.carrier = dev_now.carrier
net_now.state = dev_now.state net_now.state = dev_now.state
net_now.devices[dev] = dev_now net_now.devices[dev] = dev_now
-- new_now.sent and net_now.received will be -- net_now.sent and net_now.received will be
-- the totals across all specified devices -- the totals across all specified devices
end end