Add index filter
This commit is contained in:
@@ -2,10 +2,12 @@
|
||||
struct HBMustacheContext: HBMustacheMethods {
|
||||
var first: Bool
|
||||
var last: Bool
|
||||
var index: Int
|
||||
|
||||
init(first: Bool = false, last: Bool = false) {
|
||||
self.first = first
|
||||
self.last = last
|
||||
self.index = 0
|
||||
}
|
||||
|
||||
func runMethod(_ name: String) -> Any? {
|
||||
@@ -14,6 +16,8 @@ struct HBMustacheContext: HBMustacheMethods {
|
||||
return self.first
|
||||
case "last":
|
||||
return self.last
|
||||
case "index":
|
||||
return self.index
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -39,3 +39,14 @@ extension Dictionary: HBMustacheMethods {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
extension Int: HBMustacheMethods {
|
||||
func runMethod(_ name: String) -> Any? {
|
||||
switch name {
|
||||
case "plus1":
|
||||
return self + 1
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -15,6 +15,7 @@ extension Sequence {
|
||||
string += template.render(currentObject, context: context)
|
||||
currentObject = object
|
||||
context.first = false
|
||||
context.index += 1
|
||||
}
|
||||
|
||||
context.last = true
|
||||
|
||||
@@ -33,6 +33,21 @@ final class MethodTests: XCTestCase {
|
||||
""")
|
||||
}
|
||||
|
||||
func testIndex() throws {
|
||||
let template = try HBMustacheTemplate(string: """
|
||||
{{#repo}}
|
||||
<b>{{#index()}}{{plus1(.)}}) {{/}}{{ name }}</b>
|
||||
{{/repo}}
|
||||
""")
|
||||
let object: [String: Any] = ["repo": [["name": "resque"], ["name": "hub"], ["name": "rip"]]]
|
||||
XCTAssertEqual(template.render(object), """
|
||||
<b>1) resque</b>
|
||||
<b>2) hub</b>
|
||||
<b>3) rip</b>
|
||||
|
||||
""")
|
||||
}
|
||||
|
||||
func testReversed() throws {
|
||||
let template = try HBMustacheTemplate(string: """
|
||||
{{#reversed(repo)}}
|
||||
|
||||
Reference in New Issue
Block a user