diff options
Diffstat (limited to 'desktop/onionshare/__init__.py')
| -rw-r--r-- | desktop/onionshare/__init__.py | 261 |
1 files changed, 261 insertions, 0 deletions
diff --git a/desktop/onionshare/__init__.py b/desktop/onionshare/__init__.py new file mode 100644 index 00000000..40a91913 --- /dev/null +++ b/desktop/onionshare/__init__.py @@ -0,0 +1,261 @@ +# -*- coding: utf-8 -*- +""" +OnionShare | https://onionshare.org/ + +Copyright (C) 2014-2021 Micah Lee, et al. <micah@micahflee.com> + +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 <http://www.gnu.org/licenses/>. +""" + +from __future__ import division +import os +import sys +import argparse +import signal +import json +import psutil +import getpass +from PySide2 import QtCore, QtWidgets, QtGui + +from PySide2.QtCore import Slot,Qt +from PySide2.QtGui import QPalette, QColor + +from onionshare_cli.common import Common +from onionshare_cli.settings import Settings + +from .gui_common import GuiCommon +from .widgets import Alert +from .main_window import MainWindow + + +class Application(QtWidgets.QApplication): + """ + This is Qt's QApplication class. It has been overridden to support threads + and the quick keyboard shortcut. + """ + + def __init__(self, common): + if common.platform == "Linux" or common.platform == "BSD": + self.setAttribute(QtCore.Qt.AA_X11InitThreads, True) + QtWidgets.QApplication.__init__(self, sys.argv) + self.setStyle("Fusion") + + # Check color mode on starting the app + self.color_mode = self.get_color_mode(common) + + # Enable Dark Theme + if self.color_mode == "dark": + self.setDarkMode() + else: + self.setLightMode() + + self.installEventFilter(self) + + def eventFilter(self, obj, event): + if ( + event.type() == QtCore.QEvent.KeyPress + and event.key() == QtCore.Qt.Key_Q + and event.modifiers() == QtCore.Qt.ControlModifier + ): + self.quit() + return False + + def is_dark_mode(self): + baseColor = QtGui.QPalette().color(QtGui.QPalette.Base) + if baseColor.name().lower() == "#ffffff": + return False + return True + + def setLightMode(self): + light_palette = QPalette() + light_palette.setColor(QPalette.Window, QColor(236, 236, 236)) + light_palette.setColor(QPalette.WindowText, Qt.black) + light_palette.setColor(QPalette.Base, Qt.white) + light_palette.setColor(QPalette.AlternateBase, QColor(245, 245, 245)) + light_palette.setColor(QPalette.ToolTipBase, Qt.white) + light_palette.setColor(QPalette.ToolTipText, Qt.black) + light_palette.setColor(QPalette.Text, Qt.black) + light_palette.setColor(QPalette.Button, QColor(236, 236, 236)) + light_palette.setColor(QPalette.ButtonText, Qt.black) + light_palette.setColor(QPalette.BrightText, Qt.white) + light_palette.setColor(QPalette.Link, QColor(0, 104, 218)) + light_palette.setColor(QPalette.Highlight, QColor(179, 215, 255)) + light_palette.setColor(QPalette.HighlightedText, Qt.black) + light_palette.setColor(QPalette.Light, Qt.white) + light_palette.setColor(QPalette.Midlight, QColor(245, 245, 245)) + light_palette.setColor(QPalette.Dark, QColor(191, 191, 191)) + light_palette.setColor(QPalette.Mid, QColor(169, 169, 169)) + light_palette.setColor(QPalette.Shadow, Qt.black) + self.setPalette(light_palette) + + def setDarkMode(self): + dark_palette = QPalette() + dark_palette.setColor(QPalette.Window, QColor(53, 53, 53)) + dark_palette.setColor(QPalette.WindowText, Qt.white) + dark_palette.setColor(QPalette.Base, QColor(25, 25, 25)) + dark_palette.setColor(QPalette.AlternateBase, QColor(53, 53, 53)) + dark_palette.setColor(QPalette.ToolTipBase, Qt.white) + dark_palette.setColor(QPalette.ToolTipText, Qt.white) + dark_palette.setColor(QPalette.Text, Qt.white) + dark_palette.setColor(QPalette.Button, QColor(53, 53, 53)) + dark_palette.setColor(QPalette.ButtonText, Qt.white) + dark_palette.setColor(QPalette.BrightText, Qt.red) + dark_palette.setColor(QPalette.Link, QColor(42, 130, 218)) + dark_palette.setColor(QPalette.Highlight, QColor(42, 130, 218)) + dark_palette.setColor(QPalette.HighlightedText, Qt.black) + self.setPalette(dark_palette) + self.setStyleSheet("QToolTip { color: #ffffff; background-color: #2a82da; border: 1px solid white; }") + + def get_color_mode(self, common): + curr_settings = Settings(common) + curr_settings.load() + current_theme = curr_settings.get("theme") + + if current_theme == 1: + return "light" + elif current_theme == 2: + return "dark" + else: + return "dark" if self.is_dark_mode() else "light" + +def main(): + """ + The main() function implements all of the logic that the GUI version of onionshare uses. + """ + common = Common() + common.display_banner() + + # Required for macOS Big Sur: https://stackoverflow.com/a/64878899 + if common.platform == "Darwin": + os.environ["QT_MAC_WANTS_LAYER"] = "1" + + # Start the Qt app + global qtapp + qtapp = Application(common) + + # Parse arguments + parser = argparse.ArgumentParser( + formatter_class=lambda prog: argparse.HelpFormatter(prog, max_help_position=48) + ) + parser.add_argument( + "--local-only", + action="store_true", + dest="local_only", + help="Don't use Tor (only for development)", + ) + parser.add_argument( + "-v", + "--verbose", + action="store_true", + dest="verbose", + help="Log OnionShare errors to stdout, and web errors to disk", + ) + parser.add_argument( + "--filenames", + metavar="filenames", + nargs="+", + help="List of files or folders to share", + ) + args = parser.parse_args() + + filenames = args.filenames + if filenames: + for i in range(len(filenames)): + filenames[i] = os.path.abspath(filenames[i]) + + local_only = bool(args.local_only) + verbose = bool(args.verbose) + + # Verbose mode? + common.verbose = verbose + + # Attach the GUI common parts to the common object + common.gui = GuiCommon(common, qtapp, local_only) + + # Validation + if filenames: + valid = True + for filename in filenames: + if not os.path.isfile(filename) and not os.path.isdir(filename): + Alert(common, f"{filename} is not a valid file.") + valid = False + if not os.access(filename, os.R_OK): + Alert(common, f"{filename} is not a readable file.") + valid = False + if not valid: + sys.exit() + + # Is there another onionshare-gui running? + if os.path.exists(common.gui.lock_filename): + with open(common.gui.lock_filename, "r") as f: + existing_pid = int(f.read()) + + # Is this process actually still running? + still_running = True + if not psutil.pid_exists(existing_pid): + still_running = False + else: + for proc in psutil.process_iter(["pid", "name", "username"]): + if proc.pid == existing_pid: + if ( + proc.username() != getpass.getuser() + or "onionshare" not in " ".join(proc.cmdline()).lower() + ): + still_running = False + + if still_running: + print(f"Opening tab in existing OnionShare window (pid {existing_pid})") + + # Make an event for the existing OnionShare window + if filenames: + obj = {"type": "new_share_tab", "filenames": filenames} + else: + obj = {"type": "new_tab"} + + # Write that event to disk + with open(common.gui.events_filename, "a") as f: + f.write(json.dumps(obj) + "\n") + return + else: + os.remove(common.gui.lock_filename) + + # Write the lock file + with open(common.gui.lock_filename, "w") as f: + f.write(f"{os.getpid()}\n") + + # Allow Ctrl-C to smoothly quit the program instead of throwing an exception + def signal_handler(s, frame): + print("\nCtrl-C pressed, quitting") + if os.path.exists(common.gui.lock_filename): + os.remove(common.gui.lock_filename) + sys.exit(0) + + signal.signal(signal.SIGINT, signal_handler) + + # Launch the gui + main_window = MainWindow(common, filenames) + + # If filenames were passed in, open them in a tab + if filenames: + main_window.tabs.new_share_tab(filenames) + + # Clean up when app quits + def shutdown(): + main_window.cleanup() + os.remove(common.gui.lock_filename) + + qtapp.aboutToQuit.connect(shutdown) + + # All done + sys.exit(qtapp.exec_()) |