Mirror/intercept
1
0
Fork 0
mirror of https://github.com/smittix/intercept.git synced 2026-04-29 00:59:59 -07:00
Code Issues Packages Projects Releases Wiki Activity

Add distributed agent architecture for multi-node signal intelligence

Browse Source 
Features:
- Standalone agent server (intercept_agent.py) for remote sensor nodes
- Controller API blueprint for agent management and data aggregation
- Push mechanism for agents to send data to controller
- Pull mechanism for controller to proxy requests to agents
- Multi-agent SSE stream for combined data view
- Agent management page at /controller/manage
- Agent selector dropdown in main UI
- GPS integration for location tagging
- API key authentication for secure agent communication
- Integration with Intercept's dependency checking system

New files:
- intercept_agent.py: Remote agent HTTP server
- intercept_agent.cfg: Agent configuration template
- routes/controller.py: Controller API endpoints
- utils/agent_client.py: HTTP client for agents
- utils/trilateration.py: Multi-agent position calculation
- static/js/core/agents.js: Frontend agent management
- templates/agents.html: Agent management page
- docs/DISTRIBUTED_AGENTS.md: System documentation

Modified:
- app.py: Register controller blueprint
- utils/database.py: Add agents and push_payloads tables
- templates/index.html: Add agent selector section
This commit is contained in:
cemaxecuter
2026-01-26 06:14:42 -05:00
parent ada6d5f1f1
commit f980e2e76d
20 changed files with 8809 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

281
utils/agent_client.py Normal file
View File

