#!/usr/bin/python3

#

# Update HIB - Scrapper for the HumbleBundle library page.

# Copyright (C) 2012, Fabien Ninoles <- fabien - AT - tzone . org ->

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

# 

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from html.parser import HTMLParser

from pprint import pprint

import xml.dom

from itertools import chain

class Node:

    def __init__(self, **args):

        self.childs = []

        self.attrs = {}

        for arg in args:

            setattr(self, arg, args[arg])

        if self.name == "div" and "class" in self.attrs:

            self.tag = self.name

            self.name = self.attrs["class"]

            del self.attrs["class"]

        else:

            self.tag = self.name

    def format(self, prefix = ""):

        res = prefix + "<" + self.name

        for attr in self.attrs:

            if self.attrs[attr]:

                res += "\n" + prefix + "  " + attr + '="' + self.attrs[attr] + '"'

            else:

                res += "\n" + prefix + "  " + attr

        if self.name == "data":

            res += ">" + self.data + "</" + self.name + ">"

        elif self.childs:

            res += ">"

            for child in self.childs:

                res += "\n" + child.format(prefix + "  ")

            res += "\n" + prefix + "</" + self.name + ">"

        else:

            res += "/>"

        return res

    def find(self, prefix):

        for child in self.childs:

            if child.name.startswith(prefix):

                yield child

    def __getattr__(self, name):

        for child in self.childs:

            if child.name == name:

                setattr(self, name, child)

                return child

        raise AttributeError(name)

    def __repr__(self):

        return self.format()

class BundleParser(HTMLParser):

    def __init__(self, **args):

        super(BundleParser, self).__init__(**args)

        self.dom = Node(name = "root",

                        childs = [],

                        parent = None)

        self.current = self.dom

        self.depth = 1

    def handle_starttag(self, tag, attrs):

        # print("+" * self.depth,tag)

        new = Node(name = tag,

                   attrs = dict(attrs),

                   childs = [],

                   parent = self.current)

        self.current.childs.append(new)

        self.current = new

        self.depth += 1

    def handle_endtag(self, tag):

        while tag != self.current.tag:

            print("*** Skipping", self.current.tag,"; looking for",tag)

            self.current = self.current.parent

            self.depth-=1

            # print("-" * self.depth,self.current.tag)

            assert(self.current != self.dom)

        assert(self.current.tag == tag)

        self.depth-=1

        # print("-" * self.depth,tag)

        self.current = self.current.parent

    def handle_data(self, data):

        if data.strip():

            self.current.childs.append(Node(name = "data", data = data, childs = []))

def findRows(dom):

    for child in dom.childs:

        try:

            if child.name[:4] == "row ":

                yield child

            else:

                for row in findRows(child):

                    yield row

        except KeyError:

            pass

            for row in findRows(child):

                yield row

class Download:

    subst = { "arc32"         : ("x86",),

              "arc64"         : ("x64",),

              "i386.deb"      : ("x86","deb"),

              "x86_64.deb"    : ("x64", "deb"),

              "i686.rpm"      : ("x86", "rpm"),

              ".i386.rpm"     : ("x86", "rpm"),

              "x86_64.rpm"    : ("x64", "rpm"),

              ".x86_64.rpm"   : ("x64", "rpm"),

              "i386.tar.gz"   : ("x86", "tgz"),

              "x86_64.tar.gz" : ("x64", "tgz"),

              ".tar.gz"       : ("tgz",),

              ".deb"          : ("deb",),

              ".rpm"          : ("rpm",),

              "32-bit"        : ("x86",),

              "64-bit"        : ("x64",),              

              }

    def __init__(self, dltype, dom):

        self.dltype = dltype

        self.id = dom.name[len("download "):]

        button = list(dom.find("flexbtn "))[0]

        desc = button.span.data.data

        self.id += " " + desc

        def cleanup(attr):

            attr = attr.strip()

            if attr not in ("Download","small",""):

                for s in self.subst.get(attr,(attr,)):

                    yield s

        self.attrs = set(chain.from_iterable(cleanup(attr) for attr in self.id.split(" ")))

        urls = button.a.attrs

        self.torrent = urls["data-bt"]

        self.web = urls["data-web"]

        details = dom.dldetails.dlsize

        if details.childs:

            self.size = details.span.data.data

            self.md5 = details.a.attrs["href"]

        else:

            self.size = "Unknown"

            self.md5 = "Unknown"

    def format(self, prefix=""):

        res = prefix + '<download id="' + self.id + '">\n'

        res += prefix + "  <web>" + self.web + "</web>\n"

        res += prefix + "  <torrent>" + self.torrent + "</torrent>\n"

        res += prefix + "  <size>" + self.size + "</size>\n"

        res += prefix + "  <md5>" + self.md5 + "</md5>\n"

        res += prefix + "</download>"

        return res

    def __repr__(self):

        return self.format()

