import re
from urllib.parse import urlencode

from cloudbot import hook
from cloudbot.util import http, web, formatting

sc_re = re.compile(r'(.*:)//(www.)?(soundcloud.com)(.*)', re.I)
api_url = "http://api.soundcloud.com"
sndsc_re = re.compile(r'(.*:)//(www.)?(snd.sc)(.*)', re.I)


def soundcloud(url, api_key):
    data = http.get_json(api_url + '/resolve.json?' + urlencode({'url': url, 'client_id': api_key}))

    if data['description']:
        desc = ": {} ".format(formatting.truncate(data['description'], 50))
    else:
        desc = ""
    if data['genre']:
        genre = "- Genre: \x02{}\x02 ".format(data['genre'])
    else:
        genre = ""

    url = web.try_shorten(data['permalink_url'])

    return "SoundCloud track: \x02{}\x02 by \x02{}\x02 {}{}- {} plays, {} downloads, {} comments - {}".format(
        data['title'], data['user']['username'], desc, genre, data['playback_count'], data['download_count'],
        data['comment_count'], url)


@hook.regex(sc_re)
def soundcloud_url(match, bot):
    api_key = bot.config.get("api_keys", {}).get("soundcloud")
    if not api_key:
        print("Error: no api key set")
        return None
    url = match.group(1).split(' ')[-1] + "//" + (match.group(2) if match.group(2) else "") + match.group(3) + \
        match.group(4).split(' ')[0]
    return soundcloud(url, api_key)


@hook.regex(sndsc_re)
def sndsc_url(match, bot):
    api_key = bot.config.get("api_keys", {}).get("soundcloud")
    if not api_key:
        print("Error: no api key set")
        return None
    url = match.group(1).split(' ')[-1] + "//" + (match.group(2) if match.group(2) else "") + match.group(3) + \
        match.group(4).split(' ')[0]
    return soundcloud(http.open(url).url, api_key)