1 files changed, 96 insertions, 0 deletions
diff --git a/src/lib/Bcfg2/Server/Admin/Bundle.py b/src/lib/Bcfg2/Server/Admin/Bundle.py
new file mode 100644
index 000000000..89c099602
--- /dev/null
+++ b/src/lib/Bcfg2/Server/Admin/Bundle.py
@@ -0,0 +1,96 @@
+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"
+    # TODO: add/del functions
+    __longhelp__ = (__shorthelp__ + "\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 __call__(self, args):
+        Bcfg2.Server.Admin.MetadataCore.__call__(self, args)
+        rg = re.compile(r'([^.]+\.(?:[a-z][a-z\-]+))(?![\w\.])',
+                        re.IGNORECASE | re.DOTALL)
+
+        # Get all bundles out of the Bundle/ directory
+        repo = self.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:
+                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:
+                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:
+                print "matching %s" % bundle_path
+                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]))
+            try:
+                lineno = raw_input("Enter the line number of a bundle for details: ")
+            except NameError:
+                lineno = input("Enter the line number of a bundle for details: ")
+            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)