royalnet/royalpack/commands/fiorygi.py

import royalnet.engineer as engi
import royalpack.database as db
import royalpack.bolts as rb
import sqlalchemy.sql as ss
import functools


@engi.use_database(db.lazy_session_class)
@rb.use_ryglogin(allow_anonymous=False)
@engi.TeleportingConversation
async def fiorygi_balance_self(*, _user: db.User, _msg: engi.Message, **__):
    """
    Visualizza il tuo portafoglio attuale di fiorygi.
    """

    await _msg.reply(text=f"💰 Attualmente, possiedi \uE01Bƒ {_user.fiorygi}\uE00B.")


@engi.use_database(db.lazy_session_class)
@rb.with_target()
@engi.TeleportingConversation
async def fiorygi_balance_other(*, _target: db.User, _session: db.SessionType, _msg: engi.Message, **__):
    """
    Visualizza il portafoglio di fiorygi di un altro membro.
    """

    await _msg.reply(text=f"💰 {_target} possiede \uE01Bƒ {_target.fiorygi}\uE00B.")


def render(transaction: db.Transaction, user: db.User):
    row = []

    if transaction.plus == user:
        is_plus = True
        other = transaction.minus
    else:
        is_plus = False
        other = transaction.plus

    if transaction.amount != 0:
        row.append(f"{'➕' if is_plus else '➖'} \uE01Bƒ {transaction.amount}\uE00B")

    if other is not None:
        row.append(f"\uE011{'da' if is_plus else 'a'} {other}\uE001")

    if transaction.reason:
        row.append(f"{transaction.reason}")

    return " - ".join(row)


@engi.use_database(db.lazy_session_class)
@rb.use_ryglogin(allow_anonymous=False)
@engi.TeleportingConversation
async def fiorygi_transactions_self(*, _session: db.SessionType, _user: db.User, _msg: engi.Message, **__):
    """
    Visualizza le ultime 10 transazioni del tuo portafoglio.
    """

    transactions = _session.execute(
        ss.select(
            db.Transaction
        ).where(
            ss.or_(
                db.Transaction.minus == _user,
                db.Transaction.plus == _user,
            )
        ).order_by(
            db.Transaction.timestamp.desc()
        ).limit(
            10
        )
    ).scalars()

    msg = map(functools.partial(render, user=_user), transactions)

    await _msg.reply(text="\n\n".join(msg))


@engi.use_database(db.lazy_session_class)
@rb.with_target()
@engi.TeleportingConversation
async def fiorygi_transactions_other(*, _session: db.SessionType, _target: db.User, _msg: engi.Message, **__):
    """
    Visualizza le ultime 10 transazioni del portafoglio di un membro.
    """

    transactions = _session.execute(
        ss.select(
            db.Transaction
        ).where(
            ss.or_(
                db.Transaction.minus == _target,
                db.Transaction.plus == _target,
            )
        ).order_by(
            db.Transaction.timestamp.desc()
        ).limit(
            10
        )
    ).scalars()

    msg = map(functools.partial(render, user=_target), transactions)

    await _msg.reply(text="\n\n".join(msg))


@engi.use_database(db.lazy_session_class)
@rb.use_ryglogin(allow_anonymous=False)
@rb.with_target()
@engi.TeleportingConversation
async def fiorygi_give(
        *,
        _user: db.User,
        _target: db.User,
        _msg: engi.Message,
        _session: db.SessionType,
        amount: int,
        reason: str,
        **__
):
    """
    Dai dei fiorygi a un altro membro.
    """

    if amount <= 0:
        await _msg.reply(text=f"⚠️ Puoi trasferire solo numeri interi positivi di fiorygi.")
        return

    if _user.fiorygi < amount:
        await _msg.reply(text=f"⚠️ Non hai sufficienti fiorygi per effettuare il trasferimento.")
        return

    if _user == _target:
        await _msg.reply(text=f"⚠️ Non puoi dare fiorygi a te stesso!")
        return

    trans = db.Transaction(
        minus=_user,
        plus=_target,
        amount=amount,
        reason=reason,
    )
    _session.add(trans)

    _user.fiorygi -= amount
    _target.fiorygi += amount

    _session.commit()

    await _msg.reply(text=f"💸 Hai trasferito \uE01Bƒ {amount}\uE00B a {_target}.")


@engi.use_database(db.lazy_session_class)
@rb.use_ryglogin(allow_anonymous=False)
@rb.with_target()
@engi.TeleportingConversation
async def fiorygi_magick(
        *,
        _user: db.User,
        _target: db.User,
        _msg: engi.Message,
        _session: db.SessionType,
        amount: int,
        reason: str,
        **__
):
    """
    Modifica il portafoglio di fiorygi di un membro.
    """

    if _user.sub != "auth0|5ed2debf7308300c1ea230c3":
        await _msg.reply(text=f"⚠️ Non sei autorizzato ad eseguire questo comando.")
        return

    if amount >= 0:
        trans = db.Transaction(
            minus=None,
            plus=_target,
            amount=amount,
            reason=reason,
        )
    else:
        trans = db.Transaction(
            minus=_target,
            plus=None,
            amount=amount,
            reason=reason,
        )

    _session.add(trans)

    _target.fiorygi += amount

    _session.commit()

    await _msg.reply(text=f"🏦 Hai modificato il portafoglio di {_target} di \uE01Bƒ {amount}\uE00B.")


__all__ = (
    "fiorygi_balance_self",
    "fiorygi_balance_other",
    "fiorygi_give",
    "fiorygi_magick",
    "fiorygi_transactions_self",
    "fiorygi_transactions_other",
)