101 lines
2.6 KiB
Swift
101 lines
2.6 KiB
Swift
import Foundation
|
|
|
|
|
|
public struct TemplateSyntaxError : Error, Equatable, CustomStringConvertible {
|
|
public let description:String
|
|
public var token: Token?
|
|
|
|
public init(_ description:String) {
|
|
self.description = description
|
|
}
|
|
|
|
public func contextAwareError(templateName: String?, templateContent: String) -> TemplateSyntaxError? {
|
|
guard let token = token, token.range != .unknown else { return nil }
|
|
let templateName = templateName.map({ "\($0):" }) ?? ""
|
|
let (line, lineNumber, offset) = templateContent.lineAndPosition(at: token.range)
|
|
let tokenContent = templateContent.substring(with: token.range)
|
|
let highlight = "\(String(Array(repeating: " ", count: offset)))^\(String(Array(repeating: "~", count: max(tokenContent.count - 1, 0))))"
|
|
let description = "\(templateName)\(lineNumber):\(offset): error: " + self.description + "\n\(line)\n\(highlight)\n"
|
|
return TemplateSyntaxError(description)
|
|
}
|
|
|
|
}
|
|
|
|
|
|
public func ==(lhs:TemplateSyntaxError, rhs:TemplateSyntaxError) -> Bool {
|
|
return lhs.description == rhs.description
|
|
}
|
|
|
|
|
|
public protocol NodeType {
|
|
/// Render the node in the given context
|
|
func render(_ context:Context) throws -> String
|
|
}
|
|
|
|
|
|
/// Render the collection of nodes in the given context
|
|
public func renderNodes(_ nodes:[NodeType], _ context:Context) throws -> String {
|
|
return try nodes.map { try $0.render(context) }.joined(separator: "")
|
|
}
|
|
|
|
public class SimpleNode : NodeType {
|
|
public let handler:(Context) throws -> String
|
|
|
|
public init(handler: @escaping (Context) throws -> String) {
|
|
self.handler = handler
|
|
}
|
|
|
|
public func render(_ context: Context) throws -> String {
|
|
return try handler(context)
|
|
}
|
|
}
|
|
|
|
|
|
public class TextNode : NodeType {
|
|
public let text:String
|
|
|
|
public init(text:String) {
|
|
self.text = text
|
|
}
|
|
|
|
public func render(_ context:Context) throws -> String {
|
|
return self.text
|
|
}
|
|
}
|
|
|
|
|
|
public protocol Resolvable {
|
|
func resolve(_ context: Context) throws -> Any?
|
|
}
|
|
|
|
|
|
public class VariableNode : NodeType {
|
|
public let variable: Resolvable
|
|
|
|
public init(variable: Resolvable) {
|
|
self.variable = variable
|
|
}
|
|
|
|
public init(variable: String) {
|
|
self.variable = Variable(variable)
|
|
}
|
|
|
|
public func render(_ context: Context) throws -> String {
|
|
let result = try variable.resolve(context)
|
|
return stringify(result)
|
|
}
|
|
}
|
|
|
|
|
|
func stringify(_ result: Any?) -> String {
|
|
if let result = result as? String {
|
|
return result
|
|
} else if let result = result as? CustomStringConvertible {
|
|
return result.description
|
|
} else if let result = result as? NSObject {
|
|
return result.description
|
|
}
|
|
|
|
return ""
|
|
}
|