278 lines
9.4 KiB
Python
278 lines
9.4 KiB
Python
"""Minimal sysmonstm gateway - standalone mode without dependencies."""
|
|
|
|
import asyncio
|
|
import json
|
|
import logging
|
|
import os
|
|
from datetime import datetime
|
|
|
|
from fastapi import FastAPI, Query, WebSocket, WebSocketDisconnect
|
|
from fastapi.responses import HTMLResponse
|
|
|
|
# Configuration
|
|
API_KEY = os.environ.get("API_KEY", "")
|
|
LOG_LEVEL = os.environ.get("LOG_LEVEL", "INFO")
|
|
|
|
# Logging setup
|
|
logging.basicConfig(
|
|
level=getattr(logging, LOG_LEVEL.upper(), logging.INFO),
|
|
format="%(asctime)s [%(levelname)s] %(message)s",
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|
)
|
|
log = logging.getLogger("gateway")
|
|
|
|
app = FastAPI(title="sysmonstm")
|
|
|
|
# Store connected websockets
|
|
connections: list[WebSocket] = []
|
|
# Store latest metrics from collectors
|
|
machines: dict = {}
|
|
|
|
HTML = """
|
|
<!DOCTYPE html>
|
|
<html lang="en">
|
|
<head>
|
|
<meta charset="UTF-8">
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
<title>sysmonstm</title>
|
|
<style>
|
|
:root {
|
|
--bg: #1a1a2e;
|
|
--bg2: #16213e;
|
|
--text: #eee;
|
|
--accent: #e94560;
|
|
--success: #4ade80;
|
|
--muted: #666;
|
|
}
|
|
* { box-sizing: border-box; margin: 0; padding: 0; }
|
|
body {
|
|
font-family: system-ui, sans-serif;
|
|
background: var(--bg);
|
|
color: var(--text);
|
|
min-height: 100vh;
|
|
padding: 2rem;
|
|
}
|
|
header {
|
|
display: flex;
|
|
justify-content: space-between;
|
|
align-items: center;
|
|
margin-bottom: 2rem;
|
|
padding-bottom: 1rem;
|
|
border-bottom: 2px solid var(--accent);
|
|
}
|
|
h1 { font-size: 1.5rem; }
|
|
.status {
|
|
display: flex;
|
|
align-items: center;
|
|
gap: 0.5rem;
|
|
font-size: 0.875rem;
|
|
}
|
|
.dot {
|
|
width: 10px;
|
|
height: 10px;
|
|
border-radius: 50%;
|
|
background: var(--accent);
|
|
}
|
|
.dot.ok { background: var(--success); }
|
|
.machines {
|
|
display: grid;
|
|
grid-template-columns: repeat(auto-fill, minmax(350px, 1fr));
|
|
gap: 1rem;
|
|
}
|
|
.machine {
|
|
background: var(--bg2);
|
|
border-radius: 8px;
|
|
padding: 1rem;
|
|
}
|
|
.machine h3 { margin-bottom: 0.5rem; }
|
|
.metric {
|
|
display: flex;
|
|
justify-content: space-between;
|
|
padding: 0.25rem 0;
|
|
border-bottom: 1px solid #333;
|
|
}
|
|
.empty {
|
|
text-align: center;
|
|
color: var(--muted);
|
|
padding: 4rem;
|
|
}
|
|
.empty p { margin-top: 1rem; }
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<header>
|
|
<h1>sysmonstm</h1>
|
|
<div class="status">
|
|
<span class="dot" id="ws-status"></span>
|
|
<span id="status-text">connecting...</span>
|
|
</div>
|
|
</header>
|
|
<main>
|
|
<div id="machines" class="machines">
|
|
<div class="empty">
|
|
<h2>No collectors connected</h2>
|
|
<p>Start a collector to see metrics</p>
|
|
</div>
|
|
</div>
|
|
</main>
|
|
<script>
|
|
const machinesEl = document.getElementById('machines');
|
|
const statusDot = document.getElementById('ws-status');
|
|
const statusText = document.getElementById('status-text');
|
|
let machines = {};
|
|
|
|
function connect() {
|
|
const protocol = location.protocol === 'https:' ? 'wss:' : 'ws:';
|
|
const ws = new WebSocket(`${protocol}//${location.host}/ws`);
|
|
|
|
ws.onopen = () => {
|
|
statusDot.classList.add('ok');
|
|
statusText.textContent = 'connected';
|
|
};
|
|
|
|
ws.onclose = () => {
|
|
statusDot.classList.remove('ok');
|
|
statusText.textContent = 'disconnected';
|
|
setTimeout(connect, 2000);
|
|
};
|
|
|
|
ws.onmessage = (e) => {
|
|
const data = JSON.parse(e.data);
|
|
if (data.type === 'metrics') {
|
|
machines[data.machine_id] = data;
|
|
render();
|
|
}
|
|
};
|
|
}
|
|
|
|
function render() {
|
|
const ids = Object.keys(machines).sort();
|
|
if (ids.length === 0) {
|
|
machinesEl.innerHTML = '<div class="empty"><h2>No collectors connected</h2><p>Start a collector to see metrics</p></div>';
|
|
return;
|
|
}
|
|
|
|
machinesEl.innerHTML = ids.map(id => {
|
|
const m = machines[id];
|
|
const ts = m.timestamp ? new Date(m.timestamp * 1000).toLocaleTimeString() : '-';
|
|
return `
|
|
<div class="machine">
|
|
<h3>${id}</h3>
|
|
<div class="metric"><span>CPU</span><span>${m.cpu?.toFixed(1) || '-'}%</span></div>
|
|
<div class="metric"><span>Memory</span><span>${m.memory?.toFixed(1) || '-'}%</span></div>
|
|
<div class="metric"><span>Disk</span><span>${m.disk?.toFixed(1) || '-'}%</span></div>
|
|
<div class="metric"><span>Load (1m)</span><span>${m.load_1m?.toFixed(2) || '-'}</span></div>
|
|
<div class="metric"><span>Processes</span><span>${m.processes || '-'}</span></div>
|
|
<div class="metric"><span>Updated</span><span>${ts}</span></div>
|
|
</div>
|
|
`;
|
|
}).join('');
|
|
}
|
|
|
|
connect();
|
|
</script>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
|
|
@app.get("/", response_class=HTMLResponse)
|
|
async def index():
|
|
return HTML
|
|
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok", "machines": len(machines)}
|
|
|
|
|
|
@app.get("/api/machines")
|
|
async def get_machines():
|
|
return machines
|
|
|
|
|
|
@app.websocket("/ws")
|
|
async def websocket_endpoint(websocket: WebSocket, key: str = Query(default="")):
|
|
# API key validation for collectors (browsers don't need key)
|
|
# Check if this looks like a collector (will send metrics) or browser (will receive)
|
|
# We validate key only when metrics are received, allowing browsers to connect freely
|
|
|
|
await websocket.accept()
|
|
connections.append(websocket)
|
|
client = websocket.client.host if websocket.client else "unknown"
|
|
log.info(f"WebSocket connected: {client}")
|
|
|
|
try:
|
|
# Send current state to new connection
|
|
for machine_id, data in machines.items():
|
|
await websocket.send_json(
|
|
{"type": "metrics", "machine_id": machine_id, **data}
|
|
)
|
|
|
|
# Main loop
|
|
while True:
|
|
try:
|
|
msg = await asyncio.wait_for(websocket.receive_text(), timeout=30)
|
|
data = json.loads(msg)
|
|
|
|
if data.get("type") == "metrics":
|
|
# Validate API key for metric submissions
|
|
if API_KEY and key != API_KEY:
|
|
log.warning(f"Invalid API key from {client}")
|
|
await websocket.close(code=4001, reason="Invalid API key")
|
|
return
|
|
|
|
# Handle both formats:
|
|
# 1. Direct: {"type": "metrics", "machine_id": "...", "cpu": ...}
|
|
# 2. Nested (from gateway): {"type": "metrics", "data": {...}, "timestamp": "..."}
|
|
if "data" in data and isinstance(data["data"], dict):
|
|
# Nested format from gateway forwarding
|
|
payload = data["data"]
|
|
machine_id = payload.get("machine_id", "unknown")
|
|
# Extract metrics from nested structure
|
|
metrics = payload.get("metrics", {})
|
|
metric_data = {
|
|
"type": "metrics",
|
|
"machine_id": machine_id,
|
|
"hostname": payload.get("hostname", ""),
|
|
"timestamp": data.get("timestamp"),
|
|
}
|
|
# Flatten metrics for dashboard display
|
|
for key_name, value in metrics.items():
|
|
metric_data[key_name.lower()] = value
|
|
machines[machine_id] = metric_data
|
|
log.debug(f"Metrics (forwarded) from {machine_id}")
|
|
else:
|
|
# Direct format from collector
|
|
machine_id = data.get("machine_id", "unknown")
|
|
machines[machine_id] = data
|
|
log.debug(f"Metrics from {machine_id}: cpu={data.get('cpu')}%")
|
|
|
|
# Broadcast to all connected clients
|
|
broadcast_data = machines[machine_id]
|
|
for conn in connections:
|
|
try:
|
|
await conn.send_json(broadcast_data)
|
|
except Exception:
|
|
pass
|
|
|
|
except asyncio.TimeoutError:
|
|
# Send ping to keep connection alive
|
|
await websocket.send_json({"type": "ping"})
|
|
|
|
except WebSocketDisconnect:
|
|
log.info(f"WebSocket disconnected: {client}")
|
|
except Exception as e:
|
|
log.error(f"WebSocket error: {e}")
|
|
finally:
|
|
if websocket in connections:
|
|
connections.remove(websocket)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
log.info("Starting sysmonstm gateway")
|
|
log.info(f" API key: {'configured' if API_KEY else 'not set (open)'}")
|
|
uvicorn.run(app, host="0.0.0.0", port=8080)
|