sign_verify.py

from typing import Dict

from multiversx_sdk import Address, Message, MessageComputer, UserVerifier

from multiversx_sdk_cli.accounts import Account


class SignedMessage:
    def __init__(self, address: str, message: str, signature: str) -> None:
        self.address = address
        self.message = message

        hex_prefixes = ["0x", "0X"]
        signature_start_sequence = signature[0:2]

        if signature_start_sequence in hex_prefixes:
            signature = signature[2:]

        self.signature = signature

    def verify_signature(self) -> bool:
        verifiable_message = Message(self.message.encode())
        verifiable_message.signature = bytes.fromhex(self.signature)
        message_computer = MessageComputer()

        verifier = UserVerifier.from_address(Address.new_from_bech32(self.address))
        is_signed = verifier.verify(message_computer.compute_bytes_for_signing(verifiable_message), verifiable_message.signature)
        return is_signed

    def to_dictionary(self) -> Dict[str, str]:
        return {
            "address": self.address,
            "message": self.message,
            "signature": "0x" + self.signature
        }


def sign_message(message: str, account: Account) -> SignedMessage:
    signature = account.sign_message(message.encode())
    return SignedMessage(account.address.to_bech32(), message, signature)