From cf1073e97336e961291126556bdfa9eb263f11ef Mon Sep 17 00:00:00 2001
From: Leonid Shalupov <leonid@shalupov.com>
Date: Mon, 14 Aug 2017 19:05:48 +0300
Subject: [PATCH] fixes after review IDEA-CR-22788

---
 .../org/jetbrains/yaml/YAMLFindUsagesProvider.java  | 13 ++++++-------
 .../src/org/jetbrains/yaml/YAMLWordsScanner.java    |  7 ++-----
 2 files changed, 8 insertions(+), 12 deletions(-)

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(), "<unnamed>");
     }
 
     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);
   }
 }