src/apps/common: refactor address_n_to_str into apps.common.layout

pull/25/head
Pavol Rusnak 6 years ago
parent cc25069064
commit addbdd8937
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D

@ -1,5 +1,3 @@
from micropython import const
from trezor import wire
from trezor.crypto import base58, crc, hashlib
@ -56,23 +54,3 @@ def derive_address_and_node(root_node, path: list):
)
)
return (address, derived_node)
def _break_address_n_to_lines(address_n: list) -> list:
def path_item(i: int):
if i & HARDENED:
return str(i ^ HARDENED) + "'"
else:
return str(i)
lines = []
path_str = "m/" + "/".join([path_item(i) for i in address_n])
per_line = const(17)
while len(path_str) > per_line:
i = path_str[:per_line].rfind("/")
lines.append(path_str[:i])
path_str = path_str[i:]
lines.append(path_str)
return lines

@ -6,11 +6,12 @@ from trezor.messages.CardanoTxRequest import CardanoTxRequest
from trezor.messages.MessageType import CardanoTxAck
from trezor.ui.text import BR
from .address import _break_address_n_to_lines, derive_address_and_node
from .address import derive_address_and_node
from .layout import confirm_with_pagination, progress
from apps.cardano import cbor
from apps.common import seed, storage
from apps.common.layout import address_n_to_str, split_address
from apps.homescreen.homescreen import display_homescreen
@ -53,7 +54,7 @@ async def show_tx(
for index, change in enumerate(change_derivation_paths):
if not await confirm_with_pagination(
ctx,
_break_address_n_to_lines(change),
split_address(address_n_to_str(change)),
"Confirm change",
ui.ICON_SEND,
ui.GREEN,

@ -8,11 +8,13 @@ from trezor.ui.qr import Qr
from trezor.ui.text import Text
from trezor.utils import chunks
from apps.common import HARDENED
from apps.common.confirm import confirm, require_confirm
async def show_address(ctx, address: str, network: str = None):
async def show_address(ctx, address: str, address_n: list, network: str = None):
text = Text("Confirm address", ui.ICON_RECEIVE, icon_color=ui.GREEN)
# TODO: print address_n via address_n_to_str(address_n)
if network:
text.normal("%s network" % network)
text.mono(*split_address(address))
@ -38,12 +40,22 @@ async def show_qr(ctx, address: str):
)
def split_address(address: str):
return chunks(address, 17)
async def show_pubkey(ctx, pubkey: bytes):
lines = chunks(hexlify(pubkey).decode(), 18)
text = Text("Confirm public key", ui.ICON_RECEIVE, icon_color=ui.GREEN)
text.mono(*lines)
return await require_confirm(ctx, text, code=ButtonRequestType.PublicKey)
def split_address(address: str):
return chunks(address, 17)
def address_n_to_str(address_n: list) -> str:
def path_item(i: int):
if i & HARDENED:
return str(i ^ HARDENED) + "'"
else:
return str(i)
return "m/" + "/".join([path_item(i) for i in address_n])

@ -24,7 +24,7 @@ async def get_address(ctx, msg):
hex_addr = _ethereum_address_hex(address, network)
while True:
if await show_address(ctx, hex_addr):
if await show_address(ctx, hex_addr, address_n):
break
if await show_qr(ctx, hex_addr):
break

@ -16,7 +16,7 @@ async def get_address(ctx, msg):
if msg.show_display:
while True:
if await show_address(ctx, address):
if await show_address(ctx, address, address_n):
break
if await show_qr(ctx, address):
break

@ -14,7 +14,9 @@ async def get_address(ctx, msg):
if msg.show_display:
while True:
if await show_address(ctx, address, get_network_str(network)):
if await show_address(
ctx, address, msg.address_n, network=get_network_str(network)
):
break
if await show_qr(ctx, address.upper()):
break

@ -14,7 +14,7 @@ async def get_address(ctx, msg: RippleGetAddress):
if msg.show_display:
while True:
if await show_address(ctx, address):
if await show_address(ctx, address, msg.address_n):
break
if await show_qr(ctx, address.upper()):
break

@ -13,7 +13,7 @@ async def get_address(ctx, msg: StellarGetAddress):
if msg.show_display:
while True:
if await show_address(ctx, address):
if await show_address(ctx, address, msg.address_n):
break
if await show_qr(ctx, address.upper()):
break

@ -20,7 +20,7 @@ async def get_address(ctx, msg):
if msg.show_display:
while True:
if await show_address(ctx, address):
if await show_address(ctx, address, address_n):
break
if await show_qr(ctx, address):
break

@ -16,7 +16,7 @@ async def get_address(ctx, msg):
if msg.show_display:
while True:
if await show_address(ctx, address_short):
if await show_address(ctx, address_short, msg.address_n):
break
if await show_qr(
ctx,

Loading…
Cancel
Save