impr: Code organization and bug fixes.

MusicBot/cogs/general.py

@@ -62,7 +62,7 @@ async def get_search_suggestions(ctx: discord.AutocompleteContext) -> list[str]:
             playlists_list = await client.users_playlists_list(client.me.account.uid)
             res = [playlist.title if playlist.title else 'Без названия' for playlist in playlists_list]
 
-    return res
+    return res[:100]
 
 class General(Cog):
     

MusicBot/cogs/settings.py

@@ -20,10 +20,12 @@ class Settings(Cog):
     @settings.command(name="show", description="Показать текущие настройки бота.")
     async def show(self, ctx: discord.ApplicationContext) -> None:
         guild = await self.db.get_guild(ctx.guild.id, projection={
-            'allow_explicit': 1, 'always_allow_menu': 1, 'vote_next_track': 1, 'vote_add_track': 1, 'vote_add_album': 1, 'vote_add_artist': 1, 'vote_add_playlist': 1
+            'allow_explicit': 1, 'always_allow_menu': 1,
+            'vote_next_track': 1, 'vote_add_track': 1, 'vote_add_album': 1, 'vote_add_artist': 1, 'vote_add_playlist': 1,
+            'allow_connect': 1, 'allow_disconnect': 1
         })
         embed = discord.Embed(title="Настройки бота", color=0xfed42b)
-        
+
         explicit = "✅ - Разрешены" if guild['allow_explicit'] else "❌ - Запрещены"
         menu = "✅ - Всегда доступно" if guild['always_allow_menu'] else "❌ - Если в канале 1 человек."
         
@@ -32,14 +34,28 @@ class Settings(Cog):
         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❌ - Добавление плейлистов"
-        
+
+        connect = "\n✅ - Разрешено всем" if guild['allow_connect'] 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)
-        
+        embed.add_field(name="__Подключение и Отключение__", value=connect, inline=False)
+
         await ctx.respond(embed=embed, ephemeral=True)
     
-    @settings.command(name="explicit", description="Разрешить или запретить воспроизведение Explicit треков.")
+    @settings.command(name="connect", description="Разрешить/запретить отключение/подключение бота к каналу участникам без прав управления каналом.")
+    async def connect(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 = await self.db.get_guild(ctx.guild.id, projection={'allow_connect': 1})
+        await self.db.update(ctx.guild.id, {'allow_connect': not guild['allow_connect']})
+        await ctx.respond(f"Отключение/подключение бота к каналу теперь {'✅ разрешено' if not guild['allow_connect'] else '❌ запрещено'} участникам без прав управления каналом.", delete_after=15, 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:
@@ -48,9 +64,9 @@ class Settings(Cog):
 
         guild = await self.db.get_guild(ctx.guild.id, projection={'allow_explicit': 1})
         await 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)
+        await ctx.respond(f"Треки с содержанием не для детей теперь {'✅ разрешены' if not guild['allow_explicit'] else '❌ запрещены'}.", delete_after=15, ephemeral=True)
 
-    @settings.command(name="menu", description="Разрешить или запретить создание меню проигрывателя, даже если в канале больше одного человека.")
+    @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:
@@ -59,7 +75,7 @@ class Settings(Cog):
 
         guild = await self.db.get_guild(ctx.guild.id, projection={'always_allow_menu': 1})
         await 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)
+        await ctx.respond(f"Меню проигрывателя теперь {'✅ доступно' if not guild['always_allow_menu'] else '❌ недоступно'} в каналах с несколькими людьми.", delete_after=15, ephemeral=True)
 
     @settings.command(name="vote", description="Настроить голосование.")
     @discord.option(
@@ -86,7 +102,7 @@ class Settings(Cog):
                 'vote_add_playlist': False
                 }
             )
-            response_message = "Голосование выключено."
+            response_message = "Голосование ❌ выключено."
         elif vote_type == '+Всё':
             await self.db.update(ctx.guild.id, {
                 'vote_next_track': True,
@@ -96,21 +112,21 @@ class Settings(Cog):
                 'vote_add_playlist': True
                }
             )
-            response_message = "Голосование включено."
+            response_message = "Голосование ✅ включено."
         elif vote_type == 'Переключение':
             await self.db.update(ctx.guild.id, {'vote_next_track': not guild['vote_next_track']})
-            response_message = "Голосование за переключение трека " + ("выключено." if guild['vote_next_track'] else "включено.")
+            response_message = "Голосование за переключение трека " + ("❌ выключено." if guild['vote_next_track'] else "✅ включено.")
         elif vote_type == 'Трек':
             await self.db.update(ctx.guild.id, {'vote_add_track': not guild['vote_add_track']})
