Files
LorenEteval_Furious/Furious/Widget/SystemTrayIcon.py
T
2024-02-07 12:33:03 +08:00

141 lines
4.6 KiB
Python

# Copyright (C) 2024 Loren Eteval <loren.eteval@proton.me>
#
# This file is part of Furious.
#
# 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 <https://www.gnu.org/licenses/>.
from Furious.PyFramework import *
from Furious.QtFramework import *
from Furious.QtFramework import gettext as _
from Furious.Utility import *
from Furious.TrayActions import *
from PySide6 import QtCore
from PySide6.QtWidgets import QSystemTrayIcon
import logging
logger = logging.getLogger(__name__)
__all__ = ['SystemTrayIcon']
class SystemTrayIcon(QTranslatable, SupportConnectedCallback, QSystemTrayIcon):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
actions = [
ConnectAction(),
RoutingAction(),
SystemProxyAction(),
AppQSeperator(),
ImportAction(),
EditConfigurationAction(),
AppQSeperator(),
LanguageAction(),
SettingsAction(),
AppQSeperator(),
ExitAction(),
]
# Some old version PySide6 does not have setMenu method
# for QAction. Protect it. Currently only used in SystemTrayIcon
if hasattr(AppQAction, 'setMenu'):
logger.info('contextMenu uses setMenu implementation')
for action in actions:
if isinstance(action, AppQAction):
if hasattr(self, f'{action}'):
logger.warning(f'{self} already has action {action}')
setattr(self, f'{action}', action)
self._menu = AppQMenu(*actions)
self.setContextMenu(self._menu)
else:
logger.info('contextMenu uses addMenu implementation')
self._refs = []
self._menu = AppQMenu()
for action in actions:
if isinstance(action, AppQAction):
if hasattr(self, f'{action}'):
logger.warning(f'{self} already has action {action}')
setattr(self, f'{action}', action)
if action._menu is None:
self._menu.addAction(action)
else:
menu = AppQMenu(*action._menu._actions, title=action.text())
menu.setIcon(action.icon())
self._refs.append(menu)
self._menu.addMenu(menu)
else:
self._menu.addSeparator()
self.setContextMenu(self._menu)
self.setDisconnectedIcon()
self.activated.connect(self.handleActivated)
def bootstrap(self):
if AppSettings.isStateON_('StartupOnBoot'):
# Rrefresh startup application location
StartupOnBoot.on_()
if AppSettings.isStateON_('Connect'):
self.ConnectAction.trigger()
def showMessage(self, message: str, *args, **kwargs):
if message:
super().showMessage(_(APPLICATION_NAME), message, *args, **kwargs)
def setDisconnectedIcon(self):
if PLATFORM == 'Darwin' or isWindows7():
# Darker
self.setIcon(bootstrapIcon('rocket-takeoff-dark.svg'))
else:
self.setIcon(bootstrapIcon('rocket-takeoff.svg'))
def setConnectedIcon(self):
if isAdministrator():
self.setIcon(bootstrapIcon('rocket-takeoff-admin-connected.svg'))
else:
if PLATFORM == 'Darwin' or isWindows7():
# Darker
self.setIcon(bootstrapIcon('rocket-takeoff-connected-dark.svg'))
else:
self.setIcon(bootstrapIcon('rocket-takeoff-connected.svg'))
@QtCore.Slot(QSystemTrayIcon.ActivationReason)
def handleActivated(self, reason):
if reason == QSystemTrayIcon.ActivationReason.DoubleClick:
APP().mainWindow.show()
def setCustomToolTip(self):
self.setToolTip(f'{_(APPLICATION_NAME)} {APPLICATION_VERSION}')
def disconnectedCallback(self):
self.setDisconnectedIcon()
def connectedCallback(self):
self.setConnectedIcon()
def retranslate(self):
# Nothing to do
pass