import discord
import asyncio
import urllib.request
import requests
from discord.ext import commands

class ShortenUrl:

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

    @commands.command(pass_context=True)
    """
    Checks if the user exists in the database and returns a shortened link on success.
    
    Database format is {"USERID":"APIKEY","117450987848990720":"B3Nm9g8L?s="}
    """
    async def ouo(self, ctx):
        with open('data.json') as data_file:
            data_loaded = json.load(data_file)
        
        if data_loaded.get(ctx.message.author.id):        
            url = ("http://ouo.io/api/" + data_loaded.get(ctx.message.author.id) + ctx.message.content)
            handle = urllib.request.Request(url, headers={'User-Agent': 'Mozilla/5.0'})
            html = urllib.request.urlopen(handle).read().decode('utf-8')
            await self.bot.say("Shortened: " + html)
        else:
            await self.bot.say("nope")

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