No public description

PiperOrigin-RevId: 633877328
google · May 16, 2024 · f833225 · f833225
1 parent 4e715f3
commit f833225
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/java/src/com/google/template/soy/jbcsrc/api/SoySauce.java b/java/src/com/google/template/soy/jbcsrc/api/SoySauce.java
@@ -24,7 +24,6 @@
 import com.google.errorprone.annotations.CanIgnoreReturnValue;
 import com.google.errorprone.annotations.CheckReturnValue;
 import com.google.template.soy.data.SanitizedContent;
-import com.google.template.soy.data.SanitizedContent.ContentKind;
 import com.google.template.soy.data.SoyInjector;
 import com.google.template.soy.data.SoyTemplate;
 import com.google.template.soy.data.SoyTemplateData;
@@ -147,6 +146,13 @@ default Renderer setIj(SoyTemplateData data) {
     @CanIgnoreReturnValue
     Renderer setPluginInstances(Map<String, ? extends Supplier<Object>> pluginInstances);
 
+    /**
+     * Overrides plugin instances associated with the SoySauce instance for a given renderer, useful
+     * in cases where you want separate instance of a soy plugin for different soy renderers.
+     */
+    @CanIgnoreReturnValue
+    Renderer overridePluginInstances(Map<String, ? extends Supplier<Object>> newPluginInstances);
+
     /** Configures the {@code {css ..}} renaming map. */
     @CanIgnoreReturnValue
     Renderer setCssRenamingMap(SoyCssRenamingMap cssRenamingMap);

diff --git a/java/src/com/google/template/soy/jbcsrc/api/SoySauceImpl.java b/java/src/com/google/template/soy/jbcsrc/api/SoySauceImpl.java
@@ -37,6 +37,7 @@
 import com.google.template.soy.data.SoyValueProvider;
 import com.google.template.soy.data.UnsafeSanitizedContentOrdainer;
 import com.google.template.soy.data.internal.ParamStore;
+import com.google.template.soy.jbcsrc.api.SoySauce.WriteContinuation;
 import com.google.template.soy.jbcsrc.shared.CompiledTemplate;
 import com.google.template.soy.jbcsrc.shared.CompiledTemplates;
 import com.google.template.soy.jbcsrc.shared.RenderContext;
@@ -213,6 +214,15 @@ public RendererImpl setPluginInstances(
       return this;
     }
 
+    @CanIgnoreReturnValue
+    @Override
+    public RendererImpl overridePluginInstances(
+        Map<String, ? extends Supplier<Object>> newPluginInstances) {
+      this.pluginInstances =
+          SoySauceImpl.this.pluginInstances.combineOrOverride(newPluginInstances);
+      return this;
+    }
+
     @CanIgnoreReturnValue
     @Override
     public RendererImpl setData(Map<String, ?> record) {

diff --git a/java/src/com/google/template/soy/plugin/java/PluginInstances.java b/java/src/com/google/template/soy/plugin/java/PluginInstances.java
@@ -18,6 +18,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.ImmutableSet;
+import java.util.HashMap;
 import java.util.Map;
 import java.util.function.Supplier;
 
@@ -56,4 +57,12 @@ public PluginInstances combine(Map<String, ? extends Supplier<Object>> morePlugi
             .putAll(morePlugins)
             .build());
   }
+
+  public PluginInstances combineOrOverride(
+      Map<String, ? extends Supplier<Object>> newPluginInstances) {
+    Map<String, Supplier<Object>> overriddenPluginInstances = new HashMap<>();
+    overriddenPluginInstances.putAll(newPluginInstances);
+    this.pluginInstances.forEach(overriddenPluginInstances::putIfAbsent);
+    return new PluginInstances(ImmutableMap.copyOf(overriddenPluginInstances));
+  }
 }