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"}

bot/api/routes/bot.py

@@ -0,0 +1,39 @@
+from fastapi import APIRouter, Depends
+from api.dependencies import get_bot
+from api.schemas import BotInfo, BotStatus
+from typing import TYPE_CHECKING
+from utils.config import *
+
+
+if TYPE_CHECKING:
+    from core.zyrox import zyrox
+
+router = APIRouter()
+
+@router.get("/status", response_model=BotStatus, summary="Get bot status", description="Returns real-time health metrics, latency, and scale information.")
+async def get_status(bot: "zyrox" = Depends(get_bot)):
+    """
+    Returns the live status of the bot.
+    """
+    return BotStatus(
+        user=str(bot.user),
+        id=str(bot.user.id) if bot.user else None,
+        latency=bot.latency * 1000,
+        guild_count=len(bot.guilds),
+        user_count=sum(g.member_count or 0 for g in bot.guilds),
+        shards=bot.shard_count
+    )
+
+@router.get("/info", response_model=BotInfo, summary="Get bot info", description="Returns general information about the bot including command count and user reach.")
+async def get_bot_info(bot: "zyrox" = Depends(get_bot)):
+    """
+    Get general information about the Discord bot.
+    """
+    return BotInfo(
+        name=bot.user.name if bot.user else BRAND_NAME,
+        id=str(bot.user.id) if bot.user else None,
+        guilds=len(bot.guilds),
+        users=sum(g.member_count or 0 for g in bot.guilds),
+        commands=len(bot.commands),
+        latency=f"{round(bot.latency * 1000, 2)}ms"
+    )