import json, re
import Utils

URL_GOOGLEBOOKS = "https://www.googleapis.com/books/v1/volumes"
URL_BOOKINFO = "https://books.google.co.uk/books?id=%s"
REGEX_BOOKID = re.compile("id=([\w\-]+)")

class Module(object):
    _name = "ISBN"
    def __init__(self, bot):
        self.bot = bot
        bot.events.on("received").on("command").on("isbn").hook(
            self.isbn, help="Get book information from a provided ISBN",
            min_args=1, usage="<isbn>")
        bot.events.on("received").on("command").on("book").hook(
            self.book, help="Get book information from a provided title",
            min_args=1, usage="<book title>")

    def get_book(self, query, event):
        page = Utils.get_url(URL_GOOGLEBOOKS, get_params={
            "q": query, "country": "us"}, json=True)
        if page:
            if page["totalItems"] > 0:
                book = page["items"][0]["volumeInfo"]
                print(book)
                title = book["title"]
                sub_title = (", %s" % book.get("subtitle")
                    ) if book.get("subtitle") else ""
                authors = ", ".join(book["authors"])
                date = book["publishedDate"]
                rating = book["averageRating"]
                id = re.search(REGEX_BOOKID, book["infoLink"]).group(1)
                info_link = URL_BOOKINFO % id
                event["stdout"].write("%s - %s (%s)%s %s (%s/5.0)" % (
                    title, authors, date, sub_title, info_link, rating))
            else:
                event["stderr"].write("Unable to find book")
        else:
            event["stderr"].write("Failed to load results")

    def isbn(self, event):
        isbn = event["args_split"][0]
        if len(isbn) == 10:
            isbn = "978%s" % isbn
        isbn = isbn.replace("-", "")
        self.get_book("isbn:%s" % isbn, event)

    def book(self, event):
        self.get_book(event["args"], event)