Discord/cogs/custom_channels.py

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 libbot.pycord.classes import PycordBot

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: PycordBot):
        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="У вас вже є особистий канал.\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="Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.",
                    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="Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.",
                        color=Color.fail,
                    )
                )
                holo_user_ctx.set("customchannel", None)
                return
            if not confirm:
                await ctx.respond(
                    embed=Embed(
                        title="Підтвердження не надано",
                        description="Для підтвердження операції додайте до команди параметр `confirm` зі значенням `True`.",
                        color=Color.fail,
                    )
                )
                return
            await custom_channel.delete(reason="Власник запросив видалення")
            holo_user_ctx.set("customchannel", None)
            await ctx.respond(
                embed=Embed(
                    title="Канал знищено",
                    description="Ви відмовились від каналу та видалили його.",
                    color=Color.default,
                )
            )
        else:
            await ctx.defer(ephemeral=True)
            await ctx.respond(
                embed=Embed(
                    title="Помилка виконання",
                    description="У вас немає особистого каналу.",
                    color=Color.fail,
                )
            )


def setup(client: PycordBot):
    client.add_cog(CustomChannels(client))