Merge pull request #150 from Vlatombe/more-view-types-in-structure

Add support for groovy views as well as help files in the structure view
jenkinsci · Apr 27, 2023 · 3452cb1 · 3452cb1
2 parents 2402c41 + 68de68c
commit 3452cb1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/java/org/kohsuke/stapler/idea/extension/JavaStructureViewExtension.java b/src/main/java/org/kohsuke/stapler/idea/extension/JavaStructureViewExtension.java
@@ -6,7 +6,7 @@
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.PsiPackage;
-import org.kohsuke.stapler.idea.psi.JellyFile;
+import java.util.Set;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -34,12 +34,16 @@ public StructureViewTreeElement[] getChildren(PsiElement parent) {
             if (psiPackage != null) {
                 PsiFile[] maybeViewFiles = psiPackage.getFiles(getCurrentScope(parent));
                 for (PsiFile file : maybeViewFiles) {
-                    if (file instanceof JellyFile) {
+                    if (isView(file)) {
                         files.add(new LeafPsiStructureViewTreeElement(file));
                     }
                 }
             }
         }
         return files.toArray(StructureViewTreeElement.EMPTY_ARRAY);
     }
+
+    private static boolean isView(PsiFile file) {
+        return Set.of("Groovy", "HTML", "Jelly").contains(file.getFileType().getName());
+    }
 }