Fix heading and list markers

2 files changed, 13 insertions, 18 deletions

diff --git a/src/syntax/lexer.rs b/src/syntax/lexer.rs
index 0bf7966f..b79b4c7f 100644
--- a/src/syntax/lexer.rs
+++ b/src/syntax/lexer.rs
@@ -191,15 +191,15 @@ impl Lexer<'_> {
             ':' => SyntaxKind::Colon,
             '=' => {
                 self.s.eat_while('=');
-                if self.space_and_more() {
+                if self.space_or_end() {
                     SyntaxKind::HeadingMarker
                 } else {
                     self.text()
                 }
             }
-            '-' if self.space_and_more() => SyntaxKind::ListMarker,
-            '+' if self.space_and_more() => SyntaxKind::EnumMarker,
-            '/' if self.space_and_more() => SyntaxKind::TermMarker,
+            '-' if self.space_or_end() => SyntaxKind::ListMarker,
+            '+' if self.space_or_end() => SyntaxKind::EnumMarker,
+            '/' if self.space_or_end() => SyntaxKind::TermMarker,
 
             _ => self.text(),
         }
@@ -363,13 +363,8 @@ impl Lexer<'_> {
         alphanum(prev) && alphanum(next)
     }
 
-    fn space_and_more(&self) -> bool {
-        let mut s = self.s;
-        if !s.at(char::is_whitespace) {
-            return false;
-        }
-        s.eat_while(|c: char| c.is_whitespace() && !is_newline(c));
-        !s.done() && !s.at(is_newline)
+    fn space_or_end(&self) -> bool {
+        self.s.done() || self.s.at(char::is_whitespace)
     }
 }
 
diff --git a/src/syntax/parser.rs b/src/syntax/parser.rs
index afb207b6..5933d481 100644
--- a/src/syntax/parser.rs
+++ b/src/syntax/parser.rs
@@ -157,7 +157,7 @@ fn emph(p: &mut Parser) {
 fn heading(p: &mut Parser) {
     let m = p.marker();
     p.assert(SyntaxKind::HeadingMarker);
-    whitespace(p);
+    whitespace_line(p);
     markup(p, false, usize::MAX, |kind| {
         kind == SyntaxKind::Label || kind == SyntaxKind::RightBracket
     });
@@ -168,7 +168,7 @@ fn list_item(p: &mut Parser) {
     let m = p.marker();
     p.assert(SyntaxKind::ListMarker);
     let min_indent = p.column(p.prev_end());
-    whitespace(p);
+    whitespace_line(p);
     markup(p, false, min_indent, |kind| kind == SyntaxKind::RightBracket);
     p.wrap(m, SyntaxKind::ListItem);
 }
@@ -177,7 +177,7 @@ fn enum_item(p: &mut Parser) {
     let m = p.marker();
     p.assert(SyntaxKind::EnumMarker);
     let min_indent = p.column(p.prev_end());
-    whitespace(p);
+    whitespace_line(p);
     markup(p, false, min_indent, |kind| kind == SyntaxKind::RightBracket);
     p.wrap(m, SyntaxKind::EnumItem);
 }
@@ -186,18 +186,18 @@ fn term_item(p: &mut Parser) {
     let m = p.marker();
     p.assert(SyntaxKind::TermMarker);
     let min_indent = p.column(p.prev_end());
-    whitespace(p);
+    whitespace_line(p);
     markup(p, false, usize::MAX, |kind| {
         kind == SyntaxKind::Colon || kind == SyntaxKind::RightBracket
     });
     p.expect(SyntaxKind::Colon);
-    whitespace(p);
+    whitespace_line(p);
     markup(p, false, min_indent, |kind| kind == SyntaxKind::RightBracket);
     p.wrap(m, SyntaxKind::TermItem);
 }
 
-fn whitespace(p: &mut Parser) {
-    while p.current().is_trivia() {
+fn whitespace_line(p: &mut Parser) {
+    while !p.newline() && p.current().is_trivia() {
         p.eat();
     }
 }