]> 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:

922e5e18a19a6b3f573b1e2b79a0f81c18c739d3
[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 local string       = { format = string.format,
14                        match  = string.match }
15 local tostring     = tostring
16 local setmetatable = setmetatable
17
18 -- Network infos
19 -- lain.widgets.net
20
21 local function worker(args)
22     local net = helpers.make_widget_textbox()
23     net.last_t = 0
24     net.last_r = 0
25     net.devices = {}
26
27     local args       = args or {}
28     local timeout    = args.timeout or 2
29     local units      = args.units or 1024 --kb
30     local notify     = args.notify or "on"
31     local screen     = args.screen or 1
32     local settings   = args.settings or function() end
33
34     -- Compatibility with old API where iface was a string corresponding to 1 interface
35     net.iface = (args.iface and (type(args.iface) == "string" and {args.iface}) or (type(args.iface) == "table" and args.iface)) or {}
36
37
38     function net.get_device()
39         helpers.async(string.format("ip link show", device_cmd), function(ws)
40             ws = ws:match("(%w+): <BROADCAST,MULTICAST,.-UP,LOWER_UP>")
41             net.iface = ws and { ws } or {}
42         end)
43     end
44
45     if #net.iface == 0 then net.get_device() end
46
47     function update()
48         -- These are the totals over all specified interfaces
49         net_now = {
50             -- New api - Current state of requested devices
51             devices  = {},
52             -- Bytes since last iteration
53             sent     = 0,
54             received = 0
55         }
56
57         -- Total bytes transfered
58         local total_t = 0
59         local total_r = 0
60
61         for i, dev in ipairs(net.iface) do
62             local dev_now    = {}
63             local dev_before = net.devices[dev] or { last_t = 0, last_r = 0 }
64             local now_t      = tonumber(helpers.first_line(string.format("/sys/class/net/%s/statistics/tx_bytes", dev)) or 0)
65             local now_r      = tonumber(helpers.first_line(string.format("/sys/class/net/%s/statistics/rx_bytes", dev)) or 0)
66
67             dev_now.carrier  = helpers.first_line(string.format("/sys/class/net/%s/carrier", dev)) or "0"
68             dev_now.state    = helpers.first_line(string.format("/sys/class/net/%s/operstate", dev)) or "down"
69
70             dev_now.sent     = (now_t - dev_before.last_t) / timeout / units
71             dev_now.received = (now_r - dev_before.last_r) / timeout / units
72
73             net_now.sent     = net_now.sent     + dev_now.sent
74             net_now.received = net_now.received + dev_now.received
75
76             dev_now.sent     = string.format('%.1f', dev_now.sent)
77             dev_now.received = string.format('%.1f', dev_now.received)
78
79             dev_now.last_t   = now_t
80             dev_now.last_r   = now_r
81
82             -- This will become dev_before in the next update/iteration
83             net.devices[dev] = dev_now
84
85             total_t  = total_t + now_t
86             total_r  = total_r + now_r
87
88             -- Notify only once when connection is loss
89             if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then
90                 naughty.notify({
91                     title    = dev,
92                     text     = "no carrier",
93                     icon     = helpers.icons_dir .. "no_net.png",
94                     screen   = screen
95                 })
96                 helpers.set_map(dev, false)
97             elseif string.match(dev_now.carrier, "1") then
98                 helpers.set_map(dev, true)
99             end
100
101             -- Old api compatibility
102             net_now.carrier      = dev_now.carrier
103             net_now.state        = dev_now.state
104             -- And new api
105             net_now.devices[dev] = dev_now
106             -- With the new api new_now.sent and net_now.received will be the
107             -- totals across all specified devices
108         end
109
110         if total_t ~= net.last_t or total_r ~= net.last_r then
111             net_now.sent     = string.format('%.1f', net_now.sent)
112             net_now.received = string.format('%.1f', net_now.received)
113             net.last_t       = total_t
114             net.last_r       = total_r
115         end
116
117         widget = net.widget
118         settings()
119     end
120
121     helpers.newtimer("net-" .. tostring(net.iface), timeout, update)
122
123     return net
124 end
125
126 return setmetatable({}, { __call = function(_, ...) return worker(...) end })