138 lines
3.5 KiB
PHP
138 lines
3.5 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Component\Ldap\Adapter\ExtLdap;
|
|
|
|
use Symfony\Component\Ldap\Adapter\CollectionInterface;
|
|
use Symfony\Component\Ldap\Entry;
|
|
use Symfony\Component\Ldap\Exception\LdapException;
|
|
|
|
/**
|
|
* @author Charles Sarrazin <charles@sarraz.in>
|
|
*/
|
|
class Collection implements CollectionInterface
|
|
{
|
|
private Connection $connection;
|
|
private Query $search;
|
|
/** @var list<Entry> */
|
|
private array $entries;
|
|
|
|
public function __construct(Connection $connection, Query $search)
|
|
{
|
|
$this->connection = $connection;
|
|
$this->search = $search;
|
|
}
|
|
|
|
public function toArray(): array
|
|
{
|
|
return $this->entries ??= iterator_to_array($this->getIterator(), false);
|
|
}
|
|
|
|
public function count(): int
|
|
{
|
|
$con = $this->connection->getResource();
|
|
$searches = $this->search->getResources();
|
|
$count = 0;
|
|
foreach ($searches as $search) {
|
|
$searchCount = ldap_count_entries($con, $search);
|
|
if (false === $searchCount) {
|
|
throw new LdapException('Error while retrieving entry count: '.ldap_error($con));
|
|
}
|
|
$count += $searchCount;
|
|
}
|
|
|
|
return $count;
|
|
}
|
|
|
|
public function getIterator(): \Traversable
|
|
{
|
|
if (0 === $this->count()) {
|
|
return;
|
|
}
|
|
|
|
$con = $this->connection->getResource();
|
|
$searches = $this->search->getResources();
|
|
foreach ($searches as $search) {
|
|
$current = ldap_first_entry($con, $search);
|
|
|
|
if (false === $current) {
|
|
throw new LdapException('Could not rewind entries array: '.ldap_error($con));
|
|
}
|
|
|
|
yield $this->getSingleEntry($con, $current);
|
|
|
|
while (false !== $current = ldap_next_entry($con, $current)) {
|
|
yield $this->getSingleEntry($con, $current);
|
|
}
|
|
}
|
|
}
|
|
|
|
public function offsetExists(mixed $offset): bool
|
|
{
|
|
$this->toArray();
|
|
|
|
return isset($this->entries[$offset]);
|
|
}
|
|
|
|
public function offsetGet(mixed $offset): ?Entry
|
|
{
|
|
$this->toArray();
|
|
|
|
return $this->entries[$offset] ?? null;
|
|
}
|
|
|
|
public function offsetSet(mixed $offset, mixed $value): void
|
|
{
|
|
$this->toArray();
|
|
|
|
$this->entries[$offset] = $value;
|
|
}
|
|
|
|
public function offsetUnset($offset): void
|
|
{
|
|
$this->toArray();
|
|
|
|
unset($this->entries[$offset]);
|
|
}
|
|
|
|
private function getSingleEntry($con, $current): Entry
|
|
{
|
|
$attributes = ldap_get_attributes($con, $current);
|
|
|
|
if (false === $attributes) {
|
|
throw new LdapException('Could not fetch attributes: '.ldap_error($con));
|
|
}
|
|
|
|
$attributes = $this->cleanupAttributes($attributes);
|
|
|
|
$dn = ldap_get_dn($con, $current);
|
|
|
|
if (false === $dn) {
|
|
throw new LdapException('Could not fetch DN: '.ldap_error($con));
|
|
}
|
|
|
|
return new Entry($dn, $attributes);
|
|
}
|
|
|
|
private function cleanupAttributes(array $entry): array
|
|
{
|
|
$attributes = array_diff_key($entry, array_flip(range(0, $entry['count'] - 1)) + [
|
|
'count' => null,
|
|
'dn' => null,
|
|
]);
|
|
array_walk($attributes, function (&$value) {
|
|
unset($value['count']);
|
|
});
|
|
|
|
return $attributes;
|
|
}
|
|
}
|