from dbtexmf.dblatex.grubber.plugins import TexModule
from dbtexmf.dblatex.grubber.bibtex import BibTex


class BibSect(BibTex):
    def __init__(self, doc, bibaux, bibfiles, bibstyle):
        self.bibfiles = bibfiles.split(",")
        self.bibstyle = bibstyle
        self.bibaux = bibaux
        BibTex.__init__(self, doc, {}, bibaux)
        for bib in self.bibfiles:
            self.add_db(bib)
        if self.bibstyle:
            self.set_style(self.bibstyle)

class BibNull(BibSect):
    """
    Null biblio section having no bibfile
    """
    def __init__(self, doc, bibaux):
        pass

    def pre_compile(self):
        return 0

    def post_compile(self):
        return 0
        

class Bibtopic(TexModule):
    def __init__(self, doc, dict):
        self.doc = doc
        self.btsects = []
        doc.parser.add_hook("begin{btSect}", self.add_sect)

        # If loaded from a sect, register this sect too
        if dict["name"] == "begin{btSect}":
            self.add_sect(dict)

    def add_sect(self, dict):
        bibaux = "%s%d" % (self.doc.src_base, len(self.btsects) + 1)
        if dict["arg"]:
            btsect = BibSect(self.doc, bibaux, dict["arg"], dict["opt"])
        else:
            btsect = BibNull(self.doc, bibaux)
        self.btsects.append(btsect)

    def pre_compile(self):
        rc = 0
        for bib in self.btsects:
            rc += bib.pre_compile()
        return rc

    def post_compile(self):
        rc = 0
        for bib in self.btsects:
            rc += bib.post_compile()
        return rc


class Module(Bibtopic):
    """
    Module to load to handle bibtopic
    """