1 files changed, 99 insertions, 0 deletions
diff --git a/library/Director/PropertyModifier/PropertyModifierExtractFromDN.php b/library/Director/PropertyModifier/PropertyModifierExtractFromDN.php
new file mode 100644
index 0000000..c79c5b2
--- /dev/null
+++ b/library/Director/PropertyModifier/PropertyModifierExtractFromDN.php
@@ -0,0 +1,99 @@
+<?php
+
+namespace Icinga\Module\Director\PropertyModifier;
+
+use Icinga\Exception\InvalidPropertyException;
+use Icinga\Module\Director\Hook\PropertyModifierHook;
+use Icinga\Module\Director\Web\Form\QuickForm;
+use Icinga\Protocol\Ldap\LdapUtils;
+
+class PropertyModifierExtractFromDN extends PropertyModifierHook
+{
+    public static function addSettingsFormFields(QuickForm $form)
+    {
+        $form->addElement('select', 'dn_component', array(
+            'label'        => $form->translate('DN component'),
+            'description'  => $form->translate('What should we extract from the DN?'),
+            'multiOptions' => $form->optionalEnum(array(
+                'cn'      => $form->translate('The first (leftmost) CN'),
+                'ou'      => $form->translate('The first (leftmost) OU'),
+                'first'   => $form->translate('Any first (leftmost) component'),
+                'last_ou' => $form->translate('The last (rightmost) OU'),
+            )),
+            'required'    => true,
+        ));
+
+        $form->addElement('select', 'on_failure', array(
+            'label'        => $form->translate('On failure'),
+            'description'  => $form->translate('What should we do if the desired part does not exist?'),
+            'multiOptions' => $form->optionalEnum(array(
+                'null' => $form->translate('Set no value (null)'),
+                'keep' => $form->translate('Keep the DN as is'),
+                'fail' => $form->translate('Let the whole import run fail'),
+            )),
+            'required'    => true,
+        ));
+    }
+
+    public function getName()
+    {
+        return 'Extract from a Distinguished Name (DN)';
+    }
+
+    public function transform($value)
+    {
+        if ($value === null) {
+            return null;
+        }
+
+        $parts = LdapUtils::explodeDN($value);
+        $result = null;
+
+        switch ($this->getSetting('dn_component')) {
+            case 'cn':
+                $result = $this->extractFirst($parts, 'cn');
+                break;
+            case 'ou':
+                $result = $this->extractFirst($parts, 'ou');
+                break;
+            case 'last_ou':
+                $result = $this->extractFirst(array_reverse($parts), 'ou');
+                break;
+            case 'first':
+                $result = $this->extractFirst($parts);
+                break;
+        }
+
+        if ($result === null) {
+            switch ($this->getSetting('on_failure')) {
+                case 'null':
+                    return null;
+                case 'keep':
+                    return $value;
+                case 'fail':
+                default:
+                    throw new InvalidPropertyException(
+                        'DN part extraction failed for %s',
+                        var_export($value, 1)
+                    );
+            }
+        }
+
+        return $result;
+    }
+
+    protected function extractFirst($parts, $what = null)
+    {
+        foreach ($parts as $part) {
+            if (false === ($pos = strpos($part, '='))) {
+                continue;
+            }
+
+            if (null === $what || strtolower(substr($part, 0, $pos)) === $what) {
+                return substr($part, $pos +  1);
+            }
+        }
+
+        return null;
+    }
+}