lnbits-legend/lnbits/extensions/streamalerts/views_api.py

from typing import Optional
from pydantic.main import BaseModel
from quart import g, redirect, request, jsonify
from http import HTTPStatus

from lnbits.decorators import api_validate_post_request, api_check_wallet_key
from lnbits.core.crud import get_user
from lnbits.utils.exchange_rates import btc_price

from . import streamalerts_ext
from .crud import (
    get_charge_details,
    get_service_redirect_uri,
    create_donation,
    post_donation,
    get_donation,
    get_donations,
    delete_donation,
    create_service,
    get_service,
    get_services,
    authenticate_service,
    update_donation,
    update_service,
    delete_service,
)
from ..satspay.crud import create_charge, get_charge

class CreateServicesData(BaseModel):
    twitchuser:  str
    client_id:  str
    client_secret:  str
    wallet:  str
    servicename:  str
    onchain:  Optional[str]

@streamalerts_ext.post("/api/v1/services")
@api_check_wallet_key("invoice")
async def api_create_service(data: CreateServicesData):
    """Create a service, which holds data about how/where to post donations"""
    try:
        service = await create_service(**data)
    except Exception as e:
        return {"message": str(e)}, HTTPStatus.INTERNAL_SERVER_ERROR

    return service._asdict(), HTTPStatus.CREATED


@streamalerts_ext.get("/api/v1/getaccess/{service_id}")
async def api_get_access(service_id):
    """Redirect to Streamlabs' Approve/Decline page for API access for Service
    with service_id
    """
    service = await get_service(service_id)
    if service:
        redirect_uri = await get_service_redirect_uri(request, service_id)
        params = {
            "response_type": "code",
            "client_id": service.client_id,
            "redirect_uri": redirect_uri,
            "scope": "donations.create",
            "state": service.state,
        }
        endpoint_url = "https://streamlabs.com/api/v1.0/authorize/?"
        querystring = "&".join([f"{key}={value}" for key, value in params.items()])
        redirect_url = endpoint_url + querystring
        return redirect(redirect_url)
    else:
        return ({"message": "Service does not exist!"}, HTTPStatus.BAD_REQUEST)


@streamalerts_ext.get("/api/v1/authenticate/{service_id}")
async def api_authenticate_service(Code: str, State: str, service_id):
    """Endpoint visited via redirect during third party API authentication

    If successful, an API access token will be added to the service, and
    the user will be redirected to index.html.
    """
    code = Code
    state = State
    service = await get_service(service_id)
    if service.state != state:
        return ({"message": "State doesn't match!"}, HTTPStatus.BAD_Request)
    redirect_uri = request.scheme + "://" + request.headers["Host"]
    redirect_uri += f"/streamalerts/api/v1/authenticate/{service_id}"
    url, success = await authenticate_service(service_id, code, redirect_uri)
    if success:
        return redirect(url)
    else:
        return ({"message": "Service already authenticated!"},
            HTTPStatus.BAD_REQUEST,
        )

class CreateDonationsData(BaseModel):
    name:  str
    sats:  int
    service:  int
    message:  str

@streamalerts_ext.post("/api/v1/donations")
async def api_create_donation(data:CreateDonationsData):
    """Take data from donation form and return satspay charge"""
    # Currency is hardcoded while frotnend is limited
    cur_code = "USD"
    sats = data.sats
    message = data.message
    # Fiat amount is calculated here while frontend is limited
    price = await btc_price(cur_code)
    amount = sats * (10 ** (-8)) * price
    webhook_base = request.scheme + "://" + request.headers["Host"]
    service_id = data.service
    service = await get_service(service_id)
    charge_details = await get_charge_details(service.id)
    name = data.name
    if not name:
        name = "Anonymous"
    description = f"{sats} sats donation from {name} to {service.twitchuser}"
    charge = await create_charge(
        amount=sats,
        completelink=f"https://twitch.tv/{service.twitchuser}",
        completelinktext="Back to Stream!",
        webhook=webhook_base + "/streamalerts/api/v1/postdonation",
        description=description,
        **charge_details,
    )
    await create_donation(
        id=charge.id,
        wallet=service.wallet,
        message=message,
        name=name,
        cur_code=cur_code,
        sats=data.sats,
        amount=amount,
        service=data.service,
    )
    return ({"redirect_url": f"/satspay/{charge.id}"}), HTTPStatus.OK


