From 2293a7d569f909f66e56dd8123b9adbbdf679b04 Mon Sep 17 00:00:00 2001
From: Will Woods <wwoods@redhat.com>
Date: Fri, 13 May 2011 23:58:46 -0400
Subject: [PATCH] treebuilder WIP: add commands for installtree

---
 src/pylorax/treebuilder.py | 37 +++++++++++++++++++++++++++++++++----
 1 file changed, 33 insertions(+), 4 deletions(-)

diff --git a/src/pylorax/treebuilder.py b/src/pylorax/treebuilder.py
index 7d9f72ae..e9f8d5b7 100644
--- a/src/pylorax/treebuilder.py
+++ b/src/pylorax/treebuilder.py
@@ -191,11 +191,8 @@ class TreeBuilder(BaseBuilder):
 # "mkdir", "treeinfo", "runcmd", "remove", "replace" will take multiple args
 
 # TODO: to replace installtree:
-#       find(glob)
-#       installpkg/removepkg pkgglob [pkgglob..]
-#       run_pkg_transaction
-#       removefrom [pkgname] glob [glob..]
 #       module modname [modname...]
+#       get yum object somehow
 
 class TemplateRunner(object):
     commands = ('install', 'mkdir', 'replace', 'append', 'treeinfo',
@@ -318,3 +315,35 @@ class TemplateRunner(object):
             chdir = lambda: os.chdir(dirname)
             cmd = cmd[1:]
         check_call(cmd, preexec_fn=chdir)
+
+    def module(self, *modnames):
+        for mod in modnames:
+            # XXX TODO surely this code is elsewhere?
+            # expand groups
+            # resolve deps
+            # get firmware
+            pass
+
+    def installpkg(self, *pkgs):
+        for p in pkgs:
+            self.yum.install(pattern=p)
+
+    def removepkg(self, *pkgs):
+        for p in pkgs:
+            self.yum.remove(pattern=p)
+
+    def run_pkg_transaction(self):
+        self.yum.buildTransaction()
+        self.yum.repos.setProgressBar(LoraxDownloadCallback())
+        self.yum.processTransaction(callback=LoraxTransactionCallback(),
+                                    rpmDisplay=LoraxRpmCallback())
+        self.yum.closeRpmDB()
+
+    def removefrom(self, pkg, *globs):
+        globs_re = re.compile("|".join([fnmatch.translate(g) for g in globs]))
+        pkg_files = []
+        for pkgobj in self.yum.doPackageLists(pkgnarrow="installed", patterns=[pkg]):
+            pkg_files += pkgobj.filelist
+        remove = filter(globs_re.match(pkg_files))
+        logger.debug("removing %i files from %s", len(remove), pkg)
+        self.remove(*remove)