summaryrefslogtreecommitdiff
path: root/src/leap/bitmask/gui
diff options
context:
space:
mode:
Diffstat (limited to 'src/leap/bitmask/gui')
-rw-r--r--src/leap/bitmask/gui/app.py37
1 files changed, 33 insertions, 4 deletions
diff --git a/src/leap/bitmask/gui/app.py b/src/leap/bitmask/gui/app.py
index ffb533b..9823d29 100644
--- a/src/leap/bitmask/gui/app.py
+++ b/src/leap/bitmask/gui/app.py
@@ -43,17 +43,17 @@ if platform.system() == 'Windows':
from PySide.QtGui import QDialog
from PySide.QtGui import QApplication
from PySide.QtWebKit import QWebView, QGraphicsWebView
- from PySide.QtCore import QSize
else:
from PyQt5 import QtCore, QtGui
- from PyQt5.QtCore import QSize
from PyQt5.QtCore import QObject, pyqtSlot
from PyQt5.QtWidgets import QApplication
from PyQt5.QtGui import QIcon
from PyQt5.QtGui import QPixmap
+ from PyQt5.QtWidgets import QAction
from PyQt5.QtWidgets import QMenu
from PyQt5.QtWidgets import QSystemTrayIcon
from PyQt5.QtWidgets import QDialog
+ from PyQt5.QtWidgets import QMessageBox
try:
from PyQt5.QtWebKitWidgets import QWebView
@@ -80,14 +80,18 @@ TRAY_ICONS = (
':/black/22/on.png',
':/black/22/off.png')
+
class WithTrayIcon(QDialog):
+ user_closed = False
+
def setupSysTray(self):
self._createIcons()
+ self._createActions()
self._createTrayIcon()
self.setVPNStatus('off')
- self.trayIcon.show()
self.setUpEventListener()
+ self.trayIcon.show()
def setVPNStatus(self, status):
seticon = self.trayIcon.setIcon
@@ -106,7 +110,9 @@ class WithTrayIcon(QDialog):
settip('VPN: Stopping')
def setUpEventListener(self):
- leap_events.register(catalog.VPN_STATUS_CHANGED, self._handle_vpn_event)
+ leap_events.register(
+ catalog.VPN_STATUS_CHANGED,
+ self._handle_vpn_event)
def _handle_vpn_event(self, *args):
status = None
@@ -119,8 +125,31 @@ class WithTrayIcon(QDialog):
self.ICON_ON = QIcon(QPixmap(TRAY_ICONS[1]))
self.ICON_OFF = QIcon(QPixmap(TRAY_ICONS[2]))
+ def _createActions(self):
+ self.quitAction = QAction(
+ "&Quit", self,
+ triggered=self.closeFromSystray)
+
+ def closeFromSystray(self):
+ self.user_closed = True
+ self.close()
+
def _createTrayIcon(self):
+ self.trayIconMenu = QMenu(self)
+ self.trayIconMenu.addAction(self.quitAction)
self.trayIcon = QSystemTrayIcon(self)
+ self.trayIcon.setContextMenu(self.trayIconMenu)
+
+ def closeEvent(self, event):
+ if self.trayIcon.isVisible() and not self.user_closed:
+ QMessageBox.information(
+ self, "Systray",
+ "The program will keep running in the system tray. To "
+ "terminate the program, choose <b>Quit</b> in the "
+ "context menu of the system tray entry.")
+ self.hide()
+ if not self.user_closed:
+ event.ignore()
class BrowserWindow(QWebView, WithTrayIcon):