feat(markdown): ast for literal and line break

support header in header

2 files changed, 9 insertions, 3 deletions

diff --git a/mardown/ast.go b/mardown/ast.go
index 76328f4..819d73a 100644
--- a/mardown/ast.go
+++ b/mardown/ast.go
@@ -37,13 +37,13 @@ func getBlock(lxs lexers) (block, error) {
 	switch lxs.Current().Type {
 	case lexerHeader:
 		b, err = header(lxs)
-	case lexerBreak:
 	case lexerExternal:
 	case lexerModifier:
 	case lexerCode:
 	case lexerEscape:
 	case lexerQuote:
-	case lexerLiteral:
+	case lexerLiteral, lexerBreak:
+		b = astLiteral(lxs.Current().Value)
 	default:
 		err = errors.Join(ErrUnkownLexType, fmt.Errorf("type received: %s", lxs.Current().Type))
 	}
diff --git a/mardown/ast_header.go b/mardown/ast_header.go
index 540f47f..c7bc85d 100644
--- a/mardown/ast_header.go
+++ b/mardown/ast_header.go
@@ -16,8 +16,14 @@ func header(lxs lexers) (block, error) {
 		if err != nil {
 			return nil, err
 		}
+		// if this is a header, just consider it as literal #
 		if h, ok := bl.(*astHeader); ok {
-			//TODO: handle
+			var s string
+			for range h.level {
+				s += "#"
+			}
+			b.content = append(b.content, astLiteral(s))
+			b.content = append(b.content, h.content...)
 		}
 		b.content = append(b.content, bl)
 	}