@@ -0,0 +1,281 @@
"""
HTTP client for communicating with remote Intercept agents.
"""
from __future__ import annotations
import logging
from typing import Any
import requests
logger = logging.getLogger('intercept.agent_client')
class AgentHTTPError(RuntimeError):
"""Exception raised when agent HTTP request fails."""
def __init__(self, message: str, status_code: int | None = None):
super().__init__(message)
self.status_code = status_code
class AgentConnectionError(AgentHTTPError):
"""Exception raised when agent is unreachable."""
pass
class AgentClient:
"""HTTP client for communicating with a remote Intercept agent."""
def __init__(
self,
base_url: str,
api_key: str | None = None,
timeout: float = 60.0
):
"""
Initialize agent client.
Args:
base_url: Base URL of the agent (e.g., http://192.168.1.50:8020)
api_key: Optional API key for authentication
timeout: Request timeout in seconds
"""
self.base_url = base_url.rstrip('/')
self.api_key = api_key
self.timeout = timeout
def _headers(self) -> dict:
"""Get request headers."""
headers = {'Content-Type': 'application/json'}
if self.api_key:
headers['X-API-Key'] = self.api_key
return headers
def _get(self, path: str, params: dict | None = None) -> dict:
"""
Perform GET request to agent.
Args:
path: URL path (e.g., /capabilities)
params: Optional query parameters
Returns:
Parsed JSON response
Raises:
AgentHTTPError: On HTTP errors
AgentConnectionError: If agent is unreachable
"""
url = f"{self.base_url}{path}"
try:
response = requests.get(
url,
headers=self._headers(),
params=params,
timeout=self.timeout
)
response.raise_for_status()
return response.json() if response.content else {}
except requests.ConnectionError as e:
raise AgentConnectionError(f"Cannot connect to agent at {self.base_url}: {e}")
except requests.Timeout:
raise AgentConnectionError(f"Request to agent timed out after {self.timeout}s")
except requests.HTTPError as e:
raise AgentHTTPError(
f"Agent returned error: {e.response.status_code}",
status_code=e.response.status_code
)
except requests.RequestException as e:
raise AgentHTTPError(f"Request failed: {e}")
def _post(self, path: str, data: dict | None = None) -> dict:
"""
Perform POST request to agent.
Args:
path: URL path (e.g., /sensor/start)
data: Optional JSON body
Returns:
Parsed JSON response
Raises:
AgentHTTPError: On HTTP errors
AgentConnectionError: If agent is unreachable
"""
url = f"{self.base_url}{path}"
try:
response = requests.post(
url,
json=data or {},
headers=self._headers(),
timeout=self.timeout
)
response.raise_for_status()
return response.json() if response.content else {}
except requests.ConnectionError as e:
raise AgentConnectionError(f"Cannot connect to agent at {self.base_url}: {e}")
except requests.Timeout:
raise AgentConnectionError(f"Request to agent timed out after {self.timeout}s")
except requests.HTTPError as e:
raise AgentHTTPError(
f"Agent returned error: {e.response.status_code}",
status_code=e.response.status_code
)
except requests.RequestException as e:
raise AgentHTTPError(f"Request failed: {e}")
# =========================================================================
# Capability & Status
# =========================================================================
def get_capabilities(self) -> dict:
"""
Get agent capabilities (available modes, devices).
Returns:
Dict with 'modes' (mode -> bool), 'devices' (list), 'agent_version'
"""
return self._get('/capabilities')
def get_status(self) -> dict:
"""
Get agent status.
Returns:
Dict with 'running_modes', 'uptime', 'push_enabled', etc.
"""
return self._get('/status')
def health_check(self) -> bool:
"""
Check if agent is healthy.
Returns:
True if agent is reachable and healthy
"""
try:
result = self._get('/health')
return result.get('status') == 'healthy'
except (AgentHTTPError, AgentConnectionError):
return False
def get_config(self) -> dict:
"""Get agent configuration (non-sensitive fields)."""
return self._get('/config')
def update_config(self, **kwargs) -> dict:
"""
Update agent configuration.
Args:
push_enabled: Enable/disable push mode
push_interval: Push interval in seconds
Returns:
Updated config
"""
return self._post('/config', kwargs)
# =========================================================================
# Mode Operations
# =========================================================================
def start_mode(self, mode: str, params: dict | None = None) -> dict:
"""
Start a mode on the agent.
Args:
mode: Mode name (e.g., 'sensor', 'adsb', 'wifi')
params: Mode-specific parameters
Returns:
Start result with 'status' field
"""
return self._post(f'/{mode}/start', params or {})
def stop_mode(self, mode: str) -> dict:
"""
Stop a running mode on the agent.
Args:
mode: Mode name
Returns:
Stop result with 'status' field
"""
return self._post(f'/{mode}/stop')
def get_mode_status(self, mode: str) -> dict:
"""
Get status of a specific mode.
Args:
mode: Mode name
Returns:
Mode status with 'running' field
"""
return self._get(f'/{mode}/status')
def get_mode_data(self, mode: str) -> dict:
"""
Get current data snapshot for a mode.
Args:
mode: Mode name
Returns:
Data snapshot with 'data' field
"""
return self._get(f'/{mode}/data')
# =========================================================================
# Convenience Methods
# =========================================================================
def refresh_metadata(self) -> dict:
"""
Fetch comprehensive metadata from agent.
Returns:
Dict with capabilities, status, and config
"""
metadata = {
'capabilities': None,
'status': None,
'config': None,
'healthy': False,
}
try:
metadata['capabilities'] = self.get_capabilities()
metadata['status'] = self.get_status()
metadata['config'] = self.get_config()
metadata['healthy'] = True
except (AgentHTTPError, AgentConnectionError) as e:
logger.warning(f"Failed to refresh agent metadata: {e}")
return metadata
def __repr__(self) -> str:
return f"AgentClient({self.base_url})"
def create_client_from_agent(agent: dict) -> AgentClient:
"""
Create an AgentClient from an agent database record.
Args:
agent: Agent dict from database
Returns:
Configured AgentClient
"""
return AgentClient(
base_url=agent['base_url'],
api_key=agent.get('api_key'),
timeout=60.0
)