import discord
import dotenv
import asyncio
from discord import app_commands
import psutil
import time
from other.ikavoidping import get_api_ping
from other.almteshdotnetping import get_api_ping as get_almtesh_ping
from other.snartfrping import get_api_ping as get_snartfr_ping
dotenv.load_dotenv()
TOKEN = dotenv.get_key(".env", "DISCORD_TOKEN")

class Bot(discord.Client):
    def __init__(self):
        super().__init__(intents=discord.Intents.all())
        self.tree = app_commands.CommandTree(self)
        self.startup_timestamp = time.time()
 
    async def setup_hook(self):
        await asyncio.sleep(1)
        await self.tree.sync()
bot = Bot()

@bot.event
async def on_ready():
    print(f'Logged in as {bot.user}')

@bot.tree.command(name="status", description="Get the bot's current status")
async def status(interaction: discord.Interaction):
    cpu_usage = await asyncio.to_thread(psutil.cpu_percent, interval=1, percpu=True)
    avg_cpu_usage = sum(cpu_usage) / len(cpu_usage)
    memory_info = psutil.virtual_memory()
    used_memory = memory_info.used
    total_memory = memory_info.total
    discord_latency = bot.latency
    uptime = time.time() - bot.startup_timestamp
    server_uptime = psutil.boot_time()
    ikavoid_ping, almtesh_ping, snartfr_ping = await asyncio.gather(
        get_api_ping(), get_almtesh_ping(), get_snartfr_ping()
    )
    await interaction.response.send_message(
        f"CPU Usage: {avg_cpu_usage:.2f}%\n"
        f"Per-Core CPU Usage: {', '.join(f'{usage}%' for usage in cpu_usage)}\n"
        f"Memory Usage: {memory_info.percent}% ({used_memory / (1024**2):.2f} MB / {total_memory / (1024**2):.2f} MB)\n"
        f"Bot Latency: {discord_latency*1000:.2f} ms\n"
        f"Uptime: {int(uptime // 86400)}d {int((uptime % 86400) // 3600)}h {int((uptime % 3600) // 60)}m {int(uptime % 60)}s\n"
        f"ikavoid.fr API Ping: {ikavoid_ping:.2f} ms\n"
        f"almtesh.net Ping: {almtesh_ping:.2f} ms\n"
        f"snart.fr Ping: {snartfr_ping:.2f} ms\n"
        f"Server Uptime: {int((time.time() - server_uptime) // 86400)}d {int(((time.time() - server_uptime) % 86400) // 3600)}h {int(((time.time() - server_uptime) % 3600) // 60)}m {int((time.time() - server_uptime) % 60)}s"
        
    )


bot.run(TOKEN)