nostrrelay/views_api.py
2023-01-30 14:40:55 +02:00

32 lines
904 B
Python

from http import HTTPStatus
from fastapi import Depends, Query, WebSocket
from loguru import logger
from starlette.exceptions import HTTPException
from lnbits.decorators import WalletTypeInfo, get_key_type, require_admin_key
from . import nostrrelay_ext
from .client_manager import NostrClientConnection, NostrClientManager
client_manager = NostrClientManager()
@nostrrelay_ext.websocket("/client")
async def websocket_endpoint(websocket: WebSocket):
client = NostrClientConnection(websocket=websocket)
client_manager.add_client(client)
try:
await client.start()
except Exception as e:
logger.warning(e)
client_manager.remove_client(client)
@nostrrelay_ext.get("/api/v1/enable", status_code=HTTPStatus.OK)
async def api_nostrrelay(enable: bool = Query(True)):
return await enable_relay(enable)
async def enable_relay(enable: bool):
return enable