mirror of https://github.com/jlelse/GoBlog
147 lines
3.7 KiB
Go
147 lines
3.7 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
kemoji "github.com/kyokomi/emoji"
|
|
"github.com/yuin/goldmark"
|
|
"github.com/yuin/goldmark-emoji"
|
|
"github.com/yuin/goldmark-emoji/definition"
|
|
"github.com/yuin/goldmark/ast"
|
|
"github.com/yuin/goldmark/extension"
|
|
"github.com/yuin/goldmark/parser"
|
|
"github.com/yuin/goldmark/renderer"
|
|
"github.com/yuin/goldmark/renderer/html"
|
|
"github.com/yuin/goldmark/util"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
var emojilib definition.Emojis
|
|
var emojiOnce sync.Once
|
|
|
|
var markdown goldmark.Markdown
|
|
|
|
func initMarkdown() {
|
|
markdown = goldmark.New(
|
|
goldmark.WithRendererOptions(
|
|
html.WithUnsafe(),
|
|
),
|
|
goldmark.WithParserOptions(
|
|
parser.WithAutoHeadingID(),
|
|
),
|
|
goldmark.WithExtensions(
|
|
extension.GFM,
|
|
extension.Footnote,
|
|
extension.Typographer,
|
|
// Emojis
|
|
emoji.New(
|
|
emoji.WithEmojis(emojiGoLib()),
|
|
),
|
|
// Links
|
|
newCustomExtension(),
|
|
),
|
|
)
|
|
}
|
|
|
|
func renderMarkdown(source string) (content []byte, err error) {
|
|
var buffer bytes.Buffer
|
|
err = markdown.Convert([]byte(source), &buffer)
|
|
content = buffer.Bytes()
|
|
return
|
|
}
|
|
|
|
// Extensions etc...
|
|
|
|
// All emojis from emoji lib
|
|
func emojiGoLib() definition.Emojis {
|
|
emojiOnce.Do(func() {
|
|
var emojis []definition.Emoji
|
|
for shotcode, e := range kemoji.CodeMap() {
|
|
emojis = append(emojis, definition.NewEmoji(e, []rune(e), strings.ReplaceAll(shotcode, ":", "")))
|
|
}
|
|
emojilib = definition.NewEmojis(emojis...)
|
|
})
|
|
return emojilib
|
|
}
|
|
|
|
// Links
|
|
type customExtension struct{}
|
|
|
|
func newCustomExtension() goldmark.Extender {
|
|
return &customExtension{}
|
|
}
|
|
|
|
func (l *customExtension) Extend(m goldmark.Markdown) {
|
|
m.Renderer().AddOptions(renderer.WithNodeRenderers(
|
|
util.Prioritized(newLinkRenderer(), 500),
|
|
))
|
|
}
|
|
|
|
type customRenderer struct{}
|
|
|
|
func (c *customRenderer) RegisterFuncs(r renderer.NodeRendererFuncRegisterer) {
|
|
r.Register(ast.KindLink, c.renderLink)
|
|
r.Register(ast.KindImage, c.renderImage)
|
|
}
|
|
|
|
func (c *customRenderer) renderLink(w util.BufWriter, _ []byte, node ast.Node, entering bool) (ast.WalkStatus, error) {
|
|
n := node.(*ast.Link)
|
|
if entering {
|
|
// Make URL absolute if it's relative
|
|
newDestination := string(util.URLEscape(n.Destination, true))
|
|
if strings.HasPrefix(newDestination, "/") {
|
|
newDestination = appConfig.Server.PublicAddress + newDestination
|
|
}
|
|
// Write URL
|
|
_, _ = w.WriteString("<a href=\"")
|
|
_, _ = w.Write(util.EscapeHTML([]byte(newDestination)))
|
|
_ = w.WriteByte('"')
|
|
// Open external links (links that start with "http") in new tab
|
|
if strings.HasPrefix(string(n.Destination), "http") {
|
|
_, _ = w.WriteString(` target="_blank" rel="noopener"`)
|
|
}
|
|
// Title
|
|
if n.Title != nil {
|
|
_, _ = w.WriteString(` title="`)
|
|
_, _ = w.Write(n.Title)
|
|
_ = w.WriteByte('"')
|
|
}
|
|
_ = w.WriteByte('>')
|
|
} else {
|
|
_, _ = w.WriteString("</a>")
|
|
}
|
|
return ast.WalkContinue, nil
|
|
}
|
|
|
|
func (c *customRenderer) renderImage(w util.BufWriter, source []byte, node ast.Node, entering bool) (ast.WalkStatus, error) {
|
|
if !entering {
|
|
return ast.WalkContinue, nil
|
|
}
|
|
n := node.(*ast.Image)
|
|
// Make URL absolute if it's relative
|
|
destination := string(util.URLEscape(n.Destination, true))
|
|
if strings.HasPrefix(destination, "/") {
|
|
destination = appConfig.Server.PublicAddress + destination
|
|
}
|
|
_, _ = w.WriteString("<a href=\"")
|
|
_, _ = w.Write(util.EscapeHTML([]byte(destination)))
|
|
_, _ = w.WriteString("\">")
|
|
_, _ = w.WriteString("<img src=\"")
|
|
_, _ = w.Write(util.EscapeHTML([]byte(destination)))
|
|
_, _ = w.WriteString(`" alt="`)
|
|
_, _ = w.Write(util.EscapeHTML(n.Text(source)))
|
|
_ = w.WriteByte('"')
|
|
_, _ = w.WriteString(" loading=\"lazy\"")
|
|
if n.Title != nil {
|
|
_, _ = w.WriteString(` title="`)
|
|
_, _ = w.Write(n.Title)
|
|
_ = w.WriteByte('"')
|
|
}
|
|
_, _ = w.WriteString("></a>")
|
|
return ast.WalkSkipChildren, nil
|
|
}
|
|
|
|
func newLinkRenderer() renderer.NodeRenderer {
|
|
return &customRenderer{}
|
|
}
|