Remove automountinformation as part of the DN for automount.

To support multiple direct maps we added description to the DN of
automount key entries. The downside of this is that to display a key
you had to know the information as well, which was rather pointless if
that is what you were trying to get.

So now both modes are supported. It will first look for just a key
in the description and fall back to including automountinformation
if it needs to.

Multiple direct maps are still supported and for those the info is
always required.

ticket 1229

1 files changed, 76 insertions, 21 deletions

diff --git a/ipalib/plugins/automount.py b/ipalib/plugins/automount.py
index 504e183bf..11cea5c90 100644
--- a/ipalib/plugins/automount.py
+++ b/ipalib/plugins/automount.py
@@ -174,6 +174,7 @@ from ipalib import Object, Command
 from ipalib import Flag, Str, IA5Str
 from ipalib.plugins.baseldap import *
 from ipalib import _, ngettext
+import ldap as _ldap
 import os
 
 
@@ -604,11 +605,13 @@ class automountkey(LDAPObject):
     num_parents = 2
     label = _('Automount Keys')
     already_exists_msg = _('The key,info pair must be unique. A key named %(key)s with info %(info)s already exists')
+    key_already_exists_msg = _('key named %(key)s already exists')
     object_not_found_msg = _('The automount key %(key)s with info %(info)s does not exist')
 
     def get_dn(self, *keys, **kwargs):
         # all commands except for create send pk in keys, too
         # create cannot due to validation in frontend.py
+        ldap = self.backend
         if len(keys) == self.num_parents:
             try:
                 pkey = kwargs[self.primary_key.name]
@@ -625,6 +628,37 @@ class automountkey(LDAPObject):
             pkey,
             parent_dn
         )
+        # If we're doing an add then just return the dn we created, there
+        # is no need to check for it.
+        if kwargs.get('add_operation', False):
+            return dn
+        # We had an older mechanism where description consisted of
+        # 'automountkey automountinformation' so we could support multiple
+        # direct maps. This made showing keys nearly impossible since it
+        # required automountinfo to show, which if you had you didn't need
+        # to look at the key. We still support existing entries but now
+        # only create this type of dn when the key is /-
+        #
+        # First we look with the information given, then try to search for
+        # the right entry.
+        try:
+            (dn, entry_attrs) = ldap.get_entry(
+                dn, ['*'], normalize=self.normalize_dn
+            )
+        except errors.NotFound:
+            if kwargs.get('automountinformation', False):
+                sfilter = '(&(automountkey=%s)(automountinformation=%s))' % \
+                    (kwargs['automountkey'], kwargs['automountinformation'])
+            else:
+                sfilter = '(automountkey=%s)' % kwargs['automountkey']
+            basedn = 'automountmapname=%s,cn=%s,%s' % (parent_keys[1], parent_keys[0], self.container_dn)
+            attrs_list = ['*']
+            (entries, truncated) = ldap.find_entries(sfilter, attrs_list,
+                basedn, _ldap.SCOPE_ONELEVEL)
+            if len(entries) > 1:
+                raise errors.NotFound(reason=_('More than one entry with key %(key)s found, use --info to select specific entry.') % dict(key=pkey))
+            dn = entries[0][0]
+
         return dn
 
     def handle_not_found(self, *keys):
@@ -641,23 +675,39 @@ class automountkey(LDAPObject):
         pkey = keys[-1]
         key = pkey.split(self.rdn_separator)[0]
         info = self.rdn_separator.join(pkey.split(self.rdn_separator)[1:])
-        raise errors.DuplicateEntry(
-            message=self.already_exists_msg % {
-                'key': key, 'info': info,
-            }
-        )
+        if info:
+            raise errors.DuplicateEntry(
+                message=self.already_exists_msg % {
+                    'key': key, 'info': info,
+                }
+            )
+        else:
+            raise errors.DuplicateEntry(
+                message=self.key_already_exists_msg % {
+                    'key': key,
+                }
+            )
 
