import lxml.etree
import Bcfg2.Server.Lint
                        
class Bundles(Bcfg2.Server.Lint.ServerPlugin):
    """ Perform various bundle checks """

    @Bcfg2.Server.Lint.returnErrors
    def Run(self):
        """ run plugin """
        if 'Bundler' in self.core.plugins:
            self.missing_bundles()
            for bundle in self.core.plugins['Bundler'].entries.values():
                if self.HandlesFile(bundle.name):
                    if (Bcfg2.Server.Plugins.Bundler.have_genshi and
                        type(bundle) is
                        Bcfg2.Server.Plugins.SGenshi.SGenshiTemplateFile):
                        self.sgenshi_groups(bundle)
                    else:
                        self.bundle_names(bundle)

    def missing_bundles(self):
        """ find bundles listed in Metadata but not implemented in Bundler """
        if self.files is None:
            # when given a list of files on stdin, this check is
            # useless, so skip it
            groupdata = self.metadata.groups_xml.xdata
            ref_bundles = set([b.get("name")
                               for b in groupdata.findall("//Bundle")])

            allbundles = self.core.plugins['Bundler'].entries.keys()
            for bundle in ref_bundles:
                xmlbundle = "%s.xml" % bundle
                genshibundle = "%s.genshi" % bundle
                if (xmlbundle not in allbundles and
                    genshibundle not in allbundles):
                    self.LintError("bundle-not-found",
                                   "Bundle %s referenced, but does not exist" %
                                   bundle)

    def bundle_names(self, bundle):
        """ verify bundle name attribute matches filename """
        try:
            xdata = lxml.etree.XML(bundle.data)
        except AttributeError:
            # genshi template
            xdata = lxml.etree.parse(bundle.template.filepath).getroot()
            
        fname = bundle.name.split('Bundler/')[1].split('.')[0]
        bname = xdata.get('name')
        if fname != bname:
            self.LintError("inconsistent-bundle-name",
                           "Inconsistent bundle name: filename is %s, bundle name is %s" %
                           (fname, bname))

    def sgenshi_groups(self, bundle):
        """ ensure that Genshi Bundles do not include <Group> tags,
        which are not supported  """
        xdata = lxml.etree.parse(bundle.name)
        groups = [self.RenderXML(g)
                  for g in xdata.getroottree().findall("//Group")]
        if groups:
            self.LintError("group-tag-not-allowed",
                           "<Group> tag is not allowed in SGenshi Bundle:\n%s" %
                           "\n".join(groups))