You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
170 lines
4.9 KiB
170 lines
4.9 KiB
from typing import Callable, Iterable
|
|
|
|
import trezorui2
|
|
from trezor import TR
|
|
from trezor.enums import ButtonRequestType
|
|
from trezor.wire.context import wait as ctx_wait
|
|
|
|
from ..common import interact
|
|
from . import RustLayout, raise_if_not_confirmed
|
|
|
|
CONFIRMED = trezorui2.CONFIRMED # global_import_cache
|
|
INFO = trezorui2.INFO # global_import_cache
|
|
|
|
|
|
async def _is_confirmed_info(
|
|
dialog: RustLayout,
|
|
info_func: Callable,
|
|
) -> bool:
|
|
while True:
|
|
result = await ctx_wait(dialog)
|
|
|
|
if result is trezorui2.INFO:
|
|
await info_func()
|
|
dialog.request_complete_repaint()
|
|
else:
|
|
return result is CONFIRMED
|
|
|
|
|
|
async def request_word_count(dry_run: bool) -> int:
|
|
selector = RustLayout(trezorui2.select_word_count(dry_run=dry_run))
|
|
count = await interact(selector, "word_count", ButtonRequestType.MnemonicWordCount)
|
|
return int(count)
|
|
|
|
|
|
async def request_word(
|
|
word_index: int, word_count: int, is_slip39: bool, prefill_word: str = ""
|
|
) -> str:
|
|
prompt = TR.recovery__type_word_x_of_y_template.format(word_index + 1, word_count)
|
|
can_go_back = word_index > 0
|
|
if is_slip39:
|
|
keyboard = RustLayout(
|
|
trezorui2.request_slip39(
|
|
prompt=prompt, prefill_word=prefill_word, can_go_back=can_go_back
|
|
)
|
|
)
|
|
else:
|
|
keyboard = RustLayout(
|
|
trezorui2.request_bip39(
|
|
prompt=prompt, prefill_word=prefill_word, can_go_back=can_go_back
|
|
)
|
|
)
|
|
|
|
word: str = await ctx_wait(keyboard)
|
|
return word
|
|
|
|
|
|
async def show_remaining_shares(
|
|
groups: Iterable[tuple[int, tuple[str, ...]]], # remaining + list 3 words
|
|
shares_remaining: list[int],
|
|
group_threshold: int,
|
|
) -> None:
|
|
from trezor import strings
|
|
from trezor.crypto.slip39 import MAX_SHARE_COUNT
|
|
|
|
pages: list[tuple[str, str]] = []
|
|
for remaining, group in groups:
|
|
if 0 < remaining < MAX_SHARE_COUNT:
|
|
title = strings.format_plural(
|
|
TR.recovery__x_more_items_starting_template_plural,
|
|
remaining,
|
|
TR.plurals__x_shares_needed,
|
|
)
|
|
words = "\n".join(group)
|
|
pages.append((title, words))
|
|
elif (
|
|
remaining == MAX_SHARE_COUNT and shares_remaining.count(0) < group_threshold
|
|
):
|
|
groups_remaining = group_threshold - shares_remaining.count(0)
|
|
title = strings.format_plural(
|
|
TR.recovery__x_more_items_starting_template_plural,
|
|
groups_remaining,
|
|
TR.plurals__x_groups_needed,
|
|
)
|
|
words = "\n".join(group)
|
|
pages.append((title, words))
|
|
|
|
await raise_if_not_confirmed(
|
|
interact(
|
|
RustLayout(trezorui2.show_remaining_shares(pages=pages)),
|
|
"show_shares",
|
|
ButtonRequestType.Other,
|
|
)
|
|
)
|
|
|
|
|
|
async def show_group_share_success(share_index: int, group_index: int) -> None:
|
|
await raise_if_not_confirmed(
|
|
interact(
|
|
RustLayout(
|
|
trezorui2.show_group_share_success(
|
|
lines=[
|
|
TR.recovery__you_have_entered,
|
|
TR.recovery__share_num_template.format(share_index + 1),
|
|
TR.words__from,
|
|
TR.recovery__group_num_template.format(group_index + 1),
|
|
],
|
|
)
|
|
),
|
|
"share_success",
|
|
ButtonRequestType.Other,
|
|
)
|
|
)
|
|
|
|
|
|
async def continue_recovery(
|
|
button_label: str,
|
|
text: str,
|
|
subtext: str | None,
|
|
info_func: Callable | None,
|
|
dry_run: bool,
|
|
show_info: bool = False, # unused on TT
|
|
) -> bool:
|
|
from ..common import button_request
|
|
|
|
if show_info:
|
|
# Show this just one-time
|
|
description = TR.recovery__only_first_n_letters
|
|
else:
|
|
description = subtext or ""
|
|
|
|
homepage = RustLayout(
|
|
trezorui2.confirm_recovery(
|
|
title=text,
|
|
description=description,
|
|
button=button_label.upper(),
|
|
info_button=info_func is not None,
|
|
dry_run=dry_run,
|
|
)
|
|
)
|
|
|
|
await button_request("recovery", ButtonRequestType.RecoveryHomepage)
|
|
|
|
if info_func is not None:
|
|
return await _is_confirmed_info(homepage, info_func)
|
|
else:
|
|
result = await ctx_wait(homepage)
|
|
return result is CONFIRMED
|
|
|
|
|
|
async def show_recovery_warning(
|
|
br_type: str,
|
|
content: str,
|
|
subheader: str | None = None,
|
|
button: str | None = None,
|
|
br_code: ButtonRequestType = ButtonRequestType.Warning,
|
|
) -> None:
|
|
button = button or TR.buttons__try_again # def_arg
|
|
await raise_if_not_confirmed(
|
|
interact(
|
|
RustLayout(
|
|
trezorui2.show_warning(
|
|
title=content,
|
|
value="Try again", # TODO: use TR
|
|
)
|
|
),
|
|
br_type,
|
|
br_code,
|
|
)
|
|
)
|