Add support for member variables

This commit is contained in:
Adam Fowler
2021-03-11 17:14:12 +00:00
parent 13f6d20976
commit 55245e960f
2 changed files with 35 additions and 6 deletions

View File

@@ -89,13 +89,24 @@ extension HBTemplate {
}
}
func getChild(named name: String, from: Any) -> Any? {
if name == "." { return from }
if let customBox = from as? HBMustacheParent {
return customBox.child(named: name)
func getChild(named name: String, from object: Any) -> Any? {
func _getChild(named names: ArraySlice<String>, from object: Any) -> Any? {
guard let name = names.first else { return object }
let childObject: Any?
if let customBox = object as? HBMustacheParent {
childObject = customBox.child(named: name)
} else {
let mirror = Mirror(reflecting: object)
childObject = mirror.getAttribute(forKey: name)
}
let mirror = Mirror(reflecting: from)
return mirror.getAttribute(forKey: name)
guard childObject != nil else { return nil }
let names2 = names.dropFirst()
return _getChild(named: names2, from: childObject!)
}
if name == "." { return object }
let nameSplit = name.split(separator: ".").map { String($0) }
return _getChild(named: nameSplit[...], from: object)
}
}

View File

@@ -79,4 +79,22 @@ final class TemplateRendererTests: XCTestCase {
XCTAssertEqual(template2.render(Test(string: "string")), "test ")
XCTAssertEqual(template2.render(Test(string: nil)), "test *")
}
func testDictionarySequence() throws {
let template = try HBTemplate("test {{#.}}{{value}}{{/.}}")
XCTAssert(template.render(["one": 1, "two": 2]) == "test 12" ||
template.render(["one": 1, "two": 2]) == "test 21")
}
func testStructureInStructure() throws {
struct SubTest {
let string: String?
}
struct Test {
let test: SubTest
}
let template = try HBTemplate("test {{test.string}}")
XCTAssertEqual(template.render(Test(test: .init(string: "sub"))), "test sub")
}
}