115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
import asyncio
|
|
from datetime import datetime, timezone
|
|
from typing import Dict
|
|
from services.runner import run_cmd
|
|
|
|
|
|
def container_uptime(started_at: str) -> str:
|
|
"""
|
|
started_at: 2026-02-06T21:14:33.123456789Z
|
|
"""
|
|
try:
|
|
start = datetime.fromisoformat(
|
|
started_at.replace("Z", "+00:00")
|
|
).astimezone(timezone.utc)
|
|
delta = datetime.now(timezone.utc) - start
|
|
days = delta.days
|
|
hours = delta.seconds // 3600
|
|
minutes = (delta.seconds % 3600) // 60
|
|
|
|
if days > 0:
|
|
return f"{days}d {hours}h"
|
|
if hours > 0:
|
|
return f"{hours}h {minutes}m"
|
|
return f"{minutes}m"
|
|
except Exception:
|
|
return "unknown"
|
|
|
|
|
|
async def build_docker_map(cfg) -> Dict[str, str]:
|
|
docker_cfg = cfg.get("docker", {})
|
|
|
|
result: Dict[str, str] = {}
|
|
|
|
# 1. autodiscovery
|
|
if docker_cfg.get("autodiscovery"):
|
|
rc, raw = await run_cmd(
|
|
["sudo", "docker", "ps", "--format", "{{.Names}}"],
|
|
timeout=20
|
|
)
|
|
if rc == 0:
|
|
names = raw.splitlines()
|
|
patterns = docker_cfg.get("match", [])
|
|
for name in names:
|
|
if any(p in name for p in patterns):
|
|
result[name] = name
|
|
|
|
# 2. aliases override
|
|
aliases = docker_cfg.get("aliases", {})
|
|
for alias, real in aliases.items():
|
|
result[alias] = real
|
|
|
|
return result
|
|
|
|
|
|
async def discover_containers(cfg) -> Dict[str, str]:
|
|
"""
|
|
returns: alias -> real container name
|
|
"""
|
|
docker_cfg = cfg.get("docker", {})
|
|
result: Dict[str, str] = {}
|
|
|
|
# --- autodiscovery ---
|
|
if docker_cfg.get("autodiscovery"):
|
|
rc, raw = await run_cmd(
|
|
["sudo", "docker", "ps", "--format", "{{.Names}}"],
|
|
timeout=20
|
|
)
|
|
|
|
if rc == 0:
|
|
found = raw.splitlines()
|
|
|
|
label = docker_cfg.get("label")
|
|
patterns = docker_cfg.get("match", [])
|
|
|
|
for name in found:
|
|
# label-based discovery
|
|
if label:
|
|
key, val = label.split("=", 1)
|
|
rc2, lbl = await run_cmd([
|
|
"sudo", "docker", "inspect",
|
|
"-f", f"{{{{ index .Config.Labels \"{key}\" }}}}",
|
|
name
|
|
])
|
|
if rc2 == 0 and lbl.strip() == val:
|
|
result[name] = name
|
|
continue
|
|
|
|
# name-pattern discovery
|
|
if any(p in name for p in patterns):
|
|
result[name] = name
|
|
|
|
# --- manual aliases ALWAYS override ---
|
|
aliases = docker_cfg.get("aliases", {})
|
|
for alias, real in aliases.items():
|
|
result[alias] = real
|
|
|
|
return result
|
|
|
|
|
|
async def docker_watchdog(container_map, notify, bot, chat_id):
|
|
last = {}
|
|
while True:
|
|
for alias, real in container_map.items():
|
|
rc, state = await run_cmd(
|
|
["docker", "inspect", "-f", "{{.State.Status}}", real],
|
|
timeout=10
|
|
)
|
|
if rc != 0:
|
|
state = "error"
|
|
state = state.strip()
|
|
if last.get(alias) != state:
|
|
await notify(bot, chat_id, f"🐳 {alias}: {state}")
|
|
last[alias] = state
|
|
await asyncio.sleep(120)
|