awesome-widgets/sources/ext-sysmon/extscript.cpp
2014-10-31 06:13:49 +03:00

298 lines
8.1 KiB
C++

/***************************************************************************
* This file is part of awesome-widgets *
* *
* awesome-widgets is free software: you can redistribute it and/or *
* modify it under the terms of the GNU General Public License as *
* published by the Free Software Foundation, either version 3 of the *
* License, or (at your option) any later version. *
* *
* awesome-widgets is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with awesome-widgets. If not, see http://www.gnu.org/licenses/ *
***************************************************************************/
#include "extscript.h"
#include <QDebug>
#include <QDir>
#include <QTextCodec>
#include <pdebug/pdebug.h>
#include <task/taskadds.h>
ExtScript::ExtScript(const QString scriptName, const QStringList directories, const bool debugCmd)
: QObject(0),
name(scriptName),
dirs(directories),
debug(debugCmd)
{
readConfiguration();
}
ExtScript::~ExtScript()
{
if (debug) qDebug() << PDEBUG;
}
void ExtScript::addDirectory(const QString dir)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "Directory" << dir;
QString absPath = QDir(dir).absolutePath();
if (!QDir(absPath).exists()) return;
for (int i=0; i<dirs.count(); i++)
if (dirs[i] == absPath) return;
dirs.append(absPath);
}
QStringList ExtScript::directories()
{
if (debug) qDebug() << PDEBUG;
return dirs;
}
int ExtScript::getInterval()
{
if (debug) qDebug() << PDEBUG;
return interval;
}
QString ExtScript::getName()
{
if (debug) qDebug() << PDEBUG;
return name;
}
QString ExtScript::getPrefix()
{
if (debug) qDebug() << PDEBUG;
return prefix;
}
ExtScript::Redirect ExtScript::getRedirect()
{
if (debug) qDebug() << PDEBUG;
return redirect;
}
bool ExtScript::hasOutput()
{
if (debug) qDebug() << PDEBUG;
return output;
}
bool ExtScript::isActive()
{
if (debug) qDebug() << PDEBUG;
return active;
}
void ExtScript::setActive(const bool state)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "State" << state;
active = state;
}
void ExtScript::setDirectories(const QStringList directories)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "Directories" << directories;
for (int i=0; i<directories.count(); i++)
addDirectory(directories[i]);
}
void ExtScript::setHasOutput(const bool state)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "State" << state;
output = state;
}
void ExtScript::setInterval(const int inter)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "Interval" << inter;
if (inter <= 0) return;
interval = inter;
}
void ExtScript::setPrefix(const QString pref)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "Prefix" << pref;
prefix = pref;
}
void ExtScript::setRedirect(const Redirect redir)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "Redirect" << redir;
redirect = redir;
}
void ExtScript::readConfiguration()
{
if (debug) qDebug() << PDEBUG;
QMap<QString, QString> settings;
for (int i=0; i<dirs.count(); i++) {
if (!QDir(dirs[i]).entryList(QDir::Files).contains(name + QString(".conf"))) continue;
QString fileName = dirs[i] + QDir::separator() + name + QString(".conf");
QMap<QString, QString> newSettings = getConfigurationFromFile(fileName);
for (int i=0; i<newSettings.keys().count(); i++)
settings[newSettings.keys()[i]] = newSettings[newSettings.keys()[i]];
}
fromExternalConfiguration(settings);
}
ExtScript::ScriptData ExtScript::run(const int time)
{
if (debug) qDebug() << PDEBUG;
ScriptData response;
response.active = active;
response.name = name;
response.refresh = false;
if (!active) return response;
if (time != interval) return response;
response.refresh = true;
QStringList cmdList;
if (!prefix.isEmpty())
cmdList.append(prefix);
QString fullPath = name;
for (int i=0; i<dirs.count(); i++) {
if (!QDir(dirs[i]).entryList(QDir::Files).contains(name)) continue;
fullPath = dirs[i] + QDir::separator() + name;
break;
}
cmdList.append(fullPath);
if (debug) qDebug() << PDEBUG << ":" << "cmd" << cmdList.join(QChar(' '));
TaskResult process = runTask(cmdList.join(QChar(' ')));
if (debug) qDebug() << PDEBUG << ":" << "Cmd returns" << process.exitCode;
QString info = QTextCodec::codecForMib(106)->toUnicode(process.error).trimmed();
QString qoutput = QTextCodec::codecForMib(106)->toUnicode(process.output).trimmed();
switch(redirect) {
case stdout2stderr:
if (debug) qDebug() << PDEBUG << ":" << "Debug" << info;
if (debug) qDebug() << PDEBUG << ":" << "Output" << qoutput;
break;
case stderr2stdout:
response.output = info + QString("\t") + qoutput;
break;
default:
if (debug) qDebug() << PDEBUG << ":" << "Debug" << info;
response.output = qoutput;
break;
}
if (!output)
response.output = QString::number(process.exitCode);
return response;
}
void ExtScript::fromExternalConfiguration(const QMap<QString, QString> settings)
{
if (settings.contains(QString("ACTIVE")))
setActive(settings[QString("ACTIVE")] == QString("true"));
if (settings.contains(QString("INTERVAL")))
setInterval(settings[QString("INTERVAL")].toInt());
if (settings.contains(QString("PREFIX")))
setPrefix(settings[QString("PREFIX")]);
if (settings.contains(QString("OUTPUT")))
setHasOutput(settings[QString("OUTPUT")] == QString("true"));
if (settings.contains(QString("REDIRECT")))
setRedirect((Redirect)settings[QString("REDIRECT")].toInt());
if (!output)
redirect = stdout2stderr;
}
QMap<QString, QString> ExtScript::getConfigurationFromFile(const QString fileName)
{
if (debug) qDebug() << PDEBUG;
if (debug) qDebug() << PDEBUG << ":" << "File" << fileName;
QMap<QString, QString> settings;
QFile configFile(fileName);
QString fileStr;
if (!configFile.open(QIODevice::ReadOnly)) return settings;
while (true) {
fileStr = QString(configFile.readLine()).trimmed();
if ((fileStr.isEmpty()) && (!configFile.atEnd())) continue;
if ((fileStr[0] == QChar('#')) && (!configFile.atEnd())) continue;
if ((fileStr[0] == QChar(';')) && (!configFile.atEnd())) continue;
if (fileStr.contains(QChar('=')))
settings[fileStr.split(QChar('='))[0]] = fileStr.split(QChar('='))[1];
if (configFile.atEnd()) break;
}
configFile.close();
for (int i=0; i<settings.keys().count(); i++)
if (debug) qDebug() << PDEBUG << ":" << settings.keys()[i] + QString("=") + settings[settings.keys()[i]];
return settings;
}
QMap<QString, QString> ExtScript::toExternalConfiguration()
{
if (debug) qDebug() << PDEBUG;
QMap<QString, QString> settings;
if (active)
settings[QString("ACTIVE")] = QString("true");
else
settings[QString("ACTIVE")] = QString("false");
settings[QString("INTERVAL")] = QString::number(interval);
settings[QString("PREFIX")] = prefix;
if (output)
settings[QString("OUTPUT")] = QString("true");
else
settings[QString("OUTPUT")] = QString("false");
settings[QString("REDIRECT")] = QString::number(redirect);
return settings;
}