From f796e50000e5c198a510300e2293ed460e7113aa Mon Sep 17 00:00:00 2001
From: Rob Crittenden <rcritten@redhat.com>
Date: Mon, 10 Dec 2007 16:12:58 -0500
Subject: Add simple UI for command-line programs to be able to select when
 multiple entries are returned.

---
 ipa-admintools/ipa-adddelegation | 52 +++++++++++++++++++++++++++----------
 ipa-admintools/ipa-findgroup     | 10 +++++++
 ipa-admintools/ipa-finduser      | 11 ++++++++
 ipa-admintools/ipa-moddelegation | 56 +++++++++++++++++++++++++++++-----------
 4 files changed, 101 insertions(+), 28 deletions(-)

(limited to 'ipa-admintools')

diff --git a/ipa-admintools/ipa-adddelegation b/ipa-admintools/ipa-adddelegation
index 53bd43ce..62b4b96e 100644
--- a/ipa-admintools/ipa-adddelegation
+++ b/ipa-admintools/ipa-adddelegation
@@ -23,9 +23,9 @@ from optparse import OptionParser
 import ipa
 import ipa.user
 import ipa.ipaclient as ipaclient
-import ipa.ipavalidate as ipavalidate
 import ipa.config
 import ipa.aci
+import ipa.ipaadminutil as ipaadminutil
 
 import xmlrpclib
 import kerberos
@@ -68,27 +68,53 @@ def main():
         client = ipaclient.IPAClient()
 
         source_grp = client.find_groups(options.source)
-        if source_grp[0] > 1:
-            print "Multiple matches found for %s." % options.source
-            return 2
-        elif source_grp[0] == 0:
-            print "No matches found for %s." % options.source
+        counter = source_grp[0]
+        source_grp = source_grp[1:]
+        groupindex = -1
+        if counter == 0:
+            print "No entries found for %s" % options.source
             return 2
+        elif counter == -1:
+            print "These results are truncated."
+            print "Please refine your search and try again."
+            return 3
+
+        if counter > 1:
+            print "\nMultiple entries for the source group found."
+            groupindex = ipaadminutil.select_group(counter, source_grp)
+            if groupindex == "q":
+                return 0
+
+        if groupindex >= 0:
+            source_grp = [source_grp[groupindex]]
 
         target_grp = client.find_groups(options.target)
-        if target_grp[0] > 1:
-            print "Multiple matches found for %s." % options.target
-            return 3
-        elif target_grp[0] == 0:
-            print "No matches found for %s." % options.target
+        counter = target_grp[0]
+        target_grp = target_grp[1:]
+        groupindex = -1
+        if counter == 0:
+            print "No entries found for %s" % options.target
+            return 2
+        elif counter == -1:
+            print "These results are truncated."
+            print "Please refine your search and try again."
             return 3
 
+        if counter > 1:
+            print "\nMultiple entries for the target group found."
+            groupindex = ipaadminutil.select_group(counter, target_grp)
+            if groupindex == "q":
+                return 0
+
+        if groupindex >= 0:
+            target_grp = [target_grp[groupindex]]
+
         attr_list = options.attributes.split(',')
 
         new_aci = ipa.aci.ACI()
         new_aci.name = args[1]
-        new_aci.source_group = source_grp[1].dn
-        new_aci.dest_group = target_grp[1].dn
+        new_aci.source_group = source_grp[0].dn
+        new_aci.dest_group = target_grp[0].dn
         new_aci.attrs = attr_list
 
         aci_entry = client.get_aci_entry(['*', 'aci'])
diff --git a/ipa-admintools/ipa-findgroup b/ipa-admintools/ipa-findgroup
index 73b0bb1b..b5a5f076 100644
--- a/ipa-admintools/ipa-findgroup
+++ b/ipa-admintools/ipa-findgroup
@@ -21,6 +21,7 @@
 import sys
 from optparse import OptionParser
 import ipa.ipaclient as ipaclient
+import ipa.ipaadminutil as ipaadminutil
 import ipa.config
 
 import errno
@@ -62,6 +63,7 @@ def main():
 
         counter = groups[0]
         groups = groups[1:]
+        groupindex = -1
         if counter == 0:
             print "No entries found for", args[1]
             return 2
@@ -69,6 +71,14 @@ def main():
             print "These results are truncated."
             print "Please refine your search and try again."
 
+        if counter > 1:
+            groupindex = ipaadminutil.select_group(counter, groups)
+            if groupindex == "q":
+                return 0
+
+        if groupindex >= 0:
+            groups = [groups[groupindex]]
+   
         for ent in groups:
             try:
                 members = client.group_members(ent.dn, ['dn','cn'])
diff --git a/ipa-admintools/ipa-finduser b/ipa-admintools/ipa-finduser
index 2ee19dfe..9a57087b 100644
--- a/ipa-admintools/ipa-finduser
+++ b/ipa-admintools/ipa-finduser
@@ -23,6 +23,7 @@ from optparse import OptionParser
 import ipa.ipaclient as ipaclient
 import ipa.config
 import ipa.ipautil as ipautil
+import ipa.ipaadminutil as ipaadminutil
 import base64
 
 import errno
@@ -87,6 +88,7 @@ def main():
 
         counter = users[0]
         users = users[1:]
+        userindex = 0
         if counter == 0:
             print "No entries found for", args[1]
             return 2
@@ -94,6 +96,15 @@ def main():
             print "These results are truncated."
             print "Please refine your search and try again."
 
+        if counter > 1:
+            userindex = ipaadminutil.select_user(counter, users)
+            if userindex == "q":
+                return
+
+
+        if userindex >= 0:
+            users = [users[userindex]]
+         
         for ent in users:
             attr = ent.attrList()
             attr.sort()
diff --git a/ipa-admintools/ipa-moddelegation b/ipa-admintools/ipa-moddelegation
index 103c0586..2be10b9f 100644
--- a/ipa-admintools/ipa-moddelegation
+++ b/ipa-admintools/ipa-moddelegation
@@ -23,7 +23,7 @@ from optparse import OptionParser
 import ipa
 import ipa.user
 import ipa.ipaclient as ipaclient
-import ipa.ipavalidate as ipavalidate
+import ipa.ipaadminutil as ipaadminutil
 import ipa.config
 import ipa.aci
 
@@ -75,21 +75,47 @@ def main():
 
         if options.source:
             source_grp = client.find_groups(options.source)
-            if source_grp[0] > 1:
-                print "Multiple matches found for %s." % options.source
-                return 1
-            elif source_grp[0] == 0:
-                print "No matches found for %s." % options.source
-                return 1
+            counter = source_grp[0]
+            source_grp = source_grp[1:]
+            groupindex = -1
+            if counter == 0:
+                print "No entries found for %s" % options.source
+                return 2 
+            elif counter == -1:
+                print "These results are truncated."
+                print "Please refine your search and try again."
+                return 3
+
+            if counter > 1:
+                print "\nMultiple entries for the source group found."
+                groupindex = ipaadminutil.select_group(counter, source_grp)
+                if groupindex == "q":
+                    return 0
+
+            if groupindex >= 0:
+                source_grp = [source_grp[groupindex]]
 
         if options.target:
             target_grp = client.find_groups(options.target)
-            if target_grp[0] > 1:
-                print "Multiple matches found for %s." % options.target
-                return 1
-            elif target_grp[0] == 0:
-                print "No matches found for %s." % options.target
-                return 1
+            counter = target_grp[0]
+            target_grp = target_grp[1:]
+            groupindex = -1
+            if counter == 0:
+                print "No entries found for %s" % options.target
+                return 2 
+            elif counter == -1:
+                print "These results are truncated."
+                print "Please refine your search and try again."
+                return 3
+
+            if counter > 1:
+                print "\nMultiple entries for the target group found."
+                groupindex = ipaadminutil.select_group(counter, target_grp)
+                if groupindex == "q":
+                    return 0
+
+            if groupindex >= 0:
+                target_grp = [target_grp[groupindex]]
 
         if options.attributes:
             attr_list = options.attributes.split(',')
@@ -125,11 +151,11 @@ def main():
         new_aci = ipa.aci.ACI()
         new_aci.name = args[1]
         if options.source:
-            new_aci.source_group = source_grp[1].dn
+            new_aci.source_group = source_grp[0].dn
         else:
             new_aci.source_group = old_aci.source_group
         if options.target:
-            new_aci.dest_group = target_grp[1].dn
+            new_aci.dest_group = target_grp[0].dn
         else:
             new_aci.dest_group = old_aci.dest_group
         if options.attributes:
-- 
cgit