chore: TrayLoader single process support to start multiple plugins

yixinshark · yixinshark · commit 5b413b0308df · 2024-12-04T10:31:41.000+08:00
1.Divide plugins into three groups(selfMaintenance,application,other) 2.single process support to start multiple plugins Log: as title Bug: https://pms.uniontech.com/bug-view-283901.html
diff --git a/debian/control b/debian/control
@@ -60,7 +60,7 @@ Depends: ${shlibs:Depends}, ${misc:Depends}, libdde-shell( =${binary:Version}),
   libdtk6declarative, qml6-module-qtquick-controls2-styles-chameleon,
   qml6-module-qtquick-layouts, qml6-module-qtquick-window,
   qml6-module-qt-labs-platform, qml6-module-qt-labs-qmlmodels,
-  dde-tray-loader
+  dde-tray-loader (> 1.99.6)
 Multi-Arch: same
 Description: An wrapper for developed based on dde-shell plugin system
 
diff --git a/panels/dock/dconfig/org.deepin.ds.dock.tray.json b/panels/dock/dconfig/org.deepin.ds.dock.tray.json
@@ -76,6 +76,28 @@
             "description[zh_CN]": "托盘折叠区域当前的状态",
             "permissions": "readwrite",
             "visibility": "private"
+        },
+        "selfMaintenanceTrayPlugins": {
+            "value": ["libapplication-tray.so", "libbrightness.so", "libdatetime.so", "libdnd-mode.so", "libeye-comfort-mode.so", "libmedia.so", "libnotification.so", "libonboard.so", "libshutdown.so", "libairplane-mode.so", "libbluetooth.so", "libdock-tray-network-plugin.so", "libdock-wirelesscasting-plugin.so", "libkeyboard-layout.so", "libpower.so", "libsound.so"],
+            "serial": 0,
+            "flags": [],
+            "name": "self maintenance plugins",
+            "name[zh_CN]": "自维护托盘插件",
+            "description": "self maintenance plugins",
+            "description[zh_CN]": "自维护托盘插件",
+            "permissions": "readwrite",
+            "visibility": "private"
+        },
+        "applicationTrayPlugins": {
+            "value": ["libdock-clipboard-plugin.so", "libddegrandsearch_dockplugin.so", "libdeepin-screen-recorder-plugin.so", "libdeepin-system-monitor-plugin.so", "libshot-start-plugin.so", "libshot-start-record-plugin.so", "libdde-disk-mount-plugin.so"],
+            "serial": 0,
+            "flags": [],
+            "name": "application tray plugins",
+            "name[zh_CN]": "应用托盘插件",
+            "description": "application tray plugins",
+            "description[zh_CN]": "应用托盘插件",
+            "permissions": "readwrite",
+            "visibility": "private"
         }
     }
 }
diff --git a/panels/dock/loadtrayplugins.cpp b/panels/dock/loadtrayplugins.cpp
@@ -7,6 +7,8 @@
 
 #include <signal.h>
 
+#include <DConfig>
+
 #include <QDir>
 #include <QTimer>
 #include <QGuiApplication>
@@ -32,58 +34,17 @@ LoadTrayPlugins::~LoadTrayPlugins()
 
 void LoadTrayPlugins::loadDockPlugins()
 {
-    QStringList filters;
-    filters << "*.so";
-
-    QStringList execPaths;
-    execPaths << qEnvironmentVariable("TRAY_LOADER_EXECUTE_PATH")
-              << QString("%1/trayplugin-loader").arg(CMAKE_INSTALL_FULL_LIBEXECDIR);
-
-    QString validExePath;
-    for (const QString &execPath : execPaths) {
-        if (QFile::exists(execPath)) {
-            validExePath = execPath;
-            break;
-        }
-    }
-
+    QString validExePath = loaderPath();
     if (validExePath.isEmpty()) {
         qWarning() << "No valid loader executable path found.";
         return;
     }
 
-    QStringList dirs;
-    const auto pluginsPath = qEnvironmentVariable("TRAY_DEBUG_PLUGIN_PATH");
-    if (!pluginsPath.isEmpty())
-        dirs << pluginsPath.split(QDir::listSeparator());
-
-    if (dirs.isEmpty())
-        dirs << pluginDirs;
-
-    for (auto &pluginDir : dirs) {
-        QDir dir(pluginDir);
-        if (!dir.exists()) {
-            qWarning() << "The plugin directory does not exist:" << pluginDir;
-            continue;
-        }
-
-        auto pluginFileInfos = dir.entryInfoList(filters, QDir::Files);
-        foreach (auto pluginInfo, pluginFileInfos) {
-            qInfo() << "pluginLoader load plugin" << pluginInfo.absoluteFilePath();
-
-            QProcess *process = new QProcess(this);
-            setProcessEnv(process);
-
-            connect(process, QOverload<int, QProcess::ExitStatus>::of(&QProcess::finished),
-                    this, &LoadTrayPlugins::handleProcessFinished);
-
-            ProcessInfo pInfo = { process, pluginInfo.absoluteFilePath(), 0 };
-            m_processes.append(pInfo);
-
-            process->setProgram(validExePath);
-            process->setArguments({"-p", pluginInfo.absoluteFilePath(), "-platform", "wayland"});
-            process->start();
-        }
+    auto pluginGroupMap = getPluginGroup();
+    for (auto it = pluginGroupMap.begin(); it != pluginGroupMap.end(); ++it) {
+        if (it.value().isEmpty()) continue;
+        qDebug() << "Load plugin:" << it.value() << " group:" << it.key();
+        startProcess(validExePath, it.value(), it.key());
     }
 }
 
@@ -113,6 +74,22 @@ void LoadTrayPlugins::handleProcessFinished(int exitCode, QProcess::ExitStatus e
     }
 }
 
