120 lines
3.0 KiB
Swift
120 lines
3.0 KiB
Swift
import Foundation
|
|
|
|
|
|
extension String {
|
|
/// Split a string by a separator leaving quoted phrases together
|
|
func smartSplit(separator: Character = " ") -> [String] {
|
|
var word = ""
|
|
var components: [String] = []
|
|
var separate: Character = separator
|
|
var singleQuoteCount = 0
|
|
var doubleQuoteCount = 0
|
|
|
|
for character in self.characters {
|
|
if character == "'" { singleQuoteCount += 1 }
|
|
else if character == "\"" { doubleQuoteCount += 1 }
|
|
|
|
if character == separate {
|
|
|
|
if separate != separator {
|
|
word.append(separate)
|
|
} else if singleQuoteCount % 2 == 0 && doubleQuoteCount % 2 == 0 && !word.isEmpty {
|
|
components.append(word)
|
|
word = ""
|
|
}
|
|
|
|
separate = separator
|
|
} else {
|
|
if separate == separator && (character == "'" || character == "\"") {
|
|
separate = character
|
|
}
|
|
word.append(character)
|
|
}
|
|
}
|
|
|
|
if !word.isEmpty {
|
|
components.append(word)
|
|
}
|
|
|
|
return components
|
|
}
|
|
}
|
|
|
|
public struct SourceMap: Equatable {
|
|
public let filename: String?
|
|
public let line: RangeLine
|
|
|
|
init(filename: String? = nil, line: RangeLine = ("", 0, 0)) {
|
|
self.filename = filename
|
|
self.line = line
|
|
}
|
|
|
|
static let unknown = SourceMap()
|
|
|
|
public static func ==(lhs: SourceMap, rhs: SourceMap) -> Bool {
|
|
return lhs.filename == rhs.filename && lhs.line == rhs.line
|
|
}
|
|
}
|
|
|
|
public enum Token : Equatable {
|
|
/// A token representing a piece of text.
|
|
case text(value: String, at: SourceMap)
|
|
|
|
/// A token representing a variable.
|
|
case variable(value: String, at: SourceMap)
|
|
|
|
/// A token representing a comment.
|
|
case comment(value: String, at: SourceMap)
|
|
|
|
/// A token representing a template block.
|
|
case block(value: String, at: SourceMap)
|
|
|
|
/// Returns the underlying value as an array seperated by spaces
|
|
public func components() -> [String] {
|
|
switch self {
|
|
case .block(let value, _),
|
|
.variable(let value, _),
|
|
.text(let value, _),
|
|
.comment(let value, _):
|
|
return value.smartSplit()
|
|
}
|
|
}
|
|
|
|
public var contents: String {
|
|
switch self {
|
|
case .block(let value, _),
|
|
.variable(let value, _),
|
|
.text(let value, _),
|
|
.comment(let value, _):
|
|
return value
|
|
}
|
|
}
|
|
|
|
public var sourceMap: SourceMap {
|
|
switch self {
|
|
case .block(_, let sourceMap),
|
|
.variable(_, let sourceMap),
|
|
.text(_, let sourceMap),
|
|
.comment(_, let sourceMap):
|
|
return sourceMap
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
public func == (lhs: Token, rhs: Token) -> Bool {
|
|
switch (lhs, rhs) {
|
|
case let (.text(lhsValue, lhsAt), .text(rhsValue, rhsAt)):
|
|
return lhsValue == rhsValue && lhsAt == rhsAt
|
|
case let (.variable(lhsValue, lhsAt), .variable(rhsValue, rhsAt)):
|
|
return lhsValue == rhsValue && lhsAt == rhsAt
|
|
case let (.block(lhsValue, lhsAt), .block(rhsValue, rhsAt)):
|
|
return lhsValue == rhsValue && lhsAt == rhsAt
|
|
case let (.comment(lhsValue, lhsAt), .comment(rhsValue, rhsAt)):
|
|
return lhsValue == rhsValue && lhsAt == rhsAt
|
|
default:
|
|
return false
|
|
}
|
|
}
|