Init commit

2023-05-03 19:53:01 +03:00
commit 12c87b99d3
45 changed files with 3890 additions and 0 deletions
@@ -0,0 +1,106 @@
+import discord
+from discord.ext import commands
+from discord.utils import get
+
+from cogs.checks import is_white
+import asyncio
+from cogs.emojis import check_mark
+
+from bot import db
+
+
+class privateChannels(commands.Cog, name="Приватные комнаты"):
+    """Работа с приватными комнатами"""
+
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.Cog.listener()
+    async def on_voice_state_update(self, member, before, after):
+        for chan in [i["id"] for i in db.private_channels.find()]:
+            if not self.bot.get_channel(chan):
+                db.private_channels.delete_one({"id": chan})
+
+        v_channels = [i["id"] for i in db.private_channels.find()]
+        v_categories = [self.bot.get_channel(i).category_id for i in v_channels]
+
+        if (
+                before.channel is not None and
+                len(before.channel.members) == 0 and
+                before.channel.id not in v_channels and
+                before.channel.category_id in v_categories
+        ):
+            await before.channel.delete()
+
+        if after.channel is not None and after.channel.id in v_channels:
+            #, overwrites={
+            #    member: discord.PermissionOverwrite(manage_channels=True,
+            #                                        move_members=True,
+            #                                        manage_permissions=True)
+            #}
+
+            new_private = await after.channel.category.create_voice_channel(name=member.display_name)
+
+            try:
+                await member.edit(voice_channel=new_private)
+            except discord.errors.HTTPException:
+                await new_private.delete()
+
+            try:
+                await new_private.set_permissions(member, manage_channels=True, move_members=True, manage_permissions=True)
+            except:
+                pass
+
+        try:
+            voice = before.channel.guild.me.voice
+        except:
+            voice = after.channel.guild.me.voice
+
+        if voice:
+            if len(voice.channel.members) == 1:
+                await member.guild.voice_client.disconnect()
+
+    @commands.command(brief="Присваивает комнату главной",
+                      help="Идея приватных комнат состоит в том, что при входе в ГК создается пприватная комната. "
+                           "Тот самый ГК и является главной комнатой. Эта команда присваивает ГК, в котором вы "
+                           "находитесь в главный")
+    @commands.check(is_white)
+    async def set_private(self, ctx):
+        if ctx.author.voice.channel.id in [i["id"] for i in db.private_channels.find()]:
+            await ctx.message.delete()
+            message = await ctx.send('Канал уже добавлен')
+        else:
+            db.private_channels.insert_one({"id": ctx.author.voice.channel.id})
+            await ctx.message.add_reaction(check_mark)
+            message = ctx.message
+
+        await asyncio.sleep(3)
+        await message.delete()
+
+    @commands.command(brief="Делает комнату не приватной")
+    @commands.check(is_white)
+    async def unset_private(self, ctx):
+        if ctx.author.voice is not None:
+            message = await ctx.send("Требуется выйти из ГК")
+            await self.bot.wait_for("voice_state_update", check=lambda member, _, after: \
+                member == ctx.message.author and after.channel is None)
+            await message.edit(content="Зайдите в ГК")
+            await self.bot.wait_for("voice_state_update", check=lambda member, _, after: \
+                member == ctx.message.author and after.channel is not None)
+        else:
+            message = await ctx.send("Зайдите в ГК")
+            await self.bot.wait_for("voice_state_update", check=lambda member, _, after: \
+                member == ctx.message.author and after.channel is not None)
+        if ctx.author.voice.channel.id in [i["id"] for i in db.private_channels.find()]:
+            db.private_channels.delete_one({"id": ctx.author.voice.channel.id})
+            await ctx.message.add_reaction(check_mark)
+        else:
+            await message.edit(content='Этот канал не является приватным')
+
+        await asyncio.sleep(3)
+        await ctx.message.delete()
+        await message.delete()
+
+
+async def setup(bot):
+    await bot.add_cog(privateChannels(bot))