+void LoadTrayPlugins::startProcess(const QString &loaderPath, const QString &pluginPath, const QString &groupName)
+{
+    auto *process = new QProcess(this);
+    setProcessEnv(process);
+
+    connect(process, QOverload<int, QProcess::ExitStatus>::of(&QProcess::finished),
+            this, &LoadTrayPlugins::handleProcessFinished);
+
+    ProcessInfo pInfo = { process, pluginPath, 0 };
+    m_processes.append(pInfo);
+
+    process->setProgram(loaderPath);
+    process->setArguments({"-l", pluginPath, "-g", groupName, "-platform", "wayland"});
+    process->start();
+}
+
 void LoadTrayPlugins::setProcessEnv(QProcess *process)
 {
     if (!process) return;
@@ -127,4 +104,84 @@ void LoadTrayPlugins::setProcessEnv(QProcess *process)
     process->setProcessEnvironment(env);
 }
 
+QString LoadTrayPlugins::loaderPath() const
+{
+    QStringList execPaths;
+    execPaths << qEnvironmentVariable("TRAY_LOADER_EXECUTE_PATH")
+              << QString("%1/trayplugin-loader").arg(CMAKE_INSTALL_FULL_LIBEXECDIR);
+
+    QString validExePath;
+    for (const QString &execPath : execPaths) {
+        if (QFile::exists(execPath)) {
+            validExePath = execPath;
+            break;
+        }
+    }
+
+    return validExePath;
+}
+
+QMap<QString, QString> LoadTrayPlugins::getPluginGroup() const
+{
+    const QString selfMaintenancePlugins = "selfMaintenancePlugins";
+    const QString applicationPlugins = "applicationPlugins";
+    const QString otherPlugins = "otherPlugins";
+
+    auto dConfig = Dtk::Core::DConfig::create("org.deepin.dde.shell", "org.deepin.ds.dock.tray", QString());
+    QStringList selfMaintenanceTrayPlugins = dConfig->value("selfMaintenanceTrayPlugins").toStringList();
+    QStringList applicationTrayPlugins = dConfig->value("applicationTrayPlugins").toStringList();
+    dConfig->deleteLater();
+
+    QStringList dirs;
+    const auto pluginsPath = qEnvironmentVariable("TRAY_DEBUG_PLUGIN_PATH");
+    if (!pluginsPath.isEmpty())
+        dirs << pluginsPath.split(QDir::listSeparator());
+
+    if (dirs.isEmpty())
+        dirs << pluginDirs;
+
+    QString selfMaintenancePluginPaths;
+    QString applicationPluginPaths;
+    QString otherPluginPaths;
+
+    for (auto &pluginDir : dirs) {
+        QDir dir(pluginDir);
+        if (!dir.exists()) {
+            qWarning() << "The plugin directory does not exist:" << pluginDir;
+            continue;
+        }
+
+        auto pluginFileInfos = dir.entryInfoList({"*.so"}, QDir::Files);
+        for (auto &pluginInfo : pluginFileInfos) {
+            QString pluginName = pluginInfo.absoluteFilePath().section("/", -1);
+            if (selfMaintenanceTrayPlugins.contains(pluginName)) {
+                selfMaintenancePluginPaths += pluginInfo.absoluteFilePath() + ";";
+            } else if (applicationTrayPlugins.contains(pluginName)) {
+                applicationPluginPaths += pluginInfo.absoluteFilePath() + ";";
+            } else {
+                otherPluginPaths += pluginInfo.absoluteFilePath() + ";";
+            }
+        }
+    }
+
+    QMap<QString, QString> pluginGroup;
+
+    if (!selfMaintenancePluginPaths.isEmpty()) {
+        // remove the last ';'
+        selfMaintenancePluginPaths.chop(1);
+        pluginGroup.insert(selfMaintenancePlugins, selfMaintenancePluginPaths);
+    }
+
+    if (!applicationPluginPaths.isEmpty()) {
+        applicationPluginPaths.chop(1);
+        pluginGroup.insert(applicationPlugins, applicationPluginPaths);
+    }
+
+    if (!otherPluginPaths.isEmpty()) {
+        otherPluginPaths.chop(1);
+        pluginGroup.insert(otherPlugins, otherPluginPaths);
+    }
+
+    return pluginGroup;
+}
 }
diff --git a/panels/dock/loadtrayplugins.h b/panels/dock/loadtrayplugins.h
@@ -26,6 +26,10 @@ private slots:
     void handleProcessFinished(int exitCode, QProcess::ExitStatus exitStatus);
 
 private:
+    QString loaderPath() const;
+    QMap<QString, QString> getPluginGroup() const;
+
+    void startProcess(const QString &loaderPath, const QString &pluginPath, const QString &groupName);
     void setProcessEnv(QProcess *process);
 
 private:
