From fd492bebc6e544745bd0a85f2b069b7e54abfe75 Mon Sep 17 00:00:00 2001
From: Seth Vidal <skvidal@fedoraproject.org>
Date: Thu, 2 Aug 2012 17:02:37 -0400
Subject: [PATCH] - add a check for repoquery so we can abort politely - make
 the results output a bit more readable - fix up where repoquery is looked for
 so it is easier to change

---
 library/yum | 44 +++++++++++++++++++++++++-------------------
 1 file changed, 25 insertions(+), 19 deletions(-)

diff --git a/library/yum b/library/yum
index d43419ceb8a..58a33d9bf6d 100755
--- a/library/yum
+++ b/library/yum
@@ -23,6 +23,8 @@ import traceback
 import os
 
 def_qf = "%{name}-%{version}-%{release}.%{arch}"
+repoquery='/usr/bin/repoquery'
+yumbin='/usr/bin/yum'
 
 def is_installed(repoq, pkgspec, qf=def_qf):
     cmd = repoq + "--disablerepo=\* --pkgnarrow=installed --qf '%s' %s " % (qf, pkgspec)
@@ -104,9 +106,9 @@ def repolist(repoq, qf="%{repoid}"):
     
 def list_stuff(conf_file, stuff):
     qf = "%{name}|%{epoch}|%{version}|%{release}|%{arch}|%{repoid}"
-    repoq = '/usr/bin/repoquery --plugins --quiet  -q '
+    repoq = '%s --plugins --quiet  -q ' % repoquery
     if conf_file and os.path.exists(conf_file):
-        repoq = '/usr/bin/repoquery -c %s --plugins --quiet  -q ' % conf_file
+        repoq = '%s -c %s --plugins --quiet  -q ' % (repoquery,conf_file)
 
     
     if stuff == 'installed':
@@ -146,7 +148,7 @@ def run(command):
 
 def install(module, items, repoq, yum_basecmd):
     res = {}
-    res['results'] = ''
+    res['results'] = []
     res['msg'] = ''
     res['rc'] = 0
     res['changed'] = False
@@ -185,7 +187,7 @@ def install(module, items, repoq, yum_basecmd):
             for this in pkglist:
                 if is_installed(repoq, this):
                     found = True
-                    res['results'] += '%s providing %s is already installed\n' % (this, spec)
+                    res['results'].append('%s providing %s is already installed' % (this, spec))
                     
             if found:
                 continue
@@ -195,7 +197,6 @@ def install(module, items, repoq, yum_basecmd):
             pkg = spec
         
         cmd = "%s install '%s'" % (yum_basecmd, pkg)
-        res['results'] += "\nInstalling %s\n" % pkg
         rc, out, err = run(cmd)
         # FIXME - if we did an install - go and check the rpmdb to see if it actually installed
         # look for the pkg in rpmdb
@@ -203,12 +204,12 @@ def install(module, items, repoq, yum_basecmd):
         if rc:
             res['changed'] = False
             res['rc'] = rc
-            res['results'] += out
+            res['results'].append(out)
             res['msg'] += err
         else:
             res['changed'] = True
             res['rc'] = 0
-            res['results'] += out
+            res['results'].append(out)
             res['msg'] += err
             
     module.exit_json(**res)
@@ -216,7 +217,7 @@ def install(module, items, repoq, yum_basecmd):
 
 def remove(module, items, repoq, yum_basecmd):
     res = {}
-    res['results'] = ''
+    res['results'] = []
     res['msg'] = ''
     res['changed'] = False
     res['rc'] = 0
@@ -241,6 +242,7 @@ def remove(module, items, repoq, yum_basecmd):
                     found = True
             
             if not found:
+                res['results'].append('%s is not installed' % spec)
                 continue
             pkg = spec
 
@@ -253,19 +255,19 @@ def remove(module, items, repoq, yum_basecmd):
             res['changed'] = False
             res['failed'] = True
             res['rc'] = rc
-            res['results'] += out
+            res['results'].append(out)
             res['msg'] += err
         else:
             res['changed'] = True
             res['rc'] = 0
-            res['results'] += out
+            res['results'].append(out)
             res['msg'] += err
     
     module.exit_json(**res)
 
 def latest(module, items, repoq, yum_basecmd):
     res = {}
-    res['results'] = ''
+    res['results'] = []
     res['msg'] = ''
     res['changed'] = False
     res['rc'] = 0
@@ -294,7 +296,7 @@ def latest(module, items, repoq, yum_basecmd):
                         nothing_to_do = True
                         
             if nothing_to_do:
-                res['results'] += "All packages providing %s are up to date" % spec
+                res['results'].append("All packages providing %s are up to date" % spec)
                 continue
                 
             if not found:
@@ -322,12 +324,12 @@ def latest(module, items, repoq, yum_basecmd):
             res['changed'] = False
             res['failed'] = True
             res['rc'] = rc
-            res['results'] += out
+            res['results'].append(out)
             res['msg'] += err
         else:
             res['changed'] = True
             res['rc'] = 0
-            res['results'] += out
+            res['results'].append(out)
             res['msg'] += err
     
     module.exit_json(**res)
@@ -344,11 +346,11 @@ def ensure(module, state, pkgspec, conf_file):
     if pkgspec.find(',') != -1:
         items = pkgspec.split(',')
 
-    yum_basecmd = '/usr/bin/yum -d1 -y '
-    repoq = '/usr/bin/repoquery --plugins --quiet  -q '
+    yum_basecmd = '%s -d 1 -y ' % yumbin
+    repoq = '%s --plugins --quiet  -q ' % repoquery
     if conf_file and os.path.exists(conf_file):
-        yum_basecmd = '/usr/bin/yum -c %s -d1 -y' % conf_file 
-        repoq = '/usr/bin/repoquery -c %s --plugins --quiet  -q ' % conf_file
+        yum_basecmd = '%s -c %s -d 1 -y' % (yumbin, conf_file)
+        repoq = '%s -c %s --plugins --quiet  -q ' % (repoquery,conf_file)
 
     if state in ['installed', 'present']:
         install(module, items, repoq, yum_basecmd)
@@ -392,7 +394,11 @@ def main():
 
     if params['list'] and params['pkg']:
         module.fail_json(msg="expected 'list=' or 'name=', but not both")
-    
+
+
+    if not os.path.exists(repoquery):
+        module.fail_json(msg="%s is required to run this module. Please install the yum-utils package." % repoquery)
+        
     if params['list']:
         results = dict(results=list_stuff(params['conf_file'], params['list']))
         module.exit_json(**results)