class Downloads:

    def __init__(self, dom):

        self.id = dom.name[len("downloads "):].split(" ")[0]

        self.elements = []

        self.others = []

        self.addchilds(dom)

    def addchilds(self, dom):

        for child in dom.childs:

            if child.name.startswith("downloads"):

                self.addchilds(child)

            elif child.name.startswith("download"):

                self.elements.append(Download(self.id, child))

            elif child.name == "arc-toggle":

                self.addchilds(child)

            elif child.name in ("clearfix","label"):

                pass

            else:

                self.others.append(child)

    def __iter__(self):

        return iter(self.elements)

    def format(self, prefix = ""):

        res = prefix + '<downloads id="' + self.id + '">\n'

        if self.elements:

            for el in self.elements:

                res += el.format(prefix + "  ") + "\n"

        if self.others:

            res += prefix + "  <others>\n"

            for o in self.others:

                res += o.format(prefix + "    ") + "\n"

            res += prefix + "  </others>\n"

        res += prefix + "</downloads>"

        return res

    def __repr__(self):

        return self.format()

class Game:

    def __init__(self, dom):

        self.title = "unknown"

        self.downloads = []

        self.others = []

        for child in dom.childs:

            if child.name == "gameinfo":

                self.title = dom.gameinfo.title.a.data.data.strip()

            elif child.name.startswith("downloads "):

                self.downloads.append(Downloads(child))

            elif child.name in ["icn", "clearfix"]:

                pass

            else:

                self.others.append(child)

    def __repr__(self):

        res  = "<game>\n"

        res += "  <title>" + self.title + "</title>\n"

        if self.downloads:

            res += "  <downloads>\n"

            for dl in self.downloads:

                res += dl.format("    ") + "\n"

            res += "  </downloads>\n"

        if self.others:

            res += "  <others>\n"

            for o in self.others:

                res += o.format("    ") + "\n"

            res += "  </others>\n"

        res += "</game>"

        return res

def parseGamesFromDom(dom):

    for row in findRows(dom):

        yield Game(row)

def parseGamesFromFile(filename):

    parser = BundleParser()

    for l in open(filename):

        parser.feed(l)

    for game in parseGamesFromDom(parser.dom):

        yield game

class FileSelector:    

    def scoreDownload(self, dl):

        if dl.dltype == "audio":

            if "FLAC" in dl.attrs:

                return 2

            if "MP3" in dl.attrs:

                return 1

            if "website" in dl.attrs:

                return -1

            raise Exception("Unknow audio type: %r" % (dl.attrs))

        if dl.dltype in ("mac","windows"):

            return -1

        if dl.dltype == "linux":

            score = 1

            if "x64" in dl.attrs:

                score += 1

            if "deb" in dl.attrs:

                score += 1

            return score

        if dl.dltype == "android":

            return -1

        raise Exception("Unknown dls type: %r" % (dl,))

    def chooseDownloads(self, dls):

        return sorted(((self.scoreDownload(dl),dl) for dl in dls), key=lambda x: x[0], reverse=True)

    def __call__(self, dls):

        return self.chooseDownloads(dls)

def main(fn):

    selector = FileSelector()

    downloads = []

    for game in parseGamesFromFile(fn):

        for dls in game.downloads:

            scores = selector(dls)

            choosen = list(dl for score, dl in scores if score >= 0)[:1]

            for score, dl in scores:

                print("[%s] %2d | %-20s | %-10s | %-25s | %s " % (

                        "*" if dl in choosen else " ",

                        score,

                        game.title, 

                        dls.id,

                        ", ".join(sorted(dl.attrs)),

                        dl.torrent))

                if dl in choosen:

                    downloads.append(dl)

            if not scores:

                print("No download for",dls.id)

            print("-" * 80)

    import urllib.request

    import urllib.parse

    import os

    opener = urllib.request.build_opener()

    for dl in downloads:

        fn = os.path.basename(urllib.parse.urlsplit(dl.torrent).path)

        print("Saving",dl.torrent,"as",fn)

        with opener.open(dl.torrent) as u:

            with open(fn,"wb") as f:

                f.write(u.read())

if __name__ == '__main__':

    import sys

    main(sys.argv[1])