-    def get_pk(self, key, info):
-        return self.rdn_separator.join((key,info))
+    def get_pk(self, key, info=None):
+        if info:
+            return self.rdn_separator.join((key,info))
+        else:
+            return key
 
     def check_key_uniqueness(self, location, map, **keykw):
+        info = None
         key = keykw.get('automountkey')
-        info = keykw.get('automountinformation')
-        if key is None or info is None:
+        if key is None:
             return
 
-        entries = self.methods.find(location, map, **keykw)['result']
+        entries = self.methods.find(location, map, automountkey=key)['result']
         if len(entries) > 0:
+            if key == u'/-':
+                info = keykw.get('automountinformation')
+                entries = self.methods.find(location, map, **keykw)['result']
+                if len(entries) > 0:
+                    self.handle_duplicate_entry(location, map, self.get_pk(key, info))
+                else: return
             self.handle_duplicate_entry(location, map, self.get_pk(key, info))
 
 api.register(automountkey)
@@ -676,9 +726,13 @@ class automountkey_add(LDAPCreate):
             yield key
 
     def execute(self, *keys, **options):
-        options[self.obj.primary_key.name] = self.obj.get_pk(
-                                            options['automountkey'],
-                                            options['automountinformation'])
+        key = options['automountkey']
+        info = options.get('automountinformation', None)
+        if key == '/-':
+            options[self.obj.primary_key.name] = self.obj.get_pk(key, info)
+        else:
+            options[self.obj.primary_key.name] = self.obj.get_pk(key, None)
+        options['add_operation'] = True
         return super(automountkey_add, self).execute(*keys, **options)
 
 api.register(automountkey_add)
@@ -724,7 +778,7 @@ class automountkey_del(LDAPDelete):
                label=_('Key'),
                doc=_('Automount key name.'),
         ),
-        IA5Str('automountinformation',
+        IA5Str('automountinformation?',
                cli_name='info',
                label=_('Mount information'),
         ),
@@ -736,10 +790,10 @@ class automountkey_del(LDAPDelete):
 
     def execute(self, *keys, **options):
         keys += (self.obj.get_pk(options['automountkey'],
-                                 options['automountinformation']), )
+                                 options.get('automountinformation', None)),)
         options[self.obj.primary_key.name] = self.obj.get_pk(
                                             options['automountkey'],
-                                            options['automountinformation'])
+                                            options.get('automountinformation', None))
         return super(automountkey_del, self).execute(*keys, **options)
 
 api.register(automountkey_del)
@@ -769,10 +823,10 @@ class automountkey_mod(LDAPUpdate):
 
     def execute(self, *keys, **options):
         keys += (self.obj.get_pk(options['automountkey'],
-                                 options['automountinformation']), )
+                                 options.get('automountinformation', None)), )
         options[self.obj.primary_key.name] = self.obj.get_pk(
                                             options['automountkey'],
-                                            options['automountinformation'])
+                                            options.get('automountinformation', None))
         return super(automountkey_mod, self).execute(*keys, **options)
 
 api.register(automountkey_mod)
@@ -796,7 +850,7 @@ class automountkey_show(LDAPRetrieve):
                label=_('Key'),
                doc=_('Automount key name.'),
         ),
-        IA5Str('automountinformation',
+        IA5Str('automountinformation?',
                cli_name='info',
                label=_('Mount information'),
         ),
@@ -808,10 +862,11 @@ class automountkey_show(LDAPRetrieve):
 
     def execute(self, *keys, **options):
         keys += (self.obj.get_pk(options['automountkey'],
-                                 options['automountinformation']), )
+                                 options.get('automountinformation', None)), )
         options[self.obj.primary_key.name] = self.obj.get_pk(
                                             options['automountkey'],
-                                            options['automountinformation'])
+                                            options.get('automountinformation', None))
+
         return super(automountkey_show, self).execute(*keys, **options)
 
 api.register(automountkey_show)