mirror of
https://github.com/LBRYFoundation/LBRY-Vault.git
synced 2025-08-23 17:47:31 +00:00
11d135b32d
Break out the workflow logic of the install wizard into a base class. This means reimplementing with full support in a new GUI is now easy; you just provide ways to request passwords, show messages etc. The API is fully documented in the base class. There are a couple of minor outstanding issues, including that the old messages shown when recovering a wallet are missing. I will come back to that. Ledger wallet might be broken. Other improvements: The install wizard code is now easy to follow and understand. Hardware wallets can now be restored without any need for their accompanying libraries. Various bits of trustedcoin were broken and have been fixed. Many plugin hooks can be removed. I have only started on this.
221 lines
7.8 KiB
Python
221 lines
7.8 KiB
Python
from functools import partial
|
|
import threading
|
|
|
|
from PyQt4.Qt import QGridLayout, QInputDialog, QPushButton
|
|
from PyQt4.Qt import QVBoxLayout, QLabel, SIGNAL
|
|
from trezor import TrezorPlugin
|
|
from electrum_gui.qt.main_window import StatusBarButton
|
|
from electrum_gui.qt.password_dialog import PasswordDialog
|
|
from electrum_gui.qt.util import *
|
|
|
|
from electrum.i18n import _
|
|
from electrum.plugins import hook
|
|
from electrum.util import PrintError
|
|
|
|
|
|
class QtHandler(PrintError):
|
|
'''An interface between the GUI (here, QT) and the device handling
|
|
logic for handling I/O. This is a generic implementation of the
|
|
Trezor protocol; derived classes can customize it.'''
|
|
|
|
def __init__(self, win, pin_matrix_widget_class, device):
|
|
win.connect(win, SIGNAL('message_done'), self.dialog_stop)
|
|
win.connect(win, SIGNAL('message_dialog'), self.message_dialog)
|
|
win.connect(win, SIGNAL('pin_dialog'), self.pin_dialog)
|
|
win.connect(win, SIGNAL('passphrase_dialog'), self.passphrase_dialog)
|
|
self.win = win
|
|
self.windows = [win]
|
|
self.pin_matrix_widget_class = pin_matrix_widget_class
|
|
self.device = device
|
|
self.done = threading.Event()
|
|
self.dialog = None
|
|
|
|
def stop(self):
|
|
self.win.emit(SIGNAL('message_done'))
|
|
|
|
def show_message(self, msg, cancel_callback=None):
|
|
self.win.emit(SIGNAL('message_dialog'), msg, cancel_callback)
|
|
|
|
def get_pin(self, msg):
|
|
self.done.clear()
|
|
self.win.emit(SIGNAL('pin_dialog'), msg)
|
|
self.done.wait()
|
|
return self.response
|
|
|
|
def get_passphrase(self, msg):
|
|
self.done.clear()
|
|
self.win.emit(SIGNAL('passphrase_dialog'), msg)
|
|
self.done.wait()
|
|
return self.passphrase
|
|
|
|
def pin_dialog(self, msg):
|
|
# Needed e.g. when renaming label and haven't entered PIN
|
|
self.dialog_stop()
|
|
d = WindowModalDialog(self.windows[-1], _("Enter PIN"))
|
|
matrix = self.pin_matrix_widget_class()
|
|
vbox = QVBoxLayout()
|
|
vbox.addWidget(QLabel(msg))
|
|
vbox.addWidget(matrix)
|
|
vbox.addLayout(Buttons(CancelButton(d), OkButton(d)))
|
|
d.setLayout(vbox)
|
|
if not d.exec_():
|
|
self.response = None # FIXME: this is lost?
|
|
self.response = str(matrix.get_value())
|
|
self.done.set()
|
|
|
|
def passphrase_dialog(self, msg):
|
|
self.dialog_stop()
|
|
d = PasswordDialog(self.windows[-1], None, msg,
|
|
PasswordDialog.PW_PASSHPRASE)
|
|
confirmed, p, passphrase = d.run()
|
|
if confirmed:
|
|
passphrase = TrezorPlugin.normalize_passphrase(passphrase)
|
|
self.passphrase = passphrase
|
|
self.done.set()
|
|
|
|
def message_dialog(self, msg, cancel_callback):
|
|
# Called more than once during signing, to confirm output and fee
|
|
self.dialog_stop()
|
|
title = _('Please check your %s device') % self.device
|
|
dialog = self.dialog = WindowModalDialog(self.windows[-1], title)
|
|
l = QLabel(msg)
|
|
vbox = QVBoxLayout(dialog)
|
|
if cancel_callback:
|
|
vbox.addLayout(Buttons(CancelButton(dialog)))
|
|
dialog.connect(dialog, SIGNAL('rejected()'), cancel_callback)
|
|
vbox.addWidget(l)
|
|
dialog.show()
|
|
|
|
def dialog_stop(self):
|
|
if self.dialog:
|
|
self.dialog.hide()
|
|
self.dialog = None
|
|
|
|
def pop_window(self):
|
|
self.windows.pop()
|
|
|
|
def push_window(self, window):
|
|
self.windows.append(window)
|
|
|
|
|
|
class QtPlugin(TrezorPlugin):
|
|
# Derived classes must provide the following class-static variables:
|
|
# icon_file
|
|
# pin_matrix_widget_class
|
|
|
|
def create_handler(self, window):
|
|
return QtHandler(window, self.pin_matrix_widget_class(), self.device)
|
|
|
|
@hook
|
|
def load_wallet(self, wallet, window):
|
|
if type(wallet) != self.wallet_class:
|
|
return
|
|
try:
|
|
client = self.get_client(wallet)
|
|
client.handler = self.create_handler(window)
|
|
client.check_proper_device(wallet)
|
|
self.button = StatusBarButton(QIcon(self.icon_file), self.device,
|
|
partial(self.settings_dialog, window))
|
|
window.statusBar().addPermanentWidget(self.button)
|
|
except Exception as e:
|
|
window.show_error(str(e))
|
|
|
|
def on_create_wallet(self, wallet, wizard):
|
|
client = self.get_client(wallet)
|
|
client.handler = self.create_handler(wizard)
|
|
wallet.create_main_account(None)
|
|
|
|
@staticmethod
|
|
def is_valid_seed(seed):
|
|
return True
|
|
|
|
@hook
|
|
def receive_menu(self, menu, addrs, wallet):
|
|
if type(wallet) != self.wallet_class:
|
|
return
|
|
if (not wallet.is_watching_only() and
|
|
self.atleast_version(1, 3) and len(addrs) == 1):
|
|
menu.addAction(_("Show on %s") % self.device,
|
|
lambda: self.show_address(wallet, addrs[0]))
|
|
|
|
def settings_dialog(self, window):
|
|
|
|
handler = self.get_client(window.wallet).handler
|
|
|
|
def rename():
|
|
title = _("Set Device Label")
|
|
msg = _("Enter new label:")
|
|
response = QInputDialog().getText(dialog, title, msg)
|
|
if not response[1]:
|
|
return
|
|
new_label = str(response[0])
|
|
try:
|
|
client.change_label(new_label)
|
|
finally:
|
|
handler.stop()
|
|
device_label.setText(new_label)
|
|
|
|
def update_pin_info():
|
|
features = client.features
|
|
pin_label.setText(noyes[features.pin_protection])
|
|
pin_button.setText(_("Change") if features.pin_protection
|
|
else _("Set"))
|
|
clear_pin_button.setVisible(features.pin_protection)
|
|
|
|
def set_pin(remove):
|
|
try:
|
|
client.set_pin(remove=remove)
|
|
finally:
|
|
handler.stop()
|
|
update_pin_info()
|
|
|
|
client = self.get_client()
|
|
features = client.features
|
|
noyes = [_("No"), _("Yes")]
|
|
bl_hash = features.bootloader_hash.encode('hex').upper()
|
|
bl_hash = "%s...%s" % (bl_hash[:10], bl_hash[-10:])
|
|
info_tab = QWidget()
|
|
layout = QGridLayout(info_tab)
|
|
device_label = QLabel(features.label)
|
|
rename_button = QPushButton(_("Rename"))
|
|
rename_button.clicked.connect(rename)
|
|
pin_label = QLabel()
|
|
pin_button = QPushButton()
|
|
pin_button.clicked.connect(partial(set_pin, False))
|
|
clear_pin_button = QPushButton(_("Clear"))
|
|
clear_pin_button.clicked.connect(partial(set_pin, True))
|
|
update_pin_info()
|
|
|
|
version = "%d.%d.%d" % (features.major_version,
|
|
features.minor_version,
|
|
features.patch_version)
|
|
rows = [
|
|
(_("Bootloader Hash"), bl_hash),
|
|
(_("Device ID"), features.device_id),
|
|
(_("Device Label"), device_label, rename_button),
|
|
(_("Firmware Version"), version),
|
|
(_("Language"), features.language),
|
|
(_("Has Passphrase"), noyes[features.passphrase_protection]),
|
|
(_("Has PIN"), pin_label, pin_button, clear_pin_button)
|
|
]
|
|
|
|
for row_num, items in enumerate(rows):
|
|
for col_num, item in enumerate(items):
|
|
widget = item if isinstance(item, QWidget) else QLabel(item)
|
|
layout.addWidget(widget, row_num, col_num)
|
|
|
|
dialog = WindowModalDialog(None, _("%s Settings") % self.device)
|
|
vbox = QVBoxLayout()
|
|
tabs = QTabWidget()
|
|
tabs.addTab(info_tab, _("Information"))
|
|
tabs.addTab(QWidget(), _("Advanced"))
|
|
vbox.addWidget(tabs)
|
|
vbox.addStretch(1)
|
|
vbox.addLayout(Buttons(CloseButton(dialog)))
|
|
|
|
dialog.setLayout(vbox)
|
|
handler.push_window(dialog)
|
|
try:
|
|
dialog.exec_()
|
|
finally:
|
|
handler.pop_window()
|