trezor-firmware/core/src/trezor/ui/layouts/tr/progress.py

from typing import TYPE_CHECKING

import trezorui2
from trezor import TR, ui, utils

if TYPE_CHECKING:
    from typing import Any

    from ..common import ProgressLayout


class RustProgress:
    def __init__(
        self,
        layout: Any,
    ):
        self.layout = layout
        self.layout.attach_timer_fn(self.set_timer)
        self.layout.paint()
        ui.refresh()

    def set_timer(self, token: int, deadline: int) -> None:
        raise RuntimeError  # progress layouts should not set timers

    def report(self, value: int, description: str | None = None):
        msg = self.layout.progress_event(value, description or "")
        assert msg is None
        self.layout.paint()
        ui.refresh()


def progress(
    message: str | None = None,
    description: str | None = None,
    indeterminate: bool = False,
) -> ProgressLayout:
    return RustProgress(
        layout=trezorui2.show_progress(
            title=message.upper() if message else "",
            indeterminate=indeterminate,
            description=description or "",
        )
    )


def bitcoin_progress(description: str) -> ProgressLayout:
    return progress("", description)


def coinjoin_progress(message: str) -> ProgressLayout:
    return RustProgress(
        layout=trezorui2.show_progress_coinjoin(title=message, indeterminate=False)
    )


def pin_progress(message: str, description: str) -> ProgressLayout:
    return progress(message, description)


if not utils.BITCOIN_ONLY:

    def monero_keyimage_sync_progress() -> ProgressLayout:
        return progress("", TR.progress__syncing)

    def monero_live_refresh_progress() -> ProgressLayout:
        return progress("", TR.progress__refreshing, indeterminate=True)

    def monero_transaction_progress_inner() -> ProgressLayout:
        return progress("", TR.progress__signing_transaction)