"""MIT License Copyright (c) 2023 - present Vocard Development Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. """ import discord import copy from discord.ext import commands from .utils import BaseModal from ..mongodb import MongoDBHandler from ..placeholders import PlayerPlaceholder class Dropdown(discord.ui.Select): def __init__(self): options = [ discord.SelectOption(label='Active', description='The controller embed when music is playing', emoji='🟩'), discord.SelectOption(label='Inactive', description='The controller embed when music is not playing', emoji='🟥'), ] super().__init__(placeholder='Select a embed to edit...', min_values=1, max_values=1, options=options) async def callback(self, interaction: discord.Interaction): self.view.embed_type = self.values[0].lower() if self.view.embed_type not in self.view.data: self.view.data[self.view.embed_type] = {} await interaction.response.edit_message(embed=self.view.build_embed()) class EmbedBuilderView(discord.ui.View): def __init__(self, context: commands.Context, placeholder: PlayerPlaceholder, data: dict) -> None: super().__init__(timeout=300) self.add_item(Dropdown()) self.author: discord.Member = context.author self.ph: PlayerPlaceholder = placeholder self.response: discord.Message = None self.original_data: dict = copy.deepcopy(data) self.data: dict = copy.deepcopy(data) self.embed_type: str = "active" def build_embed(self) -> discord.Embed: return PlayerPlaceholder.build_embed(self.data.get(self.embed_type, {}), self.ph) async def on_timeout(self): for child in self.children: child.disabled = True try: await self.response.edit(view=self) except: pass async def interaction_check(self, interaction: discord.Interaction): return interaction.user == self.author @discord.ui.button(label="Edit Content", style=discord.ButtonStyle.blurple) async def edit_content(self, interaction: discord.Interaction, button: discord.ui.Button): data = self.data.get(self.embed_type, {}) modal = BaseModal( title="Edit Content", custom_id="edit_content", items=[ discord.ui.TextInput( label="Title", placeholder="The title of the embed", style=discord.TextStyle.paragraph, custom_id="title", max_length=1000, default=data.get("title", {}).get("name"), required=False ), discord.ui.TextInput( label="Url", placeholder="The url of the title", style=discord.TextStyle.short, custom_id="url", max_length=100, default=data.get("title", {}).get("url"), required=False ), discord.ui.TextInput( label="Color", placeholder="The color of the embed", style=discord.TextStyle.short, custom_id="color", max_length=100, default=data.get("color"), required=False ), discord.ui.TextInput( label="Description", placeholder="The description of the title", style=discord.TextStyle.paragraph, custom_id="description", max_length=200, default=data.get("description"), required=False ) ]) await interaction.response.send_modal(modal) await modal.wait() v = modal.values try: data["description"] = v["description"] data["color"] = int(v["color"], 16) if "title" not in data: data["title"] = {} data["title"]["name"] = v['title'] data["title"]["url"] = v['url'] except: pass return await self.response.edit(embed=self.build_embed()) @discord.ui.button(label="Edit Author",) async def edit_author(self, interaction: discord.Interaction, button: discord.ui.Button): data = self.data.get(self.embed_type, {}) modal = BaseModal( title="Edit Author", custom_id="edit_author", items=[ discord.ui.TextInput( label="Name", placeholder="The name of the author", style=discord.TextStyle.paragraph, custom_id="name", max_length=200, default=data.get("author", {}).get("name"), required=False ), discord.ui.TextInput( label="Url", placeholder="The url of the author", style=discord.TextStyle.short, custom_id="url", max_length=100, default=data.get("author", {}).get("url"), required=False ), discord.ui.TextInput( label="Icon Url", placeholder="The icon url of the author", style=discord.TextStyle.short, custom_id="icon_url", max_length=100, default=data.get("author", {}).get("icon_url"), required=False ), ]) await interaction.response.send_modal(modal) await modal.wait() v = modal.values if v['name'] != "": if "author" not in data: data["author"] = {} data["author"]["name"] = v['name'] data["author"]["url"] = v['url'] data["author"]["icon_url"] = v['icon_url'] else: del data["author"] return await self.response.edit(embed=self.build_embed()) @discord.ui.button(label="Edit Image") async def edit_image(self, interaction: discord.Interaction, button: discord.ui.Button): data = self.data.get(self.embed_type, {}) modal = BaseModal( title="Edit Image", custom_id="edit_image", items=[ discord.ui.TextInput( label="Thumbnail", placeholder="The url of the thumbnail", style=discord.TextStyle.short, custom_id="thumbnail", max_length=200, default=data.get("thumbnail"), required=False ), discord.ui.TextInput( label="Image", placeholder="The url of the image", style=discord.TextStyle.short, custom_id="image", max_length=100, default=data.get("image"), required=False ) ]) await interaction.response.send_modal(modal) await modal.wait() v = modal.values data["thumbnail"] = v['thumbnail'] data["image"] = v['image'] return await self.response.edit(embed=self.build_embed()) @discord.ui.button(label="Edit Footer") async def edit_footer(self, interaction: discord.Interaction, button: discord.ui.Button): data = self.data.get(self.embed_type, {}) modal = BaseModal( title="Edit Footer", custom_id="edit_footer", items=[ discord.ui.TextInput( label="Text", placeholder="The text of the footer", style=discord.TextStyle.paragraph, custom_id="text", max_length=200, default=data.get("footer", {}).get("text"), required=False ), discord.ui.TextInput( label="Icon Url", placeholder="The url of the icon", style=discord.TextStyle.short, custom_id="icon_url", max_length=100, default=data.get("footer", {}).get("icon_url"), required=False ) ]) await interaction.response.send_modal(modal) await modal.wait() v = modal.values if "footer" not in data: data["footer"] = {} data["footer"]["text"] = v['text'] data["footer"]["icon_url"] = v['icon_url'] return await self.response.edit(embed=self.build_embed()) @discord.ui.button(label="Add Field", style=discord.ButtonStyle.green, row=1) async def add_field(self, interaction: discord.Interaction, button: discord.ui.Button): data = self.data.get(self.embed_type) if "fields" not in data: data["fields"] = [] if len(data["fields"]) >= 25: return await interaction.response.send_message("You have already reached the maximum of fields!", ephemeral=True) modal = BaseModal( title="Add Field", custom_id="add_field", items=[ discord.ui.TextInput( label="Name", placeholder="The name of the field", style=discord.TextStyle.paragraph, custom_id="name", max_length=256 ), discord.ui.TextInput( label="Value", placeholder="The value of the field", style=discord.TextStyle.long, custom_id="value", max_length=1024 ), discord.ui.TextInput( label="Inline", placeholder="The inline of the field, e.g. True or False", style=discord.TextStyle.short, custom_id="inline", ) ]) await interaction.response.send_modal(modal) await modal.wait() v = modal.values data["fields"].append({ "name": v["name"], "value": v["value"], "inline": True if v["inline"].lower() == "true" else False }) return await self.response.edit(embed=self.build_embed()) @discord.ui.button(label="Remove Field", style=discord.ButtonStyle.red, row=1) async def remove_field(self, interaction: discord.Interaction, button: discord.ui.Button): data = self.data.get(self.embed_type) if "fields" not in data: data["fields"] = [] if len(data["fields"]) == 0: return await interaction.response.send_message("There are no fields to remove!", ephemeral=True) modal = BaseModal( title="Remove Field", custom_id="remove_field", items = [ discord.ui.TextInput( label="Index", placeholder="The number of fields to remove, e.g. 1", style=discord.TextStyle.short, custom_id="index", ) ]) await interaction.response.send_modal(modal) await modal.wait() try: del data["fields"][int(modal.values["index"])] except: return await interaction.followup.send("Can't found the field", ephemeral=True) return await self.response.edit(embed=self.build_embed()) @discord.ui.button(label="Apply", style=discord.ButtonStyle.green, row=1) async def apply(self, interaction: discord.Interaction, button: discord.ui.Button): await MongoDBHandler.update_settings( interaction.guild_id, {"$set": {"default_controller.embeds": self.data}}, ) await self.on_timeout() await interaction.response.defer() self.stop() @discord.ui.button(label="Reset", style=discord.ButtonStyle.red, row=1) async def reset(self, interaction: discord.Interaction, button: discord.ui.Button): self.data.update(copy.deepcopy(self.original_data)) return await interaction.response.edit_message(embed=self.build_embed()) @discord.ui.button(emoji='🗑️', row=1) async def stop_button(self, interaction: discord.Interaction, button: discord.ui.Button): await self.response.delete() self.stop()