fix renderWidget 更合适的处理逻辑

siyuan-note · Sep 12, 2021 · fa048e4 · fa048e4
1 parent 35e9c3d
commit fa048e4
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/src/render/ocean_press_render_custom.go b/src/render/ocean_press_render_custom.go
@@ -78,16 +78,14 @@ func (r *OceanPressRender) renderWidget(node *ast.Node, entering bool) ast.WalkS
  iframe := dom.Find("iframe")
  src, exists := iframe.Attr("src")
  if exists {
- localPrefix := "http://127.0.0.1:6806/"
- if strings.HasPrefix(src, localPrefix) {
- src = src[len(localPrefix):]
- if strings.HasSuffix(src, "/") {
- src += "index.html"
- } else if !strings.HasSuffix(src, ".html") {
- src += "/index.html"
- }
- iframe.SetAttr("src", r.context.BaseEntity.RootPath()+"assets/"+src)
+ start := strings.Index(src, "/widgets/")
+ src = src[start:]
+ if strings.HasSuffix(src, "/") {
+ src += "index.html"
+ } else if !strings.HasSuffix(src, ".html") {
+ src += "/index.html"
  }
+ iframe.SetAttr("src", r.context.BaseEntity.RootPath()+"assets/"+src)
  html, err := goquery.OuterHtml(iframe)
  if err == nil {
  node.Tokens = []byte(html)