trezor-firmware/core/src/trezor/ui/layouts/tt_v2/fido.py

from typing import TYPE_CHECKING

from trezor.enums import ButtonRequestType

import trezorui2

from ..common import interact
from . import _RustLayout

if TYPE_CHECKING:
    from trezor.loop import AwaitableTask
    from trezor.wire import GenericContext


if __debug__:
    from trezor import io
    from ... import Result

    class _RustFidoLayoutImpl(_RustLayout):
        def create_tasks(self) -> tuple[AwaitableTask, ...]:
            return (
                self.handle_timers(),
                self.handle_input_and_rendering(),
                self.handle_swipe(),
                self.handle_debug_confirm(),
            )

        async def handle_debug_confirm(self) -> None:
            from apps.debug import confirm_signal

            try:
                await confirm_signal()
            except Result as r:
                if r.value is not trezorui2.CONFIRMED:
                    raise
            else:
                return

            for event, x, y in (
                (io.TOUCH_START, 220, 220),
                (io.TOUCH_END, 220, 220),
            ):
                msg = self.layout.touch_event(event, x, y)
                self.layout.paint()
                if msg is not None:
                    raise Result(msg)

    _RustFidoLayout = _RustFidoLayoutImpl

else:
    _RustFidoLayout = _RustLayout


async def confirm_fido(
    ctx: GenericContext | None,
    header: str,
    app_name: str,
    icon_name: str | None,
    accounts: list[str | None],
) -> int:
    """Webauthn confirmation for one or more credentials."""
    confirm = _RustFidoLayout(
        trezorui2.confirm_fido(
            title=header.upper(),
            app_name=app_name,
            icon_name=icon_name,
            accounts=accounts,
        )
    )

    if ctx is None:
        result = await confirm
    else:
        result = await interact(ctx, confirm, "confirm_fido", ButtonRequestType.Other)

    # The Rust side returns either an int or `CANCELLED`. We detect the int situation
    # and assume cancellation otherwise.
    if isinstance(result, int):
        return result

    # Late import won't get executed on the happy path.
    from trezor.wire import ActionCancelled

    raise ActionCancelled


async def confirm_fido_reset() -> bool:
    confirm = _RustLayout(
        trezorui2.confirm_action(
            title="FIDO2 RESET",
            action="erase all credentials?",
            description="Do you really want to",
            reverse=True,
        )
    )
    return (await confirm) is trezorui2.CONFIRMED