2016-08-11 23:25:56 -04:00
|
|
|
local Widget = require 'Widget'
|
|
|
|
local Text = require 'Text'
|
|
|
|
local Line = require 'Line'
|
|
|
|
local ScalePlot = require 'ScalePlot'
|
|
|
|
local util = require 'util'
|
|
|
|
local schema = require 'default_patterns'
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
local __string_gmatch = string.gmatch
|
2016-08-11 23:25:56 -04:00
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
local _PLOT_SEC_BREAK_ = 20
|
|
|
|
local _PLOT_HEIGHT_ = 56
|
2016-08-11 23:25:56 -04:00
|
|
|
|
|
|
|
local SYSFS_NET = '/sys/class/net/'
|
|
|
|
local STATS_RX = '/statistics/rx_bytes'
|
|
|
|
local STATS_TX = '/statistics/tx_bytes'
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
local network_label_function = function(bytes)
|
2016-08-11 23:25:56 -04:00
|
|
|
local new_unit = util.get_unit(bytes)
|
|
|
|
|
|
|
|
local converted = util.convert_bytes(bytes, 'B', new_unit)
|
|
|
|
local precision = 0
|
|
|
|
if converted < 10 then precision = 1 end
|
|
|
|
|
|
|
|
return util.round_to_string(converted, precision)..' '..new_unit..'/s'
|
|
|
|
end
|
|
|
|
|
|
|
|
local header = Widget.Header{
|
2017-07-16 14:50:33 -04:00
|
|
|
x = _G_INIT_DATA_.CENTER_RIGHT_X,
|
|
|
|
y = _G_INIT_DATA_.TOP_Y,
|
|
|
|
width = _G_INIT_DATA_.SECTION_WIDTH,
|
2017-07-17 00:21:31 -04:00
|
|
|
header = 'NETWORK'
|
2016-08-11 23:25:56 -04:00
|
|
|
}
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
local _RIGHT_X_ = _G_INIT_DATA_.CENTER_RIGHT_X + _G_INIT_DATA_.SECTION_WIDTH
|
2016-08-11 23:25:56 -04:00
|
|
|
|
|
|
|
local dnload = {
|
|
|
|
label = Widget.Text{
|
2017-07-16 14:50:33 -04:00
|
|
|
x = _G_INIT_DATA_.CENTER_RIGHT_X,
|
2016-08-11 23:25:56 -04:00
|
|
|
y = header.bottom_y,
|
|
|
|
text = 'Download',
|
|
|
|
},
|
|
|
|
speed = Widget.Text{
|
2017-07-17 00:21:31 -04:00
|
|
|
x = _RIGHT_X_,
|
2016-08-11 23:25:56 -04:00
|
|
|
y = header.bottom_y,
|
|
|
|
x_align = 'right',
|
|
|
|
text_color = schema.blue
|
|
|
|
},
|
|
|
|
plot = Widget.ScalePlot{
|
2017-07-16 14:50:33 -04:00
|
|
|
x = _G_INIT_DATA_.CENTER_RIGHT_X,
|
2017-07-17 00:21:31 -04:00
|
|
|
y = header.bottom_y + _PLOT_SEC_BREAK_,
|
2017-07-16 14:50:33 -04:00
|
|
|
width = _G_INIT_DATA_.SECTION_WIDTH,
|
2017-07-17 00:21:31 -04:00
|
|
|
height = _PLOT_HEIGHT_,
|
|
|
|
y_label_func = network_label_function
|
2016-08-11 23:25:56 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
local _UPLOAD_Y_ = header.bottom_y + _PLOT_HEIGHT_ + _PLOT_SEC_BREAK_ * 2
|
2016-08-11 23:25:56 -04:00
|
|
|
|
|
|
|
local upload = {
|
|
|
|
label = Widget.Text{
|
2017-07-16 14:50:33 -04:00
|
|
|
x = _G_INIT_DATA_.CENTER_RIGHT_X,
|
2017-07-17 00:21:31 -04:00
|
|
|
y = _UPLOAD_Y_,
|
2016-08-11 23:25:56 -04:00
|
|
|
text = 'Upload',
|
|
|
|
},
|
|
|
|
speed = Widget.Text{
|
2017-07-17 00:21:31 -04:00
|
|
|
x = _RIGHT_X_,
|
|
|
|
y = _UPLOAD_Y_,
|
2016-08-11 23:25:56 -04:00
|
|
|
x_align = 'right',
|
|
|
|
text_color = schema.blue
|
|
|
|
},
|
|
|
|
plot = Widget.ScalePlot{
|
2017-07-16 14:50:33 -04:00
|
|
|
x = _G_INIT_DATA_.CENTER_RIGHT_X,
|
2017-07-17 00:21:31 -04:00
|
|
|
y = _UPLOAD_Y_ + _PLOT_SEC_BREAK_,
|
2017-07-16 14:50:33 -04:00
|
|
|
width = _G_INIT_DATA_.SECTION_WIDTH,
|
2017-07-17 00:21:31 -04:00
|
|
|
height = _PLOT_HEIGHT_,
|
|
|
|
y_label_func = network_label_function
|
2016-08-11 23:25:56 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
local interfaces = {}
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
local add_interface = function(iface)
|
2016-08-11 23:25:56 -04:00
|
|
|
local rx_path = SYSFS_NET..iface..STATS_RX
|
|
|
|
local tx_path = SYSFS_NET..iface..STATS_TX
|
|
|
|
|
|
|
|
interfaces[iface] = {
|
|
|
|
rx_path = rx_path,
|
|
|
|
tx_path = tx_path,
|
|
|
|
rx_cumulative_bytes = 0,
|
|
|
|
tx_cumulative_bytes = 0,
|
|
|
|
prev_rx_cumulative_bytes = util.read_file(rx_path, nil, '*n'),
|
|
|
|
prev_tx_cumulative_bytes = util.read_file(tx_path, nil, '*n'),
|
|
|
|
}
|
|
|
|
end
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
for iface in io.popen('ls -1 '..SYSFS_NET):lines() do
|
|
|
|
add_interface(iface)
|
2016-08-11 23:25:56 -04:00
|
|
|
end
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
local update = function(cr, update_frequency)
|
2016-08-11 23:25:56 -04:00
|
|
|
local dspeed, uspeed = 0, 0
|
|
|
|
local glob = util.execute_cmd('ip route show')
|
|
|
|
|
|
|
|
local rx_bps, tx_bps
|
|
|
|
|
2017-07-17 00:21:31 -04:00
|
|
|
for iface in __string_gmatch(glob, 'default via %d+%.%d+%.%d+%.%d+ dev (%w+) ') do
|
2016-08-11 23:25:56 -04:00
|
|
|
local current_iface = interfaces[iface]
|
|
|
|
|
|
|
|
if not current_iface then
|
2017-07-17 00:21:31 -04:00
|
|
|
add_interface(iface)
|
2016-08-11 23:25:56 -04:00
|
|
|
current_iface = interfaces[iface]
|
|
|
|
end
|
|
|
|
|
|
|
|
local new_rx_cumulative_bytes = util.read_file(current_iface.rx_path, nil, '*n')
|
|
|
|
local new_tx_cumulative_bytes = util.read_file(current_iface.tx_path, nil, '*n')
|
|
|
|
|
|
|
|
rx_bps = (new_rx_cumulative_bytes - current_iface.prev_rx_cumulative_bytes) * update_frequency
|
|
|
|
tx_bps = (new_tx_cumulative_bytes - current_iface.prev_tx_cumulative_bytes) * update_frequency
|
|
|
|
|
|
|
|
current_iface.prev_rx_cumulative_bytes = new_rx_cumulative_bytes
|
|
|
|
current_iface.prev_tx_cumulative_bytes = new_tx_cumulative_bytes
|
|
|
|
|
|
|
|
--mask overflow
|
|
|
|
if rx_bps < 0 then rx_bps = 0 end
|
|
|
|
if tx_bps < 0 then tx_bps = 0 end
|
|
|
|
|
|
|
|
dspeed = dspeed + rx_bps
|
|
|
|
uspeed = uspeed + tx_bps
|
|
|
|
end
|
|
|
|
|
|
|
|
local dspeed_unit = util.get_unit(dspeed)
|
|
|
|
local uspeed_unit = util.get_unit(uspeed)
|
|
|
|
|
|
|
|
dnload.speed.append_end = ' '..dspeed_unit..'/s'
|
|
|
|
upload.speed.append_end = ' '..uspeed_unit..'/s'
|
|
|
|
|
|
|
|
Text.set(dnload.speed, cr, util.precision_convert_bytes(dspeed, 'B', dspeed_unit, 3))
|
|
|
|
Text.set(upload.speed, cr, util.precision_convert_bytes(uspeed, 'B', uspeed_unit, 3))
|
|
|
|
|
|
|
|
ScalePlot.update(dnload.plot, cr, dspeed)
|
|
|
|
ScalePlot.update(upload.plot, cr, uspeed)
|
|
|
|
end
|
|
|
|
|
|
|
|
Widget = nil
|
|
|
|
schema = nil
|
2017-07-17 00:21:31 -04:00
|
|
|
_PLOT_SEC_BREAK_ = nil
|
|
|
|
_PLOT_HEIGHT_ = nil
|
|
|
|
_RIGHT_X_ = nil
|
|
|
|
_UPLOAD_Y_ = nil
|
2016-08-11 23:25:56 -04:00
|
|
|
|
|
|
|
local draw = function(cr, current_interface, update_frequency)
|
2017-07-17 00:21:31 -04:00
|
|
|
update(cr, update_frequency)
|
2016-08-11 23:25:56 -04:00
|
|
|
|
|
|
|
if current_interface == 0 then
|
|
|
|
Text.draw(header.text, cr)
|
|
|
|
Line.draw(header.underline, cr)
|
|
|
|
|
|
|
|
Text.draw(dnload.label, cr)
|
|
|
|
Text.draw(dnload.speed, cr)
|
|
|
|
ScalePlot.draw(dnload.plot, cr)
|
|
|
|
|
|
|
|
Text.draw(upload.label, cr)
|
|
|
|
Text.draw(upload.speed, cr)
|
|
|
|
ScalePlot.draw(upload.plot, cr)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
return draw
|