swiftpm-mustache/Sources/HummingbirdMustache/Template.swift

/// 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, method: String? = nil)
        case unescapedVariable(name: String, method: String? = nil)
        case section(name: String, method: String? = nil, template: HBMustacheTemplate)
        case invertedSection(name: String, method: String? = nil, template: HBMustacheTemplate)
        case inheritedSection(name: String, template: HBMustacheTemplate)
        case partial(String, indentation: String?, inherits: [String: HBMustacheTemplate]?)
    }

    let tokens: [Token]
    var library: HBMustacheLibrary?
}