import hikari
import lightbulb

plugin = lightbulb.Plugin("ProfilePlugin")


@plugin.command
@lightbulb.option("img", "image to set as new avatar", type=hikari.Attachment, required=True)
@lightbulb.add_checks(lightbulb.owner_only)
@lightbulb.command("avatar", "set the bot avatar.", ephemeral=True)
@lightbulb.implements(lightbulb.SlashCommand)
async def set_avatar(ctx: lightbulb.Context) -> None:
    await ctx.bot.rest.edit_my_user(avatar=ctx.options.img)
    embed = hikari.Embed(title="New avatar set!")
    embed.set_image(ctx.options.img)
    await ctx.respond(embed)


@plugin.command
@lightbulb.option("text", "text to set as custom status", type=str, required=True)
@lightbulb.add_checks(lightbulb.owner_only)
@lightbulb.command("status", "set the bot status.", ephemeral=True)
@lightbulb.implements(lightbulb.SlashCommand)
async def set_status(ctx: lightbulb.Context) -> None:
    await ctx.bot.update_presence(activity=hikari.Activity(name=ctx.options.text, type=hikari.ActivityType.CUSTOM))
    embed = hikari.Embed(title="New status set!", description=f"New Status: {ctx.options.text}")
    await ctx.respond(embed)


def load(bot: lightbulb.BotApp):
    bot.add_plugin(plugin)


def unload(bot: lightbulb.BotApp):
    bot.remove_plugin(plugin)