123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293 |
- <?php
- /*
- * This file is part of Object Enumerator.
- *
- * (c) Sebastian Bergmann <sebastian@phpunit.de>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace SebastianBergmann\ObjectEnumerator;
- use SebastianBergmann\RecursionContext\Context;
- /**
- * Traverses array structures and object graphs
- * to enumerate all referenced objects.
- */
- class Enumerator
- {
- /**
- * Returns an array of all objects referenced either
- * directly or indirectly by a variable.
- *
- * @param array|object $variable
- *
- * @return object[]
- */
- public function enumerate($variable)
- {
- if (!is_array($variable) && !is_object($variable)) {
- throw new InvalidArgumentException;
- }
- if (isset(func_get_args()[1])) {
- if (!func_get_args()[1] instanceof Context) {
- throw new InvalidArgumentException;
- }
- $processed = func_get_args()[1];
- } else {
- $processed = new Context;
- }
- $objects = [];
- if ($processed->contains($variable)) {
- return $objects;
- }
- $array = $variable;
- $processed->add($variable);
- if (is_array($variable)) {
- foreach ($array as $element) {
- if (!is_array($element) && !is_object($element)) {
- continue;
- }
- $objects = array_merge(
- $objects,
- $this->enumerate($element, $processed)
- );
- }
- } else {
- $objects[] = $variable;
- $reflector = new \ReflectionObject($variable);
- foreach ($reflector->getProperties() as $attribute) {
- $attribute->setAccessible(true);
- try {
- $value = $attribute->getValue($variable);
- } catch (\Throwable $e) {
- continue;
- } catch (\Exception $e) {
- continue;
- }
- if (!is_array($value) && !is_object($value)) {
- continue;
- }
- $objects = array_merge(
- $objects,
- $this->enumerate($value, $processed)
- );
- }
- }
- return $objects;
- }
- }
|