Don't include declarations in usages (fixes #629).

Signed-off-by: Ilya Rodionov <[email protected]>
antlr · Jul 7, 2023 · 7227cbc · 7227cbc
1 parent cd9a2c8
commit 7227cbc
Show file tree

Hide file tree

Showing 5 changed files with 40 additions and 6 deletions.
diff --git a/src/main/java/org/antlr/intellij/plugin/ANTLRv4ASTFactory.java b/src/main/java/org/antlr/intellij/plugin/ANTLRv4ASTFactory.java
@@ -29,6 +29,7 @@ public class ANTLRv4ASTFactory extends ASTFactory {
 		ruleElementTypeToPsiFactory.put(ANTLRv4TokenTypes.RULE_ELEMENT_TYPES.get(ANTLRv4Parser.RULE_modeSpec), ModeSpecNode.Factory.INSTANCE);
 		ruleElementTypeToPsiFactory.put(ANTLRv4TokenTypes.RULE_ELEMENT_TYPES.get(ANTLRv4Parser.RULE_action), AtAction.Factory.INSTANCE);
 		ruleElementTypeToPsiFactory.put(ANTLRv4TokenTypes.RULE_ELEMENT_TYPES.get(ANTLRv4Parser.RULE_identifier), TokenSpecNode.Factory.INSTANCE);
+		ruleElementTypeToPsiFactory.put(ANTLRv4TokenTypes.RULE_ELEMENT_TYPES.get(ANTLRv4Parser.RULE_tokensSpec), TokensSpecNode.Factory.INSTANCE);
 	}
 
 	/** Create a FileElement for root or a parse tree CompositeElement (not

diff --git a/src/main/java/org/antlr/intellij/plugin/psi/GrammarElementRefNode.java b/src/main/java/org/antlr/intellij/plugin/psi/GrammarElementRefNode.java
@@ -1,5 +1,6 @@
 package org.antlr.intellij.plugin.psi;
 
+import com.intellij.psi.PsiReference;
 import com.intellij.psi.impl.source.tree.LeafPsiElement;
 import com.intellij.psi.tree.IElementType;
 
@@ -11,6 +12,16 @@ public GrammarElementRefNode(IElementType type, CharSequence text) {
 		super(type, text);
 	}
 
+	protected abstract boolean isDeclaration();
+
+	@Override
+	public PsiReference getReference() {
+		if (isDeclaration()) {
+			return null;
+		}
+		return new GrammarElementRef(this, getText());
+	}
+
 	@Override
 	public String getName() {
 		return getText();

diff --git a/src/main/java/org/antlr/intellij/plugin/psi/LexerRuleRefNode.java b/src/main/java/org/antlr/intellij/plugin/psi/LexerRuleRefNode.java
@@ -1,15 +1,16 @@
 package org.antlr.intellij.plugin.psi;
 
-import com.intellij.psi.PsiReference;
 import com.intellij.psi.tree.IElementType;
+import com.intellij.psi.util.PsiTreeUtil;
 
 public class LexerRuleRefNode extends GrammarElementRefNode {
 	public LexerRuleRefNode(IElementType type, CharSequence text) {
 		super(type, text);
 	}
 
 	@Override
-	public PsiReference getReference() {
-		return new GrammarElementRef(this, getText());
+	protected boolean isDeclaration() {
+		return getParent() instanceof LexerRuleSpecNode
+			|| PsiTreeUtil.getParentOfType(this, TokensSpecNode.class) != null;
 	}
 }
diff --git a/src/main/java/org/antlr/intellij/plugin/psi/ParserRuleRefNode.java b/src/main/java/org/antlr/intellij/plugin/psi/ParserRuleRefNode.java
@@ -1,6 +1,5 @@
 package org.antlr.intellij.plugin.psi;
 
-import com.intellij.psi.PsiReference;
 import com.intellij.psi.tree.IElementType;
 
 public class ParserRuleRefNode extends GrammarElementRefNode {
@@ -9,7 +8,7 @@ public ParserRuleRefNode(IElementType type, CharSequence text) {
 	}
 
 	@Override
-	public PsiReference getReference() {
-		return new GrammarElementRef(this, getText());
+	protected boolean isDeclaration() {
+		return getParent() instanceof ParserRuleSpecNode;
 	}
 }
diff --git a/src/main/java/org/antlr/intellij/plugin/psi/TokensSpecNode.java b/src/main/java/org/antlr/intellij/plugin/psi/TokensSpecNode.java
@@ -0,0 +1,22 @@
+package org.antlr.intellij.plugin.psi;
+
+import com.intellij.extapi.psi.ASTWrapperPsiElement;
+import com.intellij.lang.ASTNode;
+import com.intellij.psi.PsiElement;
+import org.antlr.intellij.adaptor.parser.PsiElementFactory;
+import org.jetbrains.annotations.NotNull;
+
+public class TokensSpecNode extends ASTWrapperPsiElement {
+    public TokensSpecNode(@NotNull ASTNode node) {
+        super(node);
+    }
+
+    public static class Factory implements PsiElementFactory {
+        public static TokensSpecNode.Factory INSTANCE = new TokensSpecNode.Factory();
+
+        @Override
+        public PsiElement createElement(ASTNode node) {
+            return new TokensSpecNode(node);
+        }
+    }
+}