fix: Mark package with BUILD.bazel file in completion

bazelbuild · Oct 24, 2023 · 5cd29ec · 5cd29ec
1 parent a00e775
commit 5cd29ec
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/base/src/com/google/idea/blaze/base/lang/buildfile/references/FileLookupData.java b/base/src/com/google/idea/blaze/base/lang/buildfile/references/FileLookupData.java
@@ -15,6 +15,7 @@
  */
 package com.google.idea.blaze.base.lang.buildfile.references;
 
+import com.google.common.collect.ImmutableSet;
 import com.google.idea.blaze.base.lang.buildfile.completion.FilePathLookupElement;
 import com.google.idea.blaze.base.lang.buildfile.psi.BuildFile;
 import com.google.idea.blaze.base.lang.buildfile.psi.StringLiteral;
@@ -38,6 +39,7 @@
 
 /** The data relevant to finding file lookups. */
 public class FileLookupData {
+  private static final ImmutableSet<String> BUILDFILE_NAMES = ImmutableSet.of("BUILD", "BUILD.bazel");
 
   /** The type of path string format */
   public enum PathFormat {
@@ -163,8 +165,10 @@ public FilePathLookupElement lookupElementForFile(
         new NullableLazyValue<Icon>() {
           @Override
           protected Icon compute() {
-            if (file.findChild("BUILD") != null) {
-              return BlazeIcons.BuildFile;
+            for (String buildfileName : BUILDFILE_NAMES) {
+              if (file.findChild(buildfileName) != null) {
+                return BlazeIcons.BuildFile;
+              }
             }
             if (file.isDirectory()) {
               return PlatformIcons.FOLDER_ICON;