mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-09-08 00:03:28 +00:00
143 lines
4.4 KiB
Go
143 lines
4.4 KiB
Go
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||
|
// Copyright 2024 The Forgejo Authors c/o Codeberg e.V.. All rights reserved.
|
||
|
// SPDX-License-Identifier: MIT
|
||
|
|
||
|
package callout
|
||
|
|
||
|
import (
|
||
|
"strings"
|
||
|
|
||
|
"code.gitea.io/gitea/modules/svg"
|
||
|
|
||
|
"github.com/yuin/goldmark/ast"
|
||
|
"github.com/yuin/goldmark/parser"
|
||
|
"github.com/yuin/goldmark/renderer"
|
||
|
"github.com/yuin/goldmark/renderer/html"
|
||
|
"github.com/yuin/goldmark/text"
|
||
|
"github.com/yuin/goldmark/util"
|
||
|
)
|
||
|
|
||
|
type GitHubCalloutTransformer struct{}
|
||
|
|
||
|
// Transform transforms the given AST tree.
|
||
|
func (g *GitHubCalloutTransformer) Transform(node *ast.Document, reader text.Reader, pc parser.Context) {
|
||
|
supportedAttentionTypes := map[string]bool{
|
||
|
"note": true,
|
||
|
"tip": true,
|
||
|
"important": true,
|
||
|
"warning": true,
|
||
|
"caution": true,
|
||
|
}
|
||
|
|
||
|
_ = ast.Walk(node, func(n ast.Node, entering bool) (ast.WalkStatus, error) {
|
||
|
if !entering {
|
||
|
return ast.WalkContinue, nil
|
||
|
}
|
||
|
|
||
|
switch v := n.(type) {
|
||
|
case *ast.Blockquote:
|
||
|
// We only want attention blockquotes when the AST looks like:
|
||
|
// Text: "["
|
||
|
// Text: "!TYPE"
|
||
|
// Text(SoftLineBreak): "]"
|
||
|
|
||
|
// grab these nodes and make sure we adhere to the attention blockquote structure
|
||
|
firstParagraph := v.FirstChild()
|
||
|
if firstParagraph.ChildCount() < 3 {
|
||
|
return ast.WalkContinue, nil
|
||
|
}
|
||
|
firstTextNode, ok := firstParagraph.FirstChild().(*ast.Text)
|
||
|
if !ok || string(firstTextNode.Text(reader.Source())) != "[" {
|
||
|
return ast.WalkContinue, nil
|
||
|
}
|
||
|
secondTextNode, ok := firstTextNode.NextSibling().(*ast.Text)
|
||
|
if !ok {
|
||
|
return ast.WalkContinue, nil
|
||
|
}
|
||
|
// If the second node's text isn't one of the supported attention
|
||
|
// types, continue walking.
|
||
|
secondTextNodeText := secondTextNode.Text(reader.Source())
|
||
|
attentionType := strings.ToLower(strings.TrimPrefix(string(secondTextNodeText), "!"))
|
||
|
if _, has := supportedAttentionTypes[attentionType]; !has {
|
||
|
return ast.WalkContinue, nil
|
||
|
}
|
||
|
|
||
|
thirdTextNode, ok := secondTextNode.NextSibling().(*ast.Text)
|
||
|
if !ok || string(thirdTextNode.Text(reader.Source())) != "]" {
|
||
|
return ast.WalkContinue, nil
|
||
|
}
|
||
|
|
||
|
// color the blockquote
|
||
|
v.SetAttributeString("class", []byte("gt-py-3 attention attention-"+attentionType))
|
||
|
|
||
|
// create an emphasis to make it bold
|
||
|
emphasis := ast.NewEmphasis(2)
|
||
|
emphasis.SetAttributeString("class", []byte("attention-"+attentionType))
|
||
|
firstParagraph.InsertBefore(firstParagraph, firstTextNode, emphasis)
|
||
|
|
||
|
// capitalize first letter
|
||
|
attentionText := ast.NewString([]byte(strings.ToUpper(string(attentionType[0])) + attentionType[1:]))
|
||
|
|
||
|
// replace the ![TYPE] with icon+Type
|
||
|
emphasis.AppendChild(emphasis, attentionText)
|
||
|
for i := 0; i < 2; i++ {
|
||
|
lineBreak := ast.NewText()
|
||
|
lineBreak.SetSoftLineBreak(true)
|
||
|
firstParagraph.InsertAfter(firstParagraph, emphasis, lineBreak)
|
||
|
}
|
||
|
firstParagraph.InsertBefore(firstParagraph, emphasis, NewAttention(attentionType))
|
||
|
firstParagraph.RemoveChild(firstParagraph, firstTextNode)
|
||
|
firstParagraph.RemoveChild(firstParagraph, secondTextNode)
|
||
|
firstParagraph.RemoveChild(firstParagraph, thirdTextNode)
|
||
|
}
|
||
|
return ast.WalkContinue, nil
|
||
|
})
|
||
|
}
|
||
|
|
||
|
type GitHubCalloutHTMLRenderer struct {
|
||
|
html.Config
|
||
|
}
|
||
|
|
||
|
// RegisterFuncs implements renderer.NodeRenderer.RegisterFuncs.
|
||
|
func (r *GitHubCalloutHTMLRenderer) RegisterFuncs(reg renderer.NodeRendererFuncRegisterer) {
|
||
|
reg.Register(KindAttention, r.renderAttention)
|
||
|
}
|
||
|
|
||
|
// renderAttention renders a quote marked with i.e. "> **Note**" or "> **Warning**" with a corresponding svg
|
||
|
func (r *GitHubCalloutHTMLRenderer) renderAttention(w util.BufWriter, source []byte, node ast.Node, entering bool) (ast.WalkStatus, error) {
|
||
|
if entering {
|
||
|
_, _ = w.WriteString(`<span class="gt-mr-2 gt-vm attention-`)
|
||
|
n := node.(*Attention)
|
||
|
_, _ = w.WriteString(strings.ToLower(n.AttentionType))
|
||
|
_, _ = w.WriteString(`">`)
|
||
|
|
||
|
var octiconType string
|
||
|
switch n.AttentionType {
|
||
|
case "note":
|
||
|
octiconType = "info"
|
||
|
case "tip":
|
||
|
octiconType = "light-bulb"
|
||
|
case "important":
|
||
|
octiconType = "report"
|
||
|
case "warning":
|
||
|
octiconType = "alert"
|
||
|
case "caution":
|
||
|
octiconType = "stop"
|
||
|
}
|
||
|
_, _ = w.WriteString(string(svg.RenderHTML("octicon-" + octiconType)))
|
||
|
} else {
|
||
|
_, _ = w.WriteString("</span>\n")
|
||
|
}
|
||
|
return ast.WalkContinue, nil
|
||
|
}
|
||
|
|
||
|
func NewGitHubCalloutHTMLRenderer(opts ...html.Option) renderer.NodeRenderer {
|
||
|
r := &GitHubCalloutHTMLRenderer{
|
||
|
Config: html.NewConfig(),
|
||
|
}
|
||
|
for _, opt := range opts {
|
||
|
opt.SetHTMLOption(&r.Config)
|
||
|
}
|
||
|
return r
|
||
|
}
|