diff options
Diffstat (limited to 'src/leap/bitmask/gui/app.py')
| -rw-r--r-- | src/leap/bitmask/gui/app.py | 69 |
1 files changed, 31 insertions, 38 deletions
diff --git a/src/leap/bitmask/gui/app.py b/src/leap/bitmask/gui/app.py index a616dc3b..2033fc6d 100644 --- a/src/leap/bitmask/gui/app.py +++ b/src/leap/bitmask/gui/app.py @@ -25,7 +25,6 @@ import os import platform import signal import sys -import time import webbrowser from functools import partial @@ -33,12 +32,12 @@ from multiprocessing import Process from leap.bitmask.core.launcher import run_bitmaskd, pid from leap.bitmask.gui import app_rc +from leap.bitmask.gui.systray import WithTrayIcon +from leap.bitmask.gui.housekeeping import cleanup, terminate, reset_authtoken +from leap.bitmask.gui.housekeeping import get_authenticated_url +from leap.bitmask.gui.housekeeping import NoAuthTokenError from leap.common.config import get_path_prefix -from .housekeeping import cleanup, terminate, reset_authtoken -from .housekeeping import get_authenticated_url - -from .systray import WithTrayIcon if platform.system() == 'Windows': from multiprocessing import freeze_support @@ -57,12 +56,9 @@ else: from PyQt5.QtWidgets import QDialog from PyQt5.QtWidgets import QMessageBox - try: - from PyQt5.QtWebKitWidgets import QWebView - from PyQt5.QtWebKit import QWebSettings - except ImportError: - from PyQt5.QtWebEngineWidgets import QWebEngineView as QWebView - from PyQt5.QtWebEngineWidgets import QWebEngineSettings as QWebSettings + from PyQt5.QtWebEngineWidgets import QWebEngineView as QWebView + from PyQt5.QtWebEngineWidgets import QWebEngineSettings as QWebSettings + from PyQt5.QtWebChannel import QWebChannel IS_WIN = platform.system() == "Windows" DEBUG = os.environ.get("DEBUG", False) @@ -90,8 +86,10 @@ class BrowserWindow(QWebView, WithTrayIcon): def __init__(self, *args, **kw): url = kw.pop('url', None) + first = False if not url: url = get_authenticated_url() + first = True self.url = url self.closing = False @@ -100,9 +98,13 @@ class BrowserWindow(QWebView, WithTrayIcon): self.bitmask_browser = NewPageConnector(self) if first else None self.loadPage(self.url) - self.proxy = AppProxy(self) if first else None - self.frame.addToJavaScriptWindowObject( - "bitmaskApp", self.proxy) + self.bridge = AppBridge(self) if first else None + + if self.bridge is not None: + print "[+] registering python<->js bridge" + channel = QWebChannel(self) + channel.registerObject("bitmaskApp", self.bridge) + self.page().setWebChannel(channel) icon = QtGui.QIcon() icon.addPixmap( @@ -111,26 +113,10 @@ class BrowserWindow(QWebView, WithTrayIcon): self.setWindowIcon(icon) def loadPage(self, web_page): - try: - if os.environ.get('DEBUG'): - self.settings().setAttribute( - QWebSettings.DeveloperExtrasEnabled, True) - except Exception: - pass - - if os.environ.get('DEBUG'): - self.inspector = QWebInspector(self) - self.inspector.setPage(self.page()) - self.inspector.show() - if os.path.isabs(web_page): web_page = os.path.relpath(web_page) url = QtCore.QUrl(web_page) - # TODO -- port this to QWebEngine - self.frame = self.page().mainFrame() - self.frame.addToJavaScriptWindowObject( - "bitmaskBrowser", self.bitmask_browser) self.load(url) def shutdown(self, *args): @@ -138,10 +124,12 @@ class BrowserWindow(QWebView, WithTrayIcon): if self.closing: return self.closing = True + bitmaskd.join() terminate(pid) cleanup() print('[bitmask] shutting down gui...') + try: self.stop() try: @@ -157,11 +145,11 @@ class BrowserWindow(QWebView, WithTrayIcon): sys.exit(1) -class AppProxy(QObject): +class AppBridge(QObject): @pyqtSlot() def shutdown(self): - """To be exposed from the js bridge""" + print "[+] shutdown called from js" global browser if browser: browser.user_closed = True @@ -171,8 +159,8 @@ class AppProxy(QObject): def openSystemBrowser(self, url): webbrowser.open(url) - pixbrowser = None +closing = False class NewPageConnector(QObject): @@ -185,12 +173,17 @@ class NewPageConnector(QObject): def _handle_kill(*args, **kw): + global pixbrowser + global closing + if closing: + sys.exit() win = kw.get('win') if win: + win.user_closed = True QtCore.QTimer.singleShot(0, win.close) - global pixbrowser if pixbrowser: QtCore.QTimer.singleShot(0, pixbrowser.close) + closing = True def launch_gui(): @@ -206,7 +199,7 @@ def launch_gui(): qApp = QApplication([]) try: browser = BrowserWindow(None) - except NoAuthToken as e: + except NoAuthTokenError as e: print('ERROR: ' + e.message) sys.exit(1) @@ -237,6 +230,9 @@ def start_app(): # Allow the frozen binary in the bundle double as the cli entrypoint # Why have only a user interface when you can have two? + if DEBUG: + os.environ.setdefault('QTWEBENGINE_REMOTE_DEBUGGING', '8081') + if platform.system() == 'Windows': # In windows, there are some args added to the invocation # by PyInstaller, I guess... @@ -257,9 +253,6 @@ def start_app(): launch_gui() -class NoAuthToken(Exception): - pass - if __name__ == "__main__": start_app() |