first commit

bot/api/routes/admin.py

@@ -0,0 +1,114 @@
+from fastapi import APIRouter, Depends, HTTPException
+from api.dependencies import get_bot
+from api.schemas import AdminStats, AdminNodeStatus, AdminConfig, AdminConfigUpdate
+from typing import TYPE_CHECKING, List
+import os
+import aiosqlite
+
+if TYPE_CHECKING:
+    from core.zyrox import zyrox
+
+router = APIRouter()
+
+CONFIG_DB = "db/admin_config.db"
+
+async def init_db():
+    async with aiosqlite.connect(CONFIG_DB) as db:
+        await db.execute("CREATE TABLE IF NOT EXISTS config (key TEXT PRIMARY KEY, value TEXT)")
+        # Default values
+        await db.execute("INSERT OR IGNORE INTO config (key, value) VALUES ('maintenance_mode', 'false')")
+        await db.execute("INSERT OR IGNORE INTO config (key, value) VALUES ('global_notification', '')")
+        await db.commit()
+
+import psutil
+import time
+
+@router.get("/stats", response_model=AdminStats)
+async def get_admin_stats(bot: "zyrox" = Depends(get_bot)):
+    # Calculate DB size and shard info
+    total_size: float = 0.0
+    db_count = 0
+    db_dir = "db"
+    if os.path.exists(db_dir):
+        for f in os.listdir(db_dir):
+            if f.endswith(".db"):
+                total_size += float(os.path.getsize(os.path.join(db_dir, f)))
+                db_count += 1
+    
+    mb_size = total_size / (1024 * 1024)
+    db_size_str = f"{mb_size:.2f} MB"
+    if mb_size > 1024:
+        db_size_str = f"{(mb_size / 1024):.2f} GB"
+
+    # System Metrics
+    process = psutil.Process(os.getpid())
+    # Use a non-blocking interval check or global state for CPU
+    cpu_usage = psutil.cpu_percent() 
+    ram_raw = process.memory_info().rss
+    ram_mb = ram_raw / (1024 * 1024)
+    
+    total_commands = len(bot.commands)
+    loaded_cogs = len(bot.cogs or {})
+
+    # Node Healths
+    nodes = [
+        AdminNodeStatus(
+            name="Primary API Cluster", 
+            status="Healthy", 
+            load=f"CPU: {cpu_usage}% | RAM: {ram_mb:.1f}MB", 
+            icon="Globe"
+        ),
+        AdminNodeStatus(
+            name="Database Shards", 
+            status="Healthy" if db_count > 0 else "Warning", 
+            load=f"{db_count} SQLite DBs | {db_size_str}", 
+            icon="Database"
+        ),
+        AdminNodeStatus(
+            name="Bot Microservices", 
+            status="Healthy" if bot.is_ready() else "Booting", 
+            load=f"{loaded_cogs} Modules", 
+            icon="Cpu"
+        ),
+        AdminNodeStatus(
+            name="Auth Sockets", 
+            status="Healthy", 
+            load=f"Shard: {bot.shard_count} | Latency: {round(bot.latency * 1000)}ms", 
+            icon="Lock"
+        )
+    ]
+
+    total_members = sum(g.member_count or 0 for g in bot.guilds)
+
+    return AdminStats(
+        total_users=str(total_members),
+        active_servers=str(len(bot.guilds)),
+        api_latency=f"{round(bot.latency * 1000, 2)}ms",
+        db_size=db_size_str,
+        nodes=nodes
+    )
+
+@router.get("/config", response_model=AdminConfig)
+async def get_admin_config():
+    await init_db()
+    async with aiosqlite.connect(CONFIG_DB) as db:
+        async with db.execute("SELECT value FROM config WHERE key = 'maintenance_mode'") as cursor:
+            mm = await cursor.fetchone()
+        async with db.execute("SELECT value FROM config WHERE key = 'global_notification'") as cursor:
+            gn = await cursor.fetchone()
+            
+    return AdminConfig(
+        maintenance_mode=mm[0].lower() == 'true' if mm else False,
+        global_notification=gn[0] if gn else None
+    )
+
+@router.patch("/config")
+async def patch_admin_config(data: AdminConfigUpdate):
+    await init_db()
+    async with aiosqlite.connect(CONFIG_DB) as db:
+        if data.maintenance_mode is not None:
+            await db.execute("UPDATE config SET value = ? WHERE key = 'maintenance_mode'", (str(data.maintenance_mode).lower(),))
+        if data.global_notification is not None:
+            await db.execute("UPDATE config SET value = ? WHERE key = 'global_notification'", (data.global_notification,))
+        await db.commit()
+    return {"status": "success"}