From 7679b48164331a54eb20e3a85fd2ac2b426a7390 Mon Sep 17 00:00:00 2001
From: Yonas Kolb <yonaskolb@gmail.com>
Date: Mon, 30 Apr 2018 11:45:22 +1000
Subject: [PATCH] add contex to include

---
 Sources/Include.swift                | 13 ++++++++-----
 Tests/StencilTests/IncludeSpec.swift |  9 ++++++++-
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/Sources/Include.swift b/Sources/Include.swift
index cd9cc5c..3fc9051 100644
--- a/Sources/Include.swift
+++ b/Sources/Include.swift
@@ -3,19 +3,21 @@ import PathKit
 
 class IncludeNode : NodeType {
   let templateName: Variable
+  let includeContext: String?
 
   class func parse(_ parser: TokenParser, token: Token) throws -> NodeType {
     let bits = token.components()
 
-    guard bits.count == 2 else {
-      throw TemplateSyntaxError("'include' tag takes one argument, the template file to be included")
+    guard bits.count == 2 || (bits.count == 4 && bits[2] == "using") else {
+      throw TemplateSyntaxError("'include' tag requires one argument, the template file to be included. Another optional argument can be used to specify the context that will be passed to the included file, using the format \"using myContext\"")
     }
 
-    return IncludeNode(templateName: Variable(bits[1]))
+    return IncludeNode(templateName: Variable(bits[1]), includeContext: bits.count == 4 ? bits[3] : nil)
   }
 
-  init(templateName: Variable) {
+  init(templateName: Variable, includeContext: String? = nil) {
     self.templateName = templateName
+    self.includeContext = includeContext
   }
 
   func render(_ context: Context) throws -> String {
@@ -25,7 +27,8 @@ class IncludeNode : NodeType {
 
     let template = try context.environment.loadTemplate(name: templateName)
 
-    return try context.push {
+    let subContext = includeContext.flatMap{ context[$0] as? [String: Any] }
+    return try context.push(dictionary: subContext) {
       return try template.render(context)
     }
   }
diff --git a/Tests/StencilTests/IncludeSpec.swift b/Tests/StencilTests/IncludeSpec.swift
index 0297896..cf785e2 100644
--- a/Tests/StencilTests/IncludeSpec.swift
+++ b/Tests/StencilTests/IncludeSpec.swift
@@ -14,7 +14,7 @@ func testInclude() {
         let tokens: [Token] = [ .block(value: "include") ]
         let parser = TokenParser(tokens: tokens, environment: Environment())
 
-        let error = TemplateSyntaxError("'include' tag takes one argument, the template file to be included")
+        let error = TemplateSyntaxError("'include' tag requires one argument, the template file to be included. Another optional argument can be used to specify the context that will be passed to the included file, using the format \"using myContext\"")
         try expect(try parser.parse()).toThrow(error)
       }
 
@@ -56,6 +56,13 @@ func testInclude() {
         let value = try node.render(context)
         try expect(value) == "Hello World!"
       }
+
+      $0.it("successfully passes context") {
+        let template = Template(templateString: "{% include \"test.html\" using child %}")
+        let context = Context(dictionary: ["child": ["target": "World"]], environment: environment)
+        let value = try template.render(context)
+        try expect(value) == "Hello World!"
+      }
     }
   }
 }