64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""DELETE /swarm/hosts/{uuid} — decommission a worker.
|
|
|
|
Removes the DeckyShard rows bound to the host (portable cascade — MySQL
|
|
and SQLite both honor it via the repo layer), deletes the SwarmHost row,
|
|
and best-effort-cleans the per-worker bundle directory on the master.
|
|
|
|
Also asks the worker agent to wipe its own install (keeping logs). A
|
|
dead/unreachable worker does not block master-side cleanup.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import pathlib
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
|
|
from decnet.logging import get_logger
|
|
from decnet.swarm.client import AgentClient
|
|
from decnet.web.db.repository import BaseRepository
|
|
from decnet.web.dependencies import get_repo
|
|
|
|
log = get_logger("swarm.decommission")
|
|
router = APIRouter()
|
|
|
|
|
|
@router.delete(
|
|
"/hosts/{uuid}",
|
|
status_code=status.HTTP_204_NO_CONTENT,
|
|
tags=["Swarm Hosts"],
|
|
responses={404: {"description": "No host with this UUID is enrolled"}},
|
|
)
|
|
async def api_decommission_host(
|
|
uuid: str,
|
|
repo: BaseRepository = Depends(get_repo),
|
|
) -> None:
|
|
row = await repo.get_swarm_host_by_uuid(uuid)
|
|
if row is None:
|
|
raise HTTPException(status_code=404, detail="host not found")
|
|
|
|
try:
|
|
async with AgentClient(host=row) as agent:
|
|
await agent.self_destruct()
|
|
except Exception:
|
|
log.exception(
|
|
"decommission: self-destruct dispatch failed host=%s — "
|
|
"proceeding with master-side cleanup anyway",
|
|
row.get("name"),
|
|
)
|
|
|
|
await repo.delete_decky_shards_for_host(uuid)
|
|
await repo.delete_swarm_host(uuid)
|
|
|
|
# Best-effort bundle cleanup; if the dir was moved manually, don't fail.
|
|
bundle_dir = pathlib.Path(row.get("cert_bundle_path") or "")
|
|
if bundle_dir.is_dir():
|
|
for child in bundle_dir.iterdir():
|
|
try:
|
|
child.unlink()
|
|
except OSError:
|
|
pass
|
|
try:
|
|
bundle_dir.rmdir()
|
|
except OSError:
|
|
pass
|