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.
125 lines
4.2 KiB
125 lines
4.2 KiB
from trezor import wire
|
|
from trezor.messages import ButtonRequestType
|
|
from trezor.messages.ButtonAck import ButtonAck
|
|
from trezor.messages.ButtonRequest import ButtonRequest
|
|
from trezor.ui.confirm import CONFIRMED, INFO, Confirm, HoldToConfirm, InfoConfirm
|
|
|
|
if __debug__:
|
|
from apps.debug import confirm_signal
|
|
from trezor.ui.scroll import Paginated
|
|
|
|
if False:
|
|
from typing import Any, Callable, Optional
|
|
from trezor import ui
|
|
from trezor.ui.confirm import ButtonContent, ButtonStyleType
|
|
from trezor.ui.loader import LoaderStyleType
|
|
from trezor.messages.ButtonRequest import EnumTypeButtonRequestType
|
|
|
|
|
|
async def confirm(
|
|
ctx: wire.Context,
|
|
content: ui.Component,
|
|
code: EnumTypeButtonRequestType = ButtonRequestType.Other,
|
|
confirm: Optional[ButtonContent] = Confirm.DEFAULT_CONFIRM,
|
|
confirm_style: ButtonStyleType = Confirm.DEFAULT_CONFIRM_STYLE,
|
|
cancel: Optional[ButtonContent] = Confirm.DEFAULT_CANCEL,
|
|
cancel_style: ButtonStyleType = Confirm.DEFAULT_CANCEL_STYLE,
|
|
major_confirm: bool = False,
|
|
) -> bool:
|
|
await ctx.call(ButtonRequest(code=code), ButtonAck)
|
|
|
|
if content.__class__.__name__ == "Paginated":
|
|
# The following works because asserts are omitted in non-debug builds.
|
|
# IOW if the assert runs, that means __debug__ is True and Paginated is imported
|
|
assert isinstance(content, Paginated)
|
|
|
|
content.pages[-1] = Confirm(
|
|
content.pages[-1],
|
|
confirm,
|
|
confirm_style,
|
|
cancel,
|
|
cancel_style,
|
|
major_confirm,
|
|
)
|
|
dialog = content # type: ui.Layout
|
|
else:
|
|
dialog = Confirm(
|
|
content, confirm, confirm_style, cancel, cancel_style, major_confirm
|
|
)
|
|
|
|
if __debug__:
|
|
return await ctx.wait(dialog, confirm_signal()) is CONFIRMED
|
|
else:
|
|
return await ctx.wait(dialog) is CONFIRMED
|
|
|
|
|
|
async def info_confirm(
|
|
ctx: wire.Context,
|
|
content: ui.Component,
|
|
info_func: Callable,
|
|
code: EnumTypeButtonRequestType = ButtonRequestType.Other,
|
|
confirm: ButtonContent = InfoConfirm.DEFAULT_CONFIRM,
|
|
confirm_style: ButtonStyleType = InfoConfirm.DEFAULT_CONFIRM_STYLE,
|
|
cancel: ButtonContent = InfoConfirm.DEFAULT_CANCEL,
|
|
cancel_style: ButtonStyleType = InfoConfirm.DEFAULT_CANCEL_STYLE,
|
|
info: ButtonContent = InfoConfirm.DEFAULT_INFO,
|
|
info_style: ButtonStyleType = InfoConfirm.DEFAULT_INFO_STYLE,
|
|
) -> bool:
|
|
await ctx.call(ButtonRequest(code=code), ButtonAck)
|
|
|
|
dialog = InfoConfirm(
|
|
content, confirm, confirm_style, cancel, cancel_style, info, info_style
|
|
)
|
|
|
|
while True:
|
|
if __debug__:
|
|
result = await ctx.wait(dialog, confirm_signal())
|
|
else:
|
|
result = await ctx.wait(dialog)
|
|
|
|
if result is INFO:
|
|
await info_func(ctx)
|
|
|
|
else:
|
|
return result is CONFIRMED
|
|
|
|
|
|
async def hold_to_confirm(
|
|
ctx: wire.Context,
|
|
content: ui.Layout,
|
|
code: EnumTypeButtonRequestType = ButtonRequestType.Other,
|
|
confirm: str = HoldToConfirm.DEFAULT_CONFIRM,
|
|
confirm_style: ButtonStyleType = HoldToConfirm.DEFAULT_CONFIRM_STYLE,
|
|
loader_style: LoaderStyleType = HoldToConfirm.DEFAULT_LOADER_STYLE,
|
|
) -> bool:
|
|
await ctx.call(ButtonRequest(code=code), ButtonAck)
|
|
|
|
if content.__class__.__name__ == "Paginated":
|
|
# The following works because asserts are omitted in non-debug builds.
|
|
# IOW if the assert runs, that means __debug__ is True and Paginated is imported
|
|
assert isinstance(content, Paginated)
|
|
|
|
content.pages[-1] = HoldToConfirm(
|
|
content.pages[-1], confirm, confirm_style, loader_style
|
|
)
|
|
dialog = content # type: ui.Layout
|
|
else:
|
|
dialog = HoldToConfirm(content, confirm, confirm_style, loader_style)
|
|
|
|
if __debug__:
|
|
return await ctx.wait(dialog, confirm_signal()) is CONFIRMED
|
|
else:
|
|
return await ctx.wait(dialog) is CONFIRMED
|
|
|
|
|
|
async def require_confirm(*args: Any, **kwargs: Any) -> None:
|
|
confirmed = await confirm(*args, **kwargs)
|
|
if not confirmed:
|
|
raise wire.ActionCancelled("Cancelled")
|
|
|
|
|
|
async def require_hold_to_confirm(*args: Any, **kwargs: Any) -> None:
|
|
confirmed = await hold_to_confirm(*args, **kwargs)
|
|
if not confirmed:
|
|
raise wire.ActionCancelled("Cancelled")
|