src/gui/desktopintegration.cpp

   1 /*
   2  * Bittorrent Client using Qt and libtorrent.
   3  * Copyright (C) 2022  Vladimir Golovnev <glassez@yandex.ru>
   4  * Copyright (C) 2006  Christophe Dumez <chris@qbittorrent.org>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License
   8  * as published by the Free Software Foundation; either version 2
   9  * of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  *
  20  * In addition, as a special exception, the copyright holders give permission to
  21  * link this program with the OpenSSL project's "OpenSSL" library (or with
  22  * modified versions of it that use the same license as the "OpenSSL" library),
  23  * and distribute the linked executables. You must obey the GNU General Public
  24  * License in all respects for all of the code used other than "OpenSSL".  If you
  25  * modify file(s), you may extend this exception to your version of the file(s),
  26  * but you are not obligated to do so. If you do not wish to do so, delete this
  27  * exception statement from your version.
  28  */
  29
  30 #include "desktopintegration.h"
  31
  32 #include <chrono>
  33
  34 #include <QMenu>
  35 #include <QTimer>
  36
  37 #ifndef Q_OS_MACOS
  38 #include <QSystemTrayIcon>
  39 #endif
  40
  41 #include "base/preferences.h"
  42 #include "uithememanager.h"
  43
  44 #ifdef Q_OS_MACOS
  45 #include "macutilities.h"
  46 #endif
  47
  48 #ifdef QBT_USES_CUSTOMDBUSNOTIFICATIONS
  49 #include "notifications/dbusnotifier.h"
  50 #endif
  51
  52 namespace
  53 {
  54 #ifdef Q_OS_MACOS
  55     DesktopIntegration *desktopIntegrationInstance = nullptr;
  56
  57     bool handleDockClicked(id self, SEL cmd, ...)
  58     {
  59         Q_UNUSED(self);
  60         Q_UNUSED(cmd);
  61
  62         Q_ASSERT(desktopIntegrationInstance);
  63         emit desktopIntegrationInstance->activationRequested();
  64
  65         return true;
  66     }
  67 #endif
  68 }
  69
  70 using namespace std::chrono_literals;
  71
  72 #define SETTINGS_KEY(name) u"GUI/" name
  73 #define NOTIFICATIONS_SETTINGS_KEY(name) (SETTINGS_KEY(u"Notifications/"_qs) name)
  74
  75 DesktopIntegration::DesktopIntegration(QObject *parent)
  76     : QObject(parent)
  77     , m_storeNotificationEnabled {NOTIFICATIONS_SETTINGS_KEY(u"Enabled"_qs), true}
  78 #ifdef QBT_USES_CUSTOMDBUSNOTIFICATIONS
  79     , m_storeNotificationTimeOut {NOTIFICATIONS_SETTINGS_KEY(u"Timeout"_qs), -1}
  80 #endif
  81 {
  82 #ifdef Q_OS_MACOS
  83     desktopIntegrationInstance = this;
  84     MacUtils::overrideDockClickHandler(handleDockClicked);
  85 #else
  86     if (Preferences::instance()->systemTrayEnabled())
  87         createTrayIcon();
  88
  89 #ifdef QBT_USES_CUSTOMDBUSNOTIFICATIONS
  90     if (isNotificationsEnabled())
  91     {
  92         m_notifier = new DBusNotifier(this);
  93         connect(m_notifier, &DBusNotifier::messageClicked, this, &DesktopIntegration::notificationClicked);
  94     }
  95 #endif
  96 #endif
  97
  98     connect(Preferences::instance(), &Preferences::changed, this, &DesktopIntegration::onPreferencesChanged);
  99 }
 100
 101 DesktopIntegration::~DesktopIntegration()
 102 {
 103     if (m_menu)
 104         delete m_menu;
 105 }
 106
 107 bool DesktopIntegration::isActive() const
 108 {
 109 #ifdef Q_OS_MACOS
 110     return true;
 111 #else
 112     return m_systrayIcon && QSystemTrayIcon::isSystemTrayAvailable();
 113 #endif
 114 }
 115
 116 QString DesktopIntegration::toolTip() const
 117 {
 118     return m_toolTip;
 119 }
 120
 121 void DesktopIntegration::setToolTip(const QString &toolTip)
 122 {
 123     if (m_toolTip == toolTip)
 124         return;
 125
 126     m_toolTip = toolTip;
 127 #ifndef Q_OS_MACOS
 128     if (m_systrayIcon)
 129         m_systrayIcon->setToolTip(m_toolTip);
 130 #endif
 131 }
 132
 133 QMenu *DesktopIntegration::menu() const
 134 {
 135     return m_menu;
 136 }
 137
 138 void DesktopIntegration::setMenu(QMenu *menu)
 139 {
 140     if (menu == m_menu)
 141         return;
 142
 143 #if defined Q_OS_MACOS
 144     if (m_menu)
 145         delete m_menu;
 146
 147     m_menu = menu;
 148
 149     if (m_menu)
 150         m_menu->setAsDockMenu();
 151 #elif defined Q_OS_UNIX
 152     const bool systemTrayEnabled = m_systrayIcon;
 153     if (m_menu)
 154     {
 155         if (m_systrayIcon)
 156         {
 157             delete m_systrayIcon;
 158             m_systrayIcon = nullptr;
 159         }
 160         delete m_menu;
 161     }
 162
 163     m_menu = menu;
 164
 165     if (systemTrayEnabled && !m_systrayIcon)
 166         createTrayIcon();
 167 #else
 168     if (m_menu)
 169         delete m_menu;
 170
 171     m_menu = menu;
 172
 173     if (m_systrayIcon)
 174         m_systrayIcon->setContextMenu(m_menu);
 175 #endif
 176 }
 177
 178 bool DesktopIntegration::isNotificationsEnabled() const
 179 {
 180     return m_storeNotificationEnabled;
 181 }
 182
 183 void DesktopIntegration::setNotificationsEnabled(const bool value)
 184 {
 185     if (m_storeNotificationEnabled == value)
 186         return;
 187
 188     m_storeNotificationEnabled = value;
 189
 190 #ifdef QBT_USES_CUSTOMDBUSNOTIFICATIONS
 191     if (value)
 192     {
 193         m_notifier = new DBusNotifier(this);
 194         connect(m_notifier, &DBusNotifier::messageClicked, this, &DesktopIntegration::notificationClicked);
 195     }
 196     else
 197     {
 198         delete m_notifier;
 199         m_notifier = nullptr;
 200     }
 201 #endif
 202 }
 203
 204 int DesktopIntegration::notificationTimeout() const
 205 {
 206 #ifdef QBT_USES_CUSTOMDBUSNOTIFICATIONS
 207     return m_storeNotificationTimeOut;
 208 #else
 209     return 5000;
 210 #endif
 211 }
 212
 213 #ifdef QBT_USES_CUSTOMDBUSNOTIFICATIONS
 214 void DesktopIntegration::setNotificationTimeout(const int value)
 215 {
 216     m_storeNotificationTimeOut = value;
 217 }
 218 #endif
 219
 220 void DesktopIntegration::showNotification(const QString &title, const QString &msg) const
 221 {
 222     if (!isNotificationsEnabled())
 223         return;
 224
 225 #ifdef Q_OS_MACOS
 226     MacUtils::displayNotification(title, msg);
 227 #else
 228 #ifdef QBT_USES_CUSTOMDBUSNOTIFICATIONS
 229     m_notifier->showMessage(title, msg, notificationTimeout());
 230 #else
 231     if (m_systrayIcon && QSystemTrayIcon::supportsMessages())
 232         m_systrayIcon->showMessage(title, msg, QSystemTrayIcon::Information, notificationTimeout());
 233 #endif
 234 #endif
 235 }
 236
 237 void DesktopIntegration::onPreferencesChanged()
 238 {
 239 #ifndef Q_OS_MACOS
 240     if (Preferences::instance()->systemTrayEnabled())
 241     {
 242         if (m_systrayIcon)
 243         {
 244             // Reload systray icon
 245             m_systrayIcon->setIcon(getSystrayIcon());
 246         }
 247         else
 248         {
 249             createTrayIcon();
 250         }
 251     }
 252     else
 253     {
 254         delete m_systrayIcon;
 255         m_systrayIcon = nullptr;
 256         emit stateChanged();
 257     }
 258 #endif
 259 }
 260
 261 #ifndef Q_OS_MACOS
 262 void DesktopIntegration::createTrayIcon()
 263 {
 264     Q_ASSERT(!m_systrayIcon);
 265
 266     m_systrayIcon = new QSystemTrayIcon(getSystrayIcon(), this);
 267
 268     m_systrayIcon->setToolTip(m_toolTip);
 269
 270     if (m_menu)
 271         m_systrayIcon->setContextMenu(m_menu);
 272
 273     connect(m_systrayIcon, &QSystemTrayIcon::activated, this
 274             , [this](const QSystemTrayIcon::ActivationReason reason)
 275     {
 276         if (reason == QSystemTrayIcon::Trigger)
 277             emit activationRequested();
 278     });
 279 #ifndef QBT_USES_CUSTOMDBUSNOTIFICATIONS
 280     connect(m_systrayIcon, &QSystemTrayIcon::messageClicked, this, &DesktopIntegration::notificationClicked);
 281 #endif
 282
 283     m_systrayIcon->show();
 284     emit stateChanged();
 285 }
 286
 287 QIcon DesktopIntegration::getSystrayIcon() const
 288 {
 289     const TrayIcon::Style style = Preferences::instance()->trayIconStyle();
 290     switch (style)
 291     {
 292     default:
 293     case TrayIcon::Style::Normal:
 294         return UIThemeManager::instance()->getIcon(u"qbittorrent-tray"_qs);
 295
 296     case TrayIcon::Style::MonoDark:
 297         return UIThemeManager::instance()->getIcon(u"qbittorrent-tray-dark"_qs);
 298
 299     case TrayIcon::Style::MonoLight:
 300         return UIThemeManager::instance()->getIcon(u"qbittorrent-tray-light"_qs);
 301     }
 302 }
 303 #endif // Q_OS_MACOS