39 lines
1.1 KiB
Swift
39 lines
1.1 KiB
Swift
import PathKit
|
|
|
|
|
|
class IncludeNode : NodeType {
|
|
let templateName: Variable
|
|
|
|
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")
|
|
}
|
|
|
|
return IncludeNode(templateName: Variable(bits[1]))
|
|
}
|
|
|
|
init(templateName: Variable) {
|
|
self.templateName = templateName
|
|
}
|
|
|
|
func render(_ context: Context) throws -> String {
|
|
guard let loader = context["loader"] as? TemplateLoader else {
|
|
throw TemplateSyntaxError("Template loader not in context")
|
|
}
|
|
|
|
guard let templateName = try self.templateName.resolve(context) as? String else {
|
|
throw TemplateSyntaxError("'\(self.templateName)' could not be resolved as a string")
|
|
}
|
|
|
|
guard let template = loader.loadTemplate(templateName) else {
|
|
let paths = loader.paths.map { $0.description }.joined(separator: ", ")
|
|
throw TemplateSyntaxError("'\(templateName)' template not found in \(paths)")
|
|
}
|
|
|
|
return try template.render(context)
|
|
}
|
|
}
|
|
|