import discord
from discord.ext import commands
from random import randint
from imgurpython import ImgurClient
import aiohttp
import random

class Image:
    """Image related commands."""

    def __init__(self, bot):
        self.bot = bot
        #Reserved for further ... stuff

    """Commands section"""

    @commands.command(no_pm=True)
    async def imgur(self, *text):
        """Retrieves a random imgur picture.
        imgur search [keyword] - retrieves first hit of search query.
        imgur [subreddit section] [top or new] - retrieves top 3 hottest or latest pictures of today for given a subreddit section, e.g. 'funny'."""
        imgurclient = ImgurClient("1fd3ef04daf8cab", "f963e574e8e3c17993c933af4f0522e1dc01e230")
        if text == ():
            rand = randint(0, 59) #60 results per generated page
            items = imgurclient.gallery_random(page=0)
            await self.bot.say(items[rand].link)
        elif text[0] == "search":
            items = imgurclient.gallery_search(" ".join(text[1:len(text)]), advanced=None, sort='time', window='all', page=0)
            if len(items) < 1:
                await self.bot.say("Your search terms gave no results.")
            else:
                await self.bot.say(items[0].link)
        elif text[0] != ():
            if text[1] == "top":
                imgSort = "top"
            elif text[1] == "new":
                imgSort = "time"
            else:
                await self.bot.say("Only top or new is a valid subcommand.")
                return
            items = imgurclient.subreddit_gallery(text[0], sort=imgSort, window='day', page=0)
            if (len(items) < 3):
                await self.bot.say("This subreddit section does not exist, try 'funny'")
            else:
                await self.bot.say("{} {} {}".format(items[0].link, items[1].link, items[2].link))

    @commands.command(no_pm=True)
    async def gif(self, *text):
        random.seed()
        """ gif [keyword] - retrieves first search result from giphy """
        if len(text) > 0:
            if len(text[0]) > 1 and len(text[0]) < 20:
                try:
                    msg = "+".join(text)
                    search = "http://api.giphy.com/v1/gifs/search?q=" + msg + "&api_key=dc6zaTOxFJmzC"
                    async with aiohttp.get(search) as r:
                        result = await r.json()
                    if result["data"] != []:
                        maxarray = len(result)
                        url = result["data"][random.randint(0,maxarray)]["url"]
                        await self.bot.say(url)
                    else:
                        await self.bot.say("Your search terms gave no results.")
                except:
                    await self.bot.say("Error.")
            else:
                await self.bot.say("Invalid search.")
        else:
            await self.bot.say("gif [text]")

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