package main
import (
"bytes"
"strings"
"sync"
kemoji "github.com/kyokomi/emoji"
"github.com/yuin/goldmark"
emoji "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"
)
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.Table,
extension.Strikethrough,
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("')
} else {
_, _ = w.WriteString("")
}
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("")
_, _ = w.WriteString("")
return ast.WalkSkipChildren, nil
}
func newLinkRenderer() renderer.NodeRenderer {
return &customRenderer{}
}