import shutil
import psutil
from pyrogram import Client, filters
from config import ADMIN
from helper.progress import humanbytes

@Client.on_message(filters.user(ADMIN) & filters.command("usage"))
async def usage_stats(client, message):
    msg = await message.reply_text("<code>Fetching System Stats...</code>")
    
    # CPU Usage
    cpu_usage = psutil.cpu_percent(interval=0.5)
    
    # RAM Usage
    ram = psutil.virtual_memory()
    ram_total = humanbytes(ram.total)
    ram_used = humanbytes(ram.used)
    ram_usage = ram.percent
    
    # Disk Usage
    disk = shutil.disk_usage("/")
    disk_total = humanbytes(disk.total)
    disk_used = humanbytes(disk.used)
    disk_free = humanbytes(disk.free)
    disk_usage = (disk.used / disk.total) * 100
    
    stats = f"""<b><emoji id=5042290883949495533>📊</emoji> System Usage Stats</b>
    
<b><emoji id=5456140674028019486>🖥️</emoji> CPU Usage:</b> <code>{cpu_usage}%</code>
<b><emoji id=5985386442824619877>📟</emoji> RAM Usage:</b> <code>{ram_used}</code> / <code>{ram_total}</code> (<code>{ram_usage}%</code>)
<b><emoji id=5433653135799228968>📁</emoji> Disk Usage:</b> <code>{disk_used}</code> / <code>{disk_total}</code> (<code>{disk_usage:.1f}%</code>)
<b><emoji id=5787432469598835099>💿</emoji> Disk Free:</b> <code>{disk_free}</code>

<b><emoji id=5377834924776627189>⚡️</emoji> Status:</b> <code>Stable</code>"""

    await msg.edit(stats)