from typing import TYPE_CHECKING from .. import buttons if TYPE_CHECKING: from trezorlib.debuglink import DebugLink, LayoutContent def enter_word( debug: "DebugLink", word: str, is_slip39: bool = False ) -> "LayoutContent": typed_word = word[:4] for coords in buttons.type_word(typed_word, is_slip39=is_slip39): debug.click(coords) # For BIP39 - double-click on CONFIRM WORD is needed in case the word # is not already typed as a whole if not is_slip39 and typed_word != word: debug.click(buttons.CONFIRM_WORD) return debug.click(buttons.CONFIRM_WORD, wait=True) def confirm_recovery(debug: "DebugLink", legacy_ui: bool = False) -> None: layout = debug.wait_layout() if legacy_ui: layout.text.startswith("Recovery mode") else: assert layout.get_title() == "RECOVERY MODE" debug.click(buttons.OK, wait=True) def select_number_of_words( debug: "DebugLink", num_of_words: int = 20, legacy_ui: bool = False ) -> None: layout = debug.read_layout() # select number of words assert "Select number of words" in layout.get_content() layout = debug.click(buttons.OK, wait=True) if legacy_ui: assert layout.text == "WordSelector" else: # Two title options assert layout.get_title() in ("SEED CHECK", "RECOVERY MODE") # click the number word_option_offset = 6 word_options = (12, 18, 20, 24, 33) index = word_option_offset + word_options.index( num_of_words ) # raises if num of words is invalid coords = buttons.grid34(index % 3, index // 3) layout = debug.click(coords, wait=True) assert "Enter any share" in layout.get_content() def enter_share( debug: "DebugLink", share: str, legacy_ui: bool = False ) -> "LayoutContent": layout = debug.click(buttons.OK, wait=True) if legacy_ui: assert layout.text == "Slip39Keyboard" else: assert layout.text == "< MnemonicKeyboard >" for word in share.split(" "): layout = enter_word(debug, word, is_slip39=True) return layout def enter_shares(debug: "DebugLink", shares: list[str]) -> None: layout = debug.read_layout() expected_text = "Enter any share" remaining = len(shares) for share in shares: assert expected_text in layout.get_content() layout = enter_share(debug, share) remaining -= 1 expected_text = f"{remaining} more share" assert "You have successfully recovered your wallet" in layout.get_content() def finalize(debug: "DebugLink") -> None: layout = debug.click(buttons.OK, wait=True) assert layout.text.startswith("< Homescreen ")