diff --git a/plugins/yaml/src/org/jetbrains/yaml/YAMLFindUsagesProvider.java b/plugins/yaml/src/org/jetbrains/yaml/YAMLFindUsagesProvider.java index f9e8e35f9f0a..b4be7c975021 100644 --- a/plugins/yaml/src/org/jetbrains/yaml/YAMLFindUsagesProvider.java +++ b/plugins/yaml/src/org/jetbrains/yaml/YAMLFindUsagesProvider.java @@ -3,14 +3,12 @@ package org.jetbrains.yaml; import com.intellij.lang.HelpID; import com.intellij.lang.cacheBuilder.WordsScanner; import com.intellij.lang.findUsages.FindUsagesProvider; +import com.intellij.openapi.util.text.StringUtil; import com.intellij.psi.PsiElement; import com.intellij.psi.PsiNamedElement; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; -import org.jetbrains.yaml.psi.YAMLMapping; -import org.jetbrains.yaml.psi.YAMLScalar; -import org.jetbrains.yaml.psi.YAMLScalarText; -import org.jetbrains.yaml.psi.YAMLSequence; +import org.jetbrains.yaml.psi.*; /** * @author shalupov @@ -42,6 +40,8 @@ public class YAMLFindUsagesProvider implements FindUsagesProvider { return "sequence"; } else if (element instanceof YAMLMapping) { return "mapping"; + } else if (element instanceof YAMLKeyValue) { + return "key-value"; } else { return ""; } @@ -50,9 +50,8 @@ public class YAMLFindUsagesProvider implements FindUsagesProvider { @NotNull @Override public String getDescriptiveName(@NotNull PsiElement element) { - final String name = element instanceof PsiNamedElement ? ((PsiNamedElement)element).getName() : null; - if (name != null) { - return name; + if (element instanceof PsiNamedElement) { + return StringUtil.notNullize(((PsiNamedElement)element).getName(), ""); } if (element instanceof YAMLScalar) { diff --git a/plugins/yaml/src/org/jetbrains/yaml/YAMLWordsScanner.java b/plugins/yaml/src/org/jetbrains/yaml/YAMLWordsScanner.java index 895e5b3d9afc..262e7c64ec29 100644 --- a/plugins/yaml/src/org/jetbrains/yaml/YAMLWordsScanner.java +++ b/plugins/yaml/src/org/jetbrains/yaml/YAMLWordsScanner.java @@ -11,12 +11,9 @@ public class YAMLWordsScanner extends DefaultWordsScanner { public YAMLWordsScanner() { super( new YAMLFlexLexer(), - TokenSet.create( - YAMLElementTypes.SCALAR_TEXT_VALUE, - YAMLElementTypes.SCALAR_PLAIN_VALUE, - YAMLElementTypes.SCALAR_QUOTED_STRING), + TokenSet.create(YAMLTokenTypes.SCALAR_KEY), TokenSet.create(YAMLTokenTypes.COMMENT), - TokenSet.EMPTY); + YAMLElementTypes.SCALAR_VALUES); setMayHaveFileRefsInLiterals(true); } }