from discord.ext import commands
import discord
import animals


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

    @commands.slash_command(
        guild_ids=None, name="redpanda", description="Posts a photo of a redpanda"
    )
    async def redpanda(self, ctx: commands.Context):
        await ctx.respond(animals.get_red_panda())

    @commands.slash_command(
        guild_ids=None, name="dog", description="Posts a photo of a Dog"
    )
    async def dog(self, ctx: commands.Context):
        await ctx.respond(animals.get_dog())

    @commands.slash_command(
        guild_ids=None, name="sheeb", description="Posts a photo of a Sheeb"
    )
    async def sheeb(self, ctx: commands.Context):
        await ctx.defer()
        await ctx.send_followup(animals.random_sheeb())

    @commands.slash_command(
        guild_ids=None, name="cowboy", description="Posts a photo of a bad dog"
    )
    async def cowboy(self, ctx: commands.Context):
        await ctx.respond(animals.cowboy())

    @commands.slash_command(
        guild_ids=None,
        name="dale",
        description="Posts a photo of the goodest boy. Rest in power king",
    )
    async def dale(self, ctx: commands.Context):
        await ctx.defer()
        await ctx.send_followup(file=discord.File(animals.dale()))


def setup(bot):
    bot.add_cog(AnimalFunctions(bot))