JavaScript Highlighter: Handle ternary expressions

This commit is contained in:
Shadowfacts 2020-04-04 10:08:47 -04:00
父節點 5b9f59ec09
當前提交 a976cf894f
簽署人: shadowfacts
GPG 金鑰 ID: 94A5AB95422746E5
共有 2 個檔案被更改,包括 21 行新增2 行删除

查看文件

@ -19,6 +19,7 @@ fileprivate let identifierStarts: CharacterSet = {
return set
}()
fileprivate let operators = CharacterSet(charactersIn: "+-*/<>=")
fileprivate let expressionEnds = CharacterSet(charactersIn: ",]});")
class JavaScriptHighlighter {
private let text: String
@ -71,7 +72,7 @@ class JavaScriptHighlighter {
], range: fullRange)
currentIndex = text.startIndex
while currentIndex < text.endIndex {
while let char = peek(), !expressionEnds.contains(char) {
consumeExpression()
}
@ -116,6 +117,8 @@ class JavaScriptHighlighter {
consumeArray()
} else if char == "." {
consumeDotLookup()
} else if char == "?" {
consumeTernaryExpression()
} else {
consume()
}
@ -310,6 +313,22 @@ class JavaScriptHighlighter {
}
}
func consumeTernaryExpression() {
consume() // ?
print("Ternary expression")
indent += " "
print("Ternary true result")
while let char = peek(), char != ":" {
consumeExpression() // true result
}
consume() // :
print("Ternary false result")
while let char = peek(), !expressionEnds.contains(char) {
consumeExpression()
}
indent = String(indent.dropLast(2))
}
}
extension JavaScriptHighlighter {

查看文件

@ -8,6 +8,6 @@
import Foundation
let source = "{a: 1234 , 'b': 'blah', foo}"
let source = "a ? 'foo' : b ? 'bar' : 'baz'"
_ = JavaScriptHighlighter(text: source).highlight()