annoation icon for completion

Author: tangzx

src/main/java/com/tang/intellij/lua/editor/completion/LuaCommentCompletionContributor.java

@@ -79,7 +79,7 @@ public LuaCommentCompletionContributor() {
             protected void addCompletions(@NotNull CompletionParameters completionParameters, ProcessingContext processingContext, @NotNull CompletionResultSet completionResultSet) {
                 TokenSet set = LuaSyntaxHighlighter.DOC_KEYWORD_TOKENS;
                 for (IElementType type : set.getTypes()) {
-                    completionResultSet.addElement(LookupElementBuilder.create(type));
+                    completionResultSet.addElement(LookupElementBuilder.create(type).withIcon(LuaIcons.ANNOTATION));
                 }
             }
         });
@@ -148,14 +148,11 @@ protected void addCompletions(@NotNull CompletionParameters completionParameters
                 LuaDocClassDef classDef = PsiTreeUtil.findChildOfType(comment, LuaDocClassDef.class);
                 if (classDef != null) {
                     LuaType classType = classDef.getClassType();
-                    if (classType != null) {
-                        classType.processFields(new SearchContext(classDef.getProject()), (curType, field) -> {
-                            completionResultSet.addElement(LookupElementBuilder.create(field.getFieldName()));
-                        });
-                    }
+                    classType.processFields(new SearchContext(classDef.getProject()), (curType, field) -> {
+                        completionResultSet.addElement(LookupElementBuilder.create(field.getFieldName()));
+                    });
                 }
             }
         });
     }
-
 }

src/main/java/com/tang/intellij/lua/lang/LuaIcons.java

@@ -39,6 +39,7 @@ public class LuaIcons {
     public static final Icon LOCAL_FUNCTION = new LayeredIcon(AllIcons.Nodes.Function, AllIcons.Nodes.C_private);
     public static final Icon PARAMETER = AllIcons.Nodes.Parameter;
     public static final Icon WORD = AllIcons.Actions.Edit;
+    public static final Icon ANNOTATION = AllIcons.Css.Atrule;
 
     public static final Icon MODULE = IconLoader.getIcon("/icons/module.png");
 }