Create various helper functions

lnbits/extensions/TwitchAlerts/crud.py

@@ -1,9 +1,176 @@
-async def get_charge_details(service):
+from . import db
+from .models import Donations, Services
+
+from ..satspay.crud import delete_charge
+
+import httpx
+
+from typing import Optional
+from lnbits.core.crud import get_wallet
+
+
+async def get_charge_details(service_id):
     details = {
-        "user": "8ee4d2d2b788467ebbab172ed3d50aaa",
-        "description": "Testing",
-        "lnbitswallet": "e1e7c4f4f86a4905ab413f96fb78aabc",
+        "description": f"TwitchAlerts donation for service {str(service_id)}",
         "time": 1440,
-        "amount": 1
     }
+    service = get_service(service_id)
+    wallet_id = service.wallet
+    wallet = await get_wallet(wallet_id)
+    user = wallet.user
+    details["user"] = user
+    details["lnbitswallet"] = wallet_id
+    details["onchainwallet"] = service.onchain
     return details
+
+
+async def create_donation(
+    id: str,
+    name: str,
+    cur_code: str,
+    sats: int,
+    amount: float,
+    service: int,
+    posted: bool = False,
+) -> Donations:
+    await db.execute(
+        """
+        INSERT INTO Donations (
+            id,
+            name,
+            cur_code,
+            sats,
+            amount,
+            service,
+            posted
+        )
+        VALUES (?, ?, ?)
+        """,
+        (
+            id,
+            name,
+            cur_code,
+            sats,
+            amount,
+            service,
+            posted
+        ),
+    )
+    return await get_donation(id)
+
+
+async def post_donation(donation_id: str) -> None:
+    donation = await get_donation(donation_id)
+    if donation.posted:
+        return False
+    service = await get_service(donation.service)
+    servicename = service.servicename
+    if servicename == "Streamlabs":
+        pass
+    await db.execute(
+        "UPDATE Donations SET posted = 1 WHERE id = ?",
+        (donation_id,)
+    )
+    return True
+
+
+async def create_service(
+    twitchuser: str,
+    client_id: str,
+    client_secret: str,
+    wallet: str,
+    servicename: str,
+    onchain: str = None,
+) -> Services:
+    result = await db.execute(
+        """
+        INSERT INTO Services (
+            twitchuser,
+            client_id,
+            client_secret,
+            wallet,
+            servicename,
+            authenticated,
+            onchain
+        )
+        VALUES (?, ?, ?, ?, ?, ?, ?)
+        """,
+        (
+            twitchuser,
+            client_id,
+            client_secret,
+            wallet,
+            servicename,
+            False,
+            onchain,
+        ),
+    )
+    service_id = result._result_proxy.lastrowid
+    service = await get_service(service_id)
+    return service
+
+
+async def get_service(service_id: int) -> Optional[Services]:
+    row = await db.fetchone(
+        "SELECT * FROM Services WHERE id = ?",
+        (service_id,)
+    )
+    return Services.from_row(row) if row else None
+
+
+async def authenticate_service(service_id, code, redirect_uri):
+    # The API token is passed in the querystring as 'code'
+    service = await get_service(service_id)
+    wallet = await get_wallet(service.wallet)
+    user = wallet.user
+    url = "https://streamlabs.com/api/v1.0/token"
+    data = {
+        "grant_type": "authorization_code",
+        "code": code,
+        "client_id": service.client_id,
+        "client_secret": service.client_secret,
+        "redirect_uri": redirect_uri,
+    }
+    print(data)
+    async with httpx.AsyncClient() as client:
+        response = (await client.post(url, data=data)).json()
+    print(response)
+    token = response['access_token']
+    await service_add_token(service_id, token)
+    return f"/TwitchAlerts/?usr={user}"
+
+
+async def service_add_token(service_id, token):
+    db.execute(
+        "UPDATE Services SET token = ? where id = ?",
+        (token, service_id,),
+    )
+
+
+async def delete_service(service_id: int) -> None:
+    await db.execute(
+        "DELETE FROM Services WHERE id = ?",
+        (service_id,)
+    )
+    rows = await db.fetchall(
+        "SELECT * FROM Donations WHERE service = ?",
+        (service_id,)
+    )
+    for row in rows:
+        await delete_donation(row["id"])
+
+
+async def get_donation(donation_id: str) -> Optional[Donations]:
+    row = await db.fetchone(
+        "SELECT * FROM Donations WHERE id = ?",
+        (donation_id,)
+    )
+    return Donations.from_row(row) if row else None
+
+
+async def delete_donation(donation_id: str) -> None:
+    await db.execute(
+        "DELETE FROM Donatoins WHERE id = ?",
+        (donation_id,)
+    )
+    await delete_charge(donation_id)