-            response_message = "Голосование за добавление трека " + ("выключено." if guild['vote_add_track'] else "включено.")
+            response_message = "Голосование за добавление трека " + ("❌ выключено." if guild['vote_add_track'] else "✅ включено.")
         elif vote_type == 'Альбом':
             await self.db.update(ctx.guild.id, {'vote_add_album': not guild['vote_add_album']})
-            response_message = "Голосование за добавление альбома " + ("выключено." if guild['vote_add_album'] else "включено.")
+            response_message = "Голосование за добавление альбома " + ("❌ выключено." if guild['vote_add_album'] else "✅ включено.")
         elif vote_type == 'Артист':
             await self.db.update(ctx.guild.id, {'vote_add_artist': not guild['vote_add_artist']})
-            response_message = "Голосование за добавление артиста " + ("выключено." if guild['vote_add_artist'] else "включено.")
+            response_message = "Голосование за добавление артиста " + ("❌ выключено." if guild['vote_add_artist'] else "✅ включено.")
         elif vote_type == 'Плейлист':
             await self.db.update(ctx.guild.id, {'vote_add_playlist': not guild['vote_add_playlist']})
-            response_message = "Голосование за добавление плейлиста " + ("выключено." if guild['vote_add_playlist'] else "включено.")
-        
+            response_message = "Голосование за добавление плейлиста " + ("❌ выключено." if guild['vote_add_playlist'] else "✅ включено.")
+
         await ctx.respond(response_message, delete_after=15, ephemeral=True)

MusicBot/cogs/voice.py

@@ -18,7 +18,7 @@ class Voice(Cog, VoiceExtension):
 
     def __init__(self, bot: discord.Bot):
         VoiceExtension.__init__(self, bot)
-        self.typed_bot: discord.Bot = bot  # should be removed later
+        self.typed_bot: discord.Bot = bot
 
     @Cog.listener()
     async def on_voice_state_update(self, member: discord.Member, before: discord.VoiceState, after: discord.VoiceState) -> None:
@@ -38,15 +38,21 @@ class Voice(Cog, VoiceExtension):
 
         if len(channel.members) == 1 and vc:
             logging.info(f"[VOICE] Clearing history and stopping playback for guild {gid}")
+
+            if member.guild.id in menu_views:
+                menu_views[member.guild.id].stop()
+                del menu_views[member.guild.id]
+
             if guild['current_menu']:
                 message = self.typed_bot.get_message(guild['current_menu'])
                 if message:
                     await message.delete()
-                if member.guild.id in menu_views:
-                    menu_views[member.guild.id].stop()
-                    del menu_views[member.guild.id]
 
-            await self.db.update(gid, {'previous_tracks': [], 'next_tracks': [], 'vibing': False, 'current_menu': None, 'repeat': False, 'shuffle': False})
+            await self.db.update(gid, {
+                'previous_tracks': [], 'next_tracks': [], 'votes': [],
+                'current_track': None, 'current_menu': None, 'vibing': False,
+                'repeat': False, 'shuffle': False, 'is_stopped': True
+            })
             vc.stop()
         elif len(channel.members) > 2 and not guild['always_allow_menu']:
             if current_menu:
@@ -90,14 +96,17 @@ class Voice(Cog, VoiceExtension):
         guild_id = payload.guild_id
         if not guild_id:
             return
+
         guild = await self.db.get_guild(guild_id, projection={'votes': 1, 'current_track': 1})
         votes = guild['votes']
 
-        if payload.message_id not in votes:
+        if str(payload.message_id) not in votes:
             logging.info(f"[VOICE] Message {payload.message_id} not found in votes")
             return
 
         vote_data = votes[str(payload.message_id)]
+        logging.debug(f"[VOICE] Vote data for message {payload.message_id}: {vote_data}")
+
         if payload.emoji.name == '✅':
             logging.info(f"[VOICE] User {payload.user_id} voted positively for message {payload.message_id}")
             vote_data['positive_votes'].append(payload.user_id)
@@ -113,7 +122,6 @@ class Voice(Cog, VoiceExtension):
             if vote_data['action'] == 'next':
                 logging.info(f"[VOICE] Skipping track for message {payload.message_id}")
 
-                await self.db.update(guild_id, {'is_stopped': False})
                 title = await self.next_track(payload)
                 await message.clear_reactions()
                 await message.edit(content=f"Сейчас играет: **{title}**!", delete_after=15)
