add name completion suggestion for labels after mnemonics

Author: mike42

src/main/java/org/ca65/AsmIdentifierCompletionContributor.java

@@ -0,0 +1,44 @@
+package org.ca65;
+
+import com.intellij.codeInsight.completion.*;
+import com.intellij.codeInsight.lookup.LookupElementBuilder;
+import com.intellij.psi.PsiNamedElement;
+import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.util.ProcessingContext;
+import org.ca65.psi.AsmTypes;
+import org.jetbrains.annotations.NotNull;
+
+import static com.intellij.patterns.PlatformPatterns.psiElement;
+
+public class AsmIdentifierCompletionContributor extends CompletionContributor {
+    public AsmIdentifierCompletionContributor() {
+        // Offer to auto-complete identifiers after a mnemonic
+        extend(CompletionType.BASIC, psiElement().afterLeaf(psiElement(AsmTypes.MNEMONIC)), new LabelCompletionProvider());
+    }
+}
+
+class LabelCompletionProvider extends CompletionProvider<CompletionParameters> {
+    public void addCompletions(@NotNull CompletionParameters parameters,
+                               @NotNull ProcessingContext context,
+                               @NotNull CompletionResultSet resultSet) {
+        PsiNamedElement[] namedElements = PsiTreeUtil.getChildrenOfType(parameters.getOriginalFile(), PsiNamedElement.class);
+        if (namedElements != null) {
+            for (PsiNamedElement namedElement : namedElements) {
+                String elementName = namedElement.getName();
+                if (elementName != null) {
+                    if (!namedElement.getText().startsWith("@")) {
+                        resultSet.addElement(LookupElementBuilder.createWithIcon(namedElement));
+                    } else {
+                        // Present the label name, but complete with '@' prefix.
+                        resultSet.addElement(LookupElementBuilder.create(
+                                "@" + namedElement.getName())
+                                .withPsiElement(namedElement)
+                                .withPresentableText(namedElement.getName())
+                                .withIcon(namedElement.getIcon(0))
+                        );
+                    }
+                }
+            }
+        }
+    }
+}

src/main/java/org/ca65/psi/impl/AsmLabelDefinitionImpl.java

@@ -1,15 +1,19 @@
 package org.ca65.psi.impl;
 
 import com.intellij.extapi.psi.ASTWrapperPsiElement;
+import com.intellij.icons.AllIcons;
 import com.intellij.lang.ASTNode;
 import com.intellij.openapi.util.NlsSafe;
 import com.intellij.psi.PsiElement;
 import com.intellij.util.IncorrectOperationException;
+import org.ca65.AsmIcons;
 import org.ca65.psi.AsmLabelDefinition;
 import org.ca65.psi.AsmMarker;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
+import javax.swing.*;
+
 public class AsmLabelDefinitionImpl extends ASTWrapperPsiElement implements AsmLabelDefinition {
     public AsmLabelDefinitionImpl(@NotNull ASTNode node) {
         super(node);
@@ -29,4 +33,9 @@ public AsmLabelDefinitionImpl(@NotNull ASTNode node) {
     public PsiElement setName(@NlsSafe @NotNull String name) throws IncorrectOperationException {
         return AsmPsiImplUtil.setName((AsmMarker) this, name);
     }
+
+    @Override
+    public @Nullable Icon getIcon(int flags) {
+        return this.getText().startsWith("@") || this.getText().startsWith(":") ? AllIcons.Nodes.Annotationtype : AsmIcons.LABEL;
+    }
 }

src/main/resources/META-INF/plugin.xml

@@ -40,6 +40,7 @@
     <lang.commenter language="6502 Assembly" implementationClass="org.ca65.AsmCommenter"/>
     <lang.foldingBuilder language="6502 Assembly"
                          implementationClass="org.ca65.AsmFoldingBuilder"/>
+    <completion.contributor language="6502 Assembly" implementationClass="org.ca65.AsmIdentifierCompletionContributor" />
     <projectTemplate projectType="6502 Assembly" templatePath="projectTemplates/Generic 6502 Project.zip" category="true"/>
   </extensions>