fix: Resolve nested relations

Author: silasjoisten

src/Resolver/StoryResolver.php

@@ -24,28 +24,38 @@
     public function resolve(array $target, array $relations): array
     {
         $relationMap = [];
+        $seen = [];
 
         foreach ($relations as $relation) {
             Assert::keyExists($relation, 'uuid');
             Assert::uuid($relation['uuid']);
             $relationMap[$relation['uuid']] = $relation;
         }
 
-        foreach ($target as &$value) {
-            if (\is_string($value) && \array_key_exists($value, $relationMap)) {
-                $value = $relationMap[$value];
+        return $this->doResolve($target, $relationMap, []);
+    }
 
-                continue;
-            }
+    private function doResolve(array $target, array $relationMap, array $seen): array
+    {
+        foreach ($target as $key => $value) {
+            if (\is_string($value) && isset($relationMap[$value])) {
+                if (\in_array($value, $seen, true)) {
+                    continue;
+                }
 
-            if (\is_array($value) && \array_key_exists('id', $value) && \array_key_exists($value['id'], $relationMap)) {
-                $value = $relationMap[$value['id']];
+                $seen[] = $value;
+                $target[$key] = $this->doResolve($relationMap[$value], $relationMap, $seen);
+            } elseif (\is_array($value) && isset($value['id'], $relationMap[$value['id']])) {
+                $id = $value['id'];
 
-                continue;
-            }
+                if (\in_array($id, $seen, true)) {
+                    continue;
+                }
 
-            if (\is_array($value)) {
-                $value = $this->resolve($value, $relations);
+                $seen[] = $id;
+                $target[$key] = $this->doResolve($relationMap[$id], $relationMap, $seen);
+            } elseif (\is_array($value)) {
+                $target[$key] = $this->doResolve($value, $relationMap, $seen);
             }
         }
 

tests/Unit/Resolver/StoryResolverTest.php

@@ -188,4 +188,86 @@ public function resolveReplacesMultiLinkWithLinkPayload(): void
 
         self::assertSame($expected, $resolver->resolve($story, $references));
     }
+
+    #[Test]
+    public function resolveRecursive(): void
+    {
+        $resolver = new StoryResolver();
+
+        $faker = self::faker();
+
+        $story = [
+            'name' => $faker->word(),
+            'content' => [
+                'uuid' => $faker->uuid(),
+                'reference' => $cUuid = $faker->uuid(),
+                'some_field' => $faker->word(),
+            ],
+        ];
+
+        $references = [
+            $b = [
+                'uuid' => $bUuid = $faker->uuid(),
+                'name' => $faker->word(),
+                'another_field' => $faker->sentence(),
+            ],
+            $c = [
+                'uuid' => $cUuid,
+                'name' => $faker->word(),
+                'some_field' => [
+                    'test' => $bUuid,
+                ],
+            ],
+        ];
+
+        $expected = $story;
+        $expected['content']['reference'] = $c;
+        $expected['content']['reference']['some_field']['test'] = $b;
+
+        self::assertSame($expected, $resolver->resolve($story, $references));
+    }
+
+    #[Test]
+    public function resolveWithManyNestedRels(): void
+    {
+        $resolver = new StoryResolver();
+
+        $faker = self::faker();
+
+        $story = [
+            'name' => $faker->word(),
+            'content' => [
+                'uuid' => $faker->uuid(),
+                'reference' => $cUuid = $faker->uuid(),
+                'some_field' => $faker->word(),
+            ],
+        ];
+
+        $references = [
+            $a = [
+                'uuid' => $aUuid = $faker->uuid,
+                'name' => $faker->word(),
+            ],
+            $b = [
+                'uuid' => $bUuid = $faker->uuid(),
+                'name' => $faker->word(),
+                'another_field' => $faker->sentence(),
+                'field' => $aUuid,
+            ],
+            $c = [
+                'uuid' => $cUuid,
+                'name' => $faker->word(),
+                'some_field' => [
+                    'test' => $bUuid,
+                ],
+            ],
+        ];
+
+        $expected = $story;
+        $expected['content']['reference'] = $c;
+        $expected['content']['reference']['some_field']['test'] = $b;
+        $expected['content']['reference']['some_field']['test']['field'] = $a;
+
+        self::assertSame($expected, $resolver->resolve($story, $references));
+    }
 }