#!/usr/bin/env python

from re import compile

from Generator import Generator
from GeneratorUtils import DirectoryBacked, XMLFileBacked
from Types import Package

class PackageEntry(XMLFileBacked):
    __identifier__ = 'image'
    rpm = compile('^(?P<name>[\w\+\d\.]+(-[\w\+\d\.]+)*)-(?P<version>[\w\d\.]+-([\w\d\.]+))\.(?P<arch>\w+)\.rpm$')

    def Index(self):
        XMLFileBacked.Index(self)
        self.packages = {}
        for location in self.entries:
            for pkg in location.getchildren():
                if pkg.attrib.has_key("file"):
                    m = self.rpm.match(pkg.attrib['file'])
                    if not m:
                        print "failed to rpm match %s"%(pkg.attrib['file'])
                        continue
                    self.packages[m.group('name')] = m.groupdict()
                    self.packages[m.group('name')]['file'] = pkg.attrib['file']
                    self.packages[m.group('name')]['uri'] = location.attrib['uri']
                    self.packages[m.group('name')]['type'] = 'rpm'
                else:
                    self.packages[pkg.attrib['name']] = pkg.attrib

class PackageDir(DirectoryBacked):
    __child__ = PackageEntry

class pkgmgr(Generator):
    '''This is a generator that handles service assignments'''
    __name__ = 'pkgmgr'
    __version__ = '$Id$'
    __author__ = 'bcfg-dev@mcs.anl.gov'

    def __setup__(self):
        self.pkgdir=PackageDir(self.data,self.core.fam)

    def FindHandler(self, entry):
        if entry.tag != 'Package':
            raise KeyError, (entry.tag, entry.attrib['name'])
        return self.LocatePackage

    def LocatePackage(self, entry, metadata):
        pkgname = entry.attrib['name']
        pl = self.pkgdir["%s.xml"%(metadata.image)]
        if pl.packages.has_key(pkgname):
            p = pl.packages[pkgname]
            if p['type'] == 'rpm':
                entry.attrib.update({'url':"%s/%s"%(p['uri'],p['file']), 'version':p['version']})
        else:
            raise KeyError, ("Package", name)