✨ 提供 HTML 转 AST/Text 的接口 Fix #90

h2m.go

@@ -25,8 +25,22 @@ import (
 // HTML2Markdown 将 HTML 转换为 Markdown。
 func (lute *Lute) HTML2Markdown(htmlStr string) (markdown string, err error) {
 	// 将字符串解析为 DOM 树
+	tree := lute.HTML2Tree(htmlStr)
 
-	reader := strings.NewReader(htmlStr)
+	// 将 AST 进行 Markdown 格式化渲染
+	var formatted []byte
+	renderer := render.NewFormatRenderer(tree)
+	for nodeType, rendererFunc := range lute.HTML2MdRendererFuncs {
+		renderer.ExtRendererFuncs[nodeType] = rendererFunc
+	}
+	formatted = renderer.Render()
+	markdown = util.BytesToStr(formatted)
+	return
+}
+
+// HTML2Tree 将 HTML 转换为 AST。
+func (lute *Lute) HTML2Tree(dom string) (ret *parse.Tree) {
+	reader := strings.NewReader(dom)
 	htmlRoot := &html.Node{Type: html.ElementNode}
 	htmlNodes, err := html.ParseFragment(reader, htmlRoot)
 	if nil != err {
@@ -38,13 +52,13 @@ func (lute *Lute) HTML2Markdown(htmlStr string) (markdown string, err error) {
 
 	// 将 HTML 树转换为 Markdown AST
 
-	tree := &parse.Tree{Name: "", Root: &ast.Node{Type: ast.NodeDocument}, Context: &parse.Context{Option: lute.Options}}
-	tree.Context.Tip = tree.Root
+	ret = &parse.Tree{Name: "", Root: &ast.Node{Type: ast.NodeDocument}, Context: &parse.Context{Option: lute.Options}}
+	ret.Context.Tip = ret.Root
 	for _, htmlNode := range htmlNodes {
-		lute.genASTByDOM(htmlNode, tree)
+		lute.genASTByDOM(htmlNode, ret)
 	}
 
-	ast.Walk(tree.Root, func(n *ast.Node, entering bool) ast.WalkStatus {
+	ast.Walk(ret.Root, func(n *ast.Node, entering bool) ast.WalkStatus {
 		if entering {
 			if ast.NodeList == n.Type {
 				// ul.ul => ul.li.ul
@@ -59,16 +73,6 @@ func (lute *Lute) HTML2Markdown(htmlStr string) (markdown string, err error) {
 		}
 		return ast.WalkContinue
 	})
-
-	// 将 AST 进行 Markdown 格式化渲染
-
-	var formatted []byte
-	renderer := render.NewFormatRenderer(tree)
-	for nodeType, rendererFunc := range lute.HTML2MdRendererFuncs {
-		renderer.ExtRendererFuncs[nodeType] = rendererFunc
-	}
-	formatted = renderer.Render()
-	markdown = util.BytesToStr(formatted)
 	return
 }
 

lute.go

@@ -155,6 +155,15 @@ func (lute *Lute) TextBundleStr(name, markdown string, linkPrefixes []string) (t
 	return
 }
 
+// HTML2Text 将指定的 HTMl dom 转换为文本。
+func (lute *Lute) HTML2Text(dom string) string {
+	tree := lute.HTML2Tree(dom)
+	if nil == tree {
+		return ""
+	}
+	return tree.Root.Text()
+}
+
 // Space 用于在 text 中的中西文之间插入空格。
 func (lute *Lute) Space(text string) string {
 	return render.Space0(text)