import discord
from discord.ext import commands

# Emotes for badges
badges_emotes = {
    "Active Developer": "<:activedev:1232880809078165657>",
    "Booster": "<:boostbadge:1232881130391474238>",
    "Bug Hunter": "<:bughunterbadge:1232881643841392732>",
    "Certified Discord Moderator": "<:discordcertifiedmoderator:1232882888467419190>",
    "Early Supporter": "<:earlysupporter:1232883022391676929>",
    "HypeSquad": "<:hypesquadbadge:1232883153283190914>",
    "HypeSquad Balance": "<:hypesquadbalance:1232883920316661852>",
    "HypeSquad Bravery": "<:hypesquadbravery:1232884053506654239>",
    "HypeSquad Brilliance": "<:hypesquadbrilliance:1232884259090337913>",
    "Staff": "<:staff:1232884672057573387>",
    "Partner": "<:partner:1232884540742303784>",
    "Verified Bot Developer": "<:verifiedbotdeveloper:1232884870762598481>",
    "Nitro": "<:nitro:1232884410660163636>",
    "Bot": "<:bot:1248681234633195622>"
}

class Information(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.command(name='userinfo')
    async def userinfo(self, ctx, *, user: discord.User = None):
        if user is None:
            user = ctx.author

        badges = []

        if isinstance(user, discord.Member):
            user_flags = user.public_flags
            if user_flags.staff:
                badges.append(badges_emotes["Staff"])
            if user_flags.partner:
                badges.append(badges_emotes["Partner"])
            if user_flags.hypesquad:
                badges.append(badges_emotes["HypeSquad"])
            if user_flags.bug_hunter:
                badges.append(badges_emotes["Bug Hunter"])
            if user_flags.hypesquad_balance:
                badges.append(badges_emotes["HypeSquad Balance"])
            if user_flags.hypesquad_bravery:
                badges.append(badges_emotes["HypeSquad Bravery"])
            if user_flags.hypesquad_brilliance:
                badges.append(badges_emotes["HypeSquad Brilliance"])
            if user_flags.early_supporter:
                badges.append(badges_emotes["Early Supporter"])
            if user_flags.verified_bot_developer:
                badges.append(badges_emotes["Verified Bot Developer"])
            if user_flags.active_developer:
                badges.append(badges_emotes["Active Developer"])
            if user.premium_since:
                badges.append(badges_emotes["Booster"])
            if user.premium_type:
                badges.append(badges_emotes["Nitro"])
        else:
            user_flags = await self.bot.fetch_user(user.id)
            if user_flags.public_flags.staff:
                badges.append(badges_emotes["Staff"])
            if user_flags.public_flags.partner:
                badges.append(badges_emotes["Partner"])
            if user_flags.public_flags.hypesquad:
                badges.append(badges_emotes["HypeSquad"])
            if user_flags.public_flags.bug_hunter:
                badges.append(badges_emotes["Bug Hunter"])
            if user_flags.public_flags.hypesquad_balance:
                badges.append(badges_emotes["HypeSquad Balance"])
            if user_flags.public_flags.hypesquad_bravery:
                badges.append(badges_emotes["HypeSquad Bravery"])
            if user_flags.public_flags.hypesquad_brilliance:
                badges.append(badges_emotes["HypeSquad Brilliance"])
            if user_flags.public_flags.early_supporter:
                badges.append(badges_emotes["Early Supporter"])
            if user_flags.public_flags.verified_bot_developer:
                badges.append(badges_emotes["Verified Bot Developer"])
            if user_flags.public_flags.active_developer:
                badges.append(badges_emotes["Active Developer"])
            if user_flags.premium_type:
                badges.append(badges_emotes["Nitro"])

        if user.bot:
            badges.append(badges_emotes["Bot"])

        badges_str = ' '.join(badges) if badges else 'No badges'
        embed = discord.Embed(title=f"{user}'s Badges", description=badges_str, color=discord.Color.blue())
        await ctx.send(embed=embed)

async def setup(bot):
    await bot.add_cog(Information(bot))