diff --git a/onionshare_gui/server_status.py b/onionshare_gui/server_status.py deleted file mode 100644 index 251aa024..00000000 --- a/onionshare_gui/server_status.py +++ /dev/null @@ -1,354 +0,0 @@ -# -*- coding: utf-8 -*- -""" -OnionShare | https://onionshare.org/ - -Copyright (C) 2014-2018 Micah Lee - -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 . -""" -import platform -import textwrap -from PyQt5 import QtCore, QtWidgets, QtGui -from PyQt5.QtCore import Qt - -from onionshare import strings - -from .widgets import Alert - -class ServerStatus(QtWidgets.QWidget): - """ - The server status chunk of the GUI. - """ - server_started = QtCore.pyqtSignal() - server_started_finished = QtCore.pyqtSignal() - server_stopped = QtCore.pyqtSignal() - server_canceled = QtCore.pyqtSignal() - button_clicked = QtCore.pyqtSignal() - url_copied = QtCore.pyqtSignal() - hidservauth_copied = QtCore.pyqtSignal() - - MODE_SHARE = 'share' - MODE_RECEIVE = 'receive' - - STATUS_STOPPED = 0 - STATUS_WORKING = 1 - STATUS_STARTED = 2 - - def __init__(self, common, qtapp, app, file_selection=None, local_only=False): - super(ServerStatus, self).__init__() - - self.common = common - - self.status = self.STATUS_STOPPED - self.mode = None # Gets set in self.set_mode - - self.qtapp = qtapp - self.app = app - - self.web = None - self.local_only = local_only - - self.resizeEvent(None) - - # Shutdown timeout layout - self.shutdown_timeout_label = QtWidgets.QLabel(strings._('gui_settings_shutdown_timeout')) - self.shutdown_timeout = QtWidgets.QDateTimeEdit() - self.shutdown_timeout.setDisplayFormat("hh:mm A MMM d, yy") - if self.local_only: - # For testing - self.shutdown_timeout.setDateTime(QtCore.QDateTime.currentDateTime().addSecs(15)) - self.shutdown_timeout.setMinimumDateTime(QtCore.QDateTime.currentDateTime()) - else: - # Set proposed timeout to be 5 minutes into the future - self.shutdown_timeout.setDateTime(QtCore.QDateTime.currentDateTime().addSecs(300)) - # Onion services can take a little while to start, so reduce the risk of it expiring too soon by setting the minimum to 60s from now - self.shutdown_timeout.setMinimumDateTime(QtCore.QDateTime.currentDateTime().addSecs(60)) - self.shutdown_timeout.setCurrentSection(QtWidgets.QDateTimeEdit.MinuteSection) - shutdown_timeout_layout = QtWidgets.QHBoxLayout() - shutdown_timeout_layout.addWidget(self.shutdown_timeout_label) - shutdown_timeout_layout.addWidget(self.shutdown_timeout) - - # Shutdown timeout container, so it can all be hidden and shown as a group - shutdown_timeout_container_layout = QtWidgets.QVBoxLayout() - shutdown_timeout_container_layout.addLayout(shutdown_timeout_layout) - self.shutdown_timeout_container = QtWidgets.QWidget() - self.shutdown_timeout_container.setLayout(shutdown_timeout_container_layout) - self.shutdown_timeout_container.hide() - - # Server layout - self.server_button = QtWidgets.QPushButton() - self.server_button.clicked.connect(self.server_button_clicked) - - # URL layout - url_font = QtGui.QFontDatabase.systemFont(QtGui.QFontDatabase.FixedFont) - self.url_description = QtWidgets.QLabel() - self.url_description.setWordWrap(True) - self.url_description.setMinimumHeight(50) - self.url = QtWidgets.QLabel() - self.url.setFont(url_font) - self.url.setWordWrap(True) - self.url.setMinimumSize(self.url.sizeHint()) - self.url.setStyleSheet(self.common.css['server_status_url']) - self.url.setTextInteractionFlags(Qt.TextSelectableByMouse) - - self.copy_url_button = QtWidgets.QPushButton(strings._('gui_copy_url')) - self.copy_url_button.setFlat(True) - self.copy_url_button.setStyleSheet(self.common.css['server_status_url_buttons']) - self.copy_url_button.setMinimumHeight(20) - self.copy_url_button.clicked.connect(self.copy_url) - self.copy_hidservauth_button = QtWidgets.QPushButton(strings._('gui_copy_hidservauth')) - self.copy_hidservauth_button.setFlat(True) - self.copy_hidservauth_button.setStyleSheet(self.common.css['server_status_url_buttons']) - self.copy_hidservauth_button.clicked.connect(self.copy_hidservauth) - url_buttons_layout = QtWidgets.QHBoxLayout() - url_buttons_layout.addWidget(self.copy_url_button) - url_buttons_layout.addWidget(self.copy_hidservauth_button) - url_buttons_layout.addStretch() - - url_layout = QtWidgets.QVBoxLayout() - url_layout.addWidget(self.url_description) - url_layout.addWidget(self.url) - url_layout.addLayout(url_buttons_layout) - - # Add the widgets - layout = QtWidgets.QVBoxLayout() - layout.addWidget(self.server_button) - layout.addLayout(url_layout) - layout.addWidget(self.shutdown_timeout_container) - self.setLayout(layout) - - def set_mode(self, share_mode, file_selection=None): - """ - The server status is in share mode. - """ - self.mode = share_mode - - if self.mode == ServerStatus.MODE_SHARE: - self.file_selection = file_selection - - self.update() - - def resizeEvent(self, event): - """ - When the widget is resized, try and adjust the display of a v3 onion URL. - """ - try: - # Wrap the URL label - url_length=len(self.get_url()) - if url_length > 60: - width = self.frameGeometry().width() - if width < 530: - wrapped_onion_url = textwrap.fill(self.get_url(), 46) - self.url.setText(wrapped_onion_url) - else: - self.url.setText(self.get_url()) - except: - pass - - - def shutdown_timeout_reset(self): - """ - Reset the timeout in the UI after stopping a share - """ - self.shutdown_timeout.setDateTime(QtCore.QDateTime.currentDateTime().addSecs(300)) - if not self.local_only: - self.shutdown_timeout.setMinimumDateTime(QtCore.QDateTime.currentDateTime().addSecs(60)) - - def update(self): - """ - Update the GUI elements based on the current state. - """ - # Set the URL fields - if self.status == self.STATUS_STARTED: - self.url_description.show() - - info_image = self.common.get_resource_path('images/info.png') - - if self.mode == ServerStatus.MODE_SHARE: - self.url_description.setText(strings._('gui_share_url_description').format(info_image)) - else: - self.url_description.setText(strings._('gui_receive_url_description').format(info_image)) - - # Show a Tool Tip explaining the lifecycle of this URL - if self.common.settings.get('save_private_key'): - if self.mode == ServerStatus.MODE_SHARE and self.common.settings.get('close_after_first_download'): - self.url_description.setToolTip(strings._('gui_url_label_onetime_and_persistent')) - else: - self.url_description.setToolTip(strings._('gui_url_label_persistent')) - else: - if self.mode == ServerStatus.MODE_SHARE and self.common.settings.get('close_after_first_download'): - self.url_description.setToolTip(strings._('gui_url_label_onetime')) - else: - self.url_description.setToolTip(strings._('gui_url_label_stay_open')) - - self.url.setText(self.get_url()) - self.url.show() - - self.copy_url_button.show() - - if self.common.settings.get('save_private_key'): - if not self.common.settings.get('slug'): - self.common.settings.set('slug', self.web.slug) - self.common.settings.save() - - if self.common.settings.get('shutdown_timeout'): - self.shutdown_timeout_container.hide() - - if self.app.stealth: - self.copy_hidservauth_button.show() - else: - self.copy_hidservauth_button.hide() - else: - self.url_description.hide() - self.url.hide() - self.copy_url_button.hide() - self.copy_hidservauth_button.hide() - - # Button - if self.mode == ServerStatus.MODE_SHARE and self.file_selection.get_num_files() == 0: - self.server_button.hide() - else: - self.server_button.show() - - if self.status == self.STATUS_STOPPED: - self.server_button.setStyleSheet(self.common.css['server_status_button_stopped']) - self.server_button.setEnabled(True) - if self.mode == ServerStatus.MODE_SHARE: - self.server_button.setText(strings._('gui_share_start_server')) - else: - self.server_button.setText(strings._('gui_receive_start_server')) - self.server_button.setToolTip('') - if self.common.settings.get('shutdown_timeout'): - self.shutdown_timeout_container.show() - elif self.status == self.STATUS_STARTED: - self.server_button.setStyleSheet(self.common.css['server_status_button_started']) - self.server_button.setEnabled(True) - if self.mode == ServerStatus.MODE_SHARE: - self.server_button.setText(strings._('gui_share_stop_server')) - else: - self.server_button.setText(strings._('gui_receive_stop_server')) - if self.common.settings.get('shutdown_timeout'): - self.shutdown_timeout_container.hide() - if self.mode == ServerStatus.MODE_SHARE: - self.server_button.setToolTip(strings._('gui_share_stop_server_shutdown_timeout_tooltip').format(self.timeout)) - else: - self.server_button.setToolTip(strings._('gui_receive_stop_server_shutdown_timeout_tooltip').format(self.timeout)) - - elif self.status == self.STATUS_WORKING: - self.server_button.setStyleSheet(self.common.css['server_status_button_working']) - self.server_button.setEnabled(True) - self.server_button.setText(strings._('gui_please_wait')) - if self.common.settings.get('shutdown_timeout'): - self.shutdown_timeout_container.hide() - else: - self.server_button.setStyleSheet(self.common.css['server_status_button_working']) - self.server_button.setEnabled(False) - self.server_button.setText(strings._('gui_please_wait')) - if self.common.settings.get('shutdown_timeout'): - self.shutdown_timeout_container.hide() - - def server_button_clicked(self): - """ - Toggle starting or stopping the server. - """ - if self.status == self.STATUS_STOPPED: - if self.common.settings.get('shutdown_timeout'): - if self.local_only: - self.timeout = self.shutdown_timeout.dateTime().toPyDateTime() - else: - # Get the timeout chosen, stripped of its seconds. This prevents confusion if the share stops at (say) 37 seconds past the minute chosen - self.timeout = self.shutdown_timeout.dateTime().toPyDateTime().replace(second=0, microsecond=0) - # If the timeout has actually passed already before the user hit Start, refuse to start the server. - if QtCore.QDateTime.currentDateTime().toPyDateTime() > self.timeout: - Alert(self.common, strings._('gui_server_timeout_expired'), QtWidgets.QMessageBox.Warning) - else: - self.start_server() - else: - self.start_server() - elif self.status == self.STATUS_STARTED: - self.stop_server() - elif self.status == self.STATUS_WORKING: - self.cancel_server() - self.button_clicked.emit() - - def start_server(self): - """ - Start the server. - """ - self.status = self.STATUS_WORKING - self.update() - self.server_started.emit() - - def start_server_finished(self): - """ - The server has finished starting. - """ - self.status = self.STATUS_STARTED - self.copy_url() - self.update() - self.server_started_finished.emit() - - def stop_server(self): - """ - Stop the server. - """ - self.status = self.STATUS_WORKING - self.shutdown_timeout_reset() - self.update() - self.server_stopped.emit() - - def cancel_server(self): - """ - Cancel the server. - """ - self.common.log('ServerStatus', 'cancel_server', 'Canceling the server mid-startup') - self.status = self.STATUS_WORKING - self.shutdown_timeout_reset() - self.update() - self.server_canceled.emit() - - def stop_server_finished(self): - """ - The server has finished stopping. - """ - self.status = self.STATUS_STOPPED - self.update() - - def copy_url(self): - """ - Copy the onionshare URL to the clipboard. - """ - clipboard = self.qtapp.clipboard() - clipboard.setText(self.get_url()) - - self.url_copied.emit() - - def copy_hidservauth(self): - """ - Copy the HidServAuth line to the clipboard. - """ - clipboard = self.qtapp.clipboard() - clipboard.setText(self.app.auth_string) - - self.hidservauth_copied.emit() - - def get_url(self): - """ - Returns the OnionShare URL. - """ - if self.common.settings.get('public_mode'): - url = 'http://{0:s}'.format(self.app.onion_host) - else: - url = 'http://{0:s}/{1:s}'.format(self.app.onion_host, self.web.slug) - return url