Add support for Twig attributes extension inside the TwigExtension template (NewTwigExtensionAction)

Author: Haehnchen

src/main/java/fr/adrienbrault/idea/symfony2plugin/action/NewTwigExtensionAction.java

@@ -11,13 +11,15 @@
 import com.jetbrains.php.roots.PhpNamespaceCompositeProvider;
 import fr.adrienbrault.idea.symfony2plugin.Symfony2Icons;
 import fr.adrienbrault.idea.symfony2plugin.Symfony2ProjectComponent;
+import fr.adrienbrault.idea.symfony2plugin.util.PhpElementsUtil;
 import fr.adrienbrault.idea.symfony2plugin.util.psi.PhpBundleFileFactory;
 import org.apache.commons.lang3.StringUtils;
 import org.jetbrains.annotations.NotNull;
 
 import javax.swing.*;
 import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author Daniel Espendiller <daniel@espendiller.net>
@@ -63,16 +65,19 @@ public void actionPerformed(@NotNull AnActionEvent event) {
             return;
         }
 
+        // Detect available Twig attributes in the project
+        String templateName = detectTemplate(project);
+
         ApplicationManager.getApplication().runWriteAction(() -> {
-            HashMap<String, String> hashMap = new HashMap<>() {{
+            Map<String, String> hashMap = new HashMap<>() {{
                 put("class", className);
                 put("namespace", strings.get(0));
             }};
 
             PsiElement commandAttributes = PhpBundleFileFactory.createFile(
                 project,
                 directory.getVirtualFile(),
-                "twig_extension",
+                templateName,
                 className,
                 hashMap
             );
@@ -81,6 +86,17 @@ public void actionPerformed(@NotNull AnActionEvent event) {
         });
     }
 
+    @NotNull
+    private static String detectTemplate(@NotNull Project project) {
+        // If attributes are available, use the new attribute-based template
+        if (PhpElementsUtil.getClassInterface(project, "\\Twig\\Attribute\\AsTwigFunction") != null) {
+            return "twig_extension_function_attribute";
+        }
+
+        // Default to traditional template
+        return "twig_extension";
+    }
+
     public static class Shortcut extends NewTwigExtensionAction {
         @Override
         public void update(@NotNull AnActionEvent event) {

src/main/resources/fileTemplates/twig_extension_function_attribute.php

@@ -0,0 +1,23 @@
+<?php
+
+declare(strict_types=1);
+
+namespace {{ namespace }};
+
+use Twig\Attribute\AsTwigFunction;
+use Twig\Attribute\AsTwigFilter;
+
+class {{ class }}
+{
+    #[AsTwigFilter('my_filter')]
+    public function myFilter(): string
+    {
+        return 'Hello from PhpStorm';
+    }
+
+    #[AsTwigFunction('my_function')]
+    public function myFunction(): string
+    {
+        return 'Hello from PhpStorm';
+    }
+}