blob: b44fa74b3ace9c3dd06ce06cce1ce10623f9cf3a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.intellij.schema.findUsages;
import com.intellij.openapi.project.DumbAware;
import com.intellij.psi.PsiElement;
import com.intellij.usages.Usage;
import com.intellij.usages.UsageGroup;
import com.intellij.usages.UsageTarget;
import com.intellij.usages.rules.PsiElementUsage;
import com.intellij.usages.rules.SingleParentUsageGroupingRule;
import ai.vespa.intellij.schema.SdLanguage;
import ai.vespa.intellij.schema.psi.SdRankProfileDefinition;
/**
* A grouping rule that groups elements in the "Find Usages" window.
*
* @author Shahar Ariel
*/
public class SdRankProfileGroupingRule extends SingleParentUsageGroupingRule implements DumbAware {
@Override
protected UsageGroup getParentGroupFor(Usage usage, UsageTarget[] targets) {
PsiElement psiElement = usage instanceof PsiElementUsage ? ((PsiElementUsage)usage).getElement() : null;
if (psiElement == null || psiElement.getLanguage() != SdLanguage.INSTANCE) return null;
while (psiElement != null) {
if (psiElement instanceof SdRankProfileDefinition) {
SdRankProfileDefinition componentElement = (SdRankProfileDefinition) psiElement;
return new SdUsageGroup(componentElement);
}
psiElement = psiElement.getParent();
}
return null;
}
}
|