import logging
import os

import psutil
from fastapi import APIRouter

from app.schemas.system import HealthResponse, SystemInfoResponse
from app.services import freepik_client, task_tracker

logger = logging.getLogger(__name__)

router = APIRouter(prefix='/api/v1', tags=['system'])


@router.get('/health', response_model=HealthResponse)
async def health_check():
    return HealthResponse()


@router.get('/system', response_model=SystemInfoResponse)
async def system_info():
    api_key_valid = await freepik_client.check_api_key()
    mem = psutil.virtual_memory()

    return SystemInfoResponse(
        api_key_valid=api_key_valid,
        active_tasks=task_tracker.active_count(),
        cpu_count=os.cpu_count() or 1,
        memory_total=mem.total,
        memory_available=mem.available,
    )