from micropython import const

from trezor.crypto import base58

from apps.common import HARDENED
from apps.common.writers import write_uint8

TEZOS_AMOUNT_DIVISIBILITY = const(6)
TEZOS_ED25519_ADDRESS_PREFIX = "tz1"
TEZOS_ORIGINATED_ADDRESS_PREFIX = "KT1"
TEZOS_PUBLICKEY_PREFIX = "edpk"
TEZOS_SIGNATURE_PREFIX = "edsig"
TEZOS_PREFIX_BYTES = {
    # addresses
    "tz1": [6, 161, 159],
    "tz2": [6, 161, 161],
    "tz3": [6, 161, 164],
    "KT1": [2, 90, 121],
    # public keys
    "edpk": [13, 15, 37, 217],
    # signatures
    "edsig": [9, 245, 205, 134, 18],
    # operation hash
    "o": [5, 116],
    # protocol hash
    "P": [2, 170],
}

# MICHELSON instruction bytes
MICHELSON_INSTRUCTION_BYTES = {
    "DROP": [3, 32],  # '0320'
    "NIL": [5, 61],  # '053d'
    "operation": [3, 109],  # '036d'
    "NONE": [5, 62],  # '053e'
    "key_hash": [3, 93],  # '035d'
    "SET_DELEGATE": [3, 78],  # '034e'
    "CONS": [3, 27],  # '031b'
    "IMPLICIT_ACCOUNT": [3, 30],  # '031e'
    "PUSH": [7, 67],  # '0743'
    "mutez": [3, 106],  # '036a'
    "UNIT": [3, 79],  # '034f'
    "TRANSFER_TOKENS": [3, 77],  # '034d'
    "SOME": [3, 70],  # '0346'
    "address": [3, 110],  # '036e'
    "CONTRACT": [5, 85],  # '0555'
    "unit": [3, 108],  # '036c'
    # ASSERT_SOME unfolded as { IF_NONE { { UNIT ; FAILWITH } } {} }
    "ASSERT_SOME": [
        2,
        0,
        0,
        0,
        21,
        7,
        47,
        2,
        0,
        0,
        0,
        9,
        2,
        0,
        0,
        0,
        4,
        3,
        79,
        3,
        39,
        2,
        0,
        0,
        0,
        0,
    ],
}

DO_ENTRYPOINT_TAG = const(2)
MICHELSON_SEQUENCE_TAG = const(2)


def base58_encode_check(payload, prefix=None):
    result = payload
    if prefix is not None:
        result = bytes(TEZOS_PREFIX_BYTES[prefix]) + payload
    return base58.encode_check(result)


def base58_decode_check(enc, prefix=None):
    decoded = base58.decode_check(enc)
    if prefix is not None:
        decoded = decoded[len(TEZOS_PREFIX_BYTES[prefix]) :]
    return decoded


def validate_full_path(path: list) -> bool:
    """
    Validates derivation path to equal 44'/1729'/a',
    where `a` is an account index from 0 to 1 000 000.
    Additional component added to allow ledger migration
    44'/1729'/0'/b' where `b` is an account index from 0 to 1 000 000
    """
    length = len(path)
    if length < 3 or length > 4:
        return False
    if path[0] != 44 | HARDENED:
        return False
    if path[1] != 1729 | HARDENED:
        return False
    if length == 3:
        if path[2] < HARDENED or path[2] > 1000000 | HARDENED:
            return False
    if length == 4:
        if path[2] != 0 | HARDENED:
            return False
        if path[3] < HARDENED or path[3] > 1000000 | HARDENED:
            return False
    return True


def write_bool(w: bytearray, boolean: bool):
    if boolean:
        write_uint8(w, 255)
    else:
        write_uint8(w, 0)