PTah
2aec488226
Detect 302→303 within 1–10s, Problem with 30s dedup, rdg_flap flag. Agent command poll API, admin qwinsta/logoff from Overview. Migration 016.
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from typing import Any
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
|
from pydantic import BaseModel, Field
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.auth.api_key import get_api_key_auth
|
|
from app.database import get_db
|
|
from app.services.agent_commands import (
|
|
command_to_dict,
|
|
complete_command,
|
|
get_command_for_host_poll,
|
|
resolve_host_by_agent_instance_id,
|
|
)
|
|
|
|
router = APIRouter(prefix="/agent", tags=["agent"])
|
|
|
|
|
|
class AgentCommandResultBody(BaseModel):
|
|
status: str = Field(pattern="^(completed|failed)$")
|
|
stdout: str | None = None
|
|
stderr: str | None = None
|
|
|
|
|
|
class AgentCommandsPollResponse(BaseModel):
|
|
commands: list[dict[str, Any]]
|
|
|
|
|
|
@router.get("/commands", response_model=AgentCommandsPollResponse)
|
|
def poll_agent_commands(
|
|
agent_instance_id: str = Query(..., min_length=1),
|
|
db: Session = Depends(get_db),
|
|
_api_key: str = Depends(get_api_key_auth),
|
|
) -> AgentCommandsPollResponse:
|
|
host = resolve_host_by_agent_instance_id(db, agent_instance_id)
|
|
if host is None:
|
|
raise HTTPException(status_code=404, detail="Unknown agent_instance_id")
|
|
pending = get_command_for_host_poll(db, host)
|
|
return AgentCommandsPollResponse(
|
|
commands=[command_to_dict(c, include_run_as=True) for c in pending]
|
|
)
|
|
|
|
|
|
@router.post("/commands/{command_uuid}/result")
|
|
def post_agent_command_result(
|
|
command_uuid: str,
|
|
body: AgentCommandResultBody,
|
|
db: Session = Depends(get_db),
|
|
_api_key: str = Depends(get_api_key_auth),
|
|
) -> dict[str, str]:
|
|
cmd = complete_command(
|
|
db,
|
|
command_uuid,
|
|
status=body.status,
|
|
stdout=body.stdout,
|
|
stderr=body.stderr,
|
|
)
|
|
if cmd is None:
|
|
raise HTTPException(status_code=404, detail="Command not found")
|
|
db.commit()
|
|
return {"status": cmd.status, "command_uuid": cmd.command_uuid}
|