Cleanup CachedContainer invalidation (#3867)

* Cleanup CachedContainer invalidation

* Update AbstractTestCase.php

* Update CachedContainerBuilder.php

* cleanup

* cleanup

* fix

* Update CacheInvalidatingContainer.php

* Update CacheInvalidatingContainer.php

* Add StaleContainerCacheException

* move

packages/Testing/PHPUnit/AbstractTestCase.php

@@ -60,15 +60,7 @@ protected function getService(string $type): object
             );
         }
 
-        try {
-            $object = self::$currentContainer->get($type);
-        } catch (Throwable $throwable) {
-            // clear compiled container cache, to trigger re-discovery
-            RectorKernel::clearCache();
-
-            throw $throwable;
-        }
-
+        $object = self::$currentContainer->get($type);
         if ($object === null) {
             $message = sprintf('Service "%s" was not found', $type);
             throw new ShouldNotHappenException($message);

src/Exception/Cache/StaleContainerCacheException.php

@@ -0,0 +1,7 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\Core\Exception\Cache;
+
+final class StaleContainerCacheException extends \RuntimeException {}

src/Kernel/CacheInvalidatingContainer.php

@@ -0,0 +1,67 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\Core\Kernel;
+
+use Rector\Core\Exception\Cache\StaleContainerCacheException;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+final class CacheInvalidatingContainer implements ContainerInterface {
+    public function __construct(
+        private readonly ContainerInterface $wrapped
+    )
+    {}
+
+    public function set(string $id, ?object $service): void
+    {
+        $this->wrapped->set($id, $service);
+    }
+
+    public function get(string $id, int $invalidBehavior = self::EXCEPTION_ON_INVALID_REFERENCE): ?object
+    {
+        try {
+            return $this->wrapped->get($id, $invalidBehavior);
+        } catch (\Throwable $throwable) {
+            // clear compiled container cache, to trigger re-discovery
+            RectorKernel::clearCache();
+
+            throw new StaleContainerCacheException(
+                'Container cache is outdated and was cleared. please re-run the command.',
+                0,
+                $throwable
+            );
+        }
+    }
+
+    public function has(string $id): bool
+    {
+        return $this->wrapped->has($id);
+    }
+
+    public function initialized(string $id): bool
+    {
+        return $this->wrapped->initialized($id);
+    }
+
+    /**
+     * @return array<mixed>|bool|float|int|string|\UnitEnum|null
+     */
+    public function getParameter(string $name)
+    {
+        return $this->wrapped->getParameter($name);
+    }
+
+    public function hasParameter(string $name): bool
+    {
+        return $this->wrapped->hasParameter($name);
+    }
+
+    /**
+     * @param \UnitEnum|float|array<mixed>|bool|int|string|null $value
+     */
+    public function setParameter(string $name, \UnitEnum|float|array|bool|int|string|null $value): void
+    {
+        $this->wrapped->setParameter($name, $value);
+    }
+}

src/Kernel/CachedContainerBuilder.php

@@ -37,10 +37,11 @@ public function build(array $configFiles, string $hash, callable $containerBuild
         if (file_exists($file)) {
             require_once $file;
             $className = '\\' . __NAMESPACE__ . '\\' . $className;
-            $container = new $className();
-            if (! $container instanceof ContainerInterface) {
+            $cachedContainer = new $className();
+            if (! $cachedContainer instanceof ContainerInterface) {
                 throw new ShouldNotHappenException();
             }
+            $container = new CacheInvalidatingContainer($cachedContainer);
         } else {
             $container = ($containerBuilderCallback)($configFiles);