rewrite: New private Room system.

2023-05-23 17:49:37 +03:00
parent 4aa2587c45
commit f0dd540876
19 changed files with 2114 additions and 8 deletions
--- a/src/modules/rooms/settingui.py
+++ b/src/modules/rooms/settingui.py
@@ -0,0 +1,101 @@
+import asyncio
+
+import discord
+from discord.ui.button import button, Button, ButtonStyle
+from discord.ui.select import select, ChannelSelect
+
+from meta import LionBot
+
+from utils.ui import ConfigUI, DashboardSection
+from utils.lib import MessageArgs
+
+from .settings import RoomSettings
+from . import babel
+
+_p = babel._p
+
+
+class RoomSettingUI(ConfigUI):
+    setting_classes = RoomSettings.model_settings
+
+    def __init__(self, bot: LionBot, guildid: int, channelid: int, **kwargs):
+        self.settings = bot.get_cog('RoomCog').settings
+        super().__init__(bot, guildid, channelid, **kwargs)
+
+    # ----- UI Components -----
+    @select(cls=ChannelSelect, channel_types=[discord.ChannelType.category],
+            min_values=0, max_values=1,
+            placeholder='CATEGORY_PLACEHOLDER')
+    async def category_menu(self, selection: discord.Interaction, selected: ChannelSelect):
+        await selection.response.defer()
+        setting = self.instances[0]
+        setting.value = selected.values[0] if selected.values else None
+        await setting.write()
+
+    async def category_menu_refresh(self):
+        self.category_menu.placeholder = self.bot.translator.t(_p(
+            'ui:room_config|menu:category|placeholder',
+            "Select Private Room Category"
+        ))
+
+    @button(label="VISIBLE_BUTTON_PLACEHOLDER", style=ButtonStyle.grey)
+    async def visible_button(self, press: discord.Interaction, pressed: Button):
+        await press.response.defer()
+        setting = next(inst for inst in self.instances if inst.setting_id == RoomSettings.Visible.setting_id)
+        setting.value = not setting.value
+        await setting.write()
+
+    async def visible_button_refresh(self):
+        button = self.visible_button
+        button.label = self.bot.translator.t(_p(
+            'ui:room_config|button:visible|label',
+            "Toggle Room Visibility"
+        ))
+        setting = next(inst for inst in self.instances if inst.setting_id == RoomSettings.Visible.setting_id)
+        button.style = ButtonStyle.green if setting.value else ButtonStyle.grey
+
+    # ----- UI Flow -----
+    async def make_message(self) -> MessageArgs:
+        t = self.bot.translator.t
+        title = t(_p(
+            'ui:rooms_config|embed|title',
+            "Private Room System Configuration Panel"
+        ))
+        embed = discord.Embed(
+            colour=discord.Colour.orange(),
+            title=title
+        )
+        for setting in self.instances:
+            embed.add_field(**setting.embed_field, inline=False)
+
+        args = MessageArgs(embed=embed)
+        return args
+
+    async def reload(self):
+        lguild = await self.bot.core.lions.fetch_guild(self.guildid)
+        self.instances = tuple(
+            lguild.config.get(setting.setting_id)
+            for setting in self.settings.model_settings
+        )
+
+    async def refresh_components(self):
+        await asyncio.gather(
+            self.category_menu_refresh(),
+            self.visible_button_refresh(),
+            self.edit_button_refresh(),
+            self.close_button_refresh(),
+            self.reset_button_refresh(),
+        )
+        self.set_layout(
+            (self.category_menu,),
+            (self.visible_button, self.edit_button, self.reset_button, self.close_button)
+        )
+
+
+class RoomDashboard(DashboardSection):
+    section_name = _p(
+        'dash:rooms|title',
+        "Private Room Configuration"
+    )
+    configui = RoomSettingUI
+    setting_classes = RoomSettingUI.setting_classes