Fix IDE aspect crash when traversing a data mode deps attr.

--
MOS_MIGRATED_REVID=134566270

1 files changed, 5 insertions, 1 deletions

diff --git a/src/main/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspect.java b/src/main/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspect.java
index 9814a938ef..1b747b9f9d 100644
--- a/src/main/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspect.java
+++ b/src/main/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspect.java
@@ -227,7 +227,11 @@ public class AndroidStudioInfoAspect extends NativeAspectClass implements Config
     ImmutableList.Builder<TransitiveInfoCollection> directDepsBuilder = ImmutableList.builder();
     for (PrerequisiteAttr prerequisiteAttr : prerequisiteAttrs) {
       if (ruleContext.attributes().has(prerequisiteAttr.name, prerequisiteAttr.type)) {
-        directDepsBuilder.addAll(ruleContext.getPrerequisites(prerequisiteAttr.name, Mode.TARGET));
+        Mode mode = ruleContext.getAttributeMode(prerequisiteAttr.name);
+        if (mode == Mode.TARGET || mode == Mode.SPLIT) {
+          directDepsBuilder
+              .addAll(ruleContext.getPrerequisites(prerequisiteAttr.name, Mode.TARGET));
+        }
       }
     }
     List<TransitiveInfoCollection> directDeps = directDepsBuilder.build();