Rework RenameReplaceEntryStrategy to allow multiple strings

Author: stloyd

examples/topics/data_frame/cache/code.php

@@ -5,6 +5,7 @@
 use function Flow\ETL\DSL\{config_builder, data_frame, filesystem_cache, from_cache, ref, to_stream};
 use Flow\ETL\Adapter\Http\DynamicExtractor\NextRequestFactory;
 use Flow\ETL\Adapter\Http\PsrHttpClientDynamicExtractor;
+use Flow\ETL\Transformer\Rename\RenameReplaceEntryStrategy;
 use Http\Client\Curl\Client;
 use Nyholm\Psr7\Factory\Psr17Factory;
 use Psr\Http\Message\{RequestInterface, ResponseInterface};
@@ -41,7 +42,7 @@ public function create(?ResponseInterface $previousResponse = null) : ?RequestIn
     ->withEntry('unpacked', ref('response_body')->jsonDecode())
     ->select('unpacked')
     ->withEntry('unpacked', ref('unpacked')->unpack())
-    ->renameAll('unpacked.', '')
+    ->renameEach(new RenameReplaceEntryStrategy('unpacked.', ''))
     ->drop('unpacked')
     ->select('name', 'html_url', 'blog', 'login', 'public_repos', 'followers', 'created_at')
     ->write(to_stream(__DIR__ . '/output.txt', truncate: false))

examples/topics/data_frame/rename_columns/code.php

@@ -3,6 +3,7 @@
 declare(strict_types=1);
 
 use function Flow\ETL\DSL\{data_frame, from_array, to_stream};
+use Flow\ETL\Transformer\Rename\RenameReplaceEntryStrategy;
 
 require __DIR__ . '/vendor/autoload.php';
 
@@ -13,7 +14,7 @@
         ['id' => 3, 'name' => 'Jane', 'joined_id' => 3, 'joined_status' => 'active'],
     ]))
     ->rename('id', 'user_id')
-    ->renameAll('joined_', '')
+    ->renameEach(new RenameReplaceEntryStrategy('joined.', ''))
     ->collect()
     ->write(to_stream(__DIR__ . '/output.txt', truncate: false))
     ->run();

examples/topics/data_reading/http_dynamic/code.php

@@ -5,6 +5,7 @@
 use function Flow\ETL\DSL\{data_frame, ref, to_stream};
 use Flow\ETL\Adapter\Http\DynamicExtractor\NextRequestFactory;
 use Flow\ETL\Adapter\Http\PsrHttpClientDynamicExtractor;
+use Flow\ETL\Transformer\Rename\RenameReplaceEntryStrategy;
 use Http\Client\Curl\Client;
 use Nyholm\Psr7\Factory\Psr17Factory;
 use Psr\Http\Message;
@@ -34,7 +35,7 @@ public function create(?Message\ResponseInterface $previousResponse = null) : ?M
     ->read($from_github_api)
     ->withEntry('unpacked', ref('response_body')->jsonDecode())
     ->withEntry('unpacked', ref('unpacked')->unpack())
-    ->renameAll('unpacked.', '')
+    ->renameEach(new RenameReplaceEntryStrategy('unpacked.', ''))
     ->drop('unpacked')
     ->select('name', 'html_url', 'blog', 'login', 'public_repos', 'followers', 'created_at')
     ->write(to_stream(__DIR__ . '/output.txt', truncate: false))

src/core/etl/src/Flow/ETL/DSL/functions.php

@@ -354,8 +354,12 @@ function rename_style(Style $style) : Transformer\Rename\RenameCaseEntryStrategy
     return new Transformer\Rename\RenameCaseEntryStrategy($style);
 }
 
+/**
+ * @param array<string>|string $search
+ * @param array<string>|string $replace
+ */
 #[DocumentationDSL(module: Module::CORE, type: DSLType::TRANSFORMER)]
-function rename_replace(string $search, string $replace) : Transformer\Rename\RenameReplaceEntryStrategy
+function rename_replace(string|array $search, string|array $replace) : Transformer\Rename\RenameReplaceEntryStrategy
 {
     return new Transformer\Rename\RenameReplaceEntryStrategy($search, $replace);
 }

src/core/etl/src/Flow/ETL/Row/Formatter/ASCIISchemaFormatter.php

