mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-21 02:54:20 +01:00
e846acf946
* add websockets for quote updates * add test (not working) * wip: emit events to everyone * wip: emit events to everyone * wip, lots of things broken but invoice callback works * wip * add wip files * tests almost passing * add task * refactor nut constants * startup fix * works with old mints * wip cli * fix mypy * remove automatic invoice test now with websockets * remove comment * better logging * send back response * add rate limiter to websocket * add rate limiter to subscriptions * refactor websocket ratelimit * websocket tests * subscription kinds * doesnt start * remove circular import * update * fix mypy * move test file in test because it fails if it runs later... dunno why * adjust websocket NUT-06 settings * local import and small fix * disable websockets in CLI if "no_check" is selected * move subscription test to where it was * check proof state with callback, add tests * tests: run mint fixture per module instead of per session * subscription command name fix * test per session again * update test race conditions * fix tests * clean up * tmp * fix db issues and remove cached secrets * fix tests * blindly try pipeline * remove comments * comments
97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
from fastapi import WebSocket, status
|
|
from fastapi.responses import JSONResponse
|
|
from limits import RateLimitItemPerMinute
|
|
from loguru import logger
|
|
from slowapi import Limiter
|
|
from slowapi.util import get_remote_address
|
|
from starlette.requests import Request
|
|
|
|
from ..core.settings import settings
|
|
|
|
|
|
def _rate_limit_exceeded_handler(request: Request, exc: Exception) -> JSONResponse:
|
|
remote_address = get_remote_address(request)
|
|
logger.warning(
|
|
f"Rate limit {settings.mint_global_rate_limit_per_minute}/minute exceeded: {remote_address}"
|
|
)
|
|
return JSONResponse(
|
|
status_code=status.HTTP_429_TOO_MANY_REQUESTS,
|
|
content={"detail": "Rate limit exceeded."},
|
|
)
|
|
|
|
|
|
def get_remote_address_excluding_local(request: Request) -> str:
|
|
remote_address = get_remote_address(request)
|
|
if remote_address == "127.0.0.1":
|
|
return ""
|
|
return remote_address
|
|
|
|
|
|
limiter_global = Limiter(
|
|
key_func=get_remote_address_excluding_local,
|
|
strategy="fixed-window-elastic-expiry",
|
|
default_limits=[f"{settings.mint_global_rate_limit_per_minute}/minute"],
|
|
enabled=settings.mint_rate_limit,
|
|
)
|
|
|
|
limiter = Limiter(
|
|
key_func=get_remote_address_excluding_local,
|
|
strategy="fixed-window-elastic-expiry",
|
|
default_limits=[f"{settings.mint_transaction_rate_limit_per_minute}/minute"],
|
|
enabled=settings.mint_rate_limit,
|
|
)
|
|
|
|
|
|
def assert_limit(identifier: str):
|
|
"""Custom rate limit handler that accepts a string identifier
|
|
and raises an exception if the rate limit is exceeded. Uses the
|
|
setting `mint_transaction_rate_limit_per_minute` for the rate limit.
|
|
|
|
Args:
|
|
identifier (str): The identifier to use for the rate limit. IP address for example.
|
|
|
|
Raises:
|
|
Exception: If the rate limit is exceeded.
|
|
"""
|
|
global limiter
|
|
success = limiter._limiter.hit(
|
|
RateLimitItemPerMinute(settings.mint_transaction_rate_limit_per_minute),
|
|
identifier,
|
|
)
|
|
if not success:
|
|
logger.warning(
|
|
f"Rate limit {settings.mint_transaction_rate_limit_per_minute}/minute exceeded: {identifier}"
|
|
)
|
|
raise Exception("Rate limit exceeded")
|
|
|
|
|
|
def get_ws_remote_address(ws: WebSocket) -> str:
|
|
"""Returns the ip address for the current websocket (or 127.0.0.1 if none found)
|
|
|
|
Args:
|
|
ws (WebSocket): The FastAPI WebSocket object.
|
|
|
|
Returns:
|
|
str: The ip address for the current websocket.
|
|
"""
|
|
if not ws.client or not ws.client.host:
|
|
return "127.0.0.1"
|
|
|
|
return ws.client.host
|
|
|
|
|
|
def limit_websocket(ws: WebSocket):
|
|
"""Websocket rate limit handler that accepts a FastAPI WebSocket object.
|
|
This function will raise an exception if the rate limit is exceeded.
|
|
|
|
Args:
|
|
ws (WebSocket): The FastAPI WebSocket object.
|
|
|
|
Raises:
|
|
Exception: If the rate limit is exceeded.
|
|
"""
|
|
remote_address = get_ws_remote_address(ws)
|
|
if remote_address == "127.0.0.1":
|
|
return
|
|
assert_limit(remote_address)
|