feat(agent): implement Agent Bridge for secure diagnostics
This commit is contained in:
60
app/routers/agent_bridge.py
Normal file
60
app/routers/agent_bridge.py
Normal file
@@ -0,0 +1,60 @@
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
import os
|
||||
import platform
|
||||
import json
|
||||
from typing import Dict, Any
|
||||
|
||||
from app.lib_general_v3 import AccountContext, get_account_context
|
||||
from app.models.response_models import Resp_Body_Base, mk_resp
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
@router.get("/status", response_model=Resp_Body_Base, tags=["Agent Bridge"])
|
||||
async def get_container_status(
|
||||
account: AccountContext = Depends(get_account_context)
|
||||
):
|
||||
"""
|
||||
Returns diagnostic information about the container environment.
|
||||
Only accessible to administrators/managers via existing hierarchy.
|
||||
"""
|
||||
# Simple check for administrative access
|
||||
if not getattr(account, "administrator", False) and not getattr(account, "manager", False):
|
||||
raise HTTPException(status_code=403, detail="Administrative access required.")
|
||||
|
||||
status_data = {
|
||||
"os": platform.system(),
|
||||
"release": platform.release(),
|
||||
"python_version": platform.python_version(),
|
||||
"hostname": platform.node(),
|
||||
"cpu_count": os.cpu_count(),
|
||||
"environment_vars": {k: v for k, v in os.environ.items() if "PASSWORD" not in k.upper() and "KEY" not in k.upper() and "SECRET" not in k.upper()},
|
||||
"cwd": os.getcwd(),
|
||||
"container": os.path.exists('/.dockerenv')
|
||||
}
|
||||
|
||||
return mk_resp(data=status_data)
|
||||
|
||||
@router.get("/logs", response_model=Resp_Body_Base, tags=["Agent Bridge"])
|
||||
async def get_latest_logs(
|
||||
lines: int = 50,
|
||||
account: AccountContext = Depends(get_account_context)
|
||||
):
|
||||
"""
|
||||
Returns the last N lines of the application log.
|
||||
"""
|
||||
if not getattr(account, "administrator", False) and not getattr(account, "manager", False):
|
||||
raise HTTPException(status_code=403, detail="Administrative access required.")
|
||||
|
||||
from app.config import settings
|
||||
log_path = settings.LOG_PATH.get('app', '/logs/aether_api.log')
|
||||
|
||||
if not os.path.exists(log_path):
|
||||
return mk_resp(data=False, status_message=f"Log file not found at {log_path}")
|
||||
|
||||
try:
|
||||
with open(log_path, 'r') as f:
|
||||
log_lines = f.readlines()
|
||||
latest = log_lines[-lines:] if len(log_lines) > lines else log_lines
|
||||
return mk_resp(data="".join(latest))
|
||||
except Exception as e:
|
||||
return mk_resp(data=False, status_message=str(e))
|
||||
Reference in New Issue
Block a user