mirror of
https://github.com/arcan1s/awesome-widgets.git
synced 2025-12-16 10:03:40 +00:00
split data engine matchers to classes
This commit is contained in:
@ -22,6 +22,7 @@
|
||||
#include "awdebug.h"
|
||||
#include "awformatterhelper.h"
|
||||
#include "formatters/formatters.h"
|
||||
#include "matchers/matchers.h"
|
||||
|
||||
|
||||
AWDataEngineMapper::AWDataEngineMapper(QObject *_parent, AWFormatterHelper *_custom)
|
||||
@ -31,10 +32,6 @@ AWDataEngineMapper::AWDataEngineMapper(QObject *_parent, AWFormatterHelper *_cus
|
||||
qCDebug(LOG_AW) << __PRETTY_FUNCTION__;
|
||||
|
||||
// default formatters
|
||||
// memory
|
||||
m_formatter["mem"] = AWPluginFormatterFloat::instance();
|
||||
m_formatter["memtotmb"] = AWPluginFormatterMemoryMB::instance();
|
||||
m_formatter["memtotgb"] = AWPluginFormatterMemoryGB::instance();
|
||||
// network
|
||||
m_formatter["down"] = AWPluginFormatterNet::instance();
|
||||
m_formatter["downkb"] = AWPluginFormatterMemory::instance();
|
||||
@ -46,10 +43,6 @@ AWDataEngineMapper::AWDataEngineMapper(QObject *_parent, AWFormatterHelper *_cus
|
||||
m_formatter["uptot"] = AWPluginFormatterMemoryMB::instance();
|
||||
m_formatter["uptotkb"] = AWPluginFormatterMemory::instance();
|
||||
m_formatter["upunits"] = AWPluginFormatterNetUnits::instance();
|
||||
// swap
|
||||
m_formatter["swap"] = AWPluginFormatterFloat::instance();
|
||||
m_formatter["swaptotmb"] = AWPluginFormatterMemoryMB::instance();
|
||||
m_formatter["swaptotgb"] = AWPluginFormatterMemoryGB::instance();
|
||||
}
|
||||
|
||||
|
||||
@ -76,377 +69,44 @@ QStringList AWDataEngineMapper::registerSource(const QString &_source, const KSy
|
||||
{
|
||||
qCDebug(LOG_AW) << "Source" << _source << "with units" << _units;
|
||||
|
||||
// regular expressions
|
||||
static auto cpuRegExp = QRegularExpression("^cpu/cpu.*/usage$");
|
||||
static auto cpuclRegExp = QRegularExpression("^cpu/cpu.*/frequency$");
|
||||
static auto cpuTempRegExp = QRegularExpression("^cpu/cpu.*/temperature$");
|
||||
static auto gpuRegExp = QRegularExpression("^gpu/gpu.*/usage$");
|
||||
static auto gpuTempRegExp = QRegularExpression("^gpu/gpu.*/temperature$");
|
||||
static auto hddrRegExp = QRegularExpression("^disk/.*/read$");
|
||||
static auto hddwRegExp = QRegularExpression("^disk/.*/write$");
|
||||
static auto mountFillRegExp = QRegularExpression("^disk/.*/usedPercent$");
|
||||
static auto mountFreeRegExp = QRegularExpression("^disk/.*/free$");
|
||||
static auto mountUsedRegExp = QRegularExpression("^disk/.*/used$");
|
||||
static auto netRegExp = QRegularExpression("^network/.*/(download|upload)$");
|
||||
static auto netTotalRegExp = QRegularExpression("^network/.*/(totalDownload|totalUpload)$");
|
||||
auto matchers = AWPluginMatchers::matchers;
|
||||
auto matcher = std::find_if(matchers.cbegin(), matchers.cend(),
|
||||
[&_source](auto matcher) { return matcher->matches(_source); });
|
||||
if (matcher == matchers.cend())
|
||||
return {};
|
||||
|
||||
if (_source == "extsysmon/battery/ac") {
|
||||
// AC
|
||||
m_map.insert(_source, "ac");
|
||||
m_formatter["ac"] = AWPluginFormatterAC::instance();
|
||||
} else if (_source.startsWith("extsysmon/battery/")) {
|
||||
// battery stats
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/battery/");
|
||||
m_map.insert(_source, key);
|
||||
if (_source.contains("rate"))
|
||||
m_formatter[key] = AWPluginFormatterFloat::instance();
|
||||
else
|
||||
m_formatter[key] = AWPluginFormatterIntegerShort::instance();
|
||||
} else if (_source == "cpu/all/usage") {
|
||||
// cpu
|
||||
m_map.insert(_source, "cpu");
|
||||
m_formatter["cpu"] = AWPluginFormatterFloat::instance();
|
||||
} else if (_source.contains(cpuRegExp)) {
|
||||
// cpus
|
||||
auto key = _source;
|
||||
key.remove("cpu/").remove("/usage");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterFloat::instance();
|
||||
} else if (_source == "cpu/all/averageFrequency") {
|
||||
// cpucl
|
||||
m_map.insert(_source, "cpucl");
|
||||
m_formatter["cpucl"] = AWPluginFormatterInteger::instance();
|
||||
} else if (_source.contains(cpuclRegExp)) {
|
||||
// cpucls
|
||||
auto key = _source;
|
||||
key.remove("cpu/cpu").remove("/frequency");
|
||||
key = QString("cpucl%1").arg(key);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterInteger::instance();
|
||||
} else if (_source.startsWith("extsysmon/custom")) {
|
||||
// custom
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/custom/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source == "extsysmon/desktop/name") {
|
||||
// current desktop name
|
||||
m_map.insert(_source, "desktop");
|
||||
m_formatter["desktop"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source == "extsysmon/desktop/number") {
|
||||
// current desktop number
|
||||
m_map.insert(_source, "ndesktop");
|
||||
m_formatter["ndesktop"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source == "extsysmon/desktop/count") {
|
||||
// desktop count
|
||||
m_map.insert(_source, "tdesktops");
|
||||
m_formatter["tdesktops"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source.contains(hddrRegExp)) {
|
||||
// read speed
|
||||
auto device = _source;
|
||||
device.remove("disk/").remove("/read");
|
||||
auto index = m_devices["disk"].indexOf(device);
|
||||
if (index > -1) {
|
||||
QString key = QString("hddr%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemory::instance();
|
||||
}
|
||||
} else if (_source.contains(hddwRegExp)) {
|
||||
// write speed
|
||||
auto device = _source;
|
||||
device.remove("disk/").remove("/write");
|
||||
auto index = m_devices["disk"].indexOf(device);
|
||||
if (index > -1) {
|
||||
QString key = QString("hddw%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemory::instance();
|
||||
}
|
||||
} else if (_source == "gpu/all/usage") {
|
||||
// gpu load
|
||||
m_map.insert(_source, "gpu");
|
||||
m_formatter["gpu"] = AWPluginFormatterFloat::instance();
|
||||
} else if (_source.contains(gpuRegExp)) {
|
||||
// gpus
|
||||
auto device = _source;
|
||||
device.remove("gpu/").remove("/usage");
|
||||
auto index = m_devices["gpu"].indexOf(device);
|
||||
if (index > -1) {
|
||||
auto key = QString("gpu%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterFloat::instance();
|
||||
}
|
||||
} else if (_source.contains(gpuTempRegExp)) {
|
||||
// gpus temps
|
||||
auto device = _source;
|
||||
device.remove("gpu/").remove("/temperature");
|
||||
auto index = m_devices["gpu"].indexOf(device);
|
||||
if (index > -1) {
|
||||
auto key = QString("gputemp%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterTemperature::instance();
|
||||
}
|
||||
} else if (_source.contains(mountFillRegExp)) {
|
||||
// fill level
|
||||
auto device = _source;
|
||||
device.remove("disk/").remove("/usedPercent");
|
||||
auto index = m_devices["mount"].indexOf(device);
|
||||
if (index > -1) {
|
||||
auto key = QString("hdd%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterFloat::instance();
|
||||
// additional keys
|
||||
m_formatter[QString("hddtotmb%1").arg(index)] = AWPluginFormatterMemoryMB::instance();
|
||||
m_formatter[QString("hddtotgb%1").arg(index)] = AWPluginFormatterMemoryGB::instance();
|
||||
}
|
||||
} else if (_source.contains(mountFreeRegExp)) {
|
||||
// free space
|
||||
auto device = _source;
|
||||
device.remove("disk/").remove("/free");
|
||||
auto index = m_devices["mount"].indexOf(device);
|
||||
if (index > -1) {
|
||||
// mb
|
||||
auto key = QString("hddfreemb%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemoryMB::instance();
|
||||
// gb
|
||||
key = QString("hddfreegb%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemoryGB::instance();
|
||||
}
|
||||
} else if (_source.contains(mountUsedRegExp)) {
|
||||
// used
|
||||
auto device = _source;
|
||||
device.remove("disk/").remove("/used");
|
||||
auto index = m_devices["mount"].indexOf(device);
|
||||
if (index > -1) {
|
||||
// mb
|
||||
auto key = QString("hddmb%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemoryMB::instance();
|
||||
// gb
|
||||
key = QString("hddgb%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemoryGB::instance();
|
||||
}
|
||||
} else if (_source.startsWith("cpu/loadaverages/loadaverage")) {
|
||||
// load average
|
||||
auto time = _source;
|
||||
time.remove("cpu/loadaverages/loadaverage");
|
||||
auto key = QString("la%1").arg(time);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterFloatPrecise::instance();
|
||||
} else if (_source == "memory/physical/application") {
|
||||
// app memory
|
||||
// mb
|
||||
m_map.insert(_source, "memmb");
|
||||
m_formatter["memmb"] = AWPluginFormatterMemoryMB::instance();
|
||||
// gb
|
||||
m_map.insert(_source, "memgb");
|
||||
m_formatter["memgb"] = AWPluginFormatterMemoryGB::instance();
|
||||
} else if (_source == "memory/physical/free") {
|
||||
// free memory
|
||||
// mb
|
||||
m_map.insert(_source, "memfreemb");
|
||||
m_formatter["memfreemb"] = AWPluginFormatterMemoryMB::instance();
|
||||
// gb
|
||||
m_map.insert(_source, "memfreegb");
|
||||
m_formatter["memfreegb"] = AWPluginFormatterMemoryGB::instance();
|
||||
} else if (_source == "memory/physical/used") {
|
||||
// used memory
|
||||
// mb
|
||||
m_map.insert(_source, "memusedmb");
|
||||
m_formatter["memusedmb"] = AWPluginFormatterMemoryMB::instance();
|
||||
// gb
|
||||
m_map.insert(_source, "memusedgb");
|
||||
m_formatter["memusedgb"] = AWPluginFormatterMemoryGB::instance();
|
||||
} else if (_source == "extsysmon/network/device") {
|
||||
// network device
|
||||
m_map.insert(_source, "netdev");
|
||||
m_formatter["netdev"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source == "extsysmon/network/ssid") {
|
||||
// current ssid
|
||||
m_map.insert(_source, "ssid");
|
||||
m_formatter["ssid"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source.startsWith("extsysmon/requests/response")) {
|
||||
// network response
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/requests/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source.contains(netRegExp)) {
|
||||
// network speed
|
||||
auto type = _source.endsWith("download") ? "down" : "up";
|
||||
auto index = m_devices["net"].indexOf(_source.split('/')[1]);
|
||||
if (index > -1) {
|
||||
// kb
|
||||
auto key = QString("%1kb%2").arg(type).arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemory::instance();
|
||||
// smart
|
||||
key = QString("%1%2").arg(type).arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterNet::instance();
|
||||
// units
|
||||
key = QString("%1units%2").arg(type).arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterNetUnits::instance();
|
||||
}
|
||||
} else if (_source.contains(netTotalRegExp)) {
|
||||
// network data total
|
||||
auto type = _source.endsWith("Download") ? "down" : "up";
|
||||
auto index = m_devices["net"].indexOf(_source.split('/')[1]);
|
||||
if (index > -1) {
|
||||
// kb
|
||||
auto key = QString("%1totkb%2").arg(type).arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemory::instance();
|
||||
// mb
|
||||
key = QString("%1tot%2").arg(type).arg(index);
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterMemoryMB::instance();
|
||||
}
|
||||
} else if (_source.startsWith("extsysmon/upgrade")) {
|
||||
// package manager
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/upgrade/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterIntegerShort::instance();
|
||||
} else if (_source.startsWith("extsysmon/player")) {
|
||||
// player
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/player/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source == "extsysmon/ps/running") {
|
||||
// running processes count
|
||||
m_map.insert(_source, "pscount");
|
||||
m_formatter["pscount"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source == "extsysmon/ps/list") {
|
||||
// list of running processes
|
||||
m_map.insert(_source, "ps");
|
||||
m_formatter["ps"] = AWPluginFormatterList::instance();
|
||||
} else if (_source == "extsysmon/ps/count") {
|
||||
// total processes count
|
||||
m_map.insert(_source, "pstot");
|
||||
m_formatter["pstot"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source.startsWith("extsysmon/quotes")) {
|
||||
// quotes
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/quotes/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterDouble::instance();
|
||||
} else if (_source == "memory/swap/free") {
|
||||
// free swap
|
||||
// mb
|
||||
m_map.insert(_source, "swapfreemb");
|
||||
m_formatter["swapfreemb"] = AWPluginFormatterMemoryMB::instance();
|
||||
// gb
|
||||
m_map.insert(_source, "swapfreegb");
|
||||
m_formatter["swapfreegb"] = AWPluginFormatterMemoryGB::instance();
|
||||
} else if (_source == "memory/swap/used") {
|
||||
// used swap
|
||||
// mb
|
||||
m_map.insert(_source, "swapmb");
|
||||
m_formatter["swapmb"] = AWPluginFormatterMemoryMB::instance();
|
||||
// gb
|
||||
m_map.insert(_source, "swapgb");
|
||||
m_formatter["swapgb"] = AWPluginFormatterMemoryGB::instance();
|
||||
} else if (_source.startsWith("lmsensors/") || _source.contains(cpuTempRegExp)
|
||||
|| _source == "cpu/all/averageTemperature") {
|
||||
// temperature
|
||||
auto index = m_devices["temp"].indexOf(_source);
|
||||
// HACK on DE initialization there are no units key
|
||||
if (_units == KSysGuard::UnitInvalid)
|
||||
return QStringList({QString("temp%1").arg(index)});
|
||||
if (index > -1) {
|
||||
auto key = QString("temp%1").arg(index);
|
||||
m_map.insert(_source, key);
|
||||
if (_units == KSysGuard::UnitCelsius)
|
||||
m_formatter[key] = AWPluginFormatterTemperature::instance();
|
||||
else
|
||||
m_formatter[key] = AWPluginFormatterInteger::instance();
|
||||
}
|
||||
} else if (_source == "extsysmon/time/now") {
|
||||
// time
|
||||
m_map.insert(_source, "time");
|
||||
m_formatter["time"] = AWPluginFormatterTime::instance();
|
||||
// custom time
|
||||
m_map.insert(_source, "ctime");
|
||||
m_formatter["ctime"] = AWPluginFormatterTimeCustom::instance();
|
||||
// ISO time
|
||||
m_map.insert(_source, "isotime");
|
||||
m_formatter["isotime"] = AWPluginFormatterTimeISO::instance();
|
||||
// long time
|
||||
m_map.insert(_source, "longtime");
|
||||
m_formatter["longtime"] = AWPluginFormatterTimeLong::instance();
|
||||
// short time
|
||||
m_map.insert(_source, "shorttime");
|
||||
m_formatter["shorttime"] = AWPluginFormatterTimeShort::instance();
|
||||
// timestamp
|
||||
m_map.insert(_source, "tstime");
|
||||
m_formatter["tstime"] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source == "extsysmon/system/brightness") {
|
||||
m_map.insert(_source, "brightness");
|
||||
m_formatter["brightness"] = AWPluginFormatterIntegerShort::instance();
|
||||
} else if (_source == "extsysmon/system/volume") {
|
||||
m_map.insert(_source, "volume");
|
||||
m_formatter["volume"] = AWPluginFormatterIntegerShort::instance();
|
||||
} else if (_source == "os/system/uptime") {
|
||||
// uptime
|
||||
m_map.insert(_source, "uptime");
|
||||
m_formatter["uptime"] = AWPluginFormatterUptime::instance();
|
||||
} else if (_source.startsWith("extsysmon/weather/temperature")) {
|
||||
// temperature
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/weather/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterTemperature::instance();
|
||||
} else if (_source.startsWith("extsysmon/weather/")) {
|
||||
// other weather
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/weather/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterNoFormat::instance();
|
||||
} else if (_source.startsWith("extsysmon/load/load")) {
|
||||
// load source
|
||||
auto key = _source;
|
||||
key.remove("extsysmon/load/");
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = AWPluginFormatterTemperature::instance();
|
||||
}
|
||||
|
||||
auto foundKeys = keysFromSource(_source);
|
||||
auto foundKeys = (*matcher)->keys(_source, _units, m_settings);
|
||||
auto keys = foundKeys.keys(); // speedup a little bit
|
||||
|
||||
// rewrite formatters for custom ones
|
||||
QStringList customFormattersKeys;
|
||||
if (m_customFormatters)
|
||||
customFormattersKeys = m_customFormatters->definedFormatters();
|
||||
qCInfo(LOG_AW) << "Looking for formatters" << foundKeys << "in" << customFormattersKeys;
|
||||
for (auto &key : foundKeys) {
|
||||
qCInfo(LOG_AW) << "Looking for formatters" << keys << "in" << customFormattersKeys;
|
||||
for (auto &key : keys) {
|
||||
if (!customFormattersKeys.contains(key))
|
||||
continue;
|
||||
m_formatter[key] = AWPluginFormatterCustom::instance();
|
||||
foundKeys[key] = AWPluginFormatterCustom::instance();
|
||||
}
|
||||
|
||||
// drop key from dictionary if no one user requested key required it
|
||||
qCInfo(LOG_AW) << "Looking for keys" << foundKeys << "in" << _keys;
|
||||
auto required = _keys.isEmpty() || std::any_of(foundKeys.cbegin(), foundKeys.cend(), [&_keys](auto &key) {
|
||||
return _keys.contains(key);
|
||||
});
|
||||
if (!required) {
|
||||
m_map.remove(_source);
|
||||
for (auto &key : foundKeys)
|
||||
m_formatter.remove(key);
|
||||
// check if keys were actually requested
|
||||
qCInfo(LOG_AW) << "Looking for keys" << keys << "in" << _keys;
|
||||
auto required = _keys.isEmpty()
|
||||
|| std::any_of(keys.cbegin(), keys.cend(), [&_keys](auto &key) { return _keys.contains(key); });
|
||||
if (!required)
|
||||
return {};
|
||||
|
||||
// insert keys into memory
|
||||
for (auto [key, formatter] : foundKeys.asKeyValueRange()) {
|
||||
m_map.insert(_source, key);
|
||||
m_formatter[key] = formatter;
|
||||
}
|
||||
|
||||
return keysFromSource(_source);
|
||||
return keys;
|
||||
}
|
||||
|
||||
|
||||
void AWDataEngineMapper::setDevices(const QHash<QString, QStringList> &_devices)
|
||||
void AWDataEngineMapper::setDevices(const AWPluginMatcherSettings &_settings)
|
||||
{
|
||||
qCDebug(LOG_AW) << "Devices" << _devices;
|
||||
|
||||
m_devices = _devices;
|
||||
m_settings = _settings;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user