mirror of
https://github.com/Rainyy21/lol_discordbot
synced 2026-10-10 23:00:32 -04:00
Rainy21
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
import discord
|
|
from discord import app_commands
|
|
from discord.ext import commands
|
|
import os
|
|
from database.db import save_user
|
|
from services.riot_api import get_account
|
|
|
|
class LoginCog(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
async def _handle_login(self, interaction: discord.Interaction, summoner: str):
|
|
await interaction.response.defer(ephemeral=True)
|
|
|
|
if "#" not in summoner:
|
|
await interaction.followup.send("❌ Use the format `Name#TAG`", ephemeral=True)
|
|
return
|
|
|
|
try:
|
|
game_name, tag_line = summoner.split("#", 1)
|
|
except ValueError:
|
|
await interaction.followup.send("❌ Invalid format. Use `Name#TAG`", ephemeral=True)
|
|
return
|
|
|
|
discord_id = str(interaction.user.id)
|
|
|
|
# Use the service instead of direct API call
|
|
data = await get_account(game_name, tag_line)
|
|
|
|
if data is None:
|
|
await interaction.followup.send("❌ Summoner not found. Check your Riot ID.", ephemeral=True)
|
|
return
|
|
|
|
if isinstance(data, dict) and "error" in data:
|
|
if data["error"] == 401:
|
|
await interaction.followup.send(
|
|
"❌ Riot API error (401): Unauthorized. "
|
|
"This usually means the bot's API key has expired. "
|
|
"Please notify the bot owner to refresh it at developer.riotgames.com.",
|
|
ephemeral=True
|
|
)
|
|
else:
|
|
await interaction.followup.send(
|
|
f"❌ Riot API error ({data['error']}). Try again later.",
|
|
ephemeral=True
|
|
)
|
|
return
|
|
|
|
puuid = data["puuid"]
|
|
save_user(discord_id, puuid, game_name, tag_line)
|
|
|
|
await interaction.followup.send(
|
|
f"✅ Linked **{game_name}#{tag_line}** to your Discord account!",
|
|
ephemeral=True
|
|
)
|
|
@app_commands.command(name="login", description="Link your Riot account")
|
|
@app_commands.describe(summoner="Your Riot ID (e.g. Name#TAG)")
|
|
async def login(self, interaction: discord.Interaction, summoner: str):
|
|
await self._handle_login(interaction, summoner)
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(LoginCog(bot))
|
|
|