runabot/Christmas/Cogs/Commands_Music.py

from discord.utils import basic_autocomplete
from discord import Member, SlashCommandGroup, Option, AutocompleteContext, Color
from discord.ext.commands import Cog, cooldown, BucketType, command, has_permissions, bot_has_permissions, Context, guild_only, bot_has_guild_permissions, check
import wavelink
import random
import pendulum
from Christmas.Database import database
from Christmas.UI.Embed import Music_Embed, ChristmasEmbed
from Christmas.config import ChristmasConfig
    
class CMusic(Cog):
    def __init__(self, bot):
        self.bot = bot
        self.config = ChristmasConfig()
        
    MUSIC = SlashCommandGroup(name="음악", description="음악을 재생합니다.")
   # 
    @MUSIC.command(name="재생", description="LOFI 음악을 재생합니다.")
    @cooldown(1, 10, BucketType.user)
    @guild_only()
    async def _play(self, ctx: Context):
        if not await database.get_guild(ctx.guild.id): return await ctx.respond(embed=ChristmasEmbed(title="❌ 에러!", description="서버가 가입되어있지 않아요! 서버를 가입해주세요!", color=Color.red()),ephemeral=True)
        await ctx.defer()
        get_guild = await database.get_guild(ctx.guild.id)
        if not get_guild["music"] == True: return await ctx.respond(embed=Music_Embed.music_notenable())
        if get_guild["admin_run"] == True:
            if not ctx.author.guild_permissions.administrator: return await ctx.respond(embed=ChristmasEmbed(title="❌ 에러!", description="관리자만 사용할수 있는 명령어에요!", color=Color.red()),ephemeral=True)
        if ctx.author.voice == None or ctx.author.voice.channel == None: return await ctx.respond(embed=Music_Embed.author_not_voice())
        player = None
        node = wavelink.Pool.get_node()
        if node.get_player(ctx.guild.id) == None:
            player = await ctx.author.voice.channel.connect(cls=wavelink.Player)
        else:  
            player = node.get_player(ctx.guild.id)
        try:
            query = None
            if pendulum.now() > pendulum.datetime(2023, 12, 24, 12, 0, 0):
                query = await wavelink.Playable.search(self.config.CHRISTMAS, source=wavelink.TrackSource.YouTube)
            else:
                query = await wavelink.Playable.search(random.choice(self.config.LOFI), source=wavelink.TrackSource.YouTube)
            await player.play(query[0])    
            await ctx.respond(embed=Music_Embed.music_play())
        except Exception as e:
            print(e)
        
        
    @MUSIC.command(name="정지", description="음악을 정지합니다.")
    @cooldown(1, 10, BucketType.user)
    @guild_only()
    async def _stop(self, ctx: Context):
        if not await database.get_guild(ctx.guild.id): return await ctx.respond(embed=ChristmasEmbed(title="❌ 에러!", description="서버가 가입되어있지 않아요! 서버를 가입해주세요!", color=Color.red()),ephemeral=True)
        await ctx.defer()
        get_guild = await database.get_guild(ctx.guild.id)
        if not get_guild["music"] == True: return await ctx.respond(embed=Music_Embed.music_notenable())
        if get_guild["admin_run"] == True:
            if not ctx.author.guild_permissions.administrator: return await ctx.respond(embed=ChristmasEmbed(title="❌ 에러!", description="관리자만 사용할수 있는 명령어에요!", color=Color.red()),ephemeral=True)
        if ctx.author.voice == None or ctx.author.voice.channel == None: return await ctx.respond(embed=Music_Embed.author_not_voice())
        node = wavelink.Pool.get_node()
        player = node.get_player(ctx.guild.id)
        if player == None: return await ctx.respond(embed=Music_Embed.music_not_playing(), ephemeral=True)
        await player.stop()
        # 음악 채널에서 나감
        await player.disconnect()
        await ctx.respond(embed=Music_Embed.music_stop())
    
    @MUSIC.command(name="설정", description="음악 설정을 변경합니다.")
    @cooldown(1, 10, BucketType.user)
    @guild_only()
    async def _setting(self, ctx: Context, setting: Option(str, name="설정", description="설정할 항목을 선택해주세요!", choices=["음악 활성화", "관리자만 재생 가능"], required=True), value: Option(str, name="값", description="설정할 값을 선택해주세요!", choices=["켜기", "끄기"], required=True)):
        if not await database.get_guild(ctx.guild.id): return await ctx.respond(embed=ChristmasEmbed(title="❌ 에러!", description="서버가 가입되어있지 않아요! 서버를 가입해주세요!", color=Color.red()),ephemeral=True)
        await ctx.defer()
        if not ctx.author.guild_permissions.administrator: return await ctx.respond(embed=ChristmasEmbed(title="❌ 에러!", description="관리자만 사용할수 있는 명령어에요!", color=Color.red()),ephemeral=True)
        if setting == "관리자만 재생 가능":
            if value == "켜기":
                await database.update_music_setting(ctx.guild.id, True, "admin_run")
                await ctx.respond(embed=Music_Embed.music_setting(True, "admin_run"), ephemeral=True)
            else:
                await database.update_music_setting(ctx.guild.id, False, "admin_run")
                await ctx.respond(embed=Music_Embed.music_setting(False, "admin_run"), ephemeral=True)
        if setting == "음악 활성화":
            if value == "켜기":
                await database.update_music_setting(ctx.guild.id, True, "music")
                await ctx.respond(embed=Music_Embed.music_setting(True, "music"), ephemeral=True)
            else:
                await database.update_music_setting(ctx.guild.id, False, "music")
                await ctx.respond(embed=Music_Embed.music_setting(False, "music"), ephemeral=True)
def setup(bot):
    bot.add_cog(CMusic(bot))