echopay_template.py

#!/usr/bin/python3.9
import asyncio
from decimal import Decimal
from typing import Union
import mc_util
from forest.core import Message, PayBot, Response, requires_admin, run_bot

FEE_PMOB = int(1e12 * 0.0004)  # Mobilecoin transaction fee in Picomob.


class Echopay(PayBot):
    """A simple Payments Enabled Bot"""

    def __init__(self) -> None:
        """Creates AND STARTS a bot that routes commands to do_x handlers"""
        super().__init__()
        asyncio.create_task(self.set_payment_address())

    @staticmethod
    def to_mob(amount_picomob: int) -> Decimal:
        """converts amount from pmob to mob"""
        return mc_util.pmob2mob(amount_picomob).quantize(Decimal("1.0000"))

    @staticmethod
    def to_picomob(amount_mob: Union[int, float, Decimal]) -> int:
        """converts amount from mob to picomob"""
        return mc_util.mob2pmob(amount_mob)

    async def set_payment_address(self) -> None:
        """Updates the Bot Signal Profile to have the correct payments address
        as specified by FS_ACCOUNT_NAME"""
        fs_address = await self.mobster.get_my_address()
        # Singal addresses require Base64 encoding, but full service uses Base58.
        # This method handles the conversion
        signal_address = mc_util.b58_wrapper_to_b64_public_address(fs_address)
        # This will set the bot's Signal profile, replace avatar.png to give your bot a custom avatar
        await self.set_profile_auxin(
            given_name="PaymeBot",
            family_name="",
            payment_address=signal_address,
            # profile_path="avatar.png",
        )

    async def do_payme(self, message: Message) -> Response:
        """Sends payment to requestee for a certain amount"""
        amount_mob = 0.001  ##payment amount in MOB
        amount_picomob = self.to_picomob(amount_mob)

        payment_status = await self.send_payment(
            message.source, amount_picomob, confirm_tx_timeout=10, receipt_message=""
        )

        # This check verifies that the payment succeeded.
        if getattr(payment_status, "status", "") == "tx_status_succeeded":
            return f"Sent you a payment for {str(amount_mob)} MOB"

        return "There was a problem processing your payment, please ask an admininstrator for help"

    @requires_admin
    async def do_pay_user(self, message: Message) -> Response:
        """Send payment to user by phone number: `pay_user +15554135555`"""
        amount_mob = 0.001
        amount_picomob = self.to_picomob(amount_mob)

        # for convenience, message.arg0 is the first word of the message in this case "pay_user"
        # and msg.arg1 is the next word after that. message.arg1 should be a phone number
        # send_payment takes care of validating that it is a proper recipient,
        # but for type safety we first check that it exists and is a string
        if not isinstance(message.arg1, str):
            response = (
                "Please specify the User to be paid as a phone number"
                " with country code example: pay_user +15554135555"
            )
            return response

        recipient = message.arg1
        payment_status = await self.send_payment(
            recipient,
            amount_picomob,
            confirm_tx_timeout=10,
            receipt_message="Here's some money from your friendly Paymebot",
        )

        # This check verifies that the payment succeeded.
        if getattr(payment_status, "status", "") == "tx_status_succeeded":
            return f"Sent Payment to {recipient} for {amount_mob} MOB"

        return "There was a problem processing your payment, please ask an admininstrator for help"

    async def payment_response(self, msg: Message, amount_pmob: int) -> Response:
        """Triggers on Succesful payment, overriden from forest.core"""
        # amounts are received in picoMob, convert to Mob for readability
        amount_mob = self.to_mob(amount_pmob)
        return f"Thank you for your payment of {str(amount_mob)} MOB"


if __name__ == "__main__":
    run_bot(Echopay)