swiftpm-stencil/Stencil/Node.swift

//
//  Node.swift
//  Stencil
//
//  Created by Kyle Fuller on 23/10/2014.
//  Copyright (c) 2014 Cocode. All rights reserved.
//

import Foundation

struct NodeError : Error {
    let token:Token
    let message:String

    init(token:Token, message:String) {
        self.token = token
        self.message = message
    }

    var description:String {
        return "\(token.components().first!): \(message)"
    }
}

public protocol Node {
    func render(context:Context) -> (String?, Error?)
}

extension Array {
    func map<U>(block:((Element) -> (U?, Error?))) -> ([U]?, Error?) {
        var results = [U]()

        for item in self {
            let (result, error) = block(item)

            if let error = error {
                return (nil, error)
            } else if (result != nil) {
                // let result = result exposing a bug in the Swift compier :(
                results.append(result!)
            }
        }

        return (results, nil)
    }
}

public func renderNodes(nodes:[Node], context:Context) -> (String?, Error?) {
    let result:(results:[String]?, error:Error?) = nodes.map {
        return $0.render(context)
    }

    if let result = result.0 {
        return ("".join(result), nil)
    }

    return (nil, result.1)
}

public class TextNode : Node {
    public let text:String

    public init(text:String) {
        self.text = text
    }

    public func render(context:Context) -> (String?, Error?) {
        return (self.text, nil)
    }
}

public class VariableNode : Node {
    public let variable:Variable

    public init(variable:Variable) {
        self.variable = variable
    }

    public init(variable:String) {
        self.variable = Variable(variable)
    }

    public func render(context:Context) -> (String?, Error?) {
        let result:AnyObject? = variable.resolve(context)

        if let result = result as? String {
            return (result, nil)
        } else if let result = result as? NSObject {
            return (result.description, nil)
        }

        return (nil, nil)
    }
}

public class NowNode : Node {
    public let format:Variable

    public class func parse(parser:TokenParser, token:Token) -> (node:Node?, error:Error?) {
        var format:Variable?

        let components = token.components()
        if components.count == 2 {
            format = Variable(components[1])
        }

        return (NowNode(format:format), nil)
    }

    public init(format:Variable?) {
        if let format = format {
            self.format = format
        } else {
            self.format = Variable("\"yyyy-MM-dd 'at' HH:mm\"")
        }
    }

    public func render(context: Context) -> (String?, Error?) {
        let date = NSDate()
        let format: AnyObject? = self.format.resolve(context)
        var formatter:NSDateFormatter?

        if let format = format as? NSDateFormatter {
            formatter = format
        } else if let format = format as? String {
            formatter = NSDateFormatter()
            formatter!.dateFormat = format
        } else {
            // TODO Error
            return (nil, nil)
        }

        return ("\(formatter!.stringFromDate(date))", nil)
    }
}

public class ForNode : Node {
    let variable:Variable
    let loopVariable:String
    let nodes:[Node]

    public class func parse(parser:TokenParser, token:Token) -> (node:Node?, error:Error?) {
        let components = token.components()
        let count = countElements(components)

        if count == 4 && components[2] == "in" {
            let loopVariable = components[1]
            let variable = components[3]
            let (nodes, error) = parser.parse(until(["endfor", "empty"]))
            var emptyNodes = [Node]()

            if let error = error {
                return (nil, error)
            }

            if let token = parser.nextToken() {
                if token.contents == "empty" {
                    let (nodes, error) = parser.parse(until(["endfor"]))
                    parser.nextToken()

                    if let error = error {
                        return (nil, error)
                    }

                    if let nodes = nodes {
                        emptyNodes = nodes
                    }
                }
            } else {
                return (nil, NodeError(token: token, message: "`endfor` was not found."))
            }

            return (ForNode(variable: variable, loopVariable: loopVariable, nodes: nodes!, emptyNodes:emptyNodes), nil)
        }

        return (nil, NodeError(token: token, message: "Invalid syntax. Expected `for x in y`."))
    }

    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)
    }
}

public class IfNode : Node {
    public let variable:Variable
    public let trueNodes:[Node]
    public let falseNodes:[Node]

    public class func parse(parser:TokenParser, token:Token) -> (node:Node?, error:Error?) {
        let variable = token.components()[1]

        let (trueNodes, error) = parser.parse(until(["endif", "else"]))
        if let error = error {
            return (nil, error)
        }

        var falseNodes = [Node]()

        if let token = parser.nextToken() {
            if token.contents == "else" {
                let (nodes, error) = parser.parse(until(["endif"]))
                parser.nextToken()

                if let error = error {
                    return (nil, error)
                }

                if let nodes = nodes {
                    falseNodes = nodes
                }
            }
        } else {
            return (nil, NodeError(token: token, message: "`endif` was not found."))
        }

        return (IfNode(variable: variable, trueNodes: trueNodes!, falseNodes: falseNodes), nil)
    }

    public class func parse_ifnot(parser:TokenParser, token:Token) -> (node:Node?, error:Error?) {
        let variable = token.components()[1]

        let (falseNodes, error) = parser.parse(until(["endif", "else"]))
        if let error = error {
            return (nil, error)
        }
        var trueNodes = [Node]()

        if let token = parser.nextToken() {
            if token.contents == "else" {
                let (nodes, error) = parser.parse(until(["endif"]))
                if let error = error {
                    return (nil, error)
                }

                if let nodes = nodes {
                    trueNodes = nodes
                }

                parser.nextToken()
            }
        } else {
            return (nil, NodeError(token: token, message: "`endif` was not found."))
        }

        return (IfNode(variable: variable, trueNodes: trueNodes, falseNodes: falseNodes!), nil)
    }

    public init(variable:String, trueNodes:[Node], falseNodes:[Node]) {
        self.variable = Variable(variable)
        self.trueNodes = trueNodes
        self.falseNodes = falseNodes
    }

    public func render(context: Context) -> (String?, Error?) {
        let result: AnyObject? = variable.resolve(context)
        var truthy = false

        if let result = result as? [AnyObject] {
            if result.count > 0 {
                truthy = true
            }
        } else if let result: AnyObject = result {
            truthy = true
        }

        context.push()
        let (string, error) = renderNodes(truthy ? trueNodes : falseNodes, context)
        context.pop()

        return (string, error)
    }
}