SeaCogs/emojiinfo/emojiinfo.py

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)