mirror of
https://github.com/Matthww/TuneBot.git
synced 2026-09-21 22:57:48 +00:00
72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
import asyncio
|
|
import typing
|
|
|
|
import lavalink
|
|
from discord.channel import TextChannel
|
|
from discord.ext import commands
|
|
|
|
from bot import TuneBot
|
|
from cogs.music import helper
|
|
from cogs.music import track_embed
|
|
from cogs.music.interactions import MusicCommands
|
|
from cogs.music.interactions import QUEUE_SIZE
|
|
from cogs.music.voice_client import LavalinkVoiceClient
|
|
from utils.classes import BaseCog
|
|
from utils.log import logger
|
|
|
|
if typing.TYPE_CHECKING:
|
|
from bot import TuneBot
|
|
|
|
|
|
class MusicCog(BaseCog, name="Music"):
|
|
def __init__(self, bot: TuneBot):
|
|
super().__init__(bot)
|
|
if not hasattr(bot, "lavalink"):
|
|
self.bot.lavalink = self.bot.create_lavalink(bot.user.id)
|
|
|
|
self.bot.lavalink.add_event_hook(self.track_hook)
|
|
|
|
@commands.Cog.listener()
|
|
async def on_ready(self):
|
|
while not self.is_lavalink_ready():
|
|
await asyncio.sleep(1)
|
|
|
|
redis_result = await self.bot.global_autojoin.fetch_channels()
|
|
for guild_id, (voicechannel_id, textchannel_id) in redis_result.items():
|
|
try:
|
|
player = self.bot.lavalink.player_manager.create(guild_id)
|
|
player.store("channel", textchannel_id)
|
|
voice_channel = await self.bot.fetch_channel(voicechannel_id)
|
|
await voice_channel.connect(cls=LavalinkVoiceClient)
|
|
if not player.is_playing:
|
|
await helper.fill_player_queue(self.bot, player, QUEUE_SIZE)
|
|
await player.play()
|
|
except:
|
|
logger.error(f"Failed to autojoin guild {guild_id}.")
|
|
|
|
async def track_hook(self, event: lavalink.Event):
|
|
if isinstance(event, lavalink.events.QueueEndEvent):
|
|
guild_id = event.player.guild_id
|
|
guild = self.bot.get_guild(guild_id)
|
|
await guild.voice_client.disconnect(force=True)
|
|
elif isinstance(event, lavalink.events.TrackEndEvent):
|
|
channel_id = int(event.player.fetch("channel"))
|
|
channel: TextChannel = self.bot.get_channel(channel_id)
|
|
await helper.fill_player_queue(self.bot, event.player, 1)
|
|
event.player.append_history(event.track)
|
|
|
|
if event.reason == "FINISHED":
|
|
queue = event.player.queue[1:]
|
|
current = event.player.queue[0]
|
|
else:
|
|
queue = event.player.queue
|
|
current = event.player.current
|
|
|
|
embed = track_embed.create_track_embed(current, queue, event.player.history)
|
|
await channel.send(embed=embed)
|
|
|
|
|
|
async def setup(bot: "TuneBot"):
|
|
await bot.add_cog(MusicCog(bot))
|
|
bot.tree.add_command(MusicCommands(bot), override=True)
|