Implement message decoding

from poolpay.model.messages.PacketDecodeError import PacketDecodeError

from poolpay.model.messages.UpdateBalanceMessage import UpdateBalanceMessage

from poolpay.model.messages.Message import Message

from poolpay.wire.Packet import Packet

MESSAGE_TYPES = (

    UpdateBalanceMessage,

)

def decode(packet: Packet) -> Message | None:

    """

    Try to decode the given packet into a semantic message. Return None if nothing could be decoded.

    """

    for ConcreteMessage in MESSAGE_TYPES:

        try:

            return ConcreteMessage.from_packet(packet)

        except PacketDecodeError:

            pass

    return None