"""GET /system/deployment-mode — tells the UI whether a deploy will shard
across SWARM workers or land on the master itself.

Logic mirrors the auto-mode branch in ``api_deploy_deckies``: master role
plus at least one reachable enrolled worker = swarm; otherwise unihost.
"""
from __future__ import annotations

import os

from fastapi import APIRouter, Depends
from pydantic import BaseModel

from decnet.web.db.repository import BaseRepository
from decnet.web.dependencies import get_repo

router = APIRouter()


class DeploymentModeResponse(BaseModel):
    mode: str  # "swarm" or "unihost"
    role: str  # "master" or "agent"
    swarm_host_count: int


@router.get("/deployment-mode", response_model=DeploymentModeResponse)
async def get_deployment_mode(
    repo: BaseRepository = Depends(get_repo),
) -> DeploymentModeResponse:
    role = os.environ.get("DECNET_MODE", "master").lower()
    hosts = 0
    if role == "master":
        hosts = sum(
            1 for h in await repo.list_swarm_hosts()
            if h.get("status") in ("active", "enrolled") and h.get("address")
        )
    return DeploymentModeResponse(
        mode="swarm" if hosts > 0 else "unihost",
        role=role,
        swarm_host_count=hosts,
    )