31 lines
984 B
Python
31 lines
984 B
Python
from fastapi import APIRouter, HTTPException, Response
|
|
|
|
from models.schemas import DriveDetail
|
|
from services.smart import get_smart_data
|
|
from services.zfs import get_zfs_pool_map
|
|
|
|
router = APIRouter(prefix="/api/drives", tags=["drives"])
|
|
|
|
|
|
@router.get("/{device}", response_model=DriveDetail)
|
|
async def get_drive_detail(device: str, response: Response):
|
|
"""Get SMART detail for a specific block device."""
|
|
try:
|
|
data, cache_hit = await get_smart_data(device)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
if "error" in data:
|
|
raise HTTPException(status_code=502, detail=data["error"])
|
|
|
|
pool_map = await get_zfs_pool_map()
|
|
zfs_info = pool_map.get(device)
|
|
if zfs_info:
|
|
data["zfs_pool"] = zfs_info["pool"]
|
|
data["zfs_vdev"] = zfs_info["vdev"]
|
|
data["zfs_state"] = zfs_info.get("state")
|
|
|
|
response.headers["X-Cache"] = "HIT" if cache_hit else "MISS"
|
|
|
|
return DriveDetail(**data)
|