123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138 |
- <?php
- /*
- * This file is part of sebastian/global-state.
- *
- * (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.
- */
- declare(strict_types=1);
- namespace SebastianBergmann\GlobalState;
- use ReflectionProperty;
- /**
- * Restorer of snapshots of global state.
- */
- class Restorer
- {
- /**
- * Deletes function definitions that are not defined in a snapshot.
- *
- * @throws RuntimeException when the uopz_delete() function is not available
- *
- * @see https://github.com/krakjoe/uopz
- */
- public function restoreFunctions(Snapshot $snapshot)
- {
- if (!\function_exists('uopz_delete')) {
- throw new RuntimeException('The uopz_delete() function is required for this operation');
- }
- $functions = \get_defined_functions();
- foreach (\array_diff($functions['user'], $snapshot->functions()) as $function) {
- uopz_delete($function);
- }
- }
- /**
- * Restores all global and super-global variables from a snapshot.
- */
- public function restoreGlobalVariables(Snapshot $snapshot)
- {
- $superGlobalArrays = $snapshot->superGlobalArrays();
- foreach ($superGlobalArrays as $superGlobalArray) {
- $this->restoreSuperGlobalArray($snapshot, $superGlobalArray);
- }
- $globalVariables = $snapshot->globalVariables();
- foreach (\array_keys($GLOBALS) as $key) {
- if ($key != 'GLOBALS' &&
- !\in_array($key, $superGlobalArrays) &&
- !$snapshot->blacklist()->isGlobalVariableBlacklisted($key)) {
- if (\array_key_exists($key, $globalVariables)) {
- $GLOBALS[$key] = $globalVariables[$key];
- } else {
- unset($GLOBALS[$key]);
- }
- }
- }
- }
- /**
- * Restores all static attributes in user-defined classes from this snapshot.
- */
- public function restoreStaticAttributes(Snapshot $snapshot)
- {
- $current = new Snapshot($snapshot->blacklist(), false, false, false, false, true, false, false, false, false);
- $newClasses = \array_diff($current->classes(), $snapshot->classes());
- unset($current);
- foreach ($snapshot->staticAttributes() as $className => $staticAttributes) {
- foreach ($staticAttributes as $name => $value) {
- $reflector = new ReflectionProperty($className, $name);
- $reflector->setAccessible(true);
- $reflector->setValue($value);
- }
- }
- foreach ($newClasses as $className) {
- $class = new \ReflectionClass($className);
- $defaults = $class->getDefaultProperties();
- foreach ($class->getProperties() as $attribute) {
- if (!$attribute->isStatic()) {
- continue;
- }
- $name = $attribute->getName();
- if ($snapshot->blacklist()->isStaticAttributeBlacklisted($className, $name)) {
- continue;
- }
- if (!isset($defaults[$name])) {
- continue;
- }
- $attribute->setAccessible(true);
- $attribute->setValue($defaults[$name]);
- }
- }
- }
- /**
- * Restores a super-global variable array from this snapshot.
- */
- private function restoreSuperGlobalArray(Snapshot $snapshot, string $superGlobalArray)
- {
- $superGlobalVariables = $snapshot->superGlobalVariables();
- if (isset($GLOBALS[$superGlobalArray]) &&
- \is_array($GLOBALS[$superGlobalArray]) &&
- isset($superGlobalVariables[$superGlobalArray])) {
- $keys = \array_keys(
- \array_merge(
- $GLOBALS[$superGlobalArray],
- $superGlobalVariables[$superGlobalArray]
- )
- );
- foreach ($keys as $key) {
- if (isset($superGlobalVariables[$superGlobalArray][$key])) {
- $GLOBALS[$superGlobalArray][$key] = $superGlobalVariables[$superGlobalArray][$key];
- } else {
- unset($GLOBALS[$superGlobalArray][$key]);
- }
- }
- }
- }
- }
|