From 64cd1a98f06470a01aadcf5f77bc488e450796d9 Mon Sep 17 00:00:00 2001
From: Mathias Quintero <me@quintero.io>
Date: Sun, 7 Feb 2021 15:09:23 +0100
Subject: [PATCH] Fixing node updates during scanning of annotated strings

---
 Sources/Syntax/Internals/ScannerState.swift | 13 +----------
 Tests/SyntaxTests/HTML/HTMLNode.swift       |  4 ++--
 Tests/SyntaxTests/HTML/HTMLNodeParser.swift | 25 +++++++++++++++++++--
 Tests/SyntaxTests/HTMLSyntaxTests.swift     | 10 ++++++++-
 4 files changed, 35 insertions(+), 17 deletions(-)

diff --git a/Sources/Syntax/Internals/ScannerState.swift b/Sources/Syntax/Internals/ScannerState.swift
index c982c2d..8320897 100644
--- a/Sources/Syntax/Internals/ScannerState.swift
+++ b/Sources/Syntax/Internals/ScannerState.swift
@@ -416,18 +416,6 @@ private final class ScanningStorage<T>: ScannerStateStorage, ScanningStorageProt
             throw state.error(reason: .failedToMatch(expression))
         }
 
-        // Upon a match start with in place storage
-//        let nextState = ScannerState(range: result.range.lowerBound..<state.range.upperBound, parent: state, storage: InPlaceStorage())
-//        nextState.range = result.range.upperBound..<state.range.upperBound
-//
-//        let (firstNode, rest) = state.node.backToFirst()
-//
-//        state.node = firstNode
-//        if let rest = rest {
-//            rest.update(from: state.range.lowerBound, to: result.range.lowerBound)
-//            nextState.node = rest
-//        }
-
         rangeMatchStart = result.range.lowerBound
         return ExpressionScanResult(match: result, state: state)
     }
@@ -552,6 +540,7 @@ private class StackedScanningStateStorage<T>: ScannerStateStorage {
 
         let result = try current.take(expression: expression, in: text, for: state)
         isInPlace = true
+        state.node.update(from: state.range.lowerBound, to: result.match.range.lowerBound)
         state.range = result.match.range.upperBound..<state.range.upperBound
         return result
     }
diff --git a/Tests/SyntaxTests/HTML/HTMLNode.swift b/Tests/SyntaxTests/HTML/HTMLNode.swift
index 8881a2e..57afbd3 100644
--- a/Tests/SyntaxTests/HTML/HTMLNode.swift
+++ b/Tests/SyntaxTests/HTML/HTMLNode.swift
@@ -10,9 +10,9 @@ public class HTMLNode {
 
     public let tag: Substring
     public let attributes: [Attribute]
-    public let contents: AnnotatedString<HTMLNode>
+    public let contents: AnnotatedString<HTMLNode>?
 
-    init(tag: Substring, attributes: [Attribute], contents: AnnotatedString<HTMLNode>) {
+    init(tag: Substring, attributes: [Attribute], contents: AnnotatedString<HTMLNode>?) {
         self.tag = tag
         self.attributes = attributes
         self.contents = contents
diff --git a/Tests/SyntaxTests/HTML/HTMLNodeParser.swift b/Tests/SyntaxTests/HTML/HTMLNodeParser.swift
index 56c59a4..c183e07 100644
--- a/Tests/SyntaxTests/HTML/HTMLNodeParser.swift
+++ b/Tests/SyntaxTests/HTML/HTMLNodeParser.swift
@@ -10,13 +10,34 @@ struct HTMLNodeParser: Parser {
 
     var body: AnyParser<HTMLNode> {
         Recursive { parser in
-            InternalHTMLNodeParser(parser: parser)
+            Either {
+                HTMLNodeWithChildrenParser(parser: parser)
+                SingleHTMLNodeParser()
+            }
+        }
+    }
+
+}
+
+private struct SingleHTMLNodeParser: Parser {
+    var body: AnyParser<HTMLNode> {
+        return Group {
+            "<"
+
+            RegularExpression("[a-zA-Z][-a-zA-Z0-9]*")
+
+            Repeat {
+                AttributeParser()
+            }
+
+            "/>"
         }
+        .map { HTMLNode(tag: $0.text, attributes: $1, contents: nil) }
     }
 
 }
 
-private struct InternalHTMLNodeParser: Parser {
+private struct HTMLNodeWithChildrenParser: Parser {
     let parser: AnyParser<HTMLNode>
 
     var body: AnyParser<HTMLNode> {
diff --git a/Tests/SyntaxTests/HTMLSyntaxTests.swift b/Tests/SyntaxTests/HTMLSyntaxTests.swift
index a207b04..7ee9fc1 100644
--- a/Tests/SyntaxTests/HTMLSyntaxTests.swift
+++ b/Tests/SyntaxTests/HTMLSyntaxTests.swift
@@ -5,7 +5,15 @@ final class HTMLSyntaxTests: XCTestCase {
 
     func testParagraph() {
         let text = """
-        <p>Hello World: <a href="link">link</a></p>
+        <p align="center">
+            <img src="https://img.shields.io/badge/Swift-5.3-orange.svg" />
+            <a href="https://swift.org/package-manager">
+                <img src="https://img.shields.io/badge/swiftpm-compatible-brightgreen.svg?style=flat" alt="Swift Package Manager" />
+            </a>
+            <a href="https://twitter.com/nerdsupremacist">
+                <img src="https://img.shields.io/badge/twitter-@nerdsupremacist-blue.svg?style=flat" alt="Twitter: @nerdsupremacist" />
+            </a>
+        </p>
         """
 
         let html = try! HTMLNodeParser().parse(text)