1 /*
2 SPDX-FileCopyrightText: 2005 Jean-Remy Falleri <jr.falleri@laposte.net>
3 SPDX-FileCopyrightText: 2005-2007 Kevin Ottens <ervin@kde.org>
4
5 SPDX-License-Identifier: LGPL-2.0-only
6 */
7
8 #include "deviceserviceaction.h"
9
10 #include <QDebug>
11
12 #include <KIO/CommandLauncherJob>
13 #include <KLocalizedString>
14 #include <KNotificationJobUiDelegate>
15 #include <kmacroexpander.h>
16 #include <solid/block.h>
17 #include <solid/storageaccess.h>
18
19 class MacroExpander : public KMacroExpanderBase
20 {
21 public:
MacroExpander(const Solid::Device & device)22 MacroExpander(const Solid::Device &device)
23 : KMacroExpanderBase('%')
24 , m_device(device)
25 {
26 }
27
28 protected:
29 int expandEscapedMacro(const QString &str, int pos, QStringList &ret) override;
30
31 private:
32 Solid::Device m_device;
33 };
34
35 class DelayedExecutor : public QObject
36 {
37 Q_OBJECT
38 public:
39 DelayedExecutor(const KServiceAction &service, Solid::Device &device);
40
41 private slots:
42 void _k_storageSetupDone(Solid::ErrorType error, QVariant errorData, const QString &udi);
43
44 private:
45 void delayedExecute(const QString &udi);
46
47 KServiceAction m_service;
48 };
49
DeviceServiceAction()50 DeviceServiceAction::DeviceServiceAction()
51 : DeviceAction()
52 {
53 DeviceAction::setIconName(QStringLiteral("dialog-cancel"));
54 DeviceAction::setLabel(i18nc("A default name for an action without proper label", "Unknown"));
55 }
56
id() const57 QString DeviceServiceAction::id() const
58 {
59 if (m_service.name().isEmpty() && m_service.exec().isEmpty()) {
60 return QString();
61 } else {
62 return "#Service:" + m_service.name() + m_service.exec();
63 }
64 }
65
execute(Solid::Device & device)66 void DeviceServiceAction::execute(Solid::Device &device)
67 {
68 new DelayedExecutor(m_service, device);
69 }
70
_k_storageSetupDone(Solid::ErrorType error,QVariant errorData,const QString & udi)71 void DelayedExecutor::_k_storageSetupDone(Solid::ErrorType error, QVariant errorData, const QString &udi)
72 {
73 Q_UNUSED(errorData);
74
75 if (!error) {
76 delayedExecute(udi);
77 }
78 }
79
setService(const KServiceAction & service)80 void DeviceServiceAction::setService(const KServiceAction &service)
81 {
82 DeviceAction::setIconName(service.icon());
83 DeviceAction::setLabel(service.text());
84
85 m_service = service;
86 }
87
service() const88 KServiceAction DeviceServiceAction::service() const
89 {
90 return m_service;
91 }
92
expandEscapedMacro(const QString & str,int pos,QStringList & ret)93 int MacroExpander::expandEscapedMacro(const QString &str, int pos, QStringList &ret)
94 {
95 ushort option = str[pos + 1].unicode();
96
97 switch (option) {
98 case 'f': // Filepath
99 case 'F': // case insensitive
100 if (m_device.is<Solid::StorageAccess>()) {
101 ret << m_device.as<Solid::StorageAccess>()->filePath();
102 } else {
103 qWarning() << "DeviceServiceAction::execute: " << m_device.udi() << " is not a StorageAccess device";
104 }
105 break;
106 case 'd': // Device node
107 case 'D': // case insensitive
108 if (m_device.is<Solid::Block>()) {
109 ret << m_device.as<Solid::Block>()->device();
110 } else {
111 qWarning() << "DeviceServiceAction::execute: " << m_device.udi() << " is not a Block device";
112 }
113 break;
114 case 'i': // UDI
115 case 'I': // case insensitive
116 ret << m_device.udi();
117 break;
118 case '%':
119 ret = QStringList(QLatin1String("%"));
120 break;
121 default:
122 return -2; // subst with same and skip
123 }
124 return 2;
125 }
126
DelayedExecutor(const KServiceAction & service,Solid::Device & device)127 DelayedExecutor::DelayedExecutor(const KServiceAction &service, Solid::Device &device)
128 : m_service(service)
129 {
130 if (device.is<Solid::StorageAccess>() && !device.as<Solid::StorageAccess>()->isAccessible()) {
131 Solid::StorageAccess *access = device.as<Solid::StorageAccess>();
132
133 connect(access, &Solid::StorageAccess::setupDone, this, &DelayedExecutor::_k_storageSetupDone);
134
135 access->setup();
136 } else {
137 delayedExecute(device.udi());
138 }
139 }
140
delayedExecute(const QString & udi)141 void DelayedExecutor::delayedExecute(const QString &udi)
142 {
143 Solid::Device device(udi);
144
145 QString exec = m_service.exec();
146 MacroExpander mx(device);
147 mx.expandMacrosShellQuote(exec);
148
149 KIO::CommandLauncherJob *job = new KIO::CommandLauncherJob(exec);
150 job->setIcon(m_service.icon());
151 job->setUiDelegate(new KNotificationJobUiDelegate(KJobUiDelegate::AutoHandlingEnabled));
152 job->start();
153
154 deleteLater();
155 }
156
157 #include "deviceserviceaction.moc"
158