fix: review

Author: nikophil

.github/workflows/ci.yml

@@ -329,7 +329,7 @@ jobs:
       - name: Setup PHP
         uses: shivammathur/setup-php@v2
         with:
-          php-version: 8.4
+          php-version: 8.3
           coverage: none
 
       - name: Install dependencies

composer.json

@@ -37,13 +37,13 @@
         "doctrine/orm": "^2.16|^3.0",
         "doctrine/persistence": "^2.0|^3.0|^4.0",
         "phpunit/phpunit": "^9.5.0 || ^10.0 || ^11.0 || ^12.0",
-        "symfony/browser-kit": "^7.2",
+        "symfony/browser-kit": "^6.4|^7.0",
         "symfony/console": "^6.4|^7.0",
         "symfony/dotenv": "^6.4|^7.0",
         "symfony/framework-bundle": "^6.4|^7.0",
         "symfony/maker-bundle": "^1.55",
         "symfony/phpunit-bridge": "^6.4|^7.0",
-        "symfony/routing": "^7.2",
+        "symfony/routing": "^6.4|^7.0",
         "symfony/runtime": "^6.4|^7.0",
         "symfony/translation-contracts": "^3.4",
         "symfony/uid": "^6.4|^7.0",

config/persistence.php

@@ -2,9 +2,11 @@
 
 namespace Symfony\Component\DependencyInjection\Loader\Configurator;
 
+use Symfony\Component\Console\Event\ConsoleTerminateEvent;
 use Symfony\Component\HttpKernel\Event\TerminateEvent;
+use Symfony\Component\Messenger\Event\WorkerMessageHandledEvent;
 use Zenstruck\Foundry\Persistence\PersistenceManager;
-use Zenstruck\Foundry\Persistence\Proxy\KernelTerminateListener;
+use Zenstruck\Foundry\Persistence\Proxy\ObjectsTracker;
 use Zenstruck\Foundry\Persistence\ResetDatabase\ResetDatabaseManager;
 
 return static function (ContainerConfigurator $container): void {
@@ -22,8 +24,11 @@
     ;
 
     if (PHP_VERSION_ID >= 80400) {
-        $container->services()->set('.foundry.proxy.kernel_terminate_listener', KernelTerminateListener::class)
-            ->tag('kernel.event_listener', ['event' => TerminateEvent::class, 'method' => '__invoke'])
+        $container->services()->set('.foundry.persistence.objects_tracker', ObjectsTracker::class)
+            ->tag('kernel.reset', ['method' => 'refresh'])
+            ->tag('kernel.event_listener', ['event' => TerminateEvent::class, 'method' => 'refresh'])
+            ->tag('kernel.event_listener', ['event' => ConsoleTerminateEvent::class, 'method' => 'refresh'])
+            ->tag('kernel.event_listener', ['event' => WorkerMessageHandledEvent::class, 'method' => 'refresh']) // @phpstan-ignore class.notFound
         ;
     }
 };

config/services.php

@@ -35,6 +35,7 @@
             param('zenstruck_foundry.persistence.flush_once'),
             '%env(default:zenstruck_foundry.faker.seed:int:FOUNDRY_FAKER_SEED)%',
             service('.zenstruck_foundry.in_memory.repository_registry'),
+            service('.foundry.persistence.objects_tracker')->nullOnInvalid(),
         ])
         ->public()
     ;

phpstan.neon

@@ -69,6 +69,11 @@ parameters:
         - identifier: missingType.callable
           path: tests/Fixture/Maker/expected/
 
+        # we're currently running static analysis with PHP 8.3
+        - message: '#Call to an undefined method ReflectionClass\<(.*)\>::isUninitializedLazyObject\(\).#'
+        - message: '#Call to an undefined method ReflectionClass\<(.*)\>::resetAsLazyProxy\(\).#'
+        - message: '#Call to an undefined method ReflectionClass\<(.*)\>::initializeLazyObject\(\).#'
+
     excludePaths:
         - tests/Fixture/Maker/expected/can_create_factory_with_auto_activated_not_persisted_option.php
         - tests/Fixture/Maker/expected/can_create_factory_interactively.php

phpunit

