Move some more functions to utils

2026-03-07 20:09:55 +03:00 · 2024-06-19 01:41:53 +03:00
parent d21ad393fd
commit 85dd68e088
4 changed files with 136 additions and 107 deletions
--- a/src/utils.cpp
+++ b/src/utils.cpp
@@ -2,7 +2,19 @@

 utils::utils() {}

-QString utils::getSmartctlPath() {
+void utils::clearButtonGroup(QButtonGroup* buttonGroup, QHBoxLayout* horizontalLayout, QSpacerItem* buttonStretch, QMenu* menuDisk)
+{
+    QList<QAbstractButton*> buttons = buttonGroup->buttons();
+    for (QAbstractButton* button : buttons) {
+        buttonGroup->removeButton(button);
+        delete button;
+    }
+    horizontalLayout->removeItem(buttonStretch);
+    delete buttonStretch;
+    menuDisk->clear();
+}
+
+QString utils::getSmartctlPath(bool initializing) {
    QStringList paths = QString::fromLocal8Bit(qgetenv("PATH")).split(QDir::listSeparator(), Qt::SkipEmptyParts);

    paths << "/usr/sbin" << "/usr/local/sbin";
@@ -16,3 +28,83 @@ QString utils::getSmartctlPath() {

    return QString();
 }
+
+QString utils::getSmartctlOutput(const QStringList &arguments, bool root, bool initializing)
+{
+    QProcess process;
+    QString command;
+
+    if (root) {
+        command = "pkexec";
+    } else {
+        command = getSmartctlPath(initializing);
+    }
+
+    if (!getSmartctlPath(initializing).isEmpty()) {
+        process.start(command, arguments);
+        process.waitForFinished(-1);
+    }
+
+    if (process.exitCode() == 127) {
+        QMessageBox::critical(nullptr, QObject::tr("KDiskInfo Error"), QObject::tr("KDiskInfo needs root access in order to read S.M.A.R.T. data!"));
+        if (initializing) {
+            QTimer::singleShot(0, qApp, &QApplication::quit);
+        }
+        return QString();
+    }
+
+    if (process.isOpen()) {
+        return process.readAllStandardOutput();
+    } else {
+        return QString();
+    }
+}
+
+
+QPair<QStringList, QJsonArray> utils::scanDevices(bool initializing)
+{
+    QString output = getSmartctlOutput({"--scan", "--json"}, false, initializing);
+    QJsonDocument doc = QJsonDocument::fromJson(output.toUtf8());
+    QJsonObject jsonObj = doc.object();
+    QJsonArray devices = jsonObj["devices"].toArray();
+    QString smartctlPath = getSmartctlPath(initializing);
+    QStringList commandList;
+    QStringList deviceOutputs;
+
+    for (const QJsonValue &value : std::as_const(devices)) {
+        QJsonObject device = value.toObject();
+        QString deviceName = device["name"].toString();
+        commandList.append(QString(smartctlPath + " --all --json=o %1").arg(deviceName));
+    }
+    QString command = commandList.join(" ; ");
+
+    if (smartctlPath.isEmpty()) {
+        QMessageBox::critical(nullptr, QObject::tr("KDiskInfo Error"), QObject::tr("smartctl was not found, please install it!"));
+        QTimer::singleShot(0, qApp, &QApplication::quit);
+    }
+
+    QString allDevicesOutput = getSmartctlOutput({"sh", "-c", command}, true, initializing);
+
+    int startIndex = 0;
+    int endIndex = 0;
+
+    static const QRegularExpression regex("\\}\\n\\{");
+
+    while ((endIndex = allDevicesOutput.indexOf(regex, startIndex)) != -1) {
+        ++endIndex;
+        QString jsonFragment = allDevicesOutput.mid(startIndex, endIndex - startIndex);
+        deviceOutputs.append(jsonFragment);
+        startIndex = endIndex;
+    }
+
+    if (startIndex < allDevicesOutput.size()) {
+        QString jsonFragment = allDevicesOutput.mid(startIndex);
+        deviceOutputs.append(jsonFragment);
+    }
+
+    if (!allDevicesOutput.isEmpty()) {
+        return QPair<QStringList, QJsonArray>(deviceOutputs, devices);
+    } else {
+        return QPair<QStringList, QJsonArray>(QStringList(), QJsonArray());
+    }
+}