trezor-firmware/src/apps/management/apply_settings.py

from trezor import ui, wire
from trezor.messages import ButtonRequestType, PassphraseSourceType
from trezor.messages.Success import Success
from trezor.ui.text import Text

from apps.common import storage
from apps.common.confirm import require_confirm


async def apply_settings(ctx, msg):
    if (
        msg.homescreen is None
        and msg.label is None
        and msg.use_passphrase is None
        and msg.passphrase_source is None
    ):
        raise wire.ProcessError("No setting provided")

    if msg.homescreen is not None:
        if len(msg.homescreen) > storage.HOMESCREEN_MAXSIZE:
            raise wire.DataError("Homescreen is too complex")
        await require_confirm_change_homescreen(ctx)

    if msg.label is not None:
        await require_confirm_change_label(ctx, msg.label)

    if msg.use_passphrase is not None:
        await require_confirm_change_passphrase(ctx, msg.use_passphrase)

    if msg.passphrase_source is not None:
        await require_confirm_change_passphrase_source(ctx, msg.passphrase_source)

    storage.load_settings(
        label=msg.label,
        use_passphrase=msg.use_passphrase,
        homescreen=msg.homescreen,
        passphrase_source=msg.passphrase_source,
    )

    return Success(message="Settings applied")


async def require_confirm_change_homescreen(ctx):
    text = Text("Change homescreen", ui.ICON_CONFIG)
    text.normal("Do you really want to", "change homescreen?")
    await require_confirm(ctx, text, code=ButtonRequestType.ProtectCall)


async def require_confirm_change_label(ctx, label):
    text = Text("Change label", ui.ICON_CONFIG)
    text.normal("Do you really want to", "change label to")
    text.bold("%s?" % label)
    await require_confirm(ctx, text, code=ButtonRequestType.ProtectCall)


async def require_confirm_change_passphrase(ctx, use):
    text = Text("Enable passphrase" if use else "Disable passphrase", ui.ICON_CONFIG)
    text.normal("Do you really want to")
    text.normal("enable passphrase" if use else "disable passphrase")
    text.normal("encryption?")
    await require_confirm(ctx, text, code=ButtonRequestType.ProtectCall)


async def require_confirm_change_passphrase_source(ctx, source):
    if source == PassphraseSourceType.DEVICE:
        desc = "ON DEVICE"
    elif source == PassphraseSourceType.HOST:
        desc = "ON HOST"
    else:
        desc = "ASK"
    text = Text("Passphrase source", ui.ICON_CONFIG)
    text.normal("Do you really want to", "change the passphrase", "source to")
    text.bold("ALWAYS %s?" % desc)
    await require_confirm(ctx, text, code=ButtonRequestType.ProtectCall)