fix: Query Sync can now work with '.' in the directories section (#6834)

bazelbuild · Oct 3, 2024 · 7b52b42 · 7b52b42
1 parent 3f676d4
commit 7b52b42
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/querysync/java/com/google/idea/blaze/qsync/BlazeQueryParser.java b/querysync/java/com/google/idea/blaze/qsync/BlazeQueryParser.java
@@ -37,6 +37,7 @@
 import com.google.idea.blaze.qsync.query.Query.Rule;
 import com.google.idea.blaze.qsync.query.QuerySummary;
 import java.nio.file.Path;
+import java.nio.file.Paths;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
@@ -97,8 +98,12 @@ public BuildGraphData parse() {
     for (Map.Entry<Label, Query.SourceFile> sourceFileEntry :
         query.getSourceFilesMap().entrySet()) {
       Location l = new Location(sourceFileEntry.getValue().getLocation());
-      if (l.file.endsWith(Path.of("BUILD"))) {
-        packages.add(l.file.getParent());
+      if (l.file.endsWith(Path.of("BUILD")) || l.file.endsWith(Path.of("BUILD.bazel"))){
+        if(l.file.getParent() == null){
+          packages.add(Paths.get(""));
+        } else{
+          packages.add(l.file.getParent());
+        }
       }
       graphBuilder.locationsBuilder().put(sourceFileEntry.getKey(), l);
       graphBuilder.fileToTargetBuilder().put(l.file, sourceFileEntry.getKey());

diff --git a/querysync/java/com/google/idea/blaze/qsync/GraphToProjectConverter.java b/querysync/java/com/google/idea/blaze/qsync/GraphToProjectConverter.java
@@ -219,7 +219,7 @@ ImmutableMap<Path, ImmutableMap<Path, String>> splitByRoot(Map<Path, String> pre
       ImmutableMap.Builder<Path, String> inRoot = ImmutableMap.builder();
       for (Entry<Path, String> pkg : prefixes.entrySet()) {
         Path rel = pkg.getKey();
-        if (rel.startsWith(root)) {
+        if (rel.startsWith(root) || root.toString().equals("")) {
           Path relToRoot = root.relativize(rel);
           inRoot.put(relToRoot, pkg.getValue());
         }