from fastapi import APIRouter
import logging
from os import getenv
from remnawave import RemnawaveSDK
from remnawave.models.hosts import GetAllHostsResponseDto

logging.basicConfig(level=logging.INFO)

router = APIRouter(prefix='/api/v1', tags=['nodes'])
remnawave = RemnawaveSDK(base_url='http://remnawave:3000', token=getenv('REMNAWAVE_API_TOKEN'))

@router.get('/nodes')
async def get_nodes():
    hosts: GetAllHostsResponseDto = await remnawave.hosts.get_all_hosts()

    return hosts.root