Gemini parser cleanup

This commit is contained in:
Shadowfacts 2020-07-13 00:12:31 -04:00
parent 1a4be887d3
commit b983f34685
Signed by: shadowfacts
GPG Key ID: 94A5AB95422746E5
1 changed files with 8 additions and 12 deletions

View File

@ -14,28 +14,24 @@ public struct GeminiParser {
public static func parse(text: String, baseURL: URL) -> Document { public static func parse(text: String, baseURL: URL) -> Document {
var doc = Document(url: baseURL) var doc = Document(url: baseURL)
var preformattingState = PreformattingState.off var inPreformattingBlock = false
text.enumerateLines { (line, stop) in text.enumerateLines { (line, stop) in
if line.starts(with: "```") { if line.starts(with: "```") {
switch preformattingState { if inPreformattingBlock {
case .off: inPreformattingBlock = false
// todo: should the toggle off line be a separate line type?
doc.lines.append(.preformattedToggle(alt: nil))
} else {
let alt: String? let alt: String?
if line.count > 3 { if line.count > 3 {
alt = String(line[line.index(line.startIndex, offsetBy: 3)...]) alt = String(line[line.index(line.startIndex, offsetBy: 3)...])
} else { } else {
alt = nil alt = nil
} }
preformattingState = .on(alt) inPreformattingBlock = true
doc.lines.append(.preformattedToggle(alt: alt)) doc.lines.append(.preformattedToggle(alt: alt))
case let .on(alt):
preformattingState = .off
// todo: should the toggle off line be a separate line type?
doc.lines.append(.preformattedToggle(alt: nil))
} }
if case .off = preformattingState { } else if inPreformattingBlock {
}
} else if case .on(_) = preformattingState {
doc.lines.append(.preformattedText(line)) doc.lines.append(.preformattedText(line))
} else if line.starts(with: "=>") { } else if line.starts(with: "=>") {
// Link line // Link line