@streamalerts_ext.post("/api/v1/postdonation")

async def api_post_donation(id: str):
    """Post a paid donation to Stremalabs/StreamElements.

    This endpoint acts as a webhook for the SatsPayServer extension."""
    data = await request.get_json(force=True)
    donation_id = id
    charge = await get_charge(donation_id)
    if charge and charge.paid:
        return await post_donation(donation_id)
    else:
        return ({"message": "Not a paid charge!"}, HTTPStatus.BAD_REQUEST)


@streamalerts_ext.get("/api/v1/services")
@api_check_wallet_key("invoice")
async def api_get_services():
    """Return list of all services assigned to wallet with given invoice key"""
    wallet_ids = (await get_user(g.wallet.user)).wallet_ids
    services = []
    for wallet_id in wallet_ids:
        new_services = await get_services(wallet_id)
        services += new_services if new_services else []
    return (
        [service._asdict() for service in services] if services else [],
        HTTPStatus.OK,
    )


@streamalerts_ext.get("/api/v1/donations")
@api_check_wallet_key("invoice")
async def api_get_donations():
    """Return list of all donations assigned to wallet with given invoice
    key
    """
    wallet_ids = (await get_user(g.wallet.user)).wallet_ids
    donations = []
    for wallet_id in wallet_ids:
        new_donations = await get_donations(wallet_id)
        donations += new_donations if new_donations else []
    return (
        [donation._asdict() for donation in donations] if donations else [],
        HTTPStatus.OK,
    )


@streamalerts_ext.put("/api/v1/donations/{donation_id}")
@api_check_wallet_key("invoice")
async def api_update_donation(donation_id=None):
    """Update a donation with the data given in the request"""
    if donation_id:
        donation = await get_donation(donation_id)

        if not donation:
            return (
                {"message": "Donation does not exist."},
                HTTPStatus.NOT_FOUND,
            )

        if donation.wallet != g.wallet.id:
            return ({"message": "Not your donation."}, HTTPStatus.FORBIDDEN)

        donation = await update_donation(donation_id, **g.data)
    else:
        return (
            {"message": "No donation ID specified"},
            HTTPStatus.BAD_REQUEST,
        )
    return donation._asdict(), HTTPStatus.CREATED


@streamalerts_ext.put("/api/v1/services/{service_id}")
@api_check_wallet_key("invoice")
async def api_update_service(service_id=None):
    """Update a service with the data given in the request"""
    if service_id:
        service = await get_service(service_id)

        if not service:
            return ({"message": "Service does not exist."},
                HTTPStatus.NOT_FOUND,
            )

        if service.wallet != g.wallet.id:
            return ({"message": "Not your service."}), HTTPStatus.FORBIDDEN

        service = await update_service(service_id, **g.data)
    else:
        return ({"message": "No service ID specified"}), HTTPStatus.BAD_REQUEST
    return service._asdict(), HTTPStatus.CREATED


@streamalerts_ext.delete("/api/v1/donations/{donation_id}")
@api_check_wallet_key("invoice")
async def api_delete_donation(donation_id):
    """Delete the donation with the given donation_id"""
    donation = await get_donation(donation_id)
    if not donation:
        return ({"message": "No donation with this ID!"}, HTTPStatus.NOT_FOUND)
    if donation.wallet != g.wallet.id:
        return ({"message": "Not authorized to delete this donation!"},
            HTTPStatus.FORBIDDEN,
        )
    await delete_donation(donation_id)

    return "", HTTPStatus.NO_CONTENT


@streamalerts_ext.delete("/api/v1/services/{service_id}")
@api_check_wallet_key("invoice")
async def api_delete_service(service_id):
    """Delete the service with the given service_id"""
    service = await get_service(service_id)
    if not service:
        return ({"message": "No service with this ID!"}, HTTPStatus.NOT_FOUND)
    if service.wallet != g.wallet.id:
        return (
            {"message": "Not authorized to delete this service!"},
            HTTPStatus.FORBIDDEN,
        )
    await delete_service(service_id)

    return "", HTTPStatus.NO_CONTENT