1 files changed, 20 insertions, 0 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/syntax/SkylarkUtils.java b/src/main/java/com/google/devtools/build/lib/syntax/SkylarkUtils.java
index 970543f60a..397768afe4 100644
--- a/src/main/java/com/google/devtools/build/lib/syntax/SkylarkUtils.java
+++ b/src/main/java/com/google/devtools/build/lib/syntax/SkylarkUtils.java
@@ -14,12 +14,15 @@
 
 package com.google.devtools.build.lib.syntax;
 
+import com.google.common.collect.ImmutableMap;
+
 /** This class contains Bazel-specific functions to extend or interoperate with Skylark. */
 public final class SkylarkUtils {
 
   /** Bazel-specific information that we store in the Environment. */
   private static class BazelInfo {
     String toolsRepository;
+    ImmutableMap<String, Class<?>> fragmentNameToClass;
   }
 
   private static final String BAZEL_INFO_KEY = "$bazel";
@@ -46,4 +49,21 @@ public final class SkylarkUtils {
   public static String getToolsRepository(Environment env) {
     return getInfo(env).toolsRepository;
   }
+
+  /**
+   * Sets, on an {@link Environment}, a {@link Map} from configuration fragment name to
+   * configuration fragment class.
+   */
+  public static void setFragmentMap(Environment env,
+      ImmutableMap<String, Class<?>> fragmentNameToClass) {
+    getInfo(env).fragmentNameToClass = fragmentNameToClass;
+  }
+
+  /*
+   * Returns the {@link Map} from configuration fragment name to configuration fragment class, as
+   * set by {@link #setFragmentMap}.
+   */
+  public static ImmutableMap<String, Class<?>> getFragmentMap(Environment env) {
+    return getInfo(env).fragmentNameToClass;
+  }
 }