@@ -10,6 +10,7 @@
 use Flow\ETL\PHP\Type\Type;
 use Flow\ETL\Schema;
 use Flow\ETL\Schema\SchemaFormatter;
+use Flow\ETL\Transformer\Rename\RenameReplaceEntryStrategy;
 
 final readonly class ASCIISchemaFormatter implements SchemaFormatter
 {
@@ -24,7 +25,7 @@ public function format(Schema $schema) : string
             df()
                 ->read(from_array($schema->normalize()))
                 ->withEntry('type', ref('type')->unpack())
-                ->renameAll('type.', '')
+                ->renameEach(new RenameReplaceEntryStrategy('type.', ''))
                 ->rename('ref', 'name')
                 ->collect()
                 ->select('name', 'type', 'nullable', 'metadata')

src/core/etl/src/Flow/ETL/Transformer/Rename/RenameReplaceEntryStrategy.php

@@ -8,9 +8,13 @@
 
 final readonly class RenameReplaceEntryStrategy implements RenameEntryStrategy
 {
+    /**
+     * @param array<string>|string $search
+     * @param array<string>|string $replace
+     */
     public function __construct(
-        private string $search,
-        private string $replace,
+        private string|array $search,
+        private string|array $replace,
     ) {
     }
 

src/core/etl/tests/Flow/ETL/Tests/Integration/DataFrame/RenameTest.php

@@ -83,6 +83,29 @@ public function test_rename_all_lower_case_i18n() : void
         );
     }
 
+    public function test_rename_all_multiple() : void
+    {
+        $rows = rows(
+            row(json_entry('array', ['id' => 1, 'name' => 'name', 'isActive' => true])),
+            row(json_entry('array', ['id' => 2, 'name' => 'name', 'isActive' => false])),
+        );
+
+        $ds = df()
+            ->read(from_rows($rows))
+            ->withEntry('row', ref('array')->unpack())
+            ->renameEach(rename_replace(['row.', 'isActive'], ['', 'active']))
+            ->drop('array')
+            ->getEachAsArray();
+
+        self::assertEquals(
+            [
+                ['id' => 1, 'name' => 'name', 'active' => true],
+                ['id' => 2, 'name' => 'name', 'active' => false],
+            ],
+            \iterator_to_array($ds)
+        );
+    }
+
     public function test_rename_all_to_ascii() : void
     {
         $rows = rows(row(int_entry('ÓSMY', 8)), row(int_entry('DZIEWIĄTY', 9)));

src/core/etl/tests/Flow/ETL/Tests/Integration/Function/ArrayUnpackTest.php

@@ -8,6 +8,7 @@
 use function Flow\ETL\DSL\{from_array, ref, to_memory};
 use Flow\ETL\Memory\ArrayMemory;
 use Flow\ETL\Tests\FlowTestCase;
+use Flow\ETL\Transformer\Rename\RenameReplaceEntryStrategy;
 
 final class ArrayUnpackTest extends FlowTestCase
 {
@@ -23,7 +24,7 @@ public function test_array_unpack() : void
                 )
             )
             ->withEntry('array', ref('array')->unpack())
-            ->renameAll('array.', '')
+            ->renameEach(new RenameReplaceEntryStrategy('array.', ''))
             ->drop('array')
             ->write(to_memory($memory = new ArrayMemory()))
             ->run();

web/landing/resources/dsl.json

@@ -8,6 +8,7 @@
 use Flow\ETL\Adapter\Http\PsrHttpClientDynamicExtractor;
 use Flow\ETL\Cache\Implementation\PSRSimpleCache;
 use Flow\ETL\Memory\ArrayMemory;
+use Flow\ETL\Transformer\Rename\RenameReplaceEntryStrategy;
 use Flow\Website\Factory\Github\ContributorsRequestFactory;
 use Http\Client\Curl\Client;
 use Http\Discovery\Psr17Factory;
@@ -55,7 +56,7 @@ public function contributors() : array
                 ->select('unpacked')
                 ->withEntry('data', ref('unpacked')->expand())
                 ->withEntry('data', ref('data')->unpack())
-                ->renameAll('data.', '')
+                ->renameEach(new RenameReplaceEntryStrategy('data.', ''))
                 ->drop('unpacked', 'data')
                 ->filter(not(ref('login')->endsWith(lit('[bot]'))))
                 ->filter(not(ref('login')->equals(lit('aeon-automation'))))