trezor-firmware/core/src/apps/common/storage/slip39.py

from micropython import const

from apps.common.storage import common, slip39_mnemonics

# Namespace:
_NAMESPACE = common._APP_SLIP39

# fmt: off
# Keys:
_SLIP39_IN_PROGRESS        = const(0x00)  # bool
_SLIP39_IDENTIFIER         = const(0x01)  # bytes
_SLIP39_THRESHOLD          = const(0x02)  # int
_SLIP39_REMAINING          = const(0x03)  # int
_SLIP39_WORDS_COUNT        = const(0x04)  # int
_SLIP39_ITERATION_EXPONENT = const(0x05)  # int
# fmt: on


def set_in_progress(val: bool):
    common._set_bool(_NAMESPACE, _SLIP39_IN_PROGRESS, val)


def is_in_progress():
    return common._get_bool(_NAMESPACE, _SLIP39_IN_PROGRESS)


def set_identifier(identifier: int):
    common._set_uint16(_NAMESPACE, _SLIP39_IDENTIFIER, identifier)


def get_identifier() -> int:
    return common._get_uint16(_NAMESPACE, _SLIP39_IDENTIFIER)


def set_threshold(threshold: int):
    common._set_uint8(_NAMESPACE, _SLIP39_THRESHOLD, threshold)


def get_threshold() -> int:
    return common._get_uint8(_NAMESPACE, _SLIP39_THRESHOLD)


def set_remaining(remaining: int):
    common._set_uint8(_NAMESPACE, _SLIP39_REMAINING, remaining)


def get_remaining() -> int:
    return common._get_uint8(_NAMESPACE, _SLIP39_REMAINING)


def set_words_count(count: int):
    common._set_uint8(_NAMESPACE, _SLIP39_WORDS_COUNT, count)


def get_words_count() -> int:
    return common._get_uint8(_NAMESPACE, _SLIP39_WORDS_COUNT)


def set_iteration_exponent(exponent: int):
    common._set_uint8(_NAMESPACE, _SLIP39_ITERATION_EXPONENT, exponent)


def get_iteration_exponent() -> int:
    return common._get_uint8(_NAMESPACE, _SLIP39_ITERATION_EXPONENT)


def delete_progress():
    common._delete(_NAMESPACE, _SLIP39_IN_PROGRESS)
    common._delete(_NAMESPACE, _SLIP39_REMAINING)
    common._delete(_NAMESPACE, _SLIP39_THRESHOLD)
    common._delete(_NAMESPACE, _SLIP39_WORDS_COUNT)
    slip39_mnemonics.delete()