Load constants from AsTwigConstant attribute

composer.json

@@ -18,6 +18,7 @@
     "homepage": "http://github.com/jdecool/TwigConstantAccessorBundle",
     "require": {
         "php": "^8.1",
+        "kcs/class-finder": "^0.5.0",
         "symfony/dependency-injection": "^5.4 || ^6.4 || ^7.0",
         "symfony/framework-bundle": "^5.4 || ^6.4 || ^7.0",
         "symfony/twig-bundle": "^5.4 || ^6.4 || ^7.0",

src/Accessor/ConstantResolver.php

@@ -8,11 +8,11 @@
 class ConstantResolver
 {
     /**
-     * @param list<ConstantConfiguration> $classes
+     * @param iterable<ConstantConfiguration> $classes
      *
      * @return array<string, mixed>
      */
-    public function fromClassList(array $classes): array
+    public function fromClassList(iterable $classes): array
     {
         $constants = [];
 

src/Attribute/AsTwigConstant.php

@@ -0,0 +1,13 @@
+<?php
+
+namespace JDecool\Bundle\TwigConstantAccessorBundle\Attribute;
+
+#[\Attribute(\Attribute::TARGET_CLASS)]
+final class AsTwigConstant
+{
+    public function __construct(
+        public readonly ?string $alias = null,
+        public readonly ?string $matches = null,
+    ) {
+    }
+}

src/DependencyInjection/TwigConstantAccessorExtension.php

@@ -3,7 +3,9 @@
 namespace JDecool\Bundle\TwigConstantAccessorBundle\DependencyInjection;
 
 use JDecool\Bundle\TwigConstantAccessorBundle\Accessor\ConstantResolver;
+use JDecool\Bundle\TwigConstantAccessorBundle\Attribute\AsTwigConstant;
 use JDecool\Bundle\TwigConstantAccessorBundle\Twig\ConstantExtension;
+use Kcs\ClassFinder\Finder\ComposerFinder;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Definition;
 use Symfony\Component\DependencyInjection\Extension\Extension;
@@ -23,8 +25,12 @@ public function load(array $configs, ContainerBuilder $container): void
 
     private function resolveConstants(array $config): array
     {
+        $finder = new ComposerFinder();
         $resolver = new ConstantResolver();
 
-        return $resolver->fromClassList($config['classes']);
+        return array_merge(
+            $resolver->fromClassList($config['classes']),
+            $resolver->fromClassList($finder->withAttribute(AsTwigConstant::class)),
+        );
     }
 }

tests/Fixtures/AttributeConstant.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Fixtures;
+
+use JDecool\Bundle\TwigConstantAccessorBundle\Attribute\AsTwigConstant;
+
+#[AsTwigConstant]
+class AttributeConstant
+{
+    public const FROM_ATTRIBUTE_1 = 'from_attribute_1';
+    public const FROM_ATTRIBUTE_2 = 'from_attribute_2';
+}

tests/Fixtures/ConstantAccessFunctionalTest.php

@@ -32,6 +32,7 @@ public function testConstantsAreAccessibleInView(): void
         $content = $this->render('index.html.twig');
 
         $this->assertStringContainsString('my_constant.my_constant_value', $content);
+        $this->assertStringContainsString('from_attribute_1', $content);
     }
 
     private function render(string $template): string

tests/Fixtures/Resources/views/index.html.twig

@@ -1 +1,5 @@
+{# From `class` configuration #}
 {{ MyConstant.MY_CONSTANT_VALUE }}
+
+{# From Fixtures\AttributeConstant class registered with AsTwigConstant attribute #}
+{{ AttributeConstant.FROM_ATTRIBUTE_1 }}

tests/Fixtures/TemplateKernel.php

@@ -61,6 +61,6 @@ public function getLogDir(): string
 
     public function getBasePath(): string
     {
-        return sprintf('%s/%s/TemplateKernel', sys_get_temp_dir(), Kernel::VERSION);
+        return sprintf('%s/TwigConstantAccessor/%s/TemplateKernel', sys_get_temp_dir(), Kernel::VERSION);
     }
 }