[Node] Add a standard for node

This commit is contained in:
Kyle Fuller
2014-10-25 15:12:46 +01:00
parent d71e0ec012
commit 46350dbd49
4 changed files with 97 additions and 0 deletions

View File

@@ -93,3 +93,64 @@ public class NowNode : Node {
return ("\(date)", nil)
}
}
public class ForNode : Node {
let variable:Variable
let loopVariable:String
let nodes:[Node]
public class func parse(parser:TokenParser, token:Token) -> Node {
let components = token.components()
let count = countElements(components)
if count == 4 && components[2] == "in" {
let loopVariable = components[1]
let variable = components[3]
let nodes = parser.parse(until(["endfor", "empty"]))
var emptyNodes = [Node]()
if let token = parser.nextToken() {
if token.contents == "empty" {
emptyNodes = parser.parse(until(["endfor"]))
parser.nextToken()
}
}
return ForNode(variable: variable, loopVariable: loopVariable, nodes: nodes, emptyNodes:emptyNodes)
} else {
// TODO error
}
return TextNode(text: "TODO return some error")
}
public init(variable:String, loopVariable:String, nodes:[Node], emptyNodes:[Node]) {
self.variable = Variable(variable)
self.loopVariable = loopVariable
self.nodes = nodes
}
public func render(context: Context) -> (String?, Error?) {
let values = variable.resolve(context) as? [AnyObject]
var result = ""
if let values = values {
for item in values {
context.push()
context[loopVariable] = item
let (string, error) = renderNodes(nodes, context)
context.pop()
if let error = error {
return (nil, error)
}
if let string = string {
result += string
}
}
}
return (result, nil)
}
}

View File

@@ -8,12 +8,25 @@
import Foundation
public func until(tags:[String])(parser:TokenParser, token:Token) -> Bool {
if let name = token.components().first {
for tag in tags {
if name == tag {
return true
}
}
}
return false
}
public class TokenParser {
private var tokens:[Token]
private var tags = Dictionary<String, ((TokenParser, Token) -> (Node))>()
public init(tokens:[Token]) {
self.tokens = tokens
tags["for"] = ForNode.parse
tags["now"] = NowNode.parse
}

View File

@@ -30,6 +30,19 @@ public enum Token : Equatable {
return value.value.stringByTrimmingCharactersInSet(characterSet).componentsSeparatedByCharactersInSet(characterSet)
}
}
var contents:String {
switch self {
case .Block(let value):
return value
case .Variable(let value):
return value
case .Text(let value):
return value
case .Comment(let value):
return value
}
}
}
public func ==(lhs:Token, rhs:Token) -> Bool {

View File

@@ -30,6 +30,7 @@ class NodeTests: XCTestCase {
context = Context(dictionary: [
"name": "Kyle",
"age": 27,
"items": [1,2,3],
])
}
}
@@ -76,3 +77,12 @@ class RenderNodeTests: NodeTests {
XCTAssertTrue(result == nil)
}
}
class ForNodeTests: NodeTests {
func testForNodeRender() {
let node = ForNode(variable: "items", loopVariable: "item", nodes: [VariableNode(variable: "item")], emptyNodes:[])
let result = node.render(context)
XCTAssertEqual(result.0!, "123")
}
}