上一个改动不小心去除了 iframe 属性的附加功能

siyuan-note · Aug 18, 2021 · 5092c03 · 5092c03
1 parent 0ed3e1a
commit 5092c03
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 14 deletions.
diff --git a/src/render/ocean_press_render.go b/src/render/ocean_press_render.go
@@ -315,19 +315,19 @@ func (r *OceanPressRender) renderAudio(node *ast.Node, entering bool) ast.WalkSt
 	return ast.WalkContinue
 }
 
-func (r *OceanPressRender) renderIFrame(node *ast.Node, entering bool) ast.WalkStatus {
-	if entering {
-		r.Tag("div", [][]string{{"class", "iframe"}}, false)
-		tokens := node.Tokens
-		if r.Options.Sanitize {
-			tokens = sanitize(tokens)
-		}
-		tokens = r.tagSrcPath(tokens)
-		r.Write(tokens)
-		r.Tag("/div", nil, false)
-	}
-	return ast.WalkContinue
-}
+// func (r *OceanPressRender) renderIFrame(node *ast.Node, entering bool) ast.WalkStatus {
+// 	if entering {
+// 		r.Tag("div", [][]string{{"class", "iframe"}}, false)
+// 		tokens := node.Tokens
+// 		if r.Options.Sanitize {
+// 			tokens = sanitize(tokens)
+// 		}
+// 		tokens = r.tagSrcPath(tokens)
+// 		r.Write(tokens)
+// 		r.Tag("/div", nil, false)
+// 	}
+// 	return ast.WalkContinue
+// }
 
 // func (r *OceanPressRender) Render() (output []byte) {
 // 	output = r.BaseRenderer.Render()

diff --git a/src/render/ocean_press_render_custom.go b/src/render/ocean_press_render_custom.go
@@ -65,7 +65,22 @@ func (r *OceanPressRender) Render() (html string, xml string) {
 
 	return html, xml
 }
-
+func (r *OceanPressRender) renderIFrame(node *ast.Node, entering bool) ast.WalkStatus {
+	if entering {
+		attr := [][]string{{"class", "iframe"}}
+		// 添加自定义属性
+		attr = append(attr, node.KramdownIAL...)
+		r.Tag("div", attr, false)
+		tokens := node.Tokens
+		if r.Options.Sanitize {
+			tokens = sanitize(tokens)
+		}
+		tokens = r.tagSrcPath(tokens)
+		r.Write(tokens)
+		r.Tag("/div", nil, false)
+	}
+	return ast.WalkContinue
+}
 // renderImage 为了实现居中效果
 func (r *OceanPressRender) renderImage(node *ast.Node, entering bool) ast.WalkStatus {
 	if entering {