88 lines
5.7 KiB
Python
88 lines
5.7 KiB
Python
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)) |