icingaweb2-module-director/library/Director/PropertyModifier/PropertyModifierExtractFrom...

100 lines
3.2 KiB
PHP

<?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;
}
}