import lxml.etree
import glob
import sys
import re
import Bcfg2.Server.Admin
import Bcfg2.Options
from Bcfg2.Server.Plugins.Metadata import MetadataConsistencyError

class Bundle(Bcfg2.Server.Admin.MetadataCore):
    __shorthelp__ = "Create or delete bundle entries"
    __longhelp__ = (__shorthelp__ + #"\n\nbcfg2-admin bundle add <bundle> "
                                    #"\n\nbcfg2-admin bundle del <bundle>"
                                    "\n\nbcfg2-admin bundle list-xml"
                                    "\nbcfg2-admin bundle list-genshi"
                                    "\nbcfg2-admin bundle show\n")
    __usage__ = ("bcfg2-admin bundle [options] [add|del] [group]")

    def __init__(self, configfile):
        Bcfg2.Server.Admin.MetadataCore.__init__(self, configfile,
                                                 self.__usage__)

    def __call__(self, args):
        Bcfg2.Server.Admin.MetadataCore.__call__(self, args)
        reg='((?:[a-z][a-z\\.\\d\\-]+)\\.(?:[a-z][a-z\\-]+))(?![\\w\\.])'

        # Get all bundles out of the Bundle/ directory
        opts = {'repo': Bcfg2.Options.SERVER_REPOSITORY}
        setup = Bcfg2.Options.OptionParser(opts)
        setup.parse(sys.argv[1:])
        repo = setup['repo']
        xml_list = glob.glob("%s/Bundler/*.xml" % repo)
        genshi_list = glob.glob("%s/Bundler/*.genshi" % repo)

        if len(args) == 0:
            self.errExit("No argument specified.\n"
                         "Please see bcfg2-admin bundle help for usage.")
#        if args[0] == 'add':
#            try:
#                self.metadata.add_bundle(args[1])
#            except MetadataConsistencyError:
#                print "Error in adding bundle."
#                raise SystemExit(1)
#        elif args[0] in ['delete', 'remove', 'del', 'rm']:
#            try:
#                self.metadata.remove_bundle(args[1])
#            except MetadataConsistencyError:
#                print "Error in deleting bundle."
#                raise SystemExit(1)
        # Lists all available xml bundles
        elif args[0] in ['list-xml', 'ls-xml']:
            bundle_name = []
            for bundle_path in xml_list:
                rg = re.compile(reg,re.IGNORECASE|re.DOTALL)
                bundle_name.append(rg.search(bundle_path).group(1))
            for bundle in bundle_name:
               print bundle.split('.')[0]
        # Lists all available genshi bundles
        elif args[0] in ['list-genshi', 'ls-gen']:
            bundle_name = []
            for bundle_path in genshi_list:
                rg = re.compile(reg,re.IGNORECASE|re.DOTALL)
                bundle_name.append(rg.search(bundle_path).group(1))
            for bundle in bundle_name:
               print bundle.split('.')[0]
        # Shows a list of all available bundles and prints bundle 
        # details after the user choose one bundle.
        # FIXME: Add support for detailed output of genshi bundles
        # FIXME: This functionality is almost identical with
        #        bcfg2-info bundles
        elif args[0] in ['show']:
            bundle_name = []
            bundle_list = xml_list + genshi_list
            for bundle_path in bundle_list:
                rg = re.compile(reg,re.IGNORECASE|re.DOTALL)
                bundle_name.append(rg.search(bundle_path).group(1))
            text = "Available bundles (Number of bundles: %s)" % \
                    (len(bundle_list))
            print text
            print "%s" % (len(text) * "-") 
            for i in range(len(bundle_list)):
                print "[%i]\t%s" % (i, bundle_name[i])
            print "Enter the line number of a bundle for details:",
            lineno = raw_input()
            if int(lineno) >= int(len(bundle_list)):
                print "No line with this number."
            else:
                if '%s/Bundler/%s' % \
                            (repo, bundle_name[int(lineno)]) in genshi_list:
                    print "Detailed output for *.genshi bundles is not supported."
                else:
                    print 'Details for the "%s" bundle:' % \
                            (bundle_name[int(lineno)].split('.')[0])
                    tree = lxml.etree.parse(bundle_list[int(lineno)])
                    #Prints bundle content
                    #print lxml.etree.tostring(tree)
                    names = ['Action', 'Package', 'Path', 'Service']
                    for name in names:
                        for node in tree.findall("//" + name):
                    	    print "%s:\t%s" % (name, node.attrib["name"])
        else:
            print "No command specified"
            raise SystemExit(1)