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.
trezor-firmware/core/src/apps/nem/transfer/layout.py

139 lines
4.3 KiB

from typing import TYPE_CHECKING
from trezor.enums import ButtonRequestType
from trezor.strings import format_amount
from ..helpers import NEM_MOSAIC_AMOUNT_DIVISOR
from ..layout import require_confirm_final
from ..mosaic.helpers import is_nem_xem_mosaic
if TYPE_CHECKING:
from trezor.messages import (
NEMImportanceTransfer,
NEMMosaic,
NEMTransactionCommon,
NEMTransfer,
)
async def ask_transfer(
common: NEMTransactionCommon,
transfer: NEMTransfer,
encrypted: bool,
) -> None:
from trezor.ui.layouts import confirm_output, confirm_text
from ..helpers import NEM_MAX_DIVISIBILITY
if transfer.payload:
# require_confirm_payload
await confirm_text(
"confirm_payload",
"Confirm payload",
bytes(transfer.payload).decode(),
"Encrypted:" if encrypted else "Unencrypted:",
ButtonRequestType.ConfirmOutput,
)
for mosaic in transfer.mosaics:
await _ask_transfer_mosaic(common, transfer, mosaic)
# require_confirm_transfer
await confirm_output(
transfer.recipient,
f"{format_amount(_get_xem_amount(transfer), NEM_MAX_DIVISIBILITY)} XEM",
)
await require_confirm_final(common.fee)
async def _ask_transfer_mosaic(
common: NEMTransactionCommon, transfer: NEMTransfer, mosaic: NEMMosaic
) -> None:
from trezor.enums import NEMMosaicLevy
from trezor.ui.layouts import confirm_action, confirm_properties
from ..mosaic.helpers import get_mosaic_definition
from ..helpers import NEM_LEVY_PERCENTILE_DIVISOR_ABSOLUTE
if is_nem_xem_mosaic(mosaic):
return
definition = get_mosaic_definition(mosaic.namespace, mosaic.mosaic, common.network)
mosaic_quantity = mosaic.quantity * transfer.amount // NEM_MOSAIC_AMOUNT_DIVISOR
if definition:
await confirm_properties(
"confirm_mosaic",
"Confirm mosaic",
(
(
"Confirm transfer of",
format_amount(mosaic_quantity, definition.divisibility)
+ definition.ticker,
),
("of", definition.name),
),
)
levy = definition.levy # local_cache_attribute
if levy is not None:
if levy == NEMMosaicLevy.MosaicLevy_Absolute:
levy_fee = levy.fee
else:
levy_fee = (
mosaic_quantity * levy.fee // NEM_LEVY_PERCENTILE_DIVISOR_ABSOLUTE
)
levy_msg = (
format_amount(levy_fee, definition.divisibility) + definition.ticker
)
await confirm_properties(
"confirm_mosaic_levy",
"Confirm mosaic",
(("Confirm mosaic\nlevy fee of", levy_msg),),
)
else:
await confirm_action(
"confirm_mosaic_unknown",
"Confirm mosaic",
"Unknown mosaic!",
"Divisibility and levy cannot be shown for unknown mosaics",
br_code=ButtonRequestType.ConfirmOutput,
)
await confirm_properties(
"confirm_mosaic_transfer",
"Confirm mosaic",
(
("Confirm transfer of", f"{mosaic_quantity} raw units"),
("of", f"{mosaic.namespace}.{mosaic.mosaic}"),
),
)
def _get_xem_amount(transfer: NEMTransfer) -> int:
# if mosaics are empty the transfer.amount denotes the xem amount
if not transfer.mosaics:
return transfer.amount
# otherwise xem amount is taken from the nem xem mosaic if present
for mosaic in transfer.mosaics:
if is_nem_xem_mosaic(mosaic):
return mosaic.quantity * transfer.amount // NEM_MOSAIC_AMOUNT_DIVISOR
# if there are mosaics but do not include xem, 0 xem is sent
return 0
async def ask_importance_transfer(
common: NEMTransactionCommon, imp: NEMImportanceTransfer
) -> None:
from trezor.enums import NEMImportanceTransferMode
from ..layout import require_confirm_text
if imp.mode == NEMImportanceTransferMode.ImportanceTransfer_Activate:
m = "Activate"
else:
m = "Deactivate"
await require_confirm_text(m + " remote harvesting?")
await require_confirm_final(common.fee)