mirror of
https://github.com/deadcxap/YandexMusicDiscordBot.git
synced 2026-01-10 02:21:45 +03:00
106 lines
6.3 KiB
Python
106 lines
6.3 KiB
Python
import logging
|
||
from typing import Literal, cast
|
||
|
||
import discord
|
||
from discord.ext.commands import Cog
|
||
|
||
from MusicBot.database import BaseUsersDatabase, BaseGuildsDatabase
|
||
from MusicBot.cogs.utils import BaseBot
|
||
|
||
def setup(bot):
|
||
bot.add_cog(Settings(bot))
|
||
|
||
class Settings(Cog, BaseBot):
|
||
|
||
settings = discord.SlashCommandGroup("settings", "Команды для изменения настроек бота.")
|
||
|
||
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:
|
||
if not ctx.guild_id:
|
||
logging.info("[SETTINGS] Show command invoked without guild_id")
|
||
await self.respond(ctx, "error", "Эта команда может быть использована только на сервере.", ephemeral=True)
|
||
return
|
||
|
||
guild = await self.db.get_guild(ctx.guild_id, projection={
|
||
'allow_change_connect': 1, 'vote_switch_track': 1, 'vote_add': 1, 'use_single_token': 1
|
||
})
|
||
|
||
vote = "✅ - Переключение" if guild['vote_switch_track'] else "❌ - Переключение"
|
||
vote += "\n✅ - Добавление в очередь" if guild['vote_add'] else "\n❌ - Добавление в очередь"
|
||
|
||
connect = "\n✅ - Разрешено всем" if guild['allow_change_connect'] else "\n❌ - Только для участникам с правами управления каналом"
|
||
|
||
token = "🔐 - Используется токен пользователя, запустившего бота" if guild['use_single_token'] else "🔒 - Используется личный токен пользователя"
|
||
|
||
embed = discord.Embed(title="Настройки бота", color=0xfed42b)
|
||
embed.set_author(name='YandexMusic', icon_url="https://github.com/Lemon4ksan/YandexMusicDiscordBot/blob/main/assets/Logo.png?raw=true")
|
||
|
||
embed.add_field(name="__Голосование__", value=vote, inline=False)
|
||
embed.add_field(name="__Подключение/Отключение__", value=connect, inline=False)
|
||
embed.add_field(name="__Токен__", value=token, inline=False)
|
||
|
||
await ctx.respond(embed=embed, ephemeral=True)
|
||
|
||
@settings.command(name="toggle", description="Переключить параметры основных настроек.")
|
||
@discord.option(
|
||
"параметр",
|
||
parameter_name="vote_type",
|
||
description="Тип голосования.",
|
||
type=discord.SlashCommandOptionType.string,
|
||
choices=[
|
||
'Переключение треков без голосования для всех',
|
||
'Добавление в очередь без голосования для всех',
|
||
'Добавление/Отключение бота от канала для всех',
|
||
'Использовать токен запустившего пользователя для всех'
|
||
]
|
||
)
|
||
async def toggle(
|
||
self,
|
||
ctx: discord.ApplicationContext,
|
||
vote_type: Literal[
|
||
'Переключение треков без голосования для всех',
|
||
'Добавление в очередь без голосования для всех',
|
||
'Добавление/Отключение бота от канала для всех',
|
||
'Использовать токен запустившего пользователя для всех'
|
||
]
|
||
) -> None:
|
||
if not ctx.guild_id:
|
||
logging.info("[SETTINGS] Toggle command invoked without guild_id")
|
||
await self.respond(ctx, "error", "Эта команда может быть использована только на сервере.", delete_after=15, ephemeral=True)
|
||
return
|
||
|
||
member = cast(discord.Member, ctx.user)
|
||
if not member.guild_permissions.manage_channels:
|
||
await self.respond(ctx, "error", "У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True)
|
||
return
|
||
|
||
guild = await self.db.get_guild(ctx.guild_id, projection={
|
||
'vote_switch_track': 1, 'vote_add': 1, 'allow_change_connect': 1, 'use_single_token': 1
|
||
})
|
||
|
||
if vote_type == 'Переключение треков без голосования для всех':
|
||
await self.db.update(ctx.guild_id, {'vote_switch_track': not guild['vote_switch_track']})
|
||
response_message = "Голосование за переключение трека " + ("❌ выключено." if guild['vote_switch_track'] else "✅ включено.")
|
||
|
||
elif vote_type == 'Добавление в очередь без голосования для всех':
|
||
await self.db.update(ctx.guild_id, {'vote_add': not guild['vote_add']})
|
||
response_message = "Голосование за добавление в очередь " + ("❌ выключено." if guild['vote_add'] else "✅ включено.")
|
||
|
||
elif vote_type == 'Добавление/Отключение бота от канала для всех':
|
||
await self.db.update(ctx.guild_id, {'allow_change_connect': not guild['allow_change_connect']})
|
||
response_message = f"Добавление/Отключение бота от канала теперь {'✅ разрешено' if not guild['allow_change_connect'] else '❌ запрещено'} участникам без прав управления каналом."
|
||
|
||
elif vote_type == 'Использовать токен запустившего пользователя для всех':
|
||
await self.db.update(ctx.guild_id, {'use_single_token': not guild['use_single_token']})
|
||
response_message = f"Использование единого токена для прослушивания теперь {'✅ включено' if not guild['use_single_token'] else '❌ выключено'}."
|
||
|
||
else:
|
||
response_message = "Неизвестный тип настроек."
|
||
|
||
await self.respond(ctx, 'info', response_message, delete_after=30, ephemeral=True)
|