mirror of
https://github.com/onionshare/onionshare.git
synced 2025-01-04 20:21:05 -05:00
271 lines
9.2 KiB
Python
271 lines
9.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
OnionShare | https://onionshare.org/
|
|
|
|
Copyright (C) 2014-2022 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)
|
|
self.setDesktopFileName("org.onionshare.OnionShare")
|
|
self.setOrganizationDomain("org.onionshare.OnionShare")
|
|
self.setOrganizationName("OnionShare")
|
|
|
|
QtWidgets.QApplication.__init__(self, sys.argv)
|
|
|
|
if common.platform == "Darwin" or common.platform == "Windows":
|
|
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_())
|