Geoffrey-Django/geoffrey/cogs/Search_Commands.py

from discord.ext import commands

from geoffrey.BotErrors import *
from geoffrey.DiscordHelperFunctions import *


class Search_Commands:
    """
    Commands to find stuff.
    """

    def __init__(self, bot):
        self.bot = bot

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def find(self, ctx, *args):
        """
        Finds all the locations matching the search term
            ?find [Search]
        """
        search = get_name(args)
        try:

            if search is None:
                raise commands.UserInputError

            result = self.bot.bot_commands.find(search)

            await ctx.send(
                '{}, The following entries match **{}**:\n{}'.format(ctx.message.author.mention, search, result))
        except LocationLookUpError:
            await ctx.send(
                '{}, no matches to **{}** were found in the database.'.format(ctx.message.author.mention, search))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def tunnel(self, ctx, player: str):
        """
        Finds all the tunnels a player owns
            ?tunnel [Player]
        """
        try:
            result = self.bot.bot_commands.tunnel(player)

            await ctx.send(
                '{}, **{}** owns the following tunnel(s): \n{}'.format(ctx.message.author.mention, player, result))
        except LocationLookUpError:
            await ctx.send('{}, no tunnels for **{}** were found in the database.'
                           .format(ctx.message.author.mention, player))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def find_around(self, ctx, x_pos: int, z_pos: int, *args):
        """
        Finds all the locations around a certain point.
        The radius defaults to 200 blocks if no value is given

        Default dimension is the overworld
            ?find_around [X Coordinate] [Z Coordinate] [Radius] [Optional Flags]

            Optional Flags:
            -d [dimension]
        """
        radius = 200
        dimension = 'Overworld'

        try:
            if len(args) > 0:
                if args[0] == '-d':
                    dimension = args[1]
                    if len(args) > 1:
                        radius = int(args[2])
                else:
                    radius = int(args[0])

                    if len(args) > 1:
                        if args[1] == '-d':
                            dimension = args[2]

            base_string = self.bot.bot_commands.find_around(x_pos, z_pos, radius, dimension)

            if len(base_string) != 0:
                await ctx.send('{}, the following locations(s) are within **{}** blocks of that point: \n {}'.format(
                    ctx.message.author.mention, radius, base_string))
            else:
                await ctx.send('{}, there are no locations within {} blocks of that point'
                               .format(ctx.message.author.mention, radius))
        except ValueError:
            await ctx.send(
                '{}, invalid radius, the radius must be a whole number.'.format(ctx.message.author.mention,
                                                                                radius))
        except InvalidDimError:
            await ctx.send('{}, {} is an invalid dimension.'.format(ctx.message.author.mention, dimension))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def selling(self, ctx, *args):
        """
        Lists all the shops selling an item

            ?selling [item]
        """
        item_name = get_name(args)
        try:
            result = self.bot.bot_commands.selling(item_name)
            await ctx.send(
                '{}, the following shop(s) sell **{}**: \n{}'.format(ctx.message.author.mention, item_name, result))
        except ItemNotFound:
            await ctx.send('{}, no shop sells **{}**.'.format(ctx.message.author.mention, item_name))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def info(self, ctx, *args):
        """
        Displays info about a location.

        If the location is a shop, it displays the shop's inventory
            ?info [Location Name]
        """
        loc = get_name(args)
        try:

            if loc is None:
                raise commands.UserInputError

            info_str = self.bot.bot_commands.info(loc)
            await ctx.send(info_str)
        except IndexError:
            await ctx.send('{}, no locations in the database match **{}**.'.format(ctx.message.author.mention, loc))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def me(self, ctx):
        """
        Displays all your locations in the database
        """
        try:
            loc_str = self.bot.bot_commands.me(discord_uuid=ctx.message.author.id)
            await ctx.send('{}, here are your location(s) in the database: \n {}'.format(ctx.message.author.mention,
                                                                                         loc_str))
        except PlayerNotFound:
            await ctx.send('{}, you don\'t have any locations in the database.'.format(ctx.message.author.mention))


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