feat: Add bot settings commands.

MusicBot/cogs/general.py

@@ -56,6 +56,7 @@ class General(Cog):
                 `help`
                 `like`
                 `queue`
+                `settings`
                 `track`
                 `voice`
                 """
@@ -76,11 +77,17 @@ class General(Cog):
             embed.description += ("Вывести список всех команд.\n```/help```\n"
                                 "Получить информацию о конкретной команде.\n```/help <команда>```")
         elif command == 'like':
-            embed.description += "Добавить трек в плейлист «Мне нравится».\n```/like```"
+            embed.description += "Добавить трек в плейлист «Мне нравится». Пользовательские треки из этого плейлиста игнорируются.\n```/like```"
         elif command == 'queue':
             embed.description += ("Получить очередь треков. По 15 элементов на страницу.\n```/queue get```\n"
                                 "Очистить очередь треков и историю прослушивания. Доступно только если вы единственный в голосовом канале"
                                 "или имеете разрешение управления каналом.\n```/queue clear```\n")
+        elif command == 'settings':
+            embed.description += ("Получить текущие настройки.\n```/settings show```\n"
+                                  "Разрешить или запретить воспроизведение Explicit треков.\n```/settings explicit```\n"
+                                  "Разрешить или запретить создание меню проигрывателя, даже если в канале больше одного человека.\n```/settings menu```\n"
+                                  "Разрешить или запретить голосование.\n```/settings vote <тип голосования>```\n"
+                                  "`Примечание`: Только пользователи с разрешением управления каналом могут менять настройки.")
         elif command == 'track':
             embed.description += ("`Примечание`: Если вы один в голосовом канале или имеете разрешение управления каналом, голосование не начинается.\n\n"
                                 "Переключиться на следующий трек в очереди. \n```/track next```\n"

MusicBot/cogs/settings.py

@@ -0,0 +1,114 @@
+from typing import Literal, cast
+
+import discord
+from discord.ext.commands import Cog
+
+from MusicBot.database import BaseUsersDatabase, BaseGuildsDatabase
+
+def setup(bot):
+    bot.add_cog(Settings(bot))
+
+class Settings(Cog):
+
+    settings = discord.SlashCommandGroup("settings", "Команды для изменения настроек бота.", guild_ids=[1247100229535141899])
+
+    def __init__(self, bot: discord.Bot):
+        self.db = BaseGuildsDatabase()
+        self.users_db = BaseUsersDatabase()
+        self.bot = bot
+
+    @settings.command(name="show", description="Показать текущие настройки бота.")
+    async def show(self, ctx: discord.ApplicationContext) -> None:
+        guild = self.db.get_guild(ctx.guild.id)
+        embed = discord.Embed(title="Настройки бота", color=0xfed42b)
+        
+        explicit = "✅ - Разрешены" if guild['allow_explicit'] else "❌ - Запрещены"
+        menu = "✅ - Всегда доступно" if guild['always_allow_menu'] else "❌ - Если в канале 1 пользователь."
+        
+        vote = "✅ - Переключение" if guild['vote_next_track'] else "❌ - Переключение"
+        vote += "\n✅ - Добавление треков" if guild['vote_add_track'] else "\n❌ - Добавление треков"
+        vote += "\n✅ - Добавление альбомов" if guild['vote_add_album'] else "\n❌ - Добавление альбомов"
+        vote += "\n✅ - Добавление артистов" if guild['vote_add_artist'] else "\n❌ - Добавление артистов"
+        vote += "\n✅ - Добавление плейлистов" if guild['vote_add_playlist'] else "\n❌ - Добавление плейлистов"
+        
+        embed.add_field(name="__Explicit треки__", value=explicit, inline=False)
+        embed.add_field(name="__Проигрыватель__", value=menu, inline=False)
+        embed.add_field(name="__Голосование__", value=vote, inline=False)
+        
+        await ctx.respond(embed=embed, ephemeral=True)
+    
+    @settings.command(name="explicit", description="Разрешить или запретить воспроизведение Explicit треков.")
+    async def explicit(self, ctx: discord.ApplicationContext) -> None:
+        member = cast(discord.Member, ctx.author)
+        if not member.guild_permissions.manage_channels:
+            await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True)
+            return
+
+        guild = self.db.get_guild(ctx.guild.id)
+        self.db.update(ctx.guild.id, {'allow_explicit': not guild['allow_explicit']})
+        await ctx.respond(f"Треки с содержанием не для детей теперь {'разрешены' if not guild['allow_explicit'] else 'запрещены'}.", delete_after=15, ephemeral=True)
+
+    @settings.command(name="menu", description="Разрешить или запретить создание меню проигрывателя, даже если в канале больше одного человека.")
+    async def menu(self, ctx: discord.ApplicationContext) -> None:
+        member = cast(discord.Member, ctx.author)
+        if not member.guild_permissions.manage_channels:
+            await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True)
+            return
+
+        guild = self.db.get_guild(ctx.guild.id)
+        self.db.update(ctx.guild.id, {'always_allow_menu': not guild['always_allow_menu']})
+        await ctx.respond(f"Меню проигрывателя теперь {'можно' if not guild['always_allow_menu'] else 'нельзя'} создавать в каналах с несколькими людьми.", delete_after=15, ephemeral=True)
+
+    @settings.command(name="vote", description="Настроить голосование.")
+    @discord.option(
+        "vote_type",
+        description="Тип голосования.",
+        type=discord.SlashCommandOptionType.string,
+        choices=['+Всё', '-Всё', 'Переключение', '+Трек', '+Альбом', '+Плейлист'],
+        default='Всё'
+    )
+    async def vote(self, ctx: discord.ApplicationContext, vote_type: Literal['+Всё', '-Всё', 'Переключение', 'Трек', 'Альбом', 'Плейлист']) -> None:
+        member = cast(discord.Member, ctx.author)
+        if not member.guild_permissions.manage_channels:
+            await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True)
+            return
+
+        guild = self.db.get_guild(ctx.guild.id)
+        
+        if vote_type == '-Всё':
+            self.db.update(ctx.guild.id, {
+                'vote_next_track': False,
+                'vote_add_track': False,
+                'vote_add_album': False,
+                'vote_add_artist': False,
+                'vote_add_playlist': False
+                }
+            )
+            response_message = "Голосование выключено."
+        elif vote_type == '+Всё':
+            self.db.update(ctx.guild.id, {
+                'vote_next_track': True,
+                'vote_add_track': True,
+                'vote_add_album': True,
+                'vote_add_artist': True,
+                'vote_add_playlist': True
+               }
+            )
+            response_message = "Голосование включено."
+        elif vote_type == 'Переключение':
+            self.db.update(ctx.guild.id, {'vote_next_track': not guild['vote_next_track']})
+            response_message = "Голосование за переключение трека " + ("включено." if guild['vote_next_track'] else "выключено.")
+        elif vote_type == 'Трек':
+            self.db.update(ctx.guild.id, {'vote_add_track': not guild['vote_add_track']})
+            response_message = "Голосование за добавление трека " + ("включено." if guild['vote_add_track'] else "выключено.")
+        elif vote_type == 'Альбом':
+            self.db.update(ctx.guild.id, {'vote_add_album': not guild['vote_add_album']})
+            response_message = "Голосование за добавление альбома " + ("включено." if guild['vote_add_album'] else "выключено.")
+        elif vote_type == 'Артист':
+            self.db.update(ctx.guild.id, {'vote_add_artist': not guild['vote_add_artist']})
+            response_message = "Голосование за добавление артиста " + ("включено." if guild['vote_add_artist'] else "выключено.")
+        elif vote_type == 'Плейлист':
+            self.db.update(ctx.guild.id, {'vote_add_playlist': not guild['vote_add_playlist']})
+            response_message = "Голосование за добавление плейлиста " + ("включено." if guild['vote_add_playlist'] else "выключено.")
+        
+        await ctx.respond(response_message, delete_after=15, ephemeral=True)

MusicBot/database/base.py

@@ -52,7 +52,20 @@ class BaseUsersDatabase:
         if not user:
             self.create_record(uid)
             user = users.find_one({'_id': uid})
-        return cast(ExplicitUser, user)
+        user =  cast(ExplicitUser, user)
+        existing_fields = user.keys()
+        fields: User = User(
+            ym_token=None,
+            playlists=[],
+            playlists_page=0,
+            queue_page=0
+        )
+        for field, default_value in fields.items():
+            if field not in existing_fields and field != '_id':
+                user[field] = default_value
+                users.update_one({'_id': uid}, {"$set": {field: default_value}})
+        
+        return user
 
     def get_ym_token(self, uid: int) -> str | None:
         user = users.find_one({'_id': uid})
@@ -78,7 +91,12 @@ class BaseGuildsDatabase:
             is_stopped=True,
             allow_explicit=True,
             always_allow_menu=False,
-            disable_vote=False,
+            vote_add=True,
+            vote_next_track=True,
+            vote_add_track=True,
+            vote_add_album=True,
+            vote_add_artist=True,
+            vote_add_playlist=True,
             shuffle=False,
             repeat=False
         ))
@@ -106,4 +124,30 @@ class BaseGuildsDatabase:
         if not guild:
             self.create_record(gid)
             guild = guilds.find_one({'_id': gid})
-        return cast(ExplicitGuild, guild)
+        
+        guild = cast(ExplicitGuild, guild)
+        existing_fields = guild.keys()
+        fields = Guild(
+            next_tracks=[],
+            previous_tracks=[],
+            current_track=None,
+            current_player=None,
+            is_stopped=True,
+            allow_explicit=True,
+            always_allow_menu=False,
+            vote_add=True,
+            vote_next_track=True,
+            vote_add_track=True,
+            vote_add_album=True,
+            vote_add_artist=True,
+            vote_add_playlist=True,
+            shuffle=False,
+            repeat=False
+        )
+        for field, default_value in fields.items():
+            if field not in existing_fields and field != '_id':
+                guild[field] = default_value
+                guilds.update_one({'_id': gid}, {"$set": {field: default_value}})
+        
+        return guild
+        

MusicBot/database/guild.py

@@ -8,7 +8,12 @@ class Guild(TypedDict, total=False):
     is_stopped: bool
     allow_explicit: bool
     always_allow_menu: bool
-    disable_vote: bool
+    vote_add: bool
+    vote_next_track: bool
+    vote_add_track: bool
+    vote_add_album: bool
+    vote_add_artist: bool
+    vote_add_playlist: bool
     shuffle: bool
     repeat: bool
 
@@ -21,6 +26,11 @@ class ExplicitGuild(TypedDict):
     is_stopped: bool  # Prevents the `after` callback of play_track
     allow_explicit: bool
     always_allow_menu: bool
-    disable_vote: bool
+    vote_add: bool
+    vote_next_track: bool
+    vote_add_track: bool
+    vote_add_album: bool
+    vote_add_artist: bool
+    vote_add_playlist: bool
     shuffle: bool
     repeat: bool

MusicBot/main.py

@@ -16,7 +16,8 @@ bot = Bot(intents=intents)
 
 cogs_list = [
     'general',
-    'voice'
+    'voice',
+    'settings'
 ]
 
 @bot.event