import os
import requests
import discord
from discord.ext import commands
from discord import app_commands
from dotenv import load_dotenv
import io
import asyncio

load_dotenv()
TOKEN = os.getenv("DISCORD_TOKEN")
GUILD_ID = discord.Object(id=1131317261202899034)
JELLYFIN_URL = os.getenv("JELLYFIN_URL")
JELLYFIN_API_KEY = os.getenv("JELLYFIN_API_KEY")

headers= {
    "X-Emby-Token": JELLYFIN_API_KEY
}

intents = discord.Intents.default()
intents.message_content = True
intents.voice_states = True

class JellyfinBot(commands.Bot):
    def __init__(self):
        super().__init__(command_prefix="/", intents=intents)

    async def setup_hook(self):
        self.tree.copy_global_to(guild=GUILD_ID)
        await self.tree.sync(guild=GUILD_ID)

bot = JellyfinBot()

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

@bot.tree.command(name="search", description="Search for a song on Jellyfin")
@app_commands.describe(title="Song title to search for")
async def search(interaction: discord.Interaction, title: str):
    await interaction.response.defer(ephemeral=True)
    url = f"{JELLYFIN_URL}/Items"
    params = {
        "searchTerm": title,
        "Recursive": True,
        "IncludeItemTypes": "Audio",
        "Limit": 5
    }
    try:
        response = requests.get(url, headers=headers, params=params)
        response.raise_for_status()
        data = response.json()
        items = data.get("Items", [])
        if not items:
            await interaction.followup.send(f"❌ No song found matching `{title}`.")
            return
        
        lines = []
        for item in items:
            title = item.get("Name")
            artist = item.get("AlbumArtist", ["Unknown Artist"])
            lines.append(f"✅ **{title}** by *{artist}*")
        await interaction.followup.send("\n".join(lines))
    
    except requests.RequestException as e:
        await interaction.followup.send("⚠️ Failed to contact Jellyfin server.")
        print(f"Error: {e}")


bot.run(TOKEN)