import asyncio import collections from typing import Any, Optional import disnake import yt_dlp import constants ytdl = yt_dlp.YoutubeDL(constants.YTDL_OPTIONS) class YTDLSource(disnake.PCMVolumeTransformer): def __init__( self, source: disnake.AudioSource, *, data: dict[str, Any], volume: float = 0.5 ): super().__init__(source, volume) self.title = data.get("title") self.original_url = data.get("original_url") self.duration = data.get("duration") @classmethod async def from_url( cls, url, *, loop: Optional[asyncio.AbstractEventLoop] = None, stream: bool = False, ): loop = loop or asyncio.get_event_loop() data: Any = await loop.run_in_executor( None, lambda: ytdl.extract_info(url, download=not stream) ) if "entries" in data: data = data["entries"][0] return cls( disnake.FFmpegPCMAudio( data["url"] if stream else ytdl.prepare_filename(data), before_options="-vn -reconnect 1", ), data=data, ) def __repr__(self): return f" duration={self.duration}>" def __str__(self): return self.__repr__() class QueuedPlayer: def __init__(self): self.queue = collections.deque() self.current = None def queue_pop(self): popped = self.queue.popleft() self.current = popped return popped def queue_add(self, item): self.queue.append(item) def queue_add_front(self, item): self.queue.appendleft(item) def __repr__(self): return f"" def __str__(self): return self.__repr__() class QueuedSong: def __init__(self, player, queuer): self.player = player self.queuer = queuer def format(self, with_queuer=False, hide_preview=False) -> str: return ( f"[`{self.player.title}`]({'<' if hide_preview else ''}{self.player.original_url}{'>' if hide_preview else ''}) [{self.format_duration(self.player.duration) if self.player.duration else 'live'}]" + (f" (<@{self.queuer}>)" if with_queuer else "") ) def format_duration(self, duration: int) -> str: hours, duration = divmod(duration, 3600) minutes, duration = divmod(duration, 60) segments = [hours, minutes, duration] if len(segments) == 3 and segments[0] == 0: del segments[0] return f"{':'.join(f'{s:0>2}' for s in segments)}" def __repr__(self): return f"" def __str__(self): return self.__repr__() def __reload_module__(): global ytdl ytdl = yt_dlp.YoutubeDL(constants.YTDL_OPTIONS)