map_entry.py

from enum import IntEnum
from typing import Optional
from .tlv import TlvSerializable
from .signing_partners import CALLDATA_PARTNER


class Tag(IntEnum):
    VERSION = 0x00
    CHAIN_ID = 0x01
    CONTRACT_ADDR = 0x02
    SELECTOR = 0x03
    ID = 0x04
    KEY = 0x05
    VALUE = 0x06
    SIGNATURE = 0xff


class MapEntry(TlvSerializable):
    version: int
    chain_id: int
    contract_addr: bytes
    selector: bytes
    id: int
    key: bytes
    value: bytes
    signature: Optional[bytes] = None

    def __init__(self,
                 version: int,
                 chain_id: int,
                 contract_addr: bytes,
                 selector: bytes,
                 id: int,
                 key: bytes,
                 value: bytes,
                 signature: Optional[bytes] = None):
        self.version = version
        self.chain_id = chain_id
        self.contract_addr = contract_addr
        self.selector = selector
        self.id = id
        self.key = key
        self.value = value
        self.signature = signature

    def serialize(self) -> bytes:
        payload = bytearray()
        payload += self.serialize_field(Tag.VERSION, self.version)
        payload += self.serialize_field(Tag.CHAIN_ID, self.chain_id)
        payload += self.serialize_field(Tag.CONTRACT_ADDR, self.contract_addr)
        payload += self.serialize_field(Tag.SELECTOR, self.selector)
        payload += self.serialize_field(Tag.ID, self.id)
        payload += self.serialize_field(Tag.KEY, self.key)
        payload += self.serialize_field(Tag.VALUE, self.value)
        sig = self.signature
        if sig is None:
            sig = CALLDATA_PARTNER.sign(bytes(payload))
        payload += self.serialize_field(Tag.SIGNATURE, sig)
        return bytes(payload)