import io from typing import Any, Literal import aiohttp import discord from colorthief import ColorThief from red_commons.logging import RedTraceLogger, getLogger from redbot.core import commands from redbot.core.bot import Red from redbot.core.utils.chat_formatting import bold, humanize_list from .model import PartialEmoji class EmojiInfo(commands.Cog): """Retrieve information about emojis.""" __author__: list[str] = ["SeaswimmerTheFsh"] __version__: str = "1.0.0" __documentation__: str = "https://seacogs.coastalcommits.com/emoji/" def __init__(self, bot: Red) -> None: super().__init__() self.bot: Red = bot self.logger: RedTraceLogger = getLogger(name="red.SeaCogs.Emoji") def format_help_for_context(self, ctx: commands.Context) -> str: pre_processed: Any | Literal[''] = super().format_help_for_context(ctx) or "" n: Literal['\n'] | Literal[''] = "\n" if "\n\n" not in pre_processed else "" text: list[str] = [ f"{pre_processed}{n}", f"Cog Version: **{self.__version__}**", f"Author: {humanize_list(items=self.__author__)}", f"Documentation: {self.__documentation__}", ] return "\n".join(text) async def fetch_twemoji(self, unicode_emoji) -> str: base_url = "https://cdn.jsdelivr.net/gh/jdecked/twemoji@latest/assets/72x72/" emoji_codepoint = "-".join([hex(ord(char))[2:] for char in unicode_emoji]) segments = emoji_codepoint.split("-") valid_segments = [seg for seg in segments if len(seg) >= 4] emoji_url = f"{base_url}{valid_segments[0]}.png" return emoji_url async def fetch_primary_color(self, emoji_url: str) -> discord.Color | None: async with aiohttp.ClientSession() as session: async with session.get(emoji_url) as response: if response.status != 200: return None image = await response.read() dominant_color = ColorThief(io.BytesIO(image)).get_color(quality=1) color = discord.Color.from_rgb(*dominant_color) return color @commands.hybrid_command(name="emoji") async def emoji(self, ctx: commands.Context, emoji: str, ephemeral: bool = False) -> None: """Retrieve information about an emoji.""" await ctx.defer(ephemeral=ephemeral) emoji: PartialEmoji = PartialEmoji.from_str(self, value=emoji) if emoji.is_unicode_emoji(): try: emoji_url = await self.fetch_twemoji(unicode_emoji=emoji.name) except Exception: await ctx.send("Please provide a valid emoji!", ephemeral=ephemeral) return else: emoji_url = emoji.url if emoji.id: emoji_id = f"{bold('ID:')} `{emoji.id}`\n" markdown = f"`<{'a' if emoji.animated else ''}:{emoji.name}:{emoji.id}>`" name = f"{bold('Name:')} {emoji.name}\n" aliases = "" else: emoji_id = "" markdown = f"`{emoji}`" name = f"{bold('Name:')} {emoji.aliases.pop(0)}\n" aliases = f"{bold('Aliases:')} {', '.join(emoji.aliases)}\n" if emoji.aliases else "" string: str = ( f"{name}" f"{emoji_id}" f"{bold('Native:')} {emoji.is_unicode_emoji()}\n" f"{aliases}" f"{bold('Animated:')} {emoji.animated}\n" f"{bold('Markdown:')} {markdown}\n" f"{bold('URL:')} [Click Here]({emoji_url})" ) if await ctx.embed_requested(): embed = embed = discord.Embed(title="Emoji Information", description=string, color = await self.fetch_primary_color(emoji_url) or await ctx.embed_color) embed.set_thumbnail(url=emoji_url) await ctx.send(embed=embed, ephemeral=ephemeral) else: await ctx.send(content=string, ephemeral=ephemeral)