199 lines
6.2 KiB
Python
199 lines
6.2 KiB
Python
"""Minimal sysmonstm gateway - standalone mode without dependencies."""
|
|
|
|
from fastapi import FastAPI, WebSocket, WebSocketDisconnect
|
|
from fastapi.responses import HTMLResponse
|
|
import json
|
|
import asyncio
|
|
from datetime import datetime
|
|
|
|
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 ws = new WebSocket(`wss://${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);
|
|
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];
|
|
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>Updated</span><span>${new Date(m.timestamp).toLocaleTimeString()}</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.websocket("/ws")
|
|
async def websocket_endpoint(websocket: WebSocket):
|
|
await websocket.accept()
|
|
connections.append(websocket)
|
|
try:
|
|
# Send current state
|
|
for machine_id, data in machines.items():
|
|
await websocket.send_json({"type": "metrics", "machine_id": machine_id, **data})
|
|
# Keep alive
|
|
while True:
|
|
try:
|
|
msg = await asyncio.wait_for(websocket.receive_text(), timeout=30)
|
|
data = json.loads(msg)
|
|
if data.get("type") == "metrics":
|
|
machine_id = data.get("machine_id", "unknown")
|
|
machines[machine_id] = {**data, "timestamp": datetime.utcnow().isoformat()}
|
|
# Broadcast to all
|
|
for conn in connections:
|
|
try:
|
|
await conn.send_json({"type": "metrics", "machine_id": machine_id, **machines[machine_id]})
|
|
except:
|
|
pass
|
|
except asyncio.TimeoutError:
|
|
await websocket.send_json({"type": "ping"})
|
|
except WebSocketDisconnect:
|
|
pass
|
|
finally:
|
|
connections.remove(websocket)
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run(app, host="0.0.0.0", port=8080)
|