150 lines
4.8 KiB
C++
150 lines
4.8 KiB
C++
/* ============================================================
|
|
* KCMOpenRC - OpenRC Service Manager
|
|
* Copyright (C) 2025 Juraj Oravec <jurajoravec@mailo.com>
|
|
*
|
|
* This program 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.
|
|
*
|
|
* This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
|
* ============================================================ */
|
|
#include "openrc_service_model.h"
|
|
|
|
#include <QBrush>
|
|
#include <QColor>
|
|
#include <QFont>
|
|
#include <QTimer>
|
|
|
|
#include <sys/queue.h>
|
|
|
|
#ifdef OPENRC_SERVICE_MODEL_DEBUG
|
|
#include <QAbstractItemModelTester>
|
|
#endif
|
|
|
|
OpenRCServiceModel::OpenRCServiceModel(QObject* parent)
|
|
: QAbstractItemModel(parent)
|
|
, rootItem(std::make_unique<OpenRCTreeItem>(QVariantList{tr("Service")}))
|
|
{
|
|
#ifdef OPENRC_SERVICE_MODEL_DEBUG
|
|
m_tester = new QAbstractItemModelTester(this, QAbstractItemModelTester::FailureReportingMode::Fatal, this);
|
|
#endif
|
|
reload_services();
|
|
}
|
|
|
|
OpenRCServiceModel::~OpenRCServiceModel() noexcept = default;
|
|
|
|
int OpenRCServiceModel::columnCount(const QModelIndex &parent) const
|
|
{
|
|
if (parent.isValid())
|
|
return static_cast<OpenRCTreeItem*>(parent.internalPointer())->columnCount();
|
|
return rootItem->columnCount();
|
|
}
|
|
|
|
QVariant OpenRCServiceModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
if (!index.isValid() || role != Qt::DisplayRole)
|
|
return {};
|
|
|
|
const auto *item = static_cast<const OpenRCTreeItem*>(index.internalPointer());
|
|
return item->data(index.column());
|
|
}
|
|
|
|
Qt::ItemFlags OpenRCServiceModel::flags(const QModelIndex &index) const
|
|
{
|
|
return index.isValid()
|
|
? QAbstractItemModel::flags(index) : Qt::ItemFlags(Qt::NoItemFlags);
|
|
}
|
|
|
|
QVariant OpenRCServiceModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
if ((orientation == Qt::Horizontal) && (role == Qt::DisplayRole)) {
|
|
return rootItem->data(section);
|
|
}
|
|
return QVariant{};
|
|
}
|
|
|
|
QModelIndex OpenRCServiceModel::index(int row, int column, const QModelIndex &parent) const
|
|
{
|
|
if (!hasIndex(row, column, parent))
|
|
return {};
|
|
|
|
OpenRCTreeItem *parentItem = parent.isValid()
|
|
? static_cast<OpenRCTreeItem*>(parent.internalPointer())
|
|
: rootItem.get();
|
|
|
|
if (auto *childItem = parentItem->child(row))
|
|
return createIndex(row, column, childItem);
|
|
return {};
|
|
}
|
|
|
|
QModelIndex OpenRCServiceModel::parent(const QModelIndex &index) const
|
|
{
|
|
if (!index.isValid())
|
|
return {};
|
|
|
|
auto *childItem = static_cast<OpenRCTreeItem*>(index.internalPointer());
|
|
OpenRCTreeItem *parentItem = childItem->parentItem();
|
|
|
|
if (parentItem != rootItem.get()) {
|
|
return createIndex(parentItem->row(), 0, parentItem);
|
|
}
|
|
return QModelIndex{};
|
|
}
|
|
|
|
int OpenRCServiceModel::rowCount(const QModelIndex &parent) const
|
|
{
|
|
if (parent.column() > 0)
|
|
return 0;
|
|
|
|
|
|
const OpenRCTreeItem *parentItem = parent.isValid()
|
|
? static_cast<const OpenRCTreeItem*>(parent.internalPointer())
|
|
: rootItem.get();
|
|
|
|
return parentItem->childCount();
|
|
}
|
|
|
|
|
|
void OpenRCServiceModel::reload_services()
|
|
{
|
|
beginResetModel();
|
|
|
|
m_runlevels.clear();
|
|
|
|
RC_STRINGLIST* list_runlevels = rc_runlevel_list();
|
|
RC_STRING* runlevel;
|
|
TAILQ_FOREACH(runlevel, list_runlevels, entries) {
|
|
std::string standard_string_name(runlevel->value);
|
|
QString runlevelStr = QString::fromStdString(standard_string_name);
|
|
|
|
QVariantList columnData;
|
|
columnData << QStringLiteral("<strong>%1</strong>").arg(runlevelStr);
|
|
|
|
rootItem.get()->appendChild(std::make_unique<OpenRCTreeItem>(columnData, rootItem.get()));
|
|
OpenRCTreeItem *parentItem = rootItem.get()->child(rootItem.get()->childCount() -1);
|
|
|
|
RC_STRINGLIST* list_services = rc_services_in_runlevel(runlevel->value);
|
|
RC_STRING* service;
|
|
TAILQ_FOREACH(service, list_services, entries) {
|
|
std::string standard_string_name(service->value);
|
|
QString serviceStr = QString::fromStdString(standard_string_name);
|
|
|
|
QVariantList serviceColumnData;
|
|
serviceColumnData << serviceStr;
|
|
|
|
parentItem->appendChild(std::make_unique<OpenRCTreeItem>(serviceColumnData, parentItem));
|
|
}
|
|
rc_stringlist_free(list_services);
|
|
}
|
|
rc_stringlist_free(list_runlevels);
|
|
|
|
endResetModel();
|
|
}
|