conky-config/drawing/network.lua

110 lines
3.0 KiB
Lua

local util = require 'util'
local common = require 'common'
local geometry = require 'geometry'
local func = require 'func'
return function(update_freq)
local PLOT_SEC_BREAK = 20
local PLOT_HEIGHT = 56
local get_interfaces = function()
local s = util.execute_cmd('realpath /sys/class/net/* | grep -v virtual')
local interfaces = {}
for iface in string.gmatch(s, '/([^/\n]+)\n') do
interfaces[#interfaces + 1] = iface
end
return interfaces
end
local INTERFACES = get_interfaces()
local INTERFACE_PATHS = func.map(
function(s)
local dir = string.format('/sys/class/net/%s/statistics/', s)
return {rx = dir..'rx_bytes', tx = dir..'tx_bytes'}
end,
INTERFACES
)
local get_bits = function(path)
return util.read_file(path, nil, '*n') * 8
end
local read_interfaces = function()
local rx = 0
local tx = 0
for i = 1, #INTERFACE_PATHS do
local p = INTERFACE_PATHS[i]
rx = rx + get_bits(p.rx)
tx = tx + get_bits(p.tx)
end
return rx, tx
end
local init_rx_bits, init_tx_bits = read_interfaces()
local value_format_function = function(bits)
local unit, value = util.convert_data_val(bits)
return util.precision_round_to_string(value, 3)..' '..unit..'b/s'
end
local make_plot = function(y, label, init)
return common.make_rate_timeseries(
geometry.CENTER_RIGHT_X,
y,
geometry.SECTION_WIDTH,
PLOT_HEIGHT,
value_format_function,
common.converted_y_label_format_generator('b'),
PLOT_SEC_BREAK,
label,
2,
update_freq,
init
)
end
-----------------------------------------------------------------------------
-- header
local header = common.make_header(
geometry.CENTER_RIGHT_X,
geometry.TOP_Y,
geometry.SECTION_WIDTH,
'NETWORK'
)
-----------------------------------------------------------------------------
-- download plot
local rx = make_plot(header.bottom_y, 'Download', init_rx_bits)
-----------------------------------------------------------------------------
-- upload plot
local TX_Y = header.bottom_y + PLOT_HEIGHT + PLOT_SEC_BREAK * 2
local tx = make_plot(TX_Y, 'Upload', init_tx_bits)
-----------------------------------------------------------------------------
-- main drawing functions
local update = function()
local rx_bits, tx_bits = read_interfaces()
common.update_rate_timeseries(rx, rx_bits)
common.update_rate_timeseries(tx, tx_bits)
end
local draw_static = function(cr)
common.draw_header(cr, header)
common.labeled_scale_plot_draw_static(rx, cr)
common.labeled_scale_plot_draw_static(tx, cr)
end
local draw_dynamic = function(cr)
common.labeled_scale_plot_draw_dynamic(rx, cr)
common.labeled_scale_plot_draw_dynamic(tx, cr)
end
return {static = draw_static, dynamic = draw_dynamic, update = update}
end