mirror of
https://github.com/kovetskiy/mark.git
synced 2025-04-23 21:32:41 +08:00
119 lines
2.8 KiB
Go
119 lines
2.8 KiB
Go
package renderer
|
|
|
|
import (
|
|
"bytes"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/kovetskiy/mark/attachment"
|
|
"github.com/kovetskiy/mark/stdlib"
|
|
"github.com/kovetskiy/mark/vfs"
|
|
|
|
"github.com/yuin/goldmark/ast"
|
|
"github.com/yuin/goldmark/renderer"
|
|
"github.com/yuin/goldmark/renderer/html"
|
|
"github.com/yuin/goldmark/util"
|
|
)
|
|
|
|
type ConfluenceImageRenderer struct {
|
|
html.Config
|
|
Stdlib *stdlib.Lib
|
|
Path string
|
|
Attachments attachment.Attacher
|
|
}
|
|
|
|
// NewConfluenceRenderer creates a new instance of the ConfluenceRenderer
|
|
func NewConfluenceImageRenderer(stdlib *stdlib.Lib, attachments attachment.Attacher, path string, opts ...html.Option) renderer.NodeRenderer {
|
|
return &ConfluenceImageRenderer{
|
|
Config: html.NewConfig(),
|
|
Stdlib: stdlib,
|
|
Path: path,
|
|
Attachments: attachments,
|
|
}
|
|
}
|
|
|
|
// RegisterFuncs implements NodeRenderer.RegisterFuncs .
|
|
func (r *ConfluenceImageRenderer) RegisterFuncs(reg renderer.NodeRendererFuncRegisterer) {
|
|
reg.Register(ast.KindImage, r.renderImage)
|
|
}
|
|
|
|
// renderImage renders an inline image
|
|
func (r *ConfluenceImageRenderer) renderImage(writer util.BufWriter, source []byte, node ast.Node, entering bool) (ast.WalkStatus, error) {
|
|
if !entering {
|
|
return ast.WalkContinue, nil
|
|
}
|
|
n := node.(*ast.Image)
|
|
|
|
attachments, err := attachment.ResolveLocalAttachments(vfs.LocalOS, filepath.Dir(r.Path), []string{string(n.Destination)})
|
|
|
|
// We were unable to resolve it locally, treat as URL
|
|
if err != nil {
|
|
escapedURL := string(n.Destination)
|
|
escapedURL = strings.ReplaceAll(escapedURL, "&", "&")
|
|
|
|
err = r.Stdlib.Templates.ExecuteTemplate(
|
|
writer,
|
|
"ac:image",
|
|
struct {
|
|
Width string
|
|
Height string
|
|
Title string
|
|
Alt string
|
|
Attachment string
|
|
Url string
|
|
}{
|
|
"",
|
|
"",
|
|
string(n.Title),
|
|
string(nodeToHTMLText(n, source)),
|
|
"",
|
|
escapedURL,
|
|
},
|
|
)
|
|
} else {
|
|
|
|
r.Attachments.Attach(attachments[0])
|
|
|
|
err = r.Stdlib.Templates.ExecuteTemplate(
|
|
writer,
|
|
"ac:image",
|
|
struct {
|
|
Width string
|
|
Height string
|
|
Title string
|
|
Alt string
|
|
Attachment string
|
|
Url string
|
|
}{
|
|
"",
|
|
"",
|
|
string(n.Title),
|
|
string(nodeToHTMLText(n, source)),
|
|
attachments[0].Filename,
|
|
"",
|
|
},
|
|
)
|
|
}
|
|
|
|
if err != nil {
|
|
return ast.WalkStop, err
|
|
}
|
|
|
|
return ast.WalkSkipChildren, nil
|
|
}
|
|
|
|
// https://github.com/yuin/goldmark/blob/c446c414ef3a41fb562da0ae5badd18f1502c42f/renderer/html/html.go
|
|
func nodeToHTMLText(n ast.Node, source []byte) []byte {
|
|
var buf bytes.Buffer
|
|
for c := n.FirstChild(); c != nil; c = c.NextSibling() {
|
|
if s, ok := c.(*ast.String); ok && s.IsCode() {
|
|
buf.Write(s.Value)
|
|
} else if t, ok := c.(*ast.Text); ok {
|
|
buf.Write(util.EscapeHTML(t.Value(source)))
|
|
} else {
|
|
buf.Write(nodeToHTMLText(c, source))
|
|
}
|
|
}
|
|
return buf.Bytes()
|
|
}
|