remove extensions
This commit is contained in:
@@ -1,170 +0,0 @@
|
||||
import hashlib
|
||||
from quart import g, jsonify, url_for
|
||||
from http import HTTPStatus
|
||||
import httpx
|
||||
|
||||
from fastapi import FastAPI, Query
|
||||
from fastapi.encoders import jsonable_encoder
|
||||
from fastapi.responses import JSONResponse
|
||||
from pydantic import BaseModel
|
||||
|
||||
from lnbits.core.crud import get_user
|
||||
from lnbits.decorators import api_check_wallet_key, api_validate_post_request
|
||||
|
||||
from lnbits.extensions.satspay import satspay_ext
|
||||
from .crud import (
|
||||
create_charge,
|
||||
update_charge,
|
||||
get_charge,
|
||||
get_charges,
|
||||
delete_charge,
|
||||
check_address_balance,
|
||||
)
|
||||
|
||||
#############################CHARGES##########################
|
||||
|
||||
class CreateData(BaseModel):
|
||||
onchainwallet: str
|
||||
lnbitswallet: str
|
||||
description: str = Query(...)
|
||||
webhook: str
|
||||
completelink: str
|
||||
completelinktext: str
|
||||
time: int = Query(..., ge=1)
|
||||
amount: int = Query(..., ge=1)
|
||||
|
||||
@satspay_ext.post("/api/v1/charge")
|
||||
@satspay_ext.put("/api/v1/charge/{charge_id}")
|
||||
@api_check_wallet_key("admin")
|
||||
# @api_validate_post_request(
|
||||
# schema={
|
||||
# "onchainwallet": {"type": "string"},
|
||||
# "lnbitswallet": {"type": "string"},
|
||||
# "description": {"type": "string", "empty": False, "required": True},
|
||||
# "webhook": {"type": "string"},
|
||||
# "completelink": {"type": "string"},
|
||||
# "completelinktext": {"type": "string"},
|
||||
# "time": {"type": "integer", "min": 1, "required": True},
|
||||
# "amount": {"type": "integer", "min": 1, "required": True},
|
||||
# }
|
||||
# )
|
||||
async def api_charge_create_or_update(data: CreateData, charge_id=None):
|
||||
if not charge_id:
|
||||
charge = await create_charge(user=g.wallet.user, **data)
|
||||
return charge._asdict(), HTTPStatus.CREATED
|
||||
else:
|
||||
charge = await update_charge(charge_id=charge_id, **data)
|
||||
return charge._asdict(), HTTPStatus.OK
|
||||
|
||||
|
||||
@satspay_ext.get("/api/v1/charges")
|
||||
@api_check_wallet_key("invoice")
|
||||
async def api_charges_retrieve():
|
||||
try:
|
||||
return (
|
||||
|
||||
[
|
||||
{
|
||||
**charge._asdict(),
|
||||
**{"time_elapsed": charge.time_elapsed},
|
||||
**{"paid": charge.paid},
|
||||
}
|
||||
for charge in await get_charges(g.wallet.user)
|
||||
]
|
||||
,
|
||||
HTTPStatus.OK,
|
||||
)
|
||||
except:
|
||||
return ""
|
||||
|
||||
|
||||
@satspay_ext.get("/api/v1/charge/{charge_id}")
|
||||
@api_check_wallet_key("invoice")
|
||||
async def api_charge_retrieve(charge_id: str):
|
||||
charge = await get_charge(charge_id)
|
||||
|
||||
if not charge:
|
||||
return {"message": "charge does not exist"}, HTTPStatus.NOT_FOUND
|
||||
|
||||
return (
|
||||
|
||||
{
|
||||
**charge._asdict(),
|
||||
**{"time_elapsed": charge.time_elapsed},
|
||||
**{"paid": charge.paid},
|
||||
}
|
||||
,
|
||||
HTTPStatus.OK,
|
||||
)
|
||||
|
||||
|
||||
@satspay_ext.delete("/api/v1/charge/{charge_id}")
|
||||
@api_check_wallet_key("invoice")
|
||||
async def api_charge_delete(charge_id: str):
|
||||
charge = await get_charge(charge_id)
|
||||
|
||||
if not charge:
|
||||
return {"message": "Wallet link does not exist."}, HTTPStatus.NOT_FOUND
|
||||
|
||||
await delete_charge(charge_id)
|
||||
|
||||
return "", HTTPStatus.NO_CONTENT
|
||||
|
||||
|
||||
#############################BALANCE##########################
|
||||
|
||||
|
||||
@satspay_ext.get("/api/v1/charges/balance/{charge_id}")
|
||||
async def api_charges_balance(charge_id: str):
|
||||
|
||||
charge = await check_address_balance(charge_id)
|
||||
|
||||
if not charge:
|
||||
return {"message": "charge does not exist"}, HTTPStatus.NOT_FOUND
|
||||
if charge.paid and charge.webhook:
|
||||
async with httpx.AsyncClient() as client:
|
||||
try:
|
||||
r = await client.post(
|
||||
charge.webhook,
|
||||
json={
|
||||
"id": charge.id,
|
||||
"description": charge.description,
|
||||
"onchainaddress": charge.onchainaddress,
|
||||
"payment_request": charge.payment_request,
|
||||
"payment_hash": charge.payment_hash,
|
||||
"time": charge.time,
|
||||
"amount": charge.amount,
|
||||
"balance": charge.balance,
|
||||
"paid": charge.paid,
|
||||
"timestamp": charge.timestamp,
|
||||
"completelink": charge.completelink,
|
||||
},
|
||||
timeout=40,
|
||||
)
|
||||
except AssertionError:
|
||||
charge.webhook = None
|
||||
return charge._asdict(), HTTPStatus.OK
|
||||
|
||||
|
||||
#############################MEMPOOL##########################
|
||||
|
||||
|
||||
@satspay_ext.put("/api/v1/mempool")
|
||||
@api_check_wallet_key("invoice")
|
||||
# @api_validate_post_request(
|
||||
# schema={
|
||||
# "endpoint": {"type": "string", "empty": False, "required": True},
|
||||
# }
|
||||
# )
|
||||
async def api_update_mempool(endpoint: str = Query(...)):
|
||||
mempool = await update_mempool(user=g.wallet.user, endpoint)
|
||||
return mempool._asdict(), HTTPStatus.OK
|
||||
|
||||
|
||||
@satspay_ext.get("/api/v1/mempool")
|
||||
@api_check_wallet_key("invoice")
|
||||
async def api_get_mempool():
|
||||
mempool = await get_mempool(g.wallet.user)
|
||||
if not mempool:
|
||||
mempool = await create_mempool(user=g.wallet.user)
|
||||
return mempool._asdict(), HTTPStatus.OK
|
||||
Reference in New Issue
Block a user