Make export of objects customizable

src/Exporter.php

@@ -37,6 +37,13 @@
 
 final readonly class Exporter
 {
+    private ?ObjectExporterChain $objectExporter;
+
+    public function __construct(?ObjectExporterChain $objectExporter = null)
+    {
+        $this->objectExporter = $objectExporter;
+    }
+
     /**
      * Exports a value as a string.
      *
@@ -247,18 +254,16 @@ private function recursiveExport(mixed &$value, int $indentation = 0, ?Recursion
             return $this->exportString($value);
         }
 
-        $whitespace = str_repeat(' ', 4 * $indentation);
-
         if (!$processed) {
             $processed = new RecursionContext;
         }
 
         if (is_array($value)) {
-            return $this->exportArray($value, $processed, $whitespace, $indentation);
+            return $this->exportArray($value, $processed, $indentation);
         }
 
         if (is_object($value)) {
-            return $this->exportObject($value, $processed, $whitespace, $indentation);
+            return $this->exportObject($value, $processed, $indentation);
         }
 
         return var_export($value, true);
@@ -303,15 +308,16 @@ private function exportString(string $value): string
             "'";
     }
 
-    private function exportArray(array &$value, RecursionContext $processed, string $whitespace, int $indentation): string
+    private function exportArray(array &$value, RecursionContext $processed, int $indentation): string
     {
         if (($key = $processed->contains($value)) !== false) {
             return 'Array &' . $key;
         }
 
-        $array  = $value;
-        $key    = $processed->add($value);
-        $values = '';
+        $array      = $value;
+        $key        = $processed->add($value);
+        $values     = '';
+        $whitespace = str_repeat(' ', 4 * $indentation);
 
         if (count($array) > 0) {
             foreach ($array as $k => $v) {
@@ -330,7 +336,7 @@ private function exportArray(array &$value, RecursionContext $processed, string
         return 'Array &' . (string) $key . ' [' . $values . ']';
     }
 
-    private function exportObject(mixed $value, RecursionContext $processed, string $whitespace, int $indentation): string
+    private function exportObject(object $value, RecursionContext $processed, int $indentation): string
     {
         $class = $value::class;
 
@@ -340,12 +346,24 @@ private function exportObject(mixed $value, RecursionContext $processed, string
 
         $processed->add($value);
 
-        $values = '';
-        $array  = $this->toArray($value);
+        if ($this->objectExporter !== null && $this->objectExporter->handles($value)) {
+            $buffer = $this->objectExporter->export($value, $this, $indentation);
+        } else {
+            $buffer = $this->defaultObjectExport($value, $processed, $indentation);
+        }
+
+        return $class . ' Object #' . spl_object_id($value) . ' (' . $buffer . ')';
+    }
+
+    private function defaultObjectExport(object $object, RecursionContext $processed, int $indentation): string
+    {
+        $array      = $this->toArray($object);
+        $buffer     = '';
+        $whitespace = str_repeat(' ', 4 * $indentation);
 
         if (count($array) > 0) {
             foreach ($array as $k => $v) {
-                $values .=
+                $buffer .=
                     $whitespace
                     . '    ' .
                     $this->recursiveExport($k, $indentation)
@@ -354,9 +372,9 @@ private function exportObject(mixed $value, RecursionContext $processed, string
                     . ",\n";
             }
 
-            $values = "\n" . $values . $whitespace;
+            $buffer = "\n" . $buffer . $whitespace;
         }
 
-        return $class . ' Object #' . spl_object_id($value) . ' (' . $values . ')';
+        return $buffer;
     }
 }

src/ObjectExporter.php

@@ -0,0 +1,17 @@
+<?php declare(strict_types=1);
+/*
+ * This file is part of sebastian/exporter.
+ *
+ * (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\Exporter;
+
+interface ObjectExporter
+{
+    public function handles(object $object): bool;
+
+    public function export(object $object, Exporter $exporter, int $indentation): string;
+}

src/ObjectExporterChain.php

@@ -0,0 +1,51 @@
+<?php declare(strict_types=1);
+/*
+ * This file is part of sebastian/exporter.
+ *
+ * (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\Exporter;
+
+final class ObjectExporterChain implements ObjectExporter
+{
+    /**
+     * @psalm-var non-empty-list<ObjectExporter>
+     */
+    private array $exporter;
+
+    /**
+     * @psalm-param non-empty-list<ObjectExporter> $exporter
+     */
+    public function __construct(array $exporter)
+    {
+        $this->exporter = $exporter;
+    }
+
+    public function handles(object $object): bool
+    {
+        foreach ($this->exporter as $exporter) {
+            if ($exporter->handles($object)) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+
+    /**
+     * @throws ObjectNotSupportedException
+     */
+    public function export(object $object, Exporter $exporter, int $indentation): string
+    {
+        foreach ($this->exporter as $objectExporter) {
+            if ($objectExporter->handles($object)) {
+                return $objectExporter->export($object, $exporter, $indentation);
+            }
+        }
+
+        throw new ObjectNotSupportedException;
+    }
+}

src/exception/Exception.php

@@ -0,0 +1,16 @@
+<?php declare(strict_types=1);
+/*
+ * This file is part of sebastian/exporter.
+ *
+ * (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\Exporter;
+
+use Throwable;
+
+interface Exception extends Throwable
+{
+}

src/exception/ObjectNotSupportedException.php

@@ -0,0 +1,16 @@
+<?php declare(strict_types=1);
+/*
+ * This file is part of sebastian/exporter.
+ *
+ * (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\Exporter;
+
+use RuntimeException;
+
+final class ObjectNotSupportedException extends RuntimeException implements Exception
+{
+}

tests/ExporterTest.php

@@ -26,12 +26,14 @@
 use PHPUnit\Framework\Attributes\DataProvider;
 use PHPUnit\Framework\Attributes\RequiresPhpExtension;
 use PHPUnit\Framework\Attributes\Small;
+use PHPUnit\Framework\Attributes\UsesClass;
 use PHPUnit\Framework\TestCase;
 use SebastianBergmann\RecursionContext\Context;
 use SplObjectStorage;
 use stdClass;
 
 #[CoversClass(Exporter::class)]
+#[UsesClass(ObjectExporterChain::class)]
 #[Small]
 final class ExporterTest extends TestCase
 {
@@ -447,6 +449,27 @@ public function testShortenedRecursiveOccurredRecursion(): void
         $this->assertEquals('*RECURSION*', (new Exporter)->shortenedRecursiveExport($value, $context));
     }
 
+    public function testExportOfObjectsCanBeCustomized(): void
+    {
+        $objectExporter = $this->createStub(ObjectExporter::class);
+        $objectExporter->method('handles')->willReturn(true);
+        $objectExporter->method('export')->willReturn('custom object export');
+
+        $exporter = new Exporter(new ObjectExporterChain([$objectExporter]));
+
+        $this->assertStringMatchesFormat(
+            <<<'EOT'
+Array &0 [
+    0 => stdClass Object #%d (custom object export),
+    1 => stdClass Object #%d (custom object export),
+]
+EOT
+            ,
+            $exporter->export([new stdClass, new stdClass]),
+        );
+
+    }
+
     private function trimNewline(string $string): string
     {
         return preg_replace('/[ ]*\n/', "\n", $string);

tests/ObjectExporterChainTest.php

@@ -0,0 +1,64 @@
+<?php declare(strict_types=1);
+/*
+ * This file is part of sebastian/exporter.
+ *
+ * (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\Exporter;
+
+use PHPUnit\Framework\Attributes\CoversClass;
+use PHPUnit\Framework\Attributes\Small;
+use PHPUnit\Framework\Attributes\UsesClass;
+use PHPUnit\Framework\TestCase;
+use stdClass;
+
+#[CoversClass(ObjectExporterChain::class)]
+#[UsesClass(Exporter::class)]
+#[Small]
+final class ObjectExporterChainTest extends TestCase
+{
+    public function testCanBeQueriedWhetherChainedExporterHandlesAnObject(): void
+    {
+        $firstExporter = $this->createStub(ObjectExporter::class);
+        $firstExporter->method('handles')->willReturn(false);
+
+        $secondExporter = $this->createStub(ObjectExporter::class);
+        $secondExporter->method('handles')->willReturn(true);
+
+        $chain = new ObjectExporterChain([$firstExporter]);
+        $this->assertFalse($chain->handles(new stdClass));
+
+        $chain = new ObjectExporterChain([$firstExporter, $secondExporter]);
+        $this->assertTrue($chain->handles(new stdClass));
+    }
+
+    public function testDelegatesExportingToFirstExporterThatHandlesAnObject(): void
+    {
+        $firstExporter = $this->createStub(ObjectExporter::class);
+        $firstExporter->method('handles')->willReturn(false);
+        $firstExporter->method('export')->willThrowException(new ObjectNotSupportedException);
+
+        $secondExporter = $this->createStub(ObjectExporter::class);
+        $secondExporter->method('handles')->willReturn(true);
+        $secondExporter->method('export')->willReturn('string');
+
+        $chain = new ObjectExporterChain([$firstExporter, $secondExporter]);
+
+        $this->assertSame('string', $chain->export(new stdClass, new Exporter, 0));
+    }
+
+    public function testCannotExportObjectWhenNoExporterHandlesIt(): void
+    {
+        $firstExporter = $this->createStub(ObjectExporter::class);
+        $firstExporter->method('handles')->willReturn(false);
+
+        $chain = new ObjectExporterChain([$firstExporter]);
+
+        $this->expectException(ObjectNotSupportedException::class);
+
+        $this->assertSame('string', $chain->export(new stdClass, new Exporter, 0));
+    }
+}