diff --git a/main.py b/main.py index d6ef9f6..9a64932 100644 --- a/main.py +++ b/main.py @@ -2,7 +2,7 @@ from fastapi import FastAPI import tomllib import logging -from routers import ip, tariffs, nodes +from routers import hosts, ip, tariffs logging.basicConfig(level=logging.INFO) @@ -23,4 +23,4 @@ async def health_check(): app.include_router(ip.router) app.include_router(tariffs.router) -app.include_router(nodes.router) \ No newline at end of file +app.include_router(hosts.router) \ No newline at end of file diff --git a/routers/nodes.py b/routers/hosts.py similarity index 57% rename from routers/nodes.py rename to routers/hosts.py index c63755c..fc471c4 100644 --- a/routers/nodes.py +++ b/routers/hosts.py @@ -6,11 +6,19 @@ from remnawave.models.hosts import GetAllHostsResponseDto logging.basicConfig(level=logging.INFO) -router = APIRouter(prefix='/api/v1', tags=['nodes']) +router = APIRouter(prefix='/api/v1', tags=['hosts']) remnawave = RemnawaveSDK(base_url='http://remnawave:3000', token=getenv('REMNAWAVE_API_TOKEN')) -@router.get('/nodes') +@router.get('/hosts') async def get_nodes(): hosts: GetAllHostsResponseDto = await remnawave.hosts.get_all_hosts() + return [ - return hosts.root \ No newline at end of file + { + 'name': host.remark, + 'address': host.address, + 'port': host.port + } + + for host in hosts.root if not host.is_disabled and not host.is_hidden + ] \ No newline at end of file