29 lines
1.1 KiB
Swift
29 lines
1.1 KiB
Swift
import Foundation
|
|
|
|
extension HBMustacheLibrary {
|
|
/// Load templates from a folder
|
|
func loadTemplates(from directory: String, withExtension extension: String = "mustache") throws {
|
|
var directory = directory
|
|
if !directory.hasSuffix("/") {
|
|
directory += "/"
|
|
}
|
|
let extWithDot = ".\(`extension`)"
|
|
let fs = FileManager()
|
|
guard let enumerator = fs.enumerator(atPath: directory) else { return }
|
|
for case let path as String in enumerator {
|
|
guard path.hasSuffix(extWithDot) else { continue }
|
|
guard let data = fs.contents(atPath: directory + path) else { continue }
|
|
let string = String(decoding: data, as: Unicode.UTF8.self)
|
|
let template: HBMustacheTemplate
|
|
do {
|
|
template = try HBMustacheTemplate(string: string)
|
|
} catch {
|
|
throw Error.failedToLoad(path, error)
|
|
}
|
|
// drop ".mustache" from path to get name
|
|
let name = String(path.dropLast(extWithDot.count))
|
|
register(template, named: name)
|
|
}
|
|
}
|
|
}
|