complete the bot

2026-10-11 00:40:31 -04:00 · 2026-04-22 00:49:43 -04:00
parent b2f09d6cca
commit c6257c5260
8 changed files with 845 additions and 42 deletions
@@ -0,0 +1,110 @@
+from discord import app_commands
+from discord.ext import commands
+import aiohttp
+import os
+
+from database.db import get_user
+
+RIOT_API_KEY = os.getenv("LEAGUEAPI")
+REGION = "na1"
+
+class ProfileCog(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @app_commands.command(name="profile", description="Show your LoL rank, level, and win rate")
+    async def profile(self, interaction: discord.Interaction):
+        await interaction.response.defer(ephemeral = False)
+        
+        # get user from the db
+        discord_id = str(interaction.user.id)
+        user = get_user(discord_id)
+
+        #check if user login
+        if not user:
+            await interaction.followup.send(
+                "❌ You haven't linked your Riot account yet. Use `/login` first."
+            )
+            return
+
+        _, puuid, game_name, tag_line = user
+        headers = {"X-Riot-Token": RIOT_API_KEY}
+        
+        async with aiohttp.ClientSession() as session:
+            # 1. Get summoner data (level) via PUUID
+            summoner = await fetch(
+                session,
+                f"https://{REGION}.api.riotgames.com/lol/summoner/v4/summoners/by-puuid/{puuid}",
+                headers
+            )
+            if not summoner:
+                await interaction.followup.send("❌ Could not fetch summoner data.")
+                return
+
+            summoner_id = summoner["id"]
+            level = summoner["summonerLevel"]
+
+            # 2. Get rank & LP (League-v4)
+            leagues = await fetch(
+                session,
+                f"https://{REGION}.api.riotgames.com/lol/league/v4/entries/by-summoner/{summoner_id}",
+                headers
+            )
+        # Parse ranked solo queue data
+        rank_info = get_solo_rank(leagues)
+
+        # Build win rate
+        wins = rank_info.get("wins", 0)
+        losses = rank_info.get("losses", 0)
+        total = wins + losses
+        win_rate = f"{round(wins / total * 100)}%" if total > 0 else "N/A"
+
+        # Build Discord embed
+        embed = discord.Embed(
+            title=f"📊 {game_name}#{tag_line}",
+            color=0x1a78c2
+        )
+        embed.add_field(name="🎮 Level", value=str(level), inline=True)
+        embed.add_field(
+            name="🏆 Rank",
+            value=format_rank(rank_info),
+            inline=True
+        )
+        embed.add_field(name="📈 Win Rate", value=win_rate, inline=True)
+        embed.add_field(
+            name="📋 Record",
+            value=f"{wins}W / {losses}L" if total > 0 else "Unranked",
+            inline=True
+        )
+        embed.set_footer(text="Powered by Riot Games API")
+
+        await interaction.followup.send(embed=embed)
+# ── Helpers ────────────────────────────────────────────────────────────────────
+#try to get the information
+async def fetch(session:aiohttp.ClientSession, url: str, headers: dict):
+    async with session.get(url, headers=headers) as resp:
+        if resp.status != 200:
+            return None
+        return await resp.json()
+
+def get_solo_rank(leagues: list) -> dict:
+    """Extract solo queue entry from league data."""
+    if not leagues:
+        return {}
+    for entry in leagues:
+        if entry.get("queueType") == "RANKED_SOLO_5x5":
+            return entry
+    return {}
+
+
+def format_rank(rank_info:dict) -> str:
+    """Format rank into a readable string."""
+    if not rank_info:
+        return "Unranked"
+    tier = rank_info.get("tier", "Unranked").capitalize()
+    division = rank_info.get("rank", "")
+    lp = rank_info.get("leaguePoints", 0)
+    return f"{tier} {division} — {lp} LP"
+
+async def setup(bot):
+    await bot.add_cog(ProfileCog(bot))