module("luci.statistics.datatree", package.seeall) local util = require("luci.util") local sys = require("luci.sys") local fs = require("luci.fs") local uci = require("luci.model.uci").Session() local sections, names = uci:sections( "luci_statistics" ) Instance = util.class() function Instance.__init__( self, host ) self._host = host or sections.collectd.Hostname or sys.hostname() self._libdir = sections.collectd.PluginDir or "/usr/lib/collectd" self._rrddir = sections.collectd_rrdtool.DataDir or "/tmp" self._libdir = self._libdir:gsub("/$","") self._rrddir = self._rrddir:gsub("/$","") self._plugins = { } self:_scan() end function Instance._mkpath( self, plugin, pinstance ) local dir = self._rrddir .. "/" .. self._host if type(plugin) == "string" and plugin:len() > 0 then dir = dir .. "/" .. plugin if type(pinstance) == "string" and pinstance:len() > 0 then dir = dir .. "-" .. pinstance end end return dir end function Instance._notzero( self, table ) for k in pairs(table) do return true end return false end function Instance._scan( self ) for i, plugin in ipairs( fs.dir( self._libdir ) ) do if plugin:match("%w+.so") then self._plugins[ plugin:gsub(".so", "") ] = { } end end for plugin, instances in pairs( self._plugins ) do for i, dir in ipairs( fs.dir( self:_mkpath() ) ) do if dir:find( plugin .. "%-" ) or dir == plugin then local instance = "" if dir ~= plugin then instance = dir:gsub( plugin .. "%-", "", 1 ) end instances[instance] = { } end end for instance, data_instances in pairs( instances ) do for i, file in ipairs( fs.dir( self:_mkpath( plugin, instance ) ) ) do if file:find("%.rrd") then file = file:gsub("%.rrd","") local data_type local data_instance if file:find("%-") then data_type = file:gsub( "%-.+","" ) data_instance = file:gsub( "[^%-]-%-", "", 1 ) else data_type = file data_instance = "" end if not data_instances[data_type] then data_instances[data_type] = { data_instance } else table.insert( data_instances[data_type], data_instance ) end end end end end end function Instance.plugins( self ) local rv = { } for plugin, val in pairs( self._plugins ) do if self:_notzero( val ) then table.insert( rv, plugin ) end end return rv end function Instance.plugin_instances( self, plugin ) local rv = { } for instance, val in pairs( self._plugins[plugin] ) do table.insert( rv, instance ) end return rv end function Instance.data_types( self, plugin, instance ) local rv = { } for type, val in pairs( self._plugins[plugin][instance] ) do table.insert( rv, type ) end return rv end function Instance.data_instances( self, plugin, instance, type ) local rv = { } for i, instance in ipairs( self._plugins[plugin][instance][type] ) do table.insert( rv, instance ) end return rv end