Source code for pons._signer

from abc import ABC, abstractmethod
from functools import cached_property

from eth_account import Account
from eth_account.signers.local import LocalAccount
from eth_account.types import TransactionDictType
from ethereum_rpc import Address




class Signer(ABC):
    """The base class for transaction signers."""

    @property
    @abstractmethod
    def address(self) -> Address:
        """Returns the address corresponding to the signer's private key."""



    @abstractmethod
    def sign_transaction(self, tx_dict: TransactionDictType) -> bytes:
        """
        Signs the given transaction and returns the RLP-packed transaction
        along with the signature.
        """






class AccountSigner(Signer):
    """A signer wrapper for ``LocalAccount`` from ``eth-account`` package."""

    def __init__(self, account: LocalAccount):
        self._account = account



    @staticmethod
    def create() -> "AccountSigner":
        """Creates an account with a random private key."""
        return AccountSigner(Account.create())


    @property
    def account(self) -> LocalAccount:
        """Returns the account object used to create this signer."""
        return self._account

    @property
    def private_key(self) -> bytes:
        """
        Returns the private key corresponding to this signer.
        Handle with care.
        """
        return bytes(self._account._private_key)  # noqa: SLF001

    @cached_property
    def address(self) -> Address:
        return Address.from_hex(self._account.address)



    def sign_transaction(self, tx_dict: TransactionDictType) -> bytes:
        return bytes(self._account.sign_transaction(tx_dict).raw_transaction)