Exposing methods to render jtwig template on specified output stream

jtwig · Feb 25, 2017 · 2e6d2bf · 2e6d2bf
1 parent f9eac3f
commit 2e6d2bf
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/main/java/org/jtwig/web/servlet/JtwigDispatcher.java b/src/main/java/org/jtwig/web/servlet/JtwigDispatcher.java
@@ -13,6 +13,7 @@
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
+import java.io.OutputStream;
 import java.util.Enumeration;
 import java.util.Map;
 
@@ -48,12 +49,17 @@ public void render(HttpServletRequest request, HttpServletResponse response) thr
 
         model.with("app", applicationFactory.create(request));
 
+        render(response.getOutputStream());
+    }
+
+    public void render (OutputStream outputStream) {
         ResourceMetadata resourceMetadata = environment.getResourceEnvironment().getResourceService().loadMetadata(resourceReference);
         if (!resourceMetadata.exists())
             throw new ResourceNotFoundException(String.format("Resource %s not found", resourceReference));
 
+
         JtwigTemplate jtwigTemplate = new JtwigTemplate(environment, resourceReference);
-        jtwigTemplate.render(model, response.getOutputStream());
+        jtwigTemplate.render(model, outputStream);
     }
 
     private void fillModelWithRequest(ServletRequest request) {

diff --git a/src/main/java/org/jtwig/web/servlet/JtwigRenderer.java b/src/main/java/org/jtwig/web/servlet/JtwigRenderer.java
@@ -48,4 +48,8 @@ public JtwigDispatcher dispatcherFor (String location) {
     public JtwigDispatcher inlineDispatcherFor (String template) {
         return new JtwigDispatcher(environment, new ResourceReference(ResourceReference.STRING, template), applicationFactory);
     }
+
+    public Environment getEnvironment() {
+        return environment;
+    }
 }