from trezor.wire import register, protobuf_workflow
from trezor.utils import unimport
from trezor.messages.wire_types import EthereumGetAddress, EthereumSignTx
# from trezor.messages.wire_types import EthereumSignMessage, EthereumVerifyMessage


@unimport
def dispatch_EthereumGetAddress(*args, **kwargs):
    from .get_address import layout_ethereum_get_address
    return layout_ethereum_get_address(*args, **kwargs)


@unimport
def dispatch_EthereumSignTx(*args, **kwargs):
    from .sign_tx import ethereum_sign_tx
    return ethereum_sign_tx(*args, **kwargs)


@unimport
def dispatch_EthereumSignMessage(*args, **kwargs):
    from .sign_message import ethereum_sign_message
    return ethereum_sign_message(*args, **kwargs)


@unimport
def dispatch_EthereumVerifyMessage(*args, **kwargs):
    from .verify_message import ethereum_verify_message
    return ethereum_verify_message(*args, **kwargs)


def boot():
    register(EthereumGetAddress, protobuf_workflow, dispatch_EthereumGetAddress)
    register(EthereumSignTx, protobuf_workflow, dispatch_EthereumSignTx)
    # TODO: re-enable once https://github.com/ethereum/EIPs/pull/712 is accepted/implemented
    # register(EthereumSignMessage, protobuf_workflow, dispatch_EthereumSignMessage)
    # register(EthereumVerifyMessage, protobuf_workflow, dispatch_EthereumVerifyMessage)