48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
from typing import Union
|
|
|
|
from ethereum_client.utils import (read, read_uint, read_varint,
|
|
write_varint, UINT64_MAX)
|
|
|
|
class Plugin:
|
|
"""Plugin class
|
|
Allows to generate an apdu of the SET_PLUGIN command or PROVIDE_NFT_INFORMATION
|
|
|
|
PROVIDE_NFT_INFORMATION
|
|
----
|
|
do not define a selector
|
|
|
|
"""
|
|
def __init__(self, type: int, version: int, name: str, addr: Union[str, bytes], selector: int = -1, chainID: int = 1, keyID: int = 0, algorithm: int = 1, sign: bytes = b'') -> None:
|
|
self.type: int = type
|
|
self.version: int = version
|
|
self.name: bytes = bytes(name, 'UTF-8')
|
|
self.addr: bytes = bytes.fromhex(addr[2:]) if isinstance(addr, str) else addr
|
|
self.selector: int = selector
|
|
self.chainID: int = chainID
|
|
self.keyID: int = keyID
|
|
self.algorithm: int = algorithm
|
|
self.sign: bytes = sign
|
|
|
|
def serialize(self) -> bytes:
|
|
return b"".join([
|
|
self.type.to_bytes(1, byteorder="big"),
|
|
|
|
self.version.to_bytes(1, byteorder="big"),
|
|
|
|
write_varint(len(self.name)),
|
|
self.name,
|
|
|
|
self.addr,
|
|
|
|
b'' if self.selector == -1 else self.selector.to_bytes(4, byteorder="big"),
|
|
|
|
self.chainID.to_bytes(8, byteorder="big"),
|
|
|
|
self.keyID.to_bytes(1, byteorder="big"),
|
|
|
|
self.algorithm.to_bytes(1, byteorder="big"),
|
|
|
|
write_varint(len(self.sign)),
|
|
self.sign,
|
|
|
|
]) |