Extend set of supported entities for GoToThriftDefinitionMarkerProvid…

…er (#94) More scrooge types, as well as all generated classes
fkorotkov · Aug 31, 2020 · bb9067e · bb9067e
1 parent 92950d1
commit bb9067e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/.../src/main/java/com/intellij/plugins/thrift/editor/GoToThriftDefinitionMarkerProvider.java b/.../src/main/java/com/intellij/plugins/thrift/editor/GoToThriftDefinitionMarkerProvider.java
@@ -34,12 +34,18 @@ public class GoToThriftDefinitionMarkerProvider extends RelatedItemLineMarkerPro
 
   private static Set<String> thriftStructBaseClasses =
       Sets.newHashSet(
+          "org.apache.thrift.TBase",
           "com.twitter.scrooge.ThriftStruct",
-          "org.apache.thrift.TBase"
+          "com.twitter.scrooge.ThriftEnum",
+          "com.twitter.scrooge.ThriftUnion",
+          "com.twitter.scrooge.ThriftException"
       );
 
   private boolean isThriftStruct(PsiClass element) {
-    return Arrays.stream(element.getSupers()).anyMatch(superClass -> thriftStructBaseClasses.contains(superClass.getQualifiedName()));
+    return Arrays
+        .stream(element.getSupers())
+        .anyMatch(superClass -> thriftStructBaseClasses.contains(superClass.getQualifiedName()))
+        || element.getAnnotation("javax.annotation.Generated") != null;
   }
 
   @Override