import re
import requests

from bs4 import BeautifulSoup

from cloudbot import hook

xkcd_re = re.compile(r'(.*:)//(www.xkcd.com|xkcd.com)(.*)', re.I)
months = {1: 'January', 2: 'February', 3: 'March', 4: 'April', 5: 'May', 6: 'June', 7: 'July', 8: 'August',
          9: 'September', 10: 'October', 11: 'November', 12: 'December'}


def xkcd_info(xkcd_id, url=False):
    """ takes an XKCD entry ID and returns a formatted string """
    request = requests.get("http://www.xkcd.com/" + xkcd_id + "/info.0.json")
    data = request.json()
    date = "{} {} {}".format(data['day'], months[int(data['month'])], data['year'])
    if url:
        url = " | http://xkcd.com/" + xkcd_id.replace("/", "")
    return "xkcd: \x02{}\x02 ({}){}".format(data['title'], date, url if url else "")


def xkcd_search(term):
    search_term = requests.utils.quote(term)
    request = requests.get("http://www.ohnorobot.com/index.pl?s={}&Search=Search&"
                           "comic=56&e=0&n=0&b=0&m=0&d=0&t=0".format(search_term))
    soup = BeautifulSoup(request.text)
    result = soup.find('li')
    if result:
        url = result.find('div', {'class': 'tinylink'}).text
        xkcd_id = url[:-1].split("/")[-1]
        print(xkcd_id)
        return xkcd_info(xkcd_id, url=True)
    else:
        return "No results found!"


@hook.regex(xkcd_re)
def xkcd_url(match):
    xkcd_id = match.group(3).split(" ")[0].split("/")[1]
    return xkcd_info(xkcd_id)


@hook.command()
def xkcd(text):
    """xkcd <search term> - Search for xkcd comic matching <search term>"""
    return xkcd_search(text)