Internal change.

PiperOrigin-RevId: 377479339
kekxv · Jun 4, 2021 · 6e21b2a · 6e21b2a
1 parent 88cd0a2
commit 6e21b2a
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/sdkcompat/v202/com/google/idea/sdkcompat/general/BaseSdkCompat.java b/sdkcompat/v202/com/google/idea/sdkcompat/general/BaseSdkCompat.java
@@ -11,6 +11,7 @@
 import com.intellij.openapi.Disposable;
 import com.intellij.openapi.editor.Document;
 import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.colors.EditorColorsScheme;
 import com.intellij.openapi.extensions.PluginId;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.project.ex.ProjectManagerEx;
@@ -20,6 +21,9 @@
 import com.intellij.psi.codeStyle.CodeStyleSettingsManager;
 import com.intellij.psi.impl.source.codeStyle.CodeFormatterFacade;
 import com.intellij.ui.content.ContentManager;
+import com.intellij.usageView.UsageTreeColors;
+import com.intellij.usageView.UsageTreeColorsScheme;
+import com.intellij.usages.TextChunk;
 import com.intellij.util.ContentUtilEx;
 import com.jetbrains.python.psi.LanguageLevel;
 import java.io.File;
@@ -123,4 +127,13 @@ public static void setSelectedTabID(SearchEverywhereManager manager, String id)
   public static Comparator<LanguageLevel> getLanguageLevelComparator() {
     return Comparator.comparingInt(LanguageLevel::getVersion);
   }
+
+  /** #api203: refactor this function back into CodesearchResultData and make it private. */
+  public static void addLineNumber(int lineNumber, List<TextChunk> chunks) {
+    EditorColorsScheme colorsScheme = UsageTreeColorsScheme.getInstance().getScheme();
+    chunks.add(
+        new TextChunk(
+            colorsScheme.getAttributes(UsageTreeColors.USAGE_LOCATION),
+            String.valueOf(lineNumber)));
+  }
 }
diff --git a/sdkcompat/v203/com/google/idea/sdkcompat/general/BaseSdkCompat.java b/sdkcompat/v203/com/google/idea/sdkcompat/general/BaseSdkCompat.java
@@ -11,6 +11,7 @@
 import com.intellij.openapi.Disposable;
 import com.intellij.openapi.editor.Document;
 import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.colors.EditorColorsScheme;
 import com.intellij.openapi.editor.ex.util.EditorFacade;
 import com.intellij.openapi.extensions.PluginId;
 import com.intellij.openapi.project.Project;
@@ -19,6 +20,9 @@
 import com.intellij.openapi.vcs.FilePath;
 import com.intellij.psi.PsiElement;
 import com.intellij.ui.content.ContentManager;
+import com.intellij.usageView.UsageTreeColors;
+import com.intellij.usageView.UsageTreeColorsScheme;
+import com.intellij.usages.TextChunk;
 import com.intellij.util.ContentUtilEx;
 import com.jetbrains.python.psi.LanguageLevel;
 import java.nio.charset.Charset;
@@ -119,4 +123,13 @@ public static Comparator<LanguageLevel> getLanguageLevelComparator() {
     return Comparator.comparingInt(LanguageLevel::getMajorVersion)
         .thenComparingInt(LanguageLevel::getMinorVersion);
   }
+
+  /** #api203: refactor this function back into CodesearchResultData and make it private. */
+  public static void addLineNumber(int lineNumber, List<TextChunk> chunks) {
+    EditorColorsScheme colorsScheme = UsageTreeColorsScheme.getInstance().getScheme();
+    chunks.add(
+        new TextChunk(
+            colorsScheme.getAttributes(UsageTreeColors.USAGE_LOCATION),
+            String.valueOf(lineNumber)));
+  }
 }
diff --git a/sdkcompat/v211/com/google/idea/sdkcompat/general/BaseSdkCompat.java b/sdkcompat/v211/com/google/idea/sdkcompat/general/BaseSdkCompat.java
@@ -19,6 +19,8 @@
 import com.intellij.openapi.vcs.FilePath;
 import com.intellij.psi.PsiElement;
 import com.intellij.ui.content.ContentManager;
+import com.intellij.usageView.UsageTreeColors;
+import com.intellij.usages.TextChunk;
 import com.intellij.util.ContentUtilEx;
 import com.jetbrains.python.psi.LanguageLevel;
 import java.nio.charset.Charset;
@@ -119,4 +121,12 @@ public static Comparator<LanguageLevel> getLanguageLevelComparator() {
     return Comparator.comparingInt(LanguageLevel::getMajorVersion)
         .thenComparingInt(LanguageLevel::getMinorVersion);
   }
+
+  /** #api203: refactor this function back into CodesearchResultData and make it private. */
+  public static void addLineNumber(int lineNumber, List<TextChunk> chunks) {
+    chunks.add(
+        new TextChunk(
+            UsageTreeColors.NUMBER_OF_USAGES_ATTRIBUTES.toTextAttributes(),
+            String.valueOf(lineNumber)));
+  }
 }