#125: Fix icons in "Go To Symbol" dialog

Author: kshchepanovskyi

src/main/java/io/protostuff/jetbrains/plugin/psi/EnumConstantNode.java

@@ -10,6 +10,8 @@
 import com.intellij.navigation.ItemPresentationProviders;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiNamedElement;
+import io.protostuff.jetbrains.plugin.Icons;
+import javax.swing.Icon;
 import org.antlr.jetbrains.adapter.psi.ScopeNode;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -59,6 +61,12 @@ public ASTNode getConstantValueNode() {
         return node.findChildByType(rule(RULE_enumFieldValue));
     }
 
+    @Nullable
+    @Override
+    public Icon getIcon(int flags) {
+        return Icons.CONSTANT;
+    }
+
     @Override
     public ItemPresentation getPresentation() {
         return ItemPresentationProviders.getItemPresentation(this);

src/main/java/io/protostuff/jetbrains/plugin/psi/EnumNode.java

@@ -7,11 +7,14 @@
 import com.intellij.psi.stubs.IStubElementType;
 import com.intellij.psi.tree.IElementType;
 import com.intellij.util.IncorrectOperationException;
+import io.protostuff.jetbrains.plugin.Icons;
 import io.protostuff.jetbrains.plugin.psi.stubs.EnumStub;
 import io.protostuff.jetbrains.plugin.psi.stubs.MessageStub;
 import java.util.Arrays;
 import java.util.List;
+import javax.swing.Icon;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 /**
  * Enum node.
@@ -40,6 +43,12 @@ public PsiElement setName(@NotNull String name) throws IncorrectOperationExcepti
         return GenericNameNode.setName(this, name);
     }
 
+    @Nullable
+    @Override
+    public Icon getIcon(int flags) {
+        return Icons.ENUM;
+    }
+
     @Override
     public ItemPresentation getPresentation() {
         return ItemPresentationProviders.getItemPresentation(this);

src/main/java/io/protostuff/jetbrains/plugin/psi/FieldNode.java

@@ -10,10 +10,13 @@
 import com.intellij.navigation.ItemPresentationProviders;
 import com.intellij.openapi.diagnostic.Logger;
 import com.intellij.psi.PsiElement;
+import io.protostuff.jetbrains.plugin.Icons;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.Optional;
+import javax.swing.Icon;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 /**
  * Field node.
@@ -85,6 +88,12 @@ public ASTNode getFieldLabelNode() {
         return node.findChildByType(R_FIELD_MODIFIER);
     }
 
+    @Nullable
+    @Override
+    public Icon getIcon(int flags) {
+        return Icons.FIELD;
+    }
+
     @Override
     public ItemPresentation getPresentation() {
         return ItemPresentationProviders.getItemPresentation(this);

src/main/java/io/protostuff/jetbrains/plugin/psi/MessageNode.java

@@ -7,12 +7,14 @@
 import com.intellij.psi.stubs.IStubElementType;
 import com.intellij.psi.tree.IElementType;
 import com.intellij.util.IncorrectOperationException;
+import io.protostuff.jetbrains.plugin.Icons;
 import io.protostuff.jetbrains.plugin.psi.stubs.DataTypeStub;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
+import javax.swing.Icon;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -89,6 +91,12 @@ public Collection<MessageField> getFields() {
         return result;
     }
 
+    @Nullable
+    @Override
+    public Icon getIcon(int flags) {
+        return Icons.MESSAGE;
+    }
+
     @Override
     public ItemPresentation getPresentation() {
         return ItemPresentationProviders.getItemPresentation(this);

src/main/java/io/protostuff/jetbrains/plugin/psi/RpcMethodNode.java

@@ -4,6 +4,8 @@
 import static io.protostuff.jetbrains.plugin.ProtoParserDefinition.rule;
 
 import com.intellij.lang.ASTNode;
+import io.protostuff.jetbrains.plugin.Icons;
+import javax.swing.Icon;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -35,4 +37,11 @@ public ASTNode getMethodNameNode() {
         ASTNode node = getNode();
         return node.findChildByType(rule(RULE_rpcName));
     }
+
+    @Nullable
+    @Override
+    public Icon getIcon(int flags) {
+        return Icons.ENUM;
+    }
+    
 }

src/main/java/io/protostuff/jetbrains/plugin/psi/ServiceNode.java

@@ -6,8 +6,10 @@
 import com.intellij.navigation.ItemPresentationProviders;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiNamedElement;
+import io.protostuff.jetbrains.plugin.Icons;
 import java.util.Arrays;
 import java.util.List;
+import javax.swing.Icon;
 import org.antlr.jetbrains.adapter.psi.ScopeNode;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -36,6 +38,12 @@ public List<RpcMethodNode> getRpcMethods() {
         return Arrays.asList(nodes);
     }
 
+    @Nullable
+    @Override
+    public Icon getIcon(int flags) {
+        return Icons.SERVICE;
+    }
+
     @Override
     public ItemPresentation getPresentation() {
         return ItemPresentationProviders.getItemPresentation(this);