112 lines
3.8 KiB
Python
112 lines
3.8 KiB
Python
from PyQt6.QtWidgets import QSystemTrayIcon, QMenu, QApplication
|
|
from PyQt6.QtGui import QAction
|
|
from PyQt6.QtCore import QTimer
|
|
|
|
from app.core.config import Config
|
|
from app.core import wireguard as wg_core
|
|
from app.ui import icons
|
|
|
|
|
|
class SystemTray(QSystemTrayIcon):
|
|
def __init__(self, config: Config, main_window, parent=None):
|
|
super().__init__(icons.icon_disconnected(), parent)
|
|
self._cfg = config
|
|
self._win = main_window
|
|
self._connected = False
|
|
self._build_menu()
|
|
self.setToolTip("WGSecure — Déconnecté")
|
|
self.activated.connect(self._on_activated)
|
|
|
|
self._poll = QTimer(parent)
|
|
self._poll.timeout.connect(self._update_status)
|
|
self._poll.start(5000)
|
|
self._update_status()
|
|
|
|
def _build_menu(self):
|
|
menu = QMenu()
|
|
|
|
# Ligne de statut (désactivée)
|
|
self._action_status = QAction("● Déconnecté", self)
|
|
self._action_status.setEnabled(False)
|
|
menu.addAction(self._action_status)
|
|
|
|
menu.addSeparator()
|
|
|
|
# Afficher la fenêtre
|
|
self._action_show = QAction(icons.icon_app(), "Afficher la fenêtre", self)
|
|
self._action_show.triggered.connect(self._show_window)
|
|
menu.addAction(self._action_show)
|
|
|
|
menu.addSeparator()
|
|
|
|
# Connexion / déconnexion
|
|
self._action_toggle = QAction(icons.icon_disconnected(), "Se connecter", self)
|
|
self._action_toggle.triggered.connect(self._toggle_connection)
|
|
menu.addAction(self._action_toggle)
|
|
|
|
menu.addSeparator()
|
|
|
|
# Panneau de configuration
|
|
self._action_config = QAction(icons.icon_admin(), "Configuration", self)
|
|
self._action_config.triggered.connect(self._win._open_admin)
|
|
menu.addAction(self._action_config)
|
|
|
|
menu.addSeparator()
|
|
|
|
# Quitter
|
|
self._action_quit = QAction("Quitter", self)
|
|
self._action_quit.triggered.connect(self._quit)
|
|
menu.addAction(self._action_quit)
|
|
|
|
self.setContextMenu(menu)
|
|
|
|
def _on_activated(self, reason: QSystemTrayIcon.ActivationReason):
|
|
if reason == QSystemTrayIcon.ActivationReason.Trigger:
|
|
self._show_window()
|
|
|
|
def _show_window(self):
|
|
self._win.show()
|
|
self._win.raise_()
|
|
self._win.activateWindow()
|
|
|
|
def _update_status(self):
|
|
prev = self._connected
|
|
try:
|
|
self._connected = wg_core.is_connected(self._cfg)
|
|
except Exception:
|
|
self._connected = False
|
|
|
|
if self._connected:
|
|
self.setIcon(icons.icon_connected())
|
|
self.setToolTip("WGSecure — Connecté")
|
|
self._action_status.setText("● Connecté")
|
|
self._action_toggle.setIcon(icons.icon_connected())
|
|
self._action_toggle.setText("Se déconnecter")
|
|
if not prev:
|
|
srv = self._cfg.wg.get("server_endpoint", "")
|
|
self.showMessage(
|
|
"WGSecure — Connecté ✓",
|
|
f"Tunnel WireGuard actif{(' → ' + srv) if srv else ''}",
|
|
QSystemTrayIcon.MessageIcon.Information, 3000,
|
|
)
|
|
else:
|
|
self.setIcon(icons.icon_disconnected())
|
|
self.setToolTip("WGSecure — Déconnecté")
|
|
self._action_status.setText("● Déconnecté")
|
|
self._action_toggle.setIcon(icons.icon_disconnected())
|
|
self._action_toggle.setText("Se connecter")
|
|
if prev:
|
|
self.showMessage(
|
|
"WGSecure — Déconnecté",
|
|
"Le tunnel WireGuard a été désactivé.",
|
|
QSystemTrayIcon.MessageIcon.Warning, 3000,
|
|
)
|
|
|
|
def _toggle_connection(self):
|
|
self._win._on_connect()
|
|
self._update_status()
|
|
|
|
def _quit(self):
|
|
self._poll.stop()
|
|
QApplication.quit()
|