113 lines
4.5 KiB
Swift
113 lines
4.5 KiB
Swift
|
|
extension HBMustacheTemplate {
|
|
enum Error: Swift.Error {
|
|
case sectionCloseNameIncorrect
|
|
case unfinishedName
|
|
case expectedSectionEnd
|
|
}
|
|
|
|
/// parse mustache text to generate a list of tokens
|
|
static func parse(_ string: String) throws -> [Token] {
|
|
var parser = HBParser(string)
|
|
return try parse(&parser, sectionName: nil)
|
|
}
|
|
|
|
/// parse section in mustache text
|
|
static func parse(_ parser: inout HBParser, sectionName: String?) throws -> [Token] {
|
|
var tokens: [Token] = []
|
|
while !parser.reachedEnd() {
|
|
let text = try parser.read(untilString: "{{", throwOnOverflow: false, skipToEnd: true)
|
|
if text.count > 0 {
|
|
tokens.append(.text(text.string))
|
|
}
|
|
if parser.reachedEnd() {
|
|
break
|
|
}
|
|
switch parser.current() {
|
|
case "#":
|
|
parser.unsafeAdvance()
|
|
let (name, method) = try parseName(&parser)
|
|
if parser.current() == "\n" {
|
|
parser.unsafeAdvance()
|
|
}
|
|
let sectionTokens = try parse(&parser, sectionName: name)
|
|
tokens.append(.section(name: name, method: method, template: HBMustacheTemplate(sectionTokens)))
|
|
|
|
case "^":
|
|
parser.unsafeAdvance()
|
|
let (name, method) = try parseName(&parser)
|
|
if parser.current() == "\n" {
|
|
parser.unsafeAdvance()
|
|
}
|
|
let sectionTokens = try parse(&parser, sectionName: name)
|
|
tokens.append(.invertedSection(name: name, method: method, template: HBMustacheTemplate(sectionTokens)))
|
|
|
|
case "/":
|
|
parser.unsafeAdvance()
|
|
let (name, _) = try parseName(&parser)
|
|
guard name == sectionName else {
|
|
throw Error.sectionCloseNameIncorrect
|
|
}
|
|
if parser.current() == "\n" {
|
|
parser.unsafeAdvance()
|
|
}
|
|
return tokens
|
|
|
|
case "{":
|
|
parser.unsafeAdvance()
|
|
let (name, method) = try parseName(&parser)
|
|
guard try parser.read("}") else { throw Error.unfinishedName }
|
|
tokens.append(.unescapedVariable(name: name, method: method))
|
|
|
|
case "!":
|
|
parser.unsafeAdvance()
|
|
_ = try parseComment(&parser)
|
|
|
|
case ">":
|
|
parser.unsafeAdvance()
|
|
let (name, _) = try parseName(&parser)
|
|
tokens.append(.partial(name))
|
|
|
|
default:
|
|
let (name, method) = try parseName(&parser)
|
|
tokens.append(.variable(name: name, method: method))
|
|
}
|
|
}
|
|
// should never get here if reading section
|
|
guard sectionName == nil else {
|
|
throw Error.expectedSectionEnd
|
|
}
|
|
return tokens
|
|
}
|
|
|
|
/// parse variable name
|
|
static func parseName(_ parser: inout HBParser) throws -> (String, String?) {
|
|
parser.read(while: \.isWhitespace)
|
|
var text = parser.read(while: sectionNameChars )
|
|
parser.read(while: \.isWhitespace)
|
|
guard try parser.read("}"), try parser.read("}") else { throw Error.unfinishedName }
|
|
// does the name include brackets. If so this is a method call
|
|
let string = text.read(while: sectionNameCharsWithoutBrackets)
|
|
if text.reachedEnd() {
|
|
return (text.string, nil)
|
|
} else {
|
|
// parse function parameter, as we have just parsed a function name
|
|
guard text.current() == "(" else { throw Error.unfinishedName }
|
|
text.unsafeAdvance()
|
|
let string2 = text.read(while: sectionNameCharsWithoutBrackets)
|
|
guard text.current() == ")" else { throw Error.unfinishedName }
|
|
text.unsafeAdvance()
|
|
guard text.reachedEnd() else { throw Error.unfinishedName }
|
|
return (string2.string, string.string)
|
|
}
|
|
}
|
|
|
|
static func parseComment(_ parser: inout HBParser) throws -> String {
|
|
let text = try parser.read(untilString: "}}", throwOnOverflow: true, skipToEnd: true)
|
|
return text.string
|
|
}
|
|
|
|
private static let sectionNameCharsWithoutBrackets = Set<Unicode.Scalar>("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ._?")
|
|
private static let sectionNameChars = Set<Unicode.Scalar>("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ._?()")
|
|
}
|