"""Contains the LookupCog, which deals with the lookup commands."""
from discord.ext import commands  # Has the cog class
from discord_slash import cog_ext  # Used for slash commands

from gwendolyn.utils import get_params  # pylint: disable=import-error

params = get_params()


class LookupCog(commands.Cog):
    """Contains the lookup commands."""

    def __init__(self, bot):
        """Initialize the cog."""
        self.bot = bot

    # Looks up a spell
    @cog_ext.cog_slash(**params["spell"])
    async def spell(self, ctx, query):
        """Look up a spell."""
        await self.bot.lookup_funcs.spellFunc(ctx, query)

    # Looks up a monster
    @cog_ext.cog_slash(**params["monster"])
    async def monster(self, ctx, query):
        """Look up a monster."""
        await self.bot.lookup_funcs.monsterFunc(ctx, query)


def setup(bot):
    """Add the cog to the bot."""
    bot.add_cog(LookupCog(bot))