import discord
import yt_dlp as youtube_dl
import asyncio
import uuid
import os

ytdl_format_options = {
    'format': 'bestaudio/best',
    'outtmpl': '%(extractor)s-%(id)s-%(title)s.%(ext)s',
    'restrictfilenames': True,
    'noplaylist': True,
    'nocheckcertificate': True,
    'ignoreerrors': False,
    'logtostderr': False,
    'quiet': True,
    'no_warnings': True,
    'default_search': 'auto',
    'source_address': '0.0.0.0',
}

ffmpeg_options = {
    'options': '-vn',
    'before_options': '-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5'
}

ytdl = youtube_dl.YoutubeDL(ytdl_format_options)


async def get_file_duration(filepath):
    try:
        proc = await asyncio.create_subprocess_exec(
            'ffprobe', '-v', 'error', '-show_entries', 'format=duration',
            '-of', 'default=noprint_wrappers=1:nokey=1', filepath,
            stdout=asyncio.subprocess.PIPE,
            stderr=asyncio.subprocess.PIPE
        )
        stdout, _ = await proc.communicate()
        if proc.returncode == 0:
            return float(stdout.decode().strip())
    except Exception as e:
        print(f"Error getting duration with ffprobe: {e}")
    return 0.0


class YTDLSource(discord.PCMVolumeTransformer):
    def __init__(self, source, *, data, volume=0.5):
        super().__init__(source, volume)
        self.data = data
        self.title = data.get('title')
        self.url = data.get('url')
        self.id = uuid.uuid4().hex  # Unique track identifier
        self.duration = data.get('duration')  # Duration in seconds
        self.elapsed = 0.0  # Elapsed time in seconds
        self.plays_left = data.get('plays_left', None)  # Remaining plays in queue

    def read(self):
        data = super().read()
        if data:
            self.elapsed += 0.02  # 20ms frames read
        return data

    @classmethod
    async def from_url(cls, url, *, loop=None, stream=False):
        loop = loop or asyncio.get_event_loop()
        data = await loop.run_in_executor(None, lambda: ytdl.extract_info(url, download=not stream))

        if 'entries' in data:
            data = data['entries'][0]

        filename = data['url'] if stream else ytdl.prepare_filename(data)
        return cls(discord.FFmpegPCMAudio(filename, **ffmpeg_options), data=data)

    @classmethod
    async def from_file(cls, filepath, *, loop=None, title=None):
        loop = loop or asyncio.get_event_loop()
        duration = await get_file_duration(filepath)
        filename_only = os.path.basename(filepath)
        display_title = title or os.path.splitext(filename_only)[0]

        data = {
            'title': display_title,
            'url': filepath,
            'webpage_url': None,
            'duration': duration,
            'is_local': True,
            'filepath': filepath
        }

        local_ffmpeg_options = {
            'options': '-vn'
        }
        return cls(discord.FFmpegPCMAudio(filepath, **local_ffmpeg_options), data=data)