mirror of
https://github.com/Qortal/Brooklyn.git
synced 2025-02-12 02:05:54 +00:00
263 lines
9.1 KiB
C++
263 lines
9.1 KiB
C++
/*
|
|
SPDX-FileCopyrightText: 2008 Rob Scheepmaker <r.scheepmaker@student.utwente.nl>
|
|
|
|
SPDX-License-Identifier: LGPL-2.0-only
|
|
*/
|
|
|
|
#include "kuiserverengine.h"
|
|
#include "jobcontrol.h"
|
|
|
|
#include <QUrl>
|
|
|
|
#include <KFormat>
|
|
#include <KJob>
|
|
#include <KLocalizedString>
|
|
|
|
#include "notifications.h"
|
|
|
|
#include <algorithm>
|
|
|
|
using namespace NotificationManager;
|
|
|
|
KuiserverEngine::KuiserverEngine(QObject *parent, const QVariantList &args)
|
|
: Plasma::DataEngine(parent, args)
|
|
{
|
|
init();
|
|
}
|
|
|
|
KuiserverEngine::~KuiserverEngine()
|
|
{
|
|
}
|
|
|
|
QString KuiserverEngine::sourceName(Job *job)
|
|
{
|
|
return QStringLiteral("Job %1").arg(job->id());
|
|
}
|
|
|
|
uint KuiserverEngine::jobId(const QString &sourceName)
|
|
{
|
|
return sourceName.midRef(4 /*length of Job + space*/).toUInt();
|
|
}
|
|
|
|
Plasma::Service *KuiserverEngine::serviceForSource(const QString &source)
|
|
{
|
|
const uint id = jobId(source);
|
|
if (!id) {
|
|
return DataEngine::serviceForSource(source);
|
|
}
|
|
|
|
auto it = std::find_if(m_jobs.constBegin(), m_jobs.constBegin(), [&id](Job *job) {
|
|
return job->id() == id;
|
|
});
|
|
|
|
if (it == m_jobs.constEnd()) {
|
|
return DataEngine::serviceForSource(source);
|
|
}
|
|
|
|
return new JobControl(this, *it);
|
|
}
|
|
|
|
void KuiserverEngine::init()
|
|
{
|
|
m_jobsModel = JobsModel::createJobsModel();
|
|
// TODO see if this causes any issues when/if other processes are using applicationjobs engine, e.g. Latte Dock
|
|
m_jobsModel->init();
|
|
|
|
connect(m_jobsModel.data(), &Notifications::rowsInserted, this, [this](const QModelIndex &parent, int first, int last) {
|
|
for (int i = first; i <= last; ++i) {
|
|
const QModelIndex idx = m_jobsModel->index(first, 0, parent);
|
|
Job *job = idx.data(Notifications::JobDetailsRole).value<Job *>();
|
|
registerJob(job);
|
|
}
|
|
});
|
|
|
|
connect(m_jobsModel.data(), &Notifications::rowsAboutToBeRemoved, this, [this](const QModelIndex &parent, int first, int last) {
|
|
for (int i = first; i <= last; ++i) {
|
|
const QModelIndex idx = m_jobsModel->index(first, 0, parent);
|
|
Job *job = idx.data(Notifications::JobDetailsRole).value<Job *>();
|
|
removeJob(job);
|
|
}
|
|
});
|
|
}
|
|
|
|
void KuiserverEngine::updateDescriptionField(Job *job, int number, QString (Job::*labelGetter)() const, QString (Job::*valueGetter)() const)
|
|
{
|
|
const QString source = sourceName(job);
|
|
const QString labelString = QStringLiteral("label%1").arg(number);
|
|
const QString labelNameString = QStringLiteral("labelName%1").arg(number);
|
|
const QString labelFileNameString = QStringLiteral("labelFileName%1").arg(number);
|
|
|
|
const QString label = ((job)->*labelGetter)();
|
|
const QString value = ((job)->*valueGetter)();
|
|
|
|
if (label.isEmpty() && value.isEmpty()) {
|
|
setData(source, labelString, QVariant());
|
|
setData(source, labelNameString, QVariant());
|
|
setData(source, labelFileNameString, QVariant());
|
|
} else {
|
|
setData(source, labelNameString, label);
|
|
setData(source, labelString, value);
|
|
|
|
const QUrl url = QUrl::fromUserInput(value, QString(), QUrl::AssumeLocalFile);
|
|
setData(source, labelFileNameString, url.toString(QUrl::PreferLocalFile | QUrl::RemoveFragment | QUrl::RemoveQuery));
|
|
}
|
|
setData(source, labelString);
|
|
}
|
|
|
|
void KuiserverEngine::updateUnit(Job *job,
|
|
int number,
|
|
const QString &unit,
|
|
qulonglong (NotificationManager::Job::*processedGetter)() const,
|
|
qulonglong (NotificationManager::Job::*totalGetter)() const)
|
|
{
|
|
const QString source = sourceName(job);
|
|
|
|
setData(source, QStringLiteral("totalUnit%1").arg(number), unit);
|
|
setData(source, QStringLiteral("totalAmount%1").arg(number), ((job)->*totalGetter)());
|
|
setData(source, QStringLiteral("processedUnit%1").arg(number), unit);
|
|
setData(source, QStringLiteral("processedAmount%1").arg(number), ((job)->*processedGetter)());
|
|
}
|
|
|
|
void KuiserverEngine::registerJob(Job *job)
|
|
{
|
|
if (m_jobs.contains(job)) { // shouldn't really happen
|
|
return;
|
|
}
|
|
|
|
const QString source = sourceName(job);
|
|
|
|
setData(source, QStringLiteral("appName"), job->desktopEntry()); // job->applicationName());
|
|
setData(source, QStringLiteral("appIconName"), job->applicationIconName());
|
|
setData(source, QStringLiteral("suspendable"), job->suspendable());
|
|
setData(source, QStringLiteral("killable"), job->killable());
|
|
updateState(job);
|
|
|
|
connect(job, &Job::stateChanged, this, [this, job] {
|
|
updateState(job);
|
|
});
|
|
connect(job, &Job::speedChanged, this, [this, job] {
|
|
updateEta(job);
|
|
});
|
|
|
|
connectJobField(job, &Job::summary, &Job::summaryChanged, QStringLiteral("infoMessage"));
|
|
connectJobField(job, &Job::percentage, &Job::percentageChanged, QStringLiteral("percentage"));
|
|
connectJobField(job, &Job::error, &Job::errorChanged, QStringLiteral("error"));
|
|
connectJobField(job, &Job::errorText, &Job::errorTextChanged, QStringLiteral("errorText"));
|
|
connectJobField(job, &Job::destUrl, &Job::destUrlChanged, QStringLiteral("destUrl"));
|
|
|
|
static const struct {
|
|
int number;
|
|
QString (Job::*labelGetter)() const;
|
|
void (Job::*labelSignal)();
|
|
QString (Job::*valueGetter)() const;
|
|
void (Job::*valueSignal)();
|
|
} s_descriptionFields[] = {
|
|
{0, &Job::descriptionLabel1, &Job::descriptionLabel1Changed, &Job::descriptionValue1, &Job::descriptionValue1Changed},
|
|
{1, &Job::descriptionLabel2, &Job::descriptionLabel2Changed, &Job::descriptionValue2, &Job::descriptionValue2Changed},
|
|
};
|
|
|
|
for (auto fields : s_descriptionFields) {
|
|
updateDescriptionField(job, fields.number, fields.labelGetter, fields.valueGetter);
|
|
connect(job, fields.labelSignal, this, [=] {
|
|
updateDescriptionField(job, fields.number, fields.labelGetter, fields.valueGetter);
|
|
});
|
|
connect(job, fields.valueSignal, this, [=] {
|
|
updateDescriptionField(job, fields.number, fields.labelGetter, fields.valueGetter);
|
|
});
|
|
}
|
|
|
|
static const struct {
|
|
// Previously the dataengine counted units up but for simplicity a fixed number is assigned to each unit
|
|
int number;
|
|
QString unit;
|
|
qulonglong (Job::*processedGetter)() const;
|
|
void (Job::*processedSignal)();
|
|
qulonglong (Job::*totalGetter)() const;
|
|
void (Job::*totalSignal)();
|
|
} s_unitsFields[] = {
|
|
{0, QStringLiteral("bytes"), &Job::processedBytes, &Job::processedBytesChanged, &Job::totalBytes, &Job::totalBytesChanged},
|
|
{1, QStringLiteral("files"), &Job::processedFiles, &Job::processedFilesChanged, &Job::totalFiles, &Job::totalFilesChanged},
|
|
{2, QStringLiteral("dirs"), &Job::processedDirectories, &Job::processedDirectoriesChanged, &Job::totalDirectories, &Job::totalDirectoriesChanged}};
|
|
|
|
for (auto fields : s_unitsFields) {
|
|
updateUnit(job, fields.number, fields.unit, fields.processedGetter, fields.totalGetter);
|
|
connect(job, fields.processedSignal, this, [=] {
|
|
updateUnit(job, fields.number, fields.unit, fields.processedGetter, fields.totalGetter);
|
|
});
|
|
connect(job, fields.totalSignal, this, [=] {
|
|
updateUnit(job, fields.number, fields.unit, fields.processedGetter, fields.totalGetter);
|
|
});
|
|
}
|
|
|
|
m_jobs.append(job);
|
|
}
|
|
|
|
void KuiserverEngine::removeJob(Job *job)
|
|
{
|
|
if (!job || !m_jobs.contains(job)) {
|
|
return;
|
|
}
|
|
|
|
m_jobs.removeOne(job);
|
|
|
|
const QString source = sourceName(job);
|
|
removeSource(source);
|
|
}
|
|
|
|
QString KuiserverEngine::speedString(qulonglong speed)
|
|
{
|
|
return i18nc("Bytes per second", "%1/s", KFormat().formatByteSize(speed));
|
|
}
|
|
|
|
void KuiserverEngine::updateState(Job *job)
|
|
{
|
|
const QString source = sourceName(job);
|
|
|
|
QString stateString;
|
|
switch (job->state()) {
|
|
case Notifications::JobStateRunning:
|
|
stateString = QStringLiteral("running");
|
|
updateSpeed(job);
|
|
break;
|
|
case Notifications::JobStateSuspended:
|
|
stateString = QStringLiteral("suspended");
|
|
setData(source, QStringLiteral("speed"), QVariant());
|
|
setData(source, QStringLiteral("numericSpeed"), QVariant());
|
|
break;
|
|
case Notifications::JobStateStopped:
|
|
stateString = QStringLiteral("stopped");
|
|
break;
|
|
}
|
|
|
|
setData(source, QStringLiteral("state"), stateString);
|
|
|
|
if (job->state() == Notifications::JobStateStopped) {
|
|
removeJob(job);
|
|
}
|
|
}
|
|
|
|
void KuiserverEngine::updateSpeed(Job *job)
|
|
{
|
|
const QString source = sourceName(job);
|
|
setData(source, QStringLiteral("speed"), speedString(job->speed()));
|
|
setData(source, QStringLiteral("numericSpeed"), job->speed());
|
|
updateEta(job);
|
|
}
|
|
|
|
void KuiserverEngine::updateEta(Job *job)
|
|
{
|
|
const QString source = sourceName(job);
|
|
|
|
if (job->speed() < 1 || job->totalBytes() < 1) {
|
|
setData(source, QStringLiteral("eta"), 0);
|
|
return;
|
|
}
|
|
|
|
const qlonglong remaining = 1000 * (job->totalBytes() - job->processedBytes());
|
|
setData(source, QStringLiteral("eta"), remaining / job->speed());
|
|
}
|
|
|
|
K_PLUGIN_CLASS_WITH_JSON(KuiserverEngine, "plasma-dataengine-applicationjobs.json")
|
|
|
|
#include "kuiserverengine.moc"
|