]> git.madduck.net Git - etc/awesome.git/blob - widgets/net.lua

madduck's git repository

Every one of the projects in this repository is available at the canonical URL git://git.madduck.net/madduck/pub/<projectpath> — see each project's metadata for the exact URL.

All patches and comments are welcome. Please squash your changes to logical commits before using git-format-patch and git-send-email to patches@git.madduck.net. If you'd read over the Git project's submission guidelines and adhered to them, I'd be especially grateful.

SSH access, as well as push access can be individually arranged.

If you use my repositories frequently, consider adding the following snippet to ~/.gitconfig and using the third clone URL listed for each project:

[url "git://git.madduck.net/madduck/"]
  insteadOf = madduck:

ed1cf4ccc2ec71fae0806d06b1cc32e4ae28a437
[etc/awesome.git] / 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 })