Introduce stimulus_controller to ease Stimulus Values API usage

Author: tgalopin

composer.json

@@ -31,6 +31,7 @@
     "require-dev": {
         "symfony/framework-bundle": "^4.4 || ^5.0",
         "symfony/phpunit-bridge": "^4.4 || ^5.0",
+        "symfony/string": "^5.0",
         "symfony/twig-bundle": "^4.4 || ^5.0",
         "symfony/web-link": "^4.4 || ^5.0"
     },

src/Exception/MissingPackageException.php

@@ -0,0 +1,14 @@
+<?php
+
+/*
+ * This file is part of the Symfony WebpackEncoreBundle package.
+ * (c) Fabien Potencier <[email protected]>
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\WebpackEncoreBundle\Exception;
+
+class MissingPackageException extends \LogicException
+{
+}

src/Resources/config/services.xml

@@ -36,6 +36,10 @@
             </argument>
         </service>
 
+        <service id="webpack_encore.twig_stimulus_extension" class="Symfony\WebpackEncoreBundle\Twig\StimulusTwigExtension">
+            <tag name="twig.extension" />
+        </service>
+
         <service id="webpack_encore.entrypoint_lookup.cache_warmer" class="Symfony\WebpackEncoreBundle\CacheWarmer\EntrypointCacheWarmer">
             <tag name="kernel.cache_warmer" />
             <argument /> <!-- build list of entrypoint paths -->

src/Twig/StimulusTwigExtension.php

@@ -0,0 +1,57 @@
+<?php
+
+/*
+ * This file is part of the Symfony WebpackEncoreBundle package.
+ * (c) Fabien Potencier <[email protected]>
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\WebpackEncoreBundle\Twig;
+
+use Symfony\Component\String\UnicodeString;
+use Symfony\WebpackEncoreBundle\Exception\MissingPackageException;
+use Twig\Environment;
+use Twig\Extension\AbstractExtension;
+use Twig\TwigFunction;
+
+final class StimulusTwigExtension extends AbstractExtension
+{
+    public function getFunctions()
+    {
+        return [
+            new TwigFunction('stimulus_controller', [$this, 'renderStimulusController'], ['needs_environment' => true, 'is_safe' => ['all']]),
+        ];
+    }
+
+    public function renderStimulusController(Environment $env, array $data): string
+    {
+        if (!class_exists(UnicodeString::class)) {
+            throw new MissingPackageException("Missing package, to use the \"stimulus_controller\" function, run:\n\ncomposer require symfony/string");
+        }
+
+        if (!$data) {
+            return '';
+        }
+
+        $controllers = [];
+        $values = [];
+
+        foreach ($data as $controllerName => $controllerValues) {
+            $controllers[] = $controllerName;
+
+            foreach ($controllerValues as $key => $value) {
+                $key = (new UnicodeString($key))->snake()->replace('_', '-')->lower();
+
+                if (!is_scalar($value)) {
+                    $value = json_encode($value);
+                }
+
+                $value = twig_escape_filter($env, $value, 'html_attr');
+                $values[] = 'data-'.$controllerName.'-'.$key.'-value="'.$value.'"';
+            }
+        }
+
+        return rtrim('data-controller="'.implode(' ', $controllers).'" '.implode(' ', $values));
+    }
+}

tests/IntegrationTest.php

@@ -28,6 +28,7 @@
 use Symfony\WebpackEncoreBundle\Asset\EntrypointLookupInterface;
 use Symfony\WebpackEncoreBundle\Asset\TagRenderer;
 use Symfony\WebpackEncoreBundle\CacheWarmer\EntrypointCacheWarmer;
+use Symfony\WebpackEncoreBundle\Twig\StimulusTwigExtension;
 use Symfony\WebpackEncoreBundle\WebpackEncoreBundle;
 
 class IntegrationTest extends TestCase
@@ -178,6 +179,64 @@ public function testAutowireDefaultBuildArgument()
         $this->assertTrue(true);
     }
 
+    public function provideRenderStimulusController()
+    {
+        yield 'empty' => [
+            'data' => [],
+            'expected' => '',
+        ];
+
+        yield 'single-controller-no-data' => [
+            'data' => [
+                'my-controller' => [],
+            ],
+            'expected' => 'data-controller="my-controller"',
+        ];
+
+        yield 'single-controller-scalar-data' => [
+            'data' => [
+                'my-controller' => [
+                    'myValue' => 'scalar-value',
+                ],
+            ],
+            'expected' => 'data-controller="my-controller" data-my-controller-my-value-value="scalar-value"',
+        ];
+
+        yield 'single-controller-nested-data' => [
+            'data' => [
+                'my-controller' => [
+                    'myValue' => ['nested' => 'array'],
+                ],
+            ],
+            'expected' => 'data-controller="my-controller" data-my-controller-my-value-value="{"nested":"array"}"',
+        ];
+
+        yield 'multiple-controllers-scalar-data' => [
+            'data' => [
+                'my-controller' => [
+                    'myValue' => 'scalar-value',
+                ],
+                'another-controller' => [
+                    'anotherValue' => 'scalar-value 2',
+                ],
+            ],
+            'expected' => 'data-controller="my-controller another-controller" data-my-controller-my-value-value="scalar-value" data-another-controller-another-value-value="scalar-value 2"',
+        ];
+    }
+
+    /**
+     * @dataProvider provideRenderStimulusController
+     */
+    public function testRenderStimulusController(array $data, string $expected)
+    {
+        $kernel = new WebpackEncoreIntegrationTestKernel(true);
+        $kernel->boot();
+        $twig = $this->getTwigEnvironmentFromBootedKernel($kernel);
+
+        $extension = new StimulusTwigExtension();
+        $this->assertSame($expected, $extension->renderStimulusController($twig, $data));
+    }
+
     private function getContainerFromBootedKernel(WebpackEncoreIntegrationTestKernel $kernel)
     {
         if ($kernel::VERSION_ID >= 40100) {