12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485 |
- <?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\ObjectReflector\ObjectReflector;
- 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 ObjectReflector;
- foreach ($reflector->getAttributes($variable) as $value) {
- if (!is_array($value) && !is_object($value)) {
- continue;
- }
- $objects = array_merge(
- $objects,
- $this->enumerate($value, $processed)
- );
- }
- }
- return $objects;
- }
- }
|