63 lines
2.2 KiB
Swift
63 lines
2.2 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the Hummingbird server framework project
|
|
//
|
|
// Copyright (c) 2021-2021 the Hummingbird authors
|
|
// Licensed under Apache License v2.0
|
|
//
|
|
// See LICENSE.txt for license information
|
|
// See hummingbird/CONTRIBUTORS.txt for the list of Hummingbird authors
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
/// Class holding Mustache template
|
|
public final class HBMustacheTemplate {
|
|
/// Initialize template
|
|
/// - Parameter string: Template text
|
|
/// - Throws: HBMustacheTemplate.Error
|
|
public init(string: String) throws {
|
|
self.tokens = try Self.parse(string)
|
|
}
|
|
|
|
/// Render object using this template
|
|
/// - Parameter object: Object to render
|
|
/// - Returns: Rendered text
|
|
public func render(_ object: Any) -> String {
|
|
self.render(context: .init(object))
|
|
}
|
|
|
|
internal init(_ tokens: [Token]) {
|
|
self.tokens = tokens
|
|
}
|
|
|
|
internal func setLibrary(_ library: HBMustacheLibrary) {
|
|
self.library = library
|
|
for token in self.tokens {
|
|
switch token {
|
|
case .section(_, _, let template), .invertedSection(_, _, let template), .inheritedSection(_, let template):
|
|
template.setLibrary(library)
|
|
case .partial(_, _, let templates):
|
|
templates?.forEach { $1.setLibrary(library) }
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
enum Token {
|
|
case text(String)
|
|
case variable(name: String, transform: String? = nil)
|
|
case unescapedVariable(name: String, transform: String? = nil)
|
|
case section(name: String, transform: String? = nil, template: HBMustacheTemplate)
|
|
case invertedSection(name: String, transform: String? = nil, template: HBMustacheTemplate)
|
|
case inheritedSection(name: String, template: HBMustacheTemplate)
|
|
case partial(String, indentation: String?, inherits: [String: HBMustacheTemplate]?)
|
|
case contentType(HBMustacheContentType)
|
|
}
|
|
|
|
let tokens: [Token]
|
|
var library: HBMustacheLibrary?
|
|
}
|