@@ -128,7 +136,6 @@ class Voice(Cog, VoiceExtension):
                     logging.info(f"[VOICE] Recieved empty vote context for message {payload.message_id}")
                     return
 
-                await self.db.update(guild_id, {'is_stopped': False})
                 await self.db.modify_track(guild_id, track, 'next', 'append')
 
                 if guild['current_track']:
@@ -179,6 +186,10 @@ class Voice(Cog, VoiceExtension):
             return
         guild = await self.db.get_guild(guild_id, projection={'votes': 1})
         votes = guild['votes']
+        
+        if str(payload.message_id) not in votes:
+            logging.info(f"[VOICE] Message {payload.message_id} not found in votes")
+            return
 
         channel = cast(discord.VoiceChannel, self.typed_bot.get_channel(payload.channel_id))
         if not channel:
@@ -219,7 +230,9 @@ class Voice(Cog, VoiceExtension):
         logging.info(f"[VOICE] Join command invoked by user {ctx.author.id} in guild {ctx.guild.id}")
 
         member = cast(discord.Member, ctx.author)
-        if not member.guild_permissions.manage_channels:
+        guild = await self.db.get_guild(ctx.guild.id, projection={'allow_connect': 1})
+
+        if not member.guild_permissions.manage_channels and not guild['allow_connect']:
             response_message = "❌ У вас нет прав для выполнения этой команды."
         elif (vc := await self.get_voice_client(ctx)) and vc.is_connected():
             response_message = "❌ Бот уже находится в голосовом канале. Выключите его с помощью команды /voice leave."
@@ -237,16 +250,21 @@ class Voice(Cog, VoiceExtension):
         logging.info(f"[VOICE] Leave command invoked by user {ctx.author.id} in guild {ctx.guild.id}")
 
         member = cast(discord.Member, ctx.author)
-        if not member.guild_permissions.manage_channels:
+        guild = await self.db.get_guild(ctx.guild.id, projection={'allow_connect': 1})
+        
+        if not member.guild_permissions.manage_channels and not guild['allow_connect']:
             logging.info(f"[VOICE] User {ctx.author.id} does not have permissions to execute leave command in guild {ctx.guild.id}")
             await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True)
             return
 
         if (vc := await self.get_voice_client(ctx)) and await self.voice_check(ctx):
-            await self.stop_playing(ctx, full=True)
-            await vc.disconnect(force=True)
-            await ctx.respond("Отключение успешно!", delete_after=15, ephemeral=True)
-            logging.info(f"[VOICE] Successfully disconnected from voice channel in guild {ctx.guild.id}")
+            res = await self.stop_playing(ctx, full=True)
+            if res:
+                await vc.disconnect(force=True)
+                await ctx.respond("Отключение успешно!", delete_after=15, ephemeral=True)
+                logging.info(f"[VOICE] Successfully disconnected from voice channel in guild {ctx.guild.id}")
+            else:
+                await ctx.respond("❌ Не удалось отключиться.", delete_after=15, ephemeral=True)
 
     @queue.command(description="Очистить очередь треков и историю прослушивания.")
     async def clear(self, ctx: discord.ApplicationContext) -> None:
@@ -294,7 +312,7 @@ class Voice(Cog, VoiceExtension):
 
                 menu = await self.db.get_current_menu(ctx.guild.id)
                 if menu:
-                    await self.update_menu_embed(ctx, menu)
+                    await self.update_menu_full(ctx, menu)
 
                 logging.info(f"[VOICE] Track paused in guild {ctx.guild.id}")
                 await ctx.respond("Воспроизведение приостановлено.", delete_after=15, ephemeral=True)
@@ -318,7 +336,7 @@ class Voice(Cog, VoiceExtension):
                 vc.resume()
                 menu = await self.db.get_current_menu(ctx.guild.id)
                 if menu:
-                    await self.update_menu_embed(ctx, menu)
+                    await self.update_menu_full(ctx, menu)
                 logging.info(f"[VOICE] Track resumed in guild {ctx.guild.id}")
                 await ctx.respond("Воспроизведение восстановлено.", delete_after=15, ephemeral=True)
             else:
@@ -337,9 +355,11 @@ class Voice(Cog, VoiceExtension):
             await ctx.respond("❌ Вы не можете остановить воспроизведение, пока в канале находятся другие пользователи.", delete_after=15, ephemeral=True)
 
         elif await self.voice_check(ctx):
-            await self.stop_playing(ctx, full=True)
-
-            await ctx.respond("Воспроизведение остановлено.", delete_after=15, ephemeral=True)
+            res = await self.stop_playing(ctx, full=True)
+            if res:
+                await ctx.respond("Воспроизведение остановлено.", delete_after=15, ephemeral=True)
+            else:
+                await ctx.respond("❌ Произошла ошибка при остановке воспроизведения.", delete_after=15, ephemeral=True)
 
     @track.command(description="Переключиться на следующую песню в очереди.")
     async def next(self, ctx: discord.ApplicationContext) -> None:
@@ -397,16 +417,18 @@ class Voice(Cog, VoiceExtension):
             await ctx.respond("Нет воспроизводимого трека.", delete_after=15, ephemeral=True)
             return
 
-        result = await self.like_track(ctx)
-        if not result:
+        result = await self.react_track(ctx, 'like')
+        if not result[0]:
             logging.warning(f"Like command failed for user {ctx.author.id} in guild {ctx.guild.id}")
             await ctx.respond("❌ Операция не удалась.", delete_after=15, ephemeral=True)
-        elif result == 'TRACK REMOVED':
+        elif result[1] == 'removed':
             logging.info(f"[VOICE] Track removed from favorites for user {ctx.author.id} in guild {ctx.guild.id}")
             await ctx.respond("Трек был удалён из избранного.", delete_after=15, ephemeral=True)
-        else:
+        elif result[1] == 'added':
             logging.info(f"[VOICE] Track added to favorites for user {ctx.author.id} in guild {ctx.guild.id}")
             await ctx.respond(f"Трек **{result}** был добавлен в избранное.", delete_after=15, ephemeral=True)
+        else:
+            raise ValueError(f"Unknown like command result: '{result}'")
     
     @track.command(name='vibe', description="Запустить Мою Волну по текущему треку.")
     async def track_vibe(self, ctx: discord.ApplicationContext) -> None:
@@ -414,7 +436,7 @@ class Voice(Cog, VoiceExtension):
         if not await self.voice_check(ctx):
             return
 
-        guild = await self.db.get_guild(ctx.guild.id, projection={'always_allow_menu': 1, 'current_track': 1})
+        guild = await self.db.get_guild(ctx.guild.id, projection={'always_allow_menu': 1, 'current_track': 1, 'current_menu': 1})
         channel = cast(discord.VoiceChannel, ctx.channel)
 
         if len(channel.members) > 2 and not guild['always_allow_menu']:
@@ -426,10 +448,17 @@ class Voice(Cog, VoiceExtension):
             await ctx.respond("❌ Нет воспроизводимого трека.", ephemeral=True)
             return
 
-        await self.send_menu_message(ctx)
         feedback = await self.update_vibe(ctx, 'track', guild['current_track']['id'])
         if not feedback:
             await ctx.respond("❌ Операция не удалась. Возможно, у вес нет подписки на Яндекс Музыку.", ephemeral=True)
+            return
+        
+        if not guild['current_menu']:
+            await self.send_menu_message(ctx)
+
+        next_track = await self.db.get_track(ctx.guild_id, 'next')
+        if next_track:
+            await self._play_next_track(ctx, next_track)
 
     @voice.command(name='vibe', description="Запустить Мою Волну.")
     async def user_vibe(self, ctx: discord.ApplicationContext) -> None:
@@ -437,7 +466,7 @@ class Voice(Cog, VoiceExtension):
         if not await self.voice_check(ctx):
             return
 
-        guild = await self.db.get_guild(ctx.guild.id, projection={'always_allow_menu': 1})
+        guild = await self.db.get_guild(ctx.guild.id, projection={'always_allow_menu': 1, 'current_menu': 1})
         channel = cast(discord.VoiceChannel, ctx.channel)
 
         if len(channel.members) > 2 and not guild['always_allow_menu']:
@@ -445,7 +474,14 @@ class Voice(Cog, VoiceExtension):
             await ctx.respond("❌ Вы не единственный в голосовом канале.", ephemeral=True)
             return
 
-        await self.send_menu_message(ctx)
         feedback = await self.update_vibe(ctx, 'user', 'onyourwave')
         if not feedback:
             await ctx.respond("❌ Операция не удалась. Возможно, у вес нет подписки на Яндекс Музыку.", ephemeral=True)
+            return
+        
+        if not guild['current_menu']:
+            await self.send_menu_message(ctx)
+
+        next_track = await self.db.get_track(ctx.guild_id, 'next')
+        if next_track:
+            await self._play_next_track(ctx, next_track)