from discord import ApplicationContext, Embed, option from discord import utils as ds_utils from discord.abc import GuildChannel from discord.commands import SlashCommandGroup from discord.ext import commands from classes.holo_user import HoloUser from enums.colors import Color from modules.database import col_users from modules.utils import config_get from modules.utils_sync import config_get_sync, guild_name class CustomChannels(commands.Cog): def __init__(self, client): self.client = client @commands.Cog.listener() async def on_guild_channel_delete(self, channel: GuildChannel): col_users.find_one_and_update( {"customchannel": channel.id}, {"$set": {"customchannel": None}} ) customchannel = SlashCommandGroup("customchannel", "Керування особистим каналом") @customchannel.command( name="get", description="Отримати персональний текстовий канал", guild_ids=[config_get_sync("guild")], ) @option("name", description="Назва каналу") @option("reactions", description="Дозволити реакції") @option("threads", description="Дозволити гілки") async def customchannel_get_cmd( self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool ): holo_user_ctx = HoloUser(ctx.user) if holo_user_ctx.customchannel == None: await ctx.defer() created_channel = await ctx.user.guild.create_text_channel( name=name, reason=f"Користувач {guild_name(ctx.user)} отримав власний приватний канал", category=ds_utils.get( ctx.author.guild.categories, id=await config_get("customchannels", "categories"), ), ) await created_channel.set_permissions( ctx.user.guild.default_role, send_messages=False, add_reactions=reactions, create_public_threads=threads, create_private_threads=threads, ) await created_channel.set_permissions( ctx.user, attach_files=True, manage_messages=True, send_messages=True, embed_links=True, manage_channels=True, ) holo_user_ctx.set("customchannel", created_channel.id) await ctx.respond( embed=Embed( title="Створено канал", description=f"Вітаємо! Ви створили канал {created_channel.mention}. Для керування ним користуйтесь меню налаштувань каналу а також командою `/customchannel edit`", color=Color.success, ) ) bots = await config_get("bots") for bot in bots: await created_channel.set_permissions( ds_utils.get(ctx.user.guild.roles, id=bots[bot]["role"]), view_channel=False, ) else: await ctx.defer(ephemeral=True) await ctx.respond( embed=Embed( title="Помилка виконання", description=f"У вас вже є особистий канал.\nДля редагування каналу є `/customchannel edit` або просто відкрийте меню керування вашим каналом.", color=Color.fail, ) ) @customchannel.command( name="edit", description="Змінити параметри особистого каналу", guild_ids=[config_get_sync("guild")], ) @option("name", description="Назва каналу") @option("reactions", description="Дозволити реакції") @option("threads", description="Дозволити гілки") async def customchannel_edit_cmd( self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool ): holo_user_ctx = HoloUser(ctx.user) custom_channel = ds_utils.get( ctx.guild.channels, id=holo_user_ctx.customchannel ) if custom_channel is None: await ctx.respond( embed=Embed( title="Канал не знайдено", description=f"Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.", color=Color.fail, ) ) return await custom_channel.edit(name=name) await custom_channel.set_permissions( ctx.user.guild.default_role, send_messages=False, add_reactions=reactions, create_public_threads=threads, create_private_threads=threads, ) await ctx.respond( embed=Embed( title="Канал змінено", description=f"Назва каналу тепер `{name}`, реакції `{reactions}` та дозволено треди `{threads}`", color=Color.fail, ) ) @customchannel.command( name="remove", description="Відібрати канал, знищуючи його, та частково повернути кошти", guild_ids=[config_get_sync("guild")], ) @option("confirm", description="Підтвердження операції") async def customchannel_remove_cmd( self, ctx: ApplicationContext, confirm: bool = False ): holo_user_ctx = HoloUser(ctx.user) if holo_user_ctx.customchannel is not None: await ctx.defer() custom_channel = ds_utils.get( ctx.guild.channels, id=holo_user_ctx.customchannel ) if custom_channel is None: await ctx.respond( embed=Embed( title="Канал не знайдено", description=f"Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.", color=Color.fail, ) ) holo_user_ctx.set("customchannel", None) return if not confirm: await ctx.respond( embed=Embed( title="Підтвердження не надано", description=f"Для підтвердження операції додайте до команди параметр `confirm` зі значенням `True`.", color=Color.fail, ) ) return await custom_channel.delete(reason="Власник запросив видалення") holo_user_ctx.set("customchannel", None) await ctx.respond( embed=Embed( title="Канал знищено", description=f"Ви відмовились від каналу та видалили його.", color=Color.default, ) ) else: await ctx.defer(ephemeral=True) await ctx.respond( embed=Embed( title="Помилка виконання", description=f"У вас немає особистого каналу.", color=Color.fail, ) )