import discord
from discord.ext import commands
from discord import FFmpegPCMAudio

import yt_dlp

# Set up intents and bot
intents = discord.Intents.default()
intents.message_content = True  # Needed to read .play/.stop commands
bot = commands.Bot(command_prefix="..", intents=intents)

# Global variable to hold the current voice clients
voice_clients = {}

# Sets the default volume to 20%, to make headphone users survive
# the volume does not change when the bot is already playing, only new audio
current_volume = 0.20

@bot.event
async def on_ready():
    # Construct the invite URL dynamically
    application_info = await bot.application_info()
    client_id = application_info.id
    invite_url = f"https://discord.com/oauth2/authorize?client_id={client_id}&permissions=3145728&scope=bot"
    
    print(f"Invite URL: {invite_url}")

@bot.command()
async def play(ctx, url: str):
    guild_id = ctx.guild.id

    if not ctx.author.voice:
        await ctx.send("lépjél már be egy kurva csatornába te szerencsétlen fasz")
        return

    if guild_id not in voice_clients or not voice_clients[guild_id].is_connected():
        voice_clients[guild_id] = await ctx.author.voice.channel.connect()

    vc = voice_clients[guild_id]

    if vc.is_playing():
        vc.stop()

    try:
        # Load blocked keywords from file
        with open("blocked_keywords.txt", "r", encoding="utf-8") as f:
            blocked_keywords = [line.strip().lower() for line in f if line.strip()]

        # YTDLP options
        ydl_opts = {
            "format": "bestaudio/best",
            "quiet": True,
            "noplaylist": True,
            "default_search": "auto",
        }

        # Use yt-dlp to extract audio URL and check for blocked keywords
        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
            info = ydl.extract_info(url, download=False)
            title = info["title"].lower()
            if any(keyword in title for keyword in blocked_keywords):
                await ctx.send(f"na ezt a fost nem jatszom le: `{info['title']}`")
                return
            audio_url = info["url"]
            print("Audio URL:", audio_url)

        # Set up FFmpeg options with the current volume
        ffmpeg_options = {
            "before_options": "-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5 -protocol_whitelist file,http,https,tcp,tls,crypto",
            "options": f"-vn -af volume={current_volume}"
        }

        # Play the audio
        vc.play(
            FFmpegPCMAudio(audio_url, **ffmpeg_options),
            after=lambda e: print(f"Finished playing: {e}")
        )
        await ctx.send(f"most ez a fos szól: {info['title']}")

    except Exception as e:
        await ctx.send(f"most ezzel mi a kurva anyámat kezdjek?: {e}")

@bot.command()
async def stop(ctx):
    guild_id = ctx.guild.id

    if guild_id in voice_clients and voice_clients[guild_id].is_playing():
        voice_clients[guild_id].stop()
        await ctx.send("hát akkor ne hallgassad basszalak szájba")
    else:
        await ctx.send("nem is szól semmi te buzeráns fasz")

#Does not work in multi-server setup, needs fix
@bot.command()
async def volume(ctx, vol: float):
    global current_volume
    if 0.0 <= vol <= 2.0:
        current_volume = vol
        await ctx.send(f"hangerő most {vol*100:.0f}%")
    else:
        await ctx.send("0.0 és 2.0 között add meg te szerencsétlen")

@bot.command()
async def leave(ctx):
    guild_id = ctx.guild.id

    if guild_id in voice_clients and voice_clients[guild_id].is_connected():
        await voice_clients[guild_id].disconnect()
        del voice_clients[guild_id]
        await ctx.send("akkor csá gecik")
    else:
        await ctx.send("nem is vagyok voice channelben basszam ki a szádat")

# Run the bot
TOKEN = open("TOKEN").read().strip()
bot.run(TOKEN)