trezor-firmware/core/src/apps/misc/get_firmware_hash.py

from typing import TYPE_CHECKING

from trezor import ui, wire, workflow
from trezor.messages import FirmwareHash, GetFirmwareHash
from trezor.ui.layouts import draw_simple_text
from trezor.utils import DISABLE_ANIMATION, firmware_hash

if TYPE_CHECKING:
    from trezor.wire import Context


async def get_firmware_hash(ctx: Context, msg: GetFirmwareHash) -> FirmwareHash:
    render_func = None
    if not DISABLE_ANIMATION:
        workflow.close_others()
        draw_simple_text("Please wait")
        render_func = _render_progress

    try:
        hash = firmware_hash(msg.challenge, render_func)
    except ValueError as e:
        raise wire.DataError(str(e))

    return FirmwareHash(hash=hash)


def _render_progress(progress: int, total: int) -> None:
    p = 1000 * progress // total
    ui.display.loader(p, False, 18, ui.WHITE, ui.BG)
    ui.refresh()