feat: added basic webview widget

aw_qt/trayicon.py

@@ -3,9 +3,9 @@
 import signal
 import os
 import subprocess
-from collections import defaultdict
-from typing import Any, DefaultDict, List, Optional, Dict
 import webbrowser
+from typing import Any, Optional, Dict
+
 
 from PyQt5 import QtCore
 from PyQt5.QtWidgets import (
@@ -21,6 +21,7 @@
 import aw_core
 
 from .manager import Manager, Module
+from .webview import create_webview
 
 logger = logging.getLogger(__name__)
 
@@ -44,7 +45,10 @@ def get_env() -> Dict[str, str]:
 
 
 def open_url(url: str) -> None:
-    if sys.platform == "linux":
+    experimental = True
+    if experimental:
+        create_webview(parent=widget)
+    elif sys.platform == "linux":
         env = get_env()
         subprocess.Popen(["xdg-open", url], env=env)
     else:
@@ -87,13 +91,17 @@ def __init__(
         self.manager = manager
         self.testing = testing
 
-        self.root_url = "http://localhost:{port}".format(port=5666 if self.testing else 5600)
+        self.root_url = "http://localhost:{port}".format(
+            port=5666 if self.testing else 5600
+        )
         self.activated.connect(self.on_activated)
 
         self._build_rootmenu()
 
     def on_activated(self, reason: QSystemTrayIcon.ActivationReason) -> None:
+        logger.info("Clicked trayicon")
         if reason == QSystemTrayIcon.DoubleClick:
+            logger.info("Double-clicked trayicon")
             open_webui(self.root_url)
 
     def _build_rootmenu(self) -> None:
@@ -229,6 +237,7 @@ def run(manager: Manager, testing: bool = False) -> Any:
         )
         sys.exit(1)
 
+    global widget
     widget = QWidget()
     if sys.platform == "darwin":
         icon = QIcon(":/black-monochrome-logo.png")

aw_qt/webview.py

@@ -0,0 +1,79 @@
+import sys
+import signal
+
+from PyQt5.QtCore import QUrl, QTimer
+from PyQt5.QtWidgets import QApplication, QWidget
+
+from PyQt5.QtWebEngineWidgets import (
+    QWebEngineView,
+    QWebEnginePage,
+    QWebEngineSettings,
+)
+
+
+class MyBrowser(QWebEnginePage):
+    """Settings for the browser."""
+
+    def userAgentForUrl(self, url):
+        """Returns a User Agent that will be seen by the website."""
+        return "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2228.0 Safari/537.36"
+
+
+class Browser(QWebEngineView):
+    def __init__(self, *args, **kwargs):
+        QWebEngineView.__init__(self, *args, **kwargs)
+        self.setPage(MyBrowser())
+
+        settings = QWebEngineSettings.globalSettings()
+        settings.setAttribute(QWebEngineSettings.JavascriptEnabled, True)
+        settings.setAttribute(QWebEngineSettings.AllowRunningInsecureContent, True)
+
+        self.setWindowTitle("Loading...")
+        self.titleChanged.connect(self.adjustTitle)
+        # super(Browser).connect(self.ui.webView,QtCore.SIGNAL("titleChanged (const QString&)"), self.adjustTitle)
+
+    def load(self, url):
+        self.setUrl(QUrl(url))
+
+    def adjustTitle(self):
+        self.setWindowTitle(self.title())
+
+
+def exit() -> None:
+    print("Shutdown initiated, stopping all services...")
+    # Terminate entire process group, just in case.
+    # os.killpg(0, signal.SIGINT)
+
+    QApplication.quit()
+
+
+def create_webview(parent):
+    view = Browser(parent=parent)
+    # view.showMaximized()
+    view.load("http://localhost:5666/")
+    view.show()
+    print("Opened webview")
+
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    app.setApplicationName("ActivityWatch")
+
+    # Without this, Ctrl+C will have no effect
+    signal.signal(signal.SIGINT, lambda *args: exit())
+    # Ensure cleanup happens on SIGTERM
+    signal.signal(signal.SIGTERM, lambda *args: exit())
+
+    timer = QTimer()
+    timer.start(100)  # You may change this if you wish.
+    timer.timeout.connect(lambda: None)  # Let the interpreter run each 500 ms.
+
+    widget = QWidget()
+    create_webview(widget)
+    widget.show()
+
+    QApplication.setQuitOnLastWindowClosed(False)
+
+    print("Initialized")
+
+    app.exec_()