@@ -4,7 +4,8 @@ set -o errexit
 set -o nounset
 
 check_phpunit_version() {
-    INSTALLED_PHPUNIT_VERSION=$(composer info phpunit/phpunit | grep versions |  cut -c 14-)
+    COMPOSER_BIN=$(which composer)
+    INSTALLED_PHPUNIT_VERSION=$(php -d xdebug.mode=off ${COMPOSER_BIN} info phpunit/phpunit | grep versions |  cut -c 14-)
 
     REQUIRED_PHPUNIT_VERSION="${1?}"
 

phpunit-deprecation-baseline.xml

@@ -3,6 +3,7 @@
  <file path="vendor/symfony/deprecation-contracts/function.php">
   <line number="25" hash="c6af5d66288d0667e424978000f29571e4954b81">
    <issue><![CDATA[Since symfony/framework-bundle 6.4: Not setting the "framework.php_errors.log" config option is deprecated. It will default to "true" in 7.0.]]></issue>
+   <issue><![CDATA[Since zenstruck/foundry 2.6: Proxy usage is deprecated in PHP 8.4. Use directly PersistentObjectFactory, Foundry now leverages the native PHP lazy system to auto-refresh objects.]]></issue>
   </line>
  </file>
 </files>

src/Configuration.php

@@ -19,6 +19,7 @@
 use Zenstruck\Foundry\InMemory\CannotEnableInMemory;
 use Zenstruck\Foundry\InMemory\InMemoryRepositoryRegistry;
 use Zenstruck\Foundry\Persistence\PersistenceManager;
+use Zenstruck\Foundry\Persistence\Proxy\ObjectsTracker;
 
 /**
  * @author Kevin Bond <[email protected]>
@@ -60,6 +61,7 @@ public function __construct(
         public readonly bool $flushOnce = false,
         ?int $forcedFakerSeed = null,
         public readonly ?InMemoryRepositoryRegistry $inMemoryRepositoryRegistry = null,
+        public readonly ?ObjectsTracker $objectsTracker = null,
     ) {
         $this->faker->seed(self::fakerSeed($forcedFakerSeed));
 

src/Persistence/PersistenceManager.php

@@ -22,6 +22,8 @@
 use Zenstruck\Foundry\Persistence\Relationship\RelationshipMetadata;
 use Zenstruck\Foundry\Persistence\ResetDatabase\ResetDatabaseManager;
 
+use function Zenstruck\Foundry\set;
+
 /**
  * @author Kevin Bond <[email protected]>
  *
@@ -147,7 +149,7 @@ public function flush(ObjectManager $om): void
      *
      * @return T
      */
-    public function refresh(object &$object, bool $force = false): object
+    public function refresh(object &$object, bool $force = false, bool $allowRefreshDeletedObject = false): object
     {
         if (!$this->flush && !$force) {
             return $object;
@@ -157,6 +159,11 @@ public function refresh(object &$object, bool $force = false): object
             return $object->_refresh();
         }
 
+        if (\PHP_VERSION_ID >= 80400 && ($reflector = new \ReflectionClass($object))->isUninitializedLazyObject($object)) {
+            /** @var T $object */
+            $object = $reflector->initializeLazyObject($object);
+        }
+
         $strategy = $this->strategyFor($object::class);
 
         if ($strategy->hasChanges($object)) {
@@ -165,26 +172,45 @@ public function refresh(object &$object, bool $force = false): object
 
         $om = $strategy->objectManagerFor($object::class);
 
-        if ($strategy->contains($object)) {
-            try {
-                $om->refresh($object);
-            } catch (\LogicException|\Error) {
-                // prevent entities/documents with readonly properties to create an error
-                // LogicException is for ORM / Error is for ODM
-                // @see https://github.com/doctrine/orm/issues/9505
-            }
-
-            return $object;
-        }
-
         if ($strategy->isEmbeddable($object)) {
             return $object;
         }
 
-        $id = $om->getClassMetadata($object::class)->getIdentifierValues($object);
+        if (!$strategy->contains($object)) {
+            $objectFromDb = null;
+
+            $id = $om->getClassMetadata($object::class)->getIdentifierValues($object);
+
+            if ($id) {
+                // "merge" object if it is not managed
+                $objectFromDb = $om->find($object::class, $id);
+            }
 
-        if (!$id || !($object = $om->find($object::class, $id))) { // @phpstan-ignore parameterByRef.type
-            throw RefreshObjectFailed::objectNoLongExists();
+            if ($objectFromDb) {
+                $object = $objectFromDb;
+            } else {
+                if ($allowRefreshDeletedObject) {
+                    $identifierFields = $om->getClassMetadata($object::class)->getIdentifierFieldNames();
+                    foreach ($identifierFields as $field) {
+                        try {
+                            set($object, $field, null);
+                        } catch (\Throwable) {
+                        }
+                    }
+
+                    return $object;
+                }
+
+                throw RefreshObjectFailed::objectNoLongExists();
+            }
+        }
+
+        try {
+            $om->refresh($object);
+        } catch (\LogicException|\Error) {
+            // prevent entities/documents with readonly properties to create an error
+            // LogicException is for ORM / Error is for ODM
+            // @see https://github.com/doctrine/orm/issues/9505
         }
 
         return $object;

src/Persistence/PersistentObjectFactory.php

@@ -23,12 +23,11 @@
 use Zenstruck\Foundry\ObjectFactory;
 use Zenstruck\Foundry\Persistence\Exception\NotEnoughObjects;
 use Zenstruck\Foundry\Persistence\Exception\RefreshObjectFailed;
+use Zenstruck\Foundry\Persistence\Proxy\ObjectsTracker;
 use Zenstruck\Foundry\Persistence\Relationship\ManyToOneRelationship;
 use Zenstruck\Foundry\Persistence\Relationship\OneToManyRelationship;
 use Zenstruck\Foundry\Persistence\Relationship\OneToOneRelationship;
 
-use Zenstruck\Foundry\Persistence\Proxy\CreatedObjectsTracker;
-
 use function Zenstruck\Foundry\force;
 use function Zenstruck\Foundry\get;
 use function Zenstruck\Foundry\set;
@@ -456,14 +455,14 @@ final protected function initializeInternal(): static
         return parent::initializeInternal()
             ->afterInstantiate(
                 static function(object $object, array $parameters, PersistentObjectFactory $factoryUsed): void {
-                    if (PHP_VERSION_ID >= 80400) {
-                        CreatedObjectsTracker::add($object);
-                    }
-
                     if (!$factoryUsed->isPersisting()) {
                         return;
                     }
 
+                    if (\PHP_VERSION_ID >= 80400 && !$factoryUsed instanceof PersistentProxyObjectFactory) {
+                        ObjectsTracker::add($object);
+                    }
+
                     $afterPersistCallbacks = [];
 
                     foreach ($factoryUsed->afterPersist as $afterPersist) {

src/Persistence/PersistentProxyObjectFactory.php

@@ -22,6 +22,19 @@
  */
 abstract class PersistentProxyObjectFactory extends PersistentObjectFactory
 {
+    public function __construct()
+    {
+        parent::__construct();
+
+        if (\PHP_VERSION_ID >= 80400) {
+            trigger_deprecation(
+                'zenstruck/foundry',
+                '2.6',
+                'Proxy usage is deprecated in PHP 8.4. Use directly PersistentObjectFactory, Foundry now leverages the native PHP lazy system to auto-refresh objects.',
+            );
+        }
+    }
+
     /**
      * @return class-string<T>
      */

src/Persistence/Proxy/CreatedObjectsTracker.php

@@ -0,0 +1,78 @@
+<?php
+
+/*
+ * This file is part of the zenstruck/foundry package.
+ *
+ * (c) Kevin Bond <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Zenstruck\Foundry\Persistence\Proxy;
+
+use Zenstruck\Foundry\Configuration;
+
+/**
+ * @internal
+ */
+final class ObjectsTracker
+{
+    /**
+     * This buffer of objects needs to be static to be kept between two kernel.reset events.
+     *
+     * @var list<\WeakReference<object>>
+     */
+    private static $buffer = [];
+
+    public function refresh(): void
+    {
+        self::proxifyObjects();
+    }
+
+    public static function add(object $object): void
+    {
+        self::$buffer[] = \WeakReference::create($object);
+    }
+
+    public function reset(): void
+    {
+        self::$buffer = [];
+    }
+
+    public static function countObjects(): int
+    {
+        return \count(
+            \array_filter(self::$buffer, static fn(\WeakReference $weakRef) => null !== $weakRef->get())
+        );
+    }
+
+    private static function proxifyObjects(): void
+    {
+        self::$buffer = \array_values(
+            \array_map(
+                static function(\WeakReference $weakRef) {
+                    $object = $weakRef->get() ?? throw new \LogicException('Object cannot be null.');
+
+                    $reflector = new \ReflectionClass($object);
+
+                    if ($reflector->isUninitializedLazyObject($object)) {
+                        return \WeakReference::create($object);
+                    }
+
+                    $clone = clone $object;
+                    $reflector->resetAsLazyProxy($object, function() use ($clone, $reflector) {
+                        Configuration::instance()->persistence()->refresh($clone, allowRefreshDeletedObject: true);
+
+                        return $reflector->initializeLazyObject($clone);
+                    });
+
+                    return \WeakReference::create($object);
+                },
+
+                // remove all empty references
+                \array_filter(self::$buffer, static fn(\WeakReference $weakRef) => null !== $weakRef->get()),
+            )
+        );
+    }
+}