mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-03-10 17:26:15 +01:00
155 lines
4.7 KiB
Python
155 lines
4.7 KiB
Python
from http import HTTPStatus
|
|
|
|
from fastapi import Depends, Query, Request, WebSocket, WebSocketDisconnect
|
|
from fastapi.templating import Jinja2Templates
|
|
from loguru import logger
|
|
from starlette.exceptions import HTTPException
|
|
from starlette.responses import HTMLResponse
|
|
|
|
from lnbits.core.models import User
|
|
from lnbits.decorators import check_user_exists
|
|
|
|
from . import market_ext, market_renderer
|
|
from .crud import (
|
|
create_market_settings,
|
|
get_market_market,
|
|
get_market_market_stalls,
|
|
get_market_order_details,
|
|
get_market_order_invoiceid,
|
|
get_market_products,
|
|
get_market_settings,
|
|
get_market_stall,
|
|
get_market_zone,
|
|
)
|
|
from .models import SetSettings
|
|
from .notifier import Notifier
|
|
|
|
templates = Jinja2Templates(directory="templates")
|
|
|
|
|
|
@market_ext.get("/", response_class=HTMLResponse)
|
|
async def index(request: Request, user: User = Depends(check_user_exists)):
|
|
settings = await get_market_settings(user=user.id)
|
|
|
|
if not settings:
|
|
await create_market_settings(
|
|
user=user.id, data=SetSettings(currency="sat", fiat_base_multiplier=1)
|
|
)
|
|
settings = await get_market_settings(user.id)
|
|
assert settings
|
|
return market_renderer().TemplateResponse(
|
|
"market/index.html",
|
|
{"request": request, "user": user.dict(), "currency": settings.currency},
|
|
)
|
|
|
|
|
|
@market_ext.get("/stalls/{stall_id}", response_class=HTMLResponse)
|
|
async def stall(request: Request, stall_id):
|
|
stall = await get_market_stall(stall_id)
|
|
products = await get_market_products(stall_id)
|
|
|
|
if not stall:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Stall does not exist."
|
|
)
|
|
|
|
zones = []
|
|
for id in stall.shippingzones.split(","):
|
|
zone = await get_market_zone(id)
|
|
assert zone
|
|
z = zone.dict()
|
|
zones.append({"label": z["countries"], "cost": z["cost"], "value": z["id"]})
|
|
|
|
_stall = stall.dict()
|
|
|
|
_stall["zones"] = zones
|
|
|
|
return market_renderer().TemplateResponse(
|
|
"market/stall.html",
|
|
{
|
|
"request": request,
|
|
"stall": _stall,
|
|
"products": [product.dict() for product in products],
|
|
},
|
|
)
|
|
|
|
|
|
@market_ext.get("/market/{market_id}", response_class=HTMLResponse)
|
|
async def market(request: Request, market_id):
|
|
market = await get_market_market(market_id)
|
|
|
|
if not market:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Marketplace doesn't exist."
|
|
)
|
|
|
|
stalls = await get_market_market_stalls(market_id)
|
|
stalls_ids = [stall.id for stall in stalls]
|
|
products = [product.dict() for product in await get_market_products(stalls_ids)]
|
|
|
|
return market_renderer().TemplateResponse(
|
|
"market/market.html",
|
|
{
|
|
"request": request,
|
|
"market": market,
|
|
"stalls": [stall.dict() for stall in stalls],
|
|
"products": products,
|
|
},
|
|
)
|
|
|
|
|
|
@market_ext.get("/order", response_class=HTMLResponse)
|
|
async def order_chat(
|
|
request: Request,
|
|
merch: str = Query(...),
|
|
invoice_id: str = Query(...),
|
|
keys: str = Query(None),
|
|
):
|
|
stall = await get_market_stall(merch)
|
|
assert stall
|
|
order = await get_market_order_invoiceid(invoice_id)
|
|
assert order
|
|
_order = await get_market_order_details(order.id)
|
|
products = await get_market_products(stall.id)
|
|
assert products
|
|
|
|
return market_renderer().TemplateResponse(
|
|
"market/order.html",
|
|
{
|
|
"request": request,
|
|
"stall": {
|
|
"id": stall.id,
|
|
"name": stall.name,
|
|
"publickey": stall.publickey,
|
|
"wallet": stall.wallet,
|
|
"currency": stall.currency,
|
|
},
|
|
"publickey": keys.split(",")[0] if keys else None,
|
|
"privatekey": keys.split(",")[1] if keys else None,
|
|
"order_id": order.invoiceid,
|
|
"order": [details.dict() for details in _order],
|
|
"products": [product.dict() for product in products],
|
|
},
|
|
)
|
|
|
|
|
|
##################WEBSOCKET ROUTES########################
|
|
|
|
# Initialize Notifier:
|
|
notifier = Notifier()
|
|
|
|
|
|
@market_ext.websocket("/ws/{room_name}")
|
|
async def websocket_endpoint(websocket: WebSocket, room_name: str):
|
|
await notifier.connect(websocket, room_name)
|
|
try:
|
|
while True:
|
|
data = await websocket.receive_text()
|
|
room_members = notifier.get_members(room_name) or []
|
|
if websocket not in room_members:
|
|
logger.warning("Sender not in room member: Reconnecting...")
|
|
await notifier.connect(websocket, room_name)
|
|
await notifier._notify(data, room_name)
|
|
|
|
except WebSocketDisconnect:
|
|
notifier.remove(websocket, room_name)
|