🎨 Vditor 支持 Vanessa219/vditor#93

88250 · Jan 29, 2020 · 5ffefa5 · 5ffefa5
1 parent cc216dd
commit 5ffefa5
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/heading.go b/heading.go
@@ -83,8 +83,12 @@ func (t *Tree) parseSetextHeading() (level int) {
 		return
 	}
 
+	var caretInLn bool
 	if t.context.option.VditorWYSIWYG {
-		ln = bytes.ReplaceAll(ln, []byte(caret), []byte(""))
+		if bytes.Contains(ln, []byte(caret)) {
+			caretInLn = true
+			ln = bytes.ReplaceAll(ln, []byte(caret), []byte(""))
+		}
 	}
 
 	markers := 0
@@ -109,5 +113,11 @@ func (t *Tree) parseSetextHeading() (level int) {
 	if itemHyphen == marker {
 		level = 2
 	}
+
+	if t.context.option.VditorWYSIWYG && caretInLn {
+		t.context.oldtip.tokens = trimWhitespace(t.context.oldtip.tokens)
+		t.context.oldtip.AppendTokens([]byte(caret))
+	}
+
 	return
 }