mirror of
https://gitea.com/Sirherobrine23/tea.git
synced 2024-08-21 20:17:26 +00:00
~~this is semi-blocked by https://github.com/charmbracelet/glamour/pull/96, but behaviour isn't really worse than the previous behaviour (most links work, some are still broken)~~ #### testcase for link resolver ``` tea pr 332 tea checkout 332 && make install && tea pr 332 ``` - [rel](./332) - [abs](/gitea/tea/pulls/332) - [full](https://gitea.com/gitea/tea/pulls/332) Co-authored-by: Norwin Roosen <git@nroo.de> Co-authored-by: 6543 <6543@obermui.de> Reviewed-on: https://gitea.com/gitea/tea/pulls/332 Reviewed-by: 6543 <6543@obermui.de> Reviewed-by: Andrew Thornton <art27@cantab.net> Co-authored-by: Norwin <noerw@noreply.gitea.io> Co-committed-by: Norwin <noerw@noreply.gitea.io>
166 lines
4.4 KiB
Go
166 lines
4.4 KiB
Go
package ansi
|
|
|
|
import (
|
|
"io"
|
|
"net/url"
|
|
|
|
"github.com/muesli/termenv"
|
|
east "github.com/yuin/goldmark-emoji/ast"
|
|
"github.com/yuin/goldmark/ast"
|
|
astext "github.com/yuin/goldmark/extension/ast"
|
|
"github.com/yuin/goldmark/renderer"
|
|
"github.com/yuin/goldmark/util"
|
|
)
|
|
|
|
// Options is used to configure an ANSIRenderer.
|
|
type Options struct {
|
|
BaseURL string
|
|
WordWrap int
|
|
ColorProfile termenv.Profile
|
|
Styles StyleConfig
|
|
}
|
|
|
|
// ANSIRenderer renders markdown content as ANSI escaped sequences.
|
|
type ANSIRenderer struct {
|
|
context RenderContext
|
|
}
|
|
|
|
// NewRenderer returns a new ANSIRenderer with style and options set.
|
|
func NewRenderer(options Options) *ANSIRenderer {
|
|
return &ANSIRenderer{
|
|
context: NewRenderContext(options),
|
|
}
|
|
}
|
|
|
|
// RegisterFuncs implements NodeRenderer.RegisterFuncs.
|
|
func (r *ANSIRenderer) RegisterFuncs(reg renderer.NodeRendererFuncRegisterer) {
|
|
// blocks
|
|
reg.Register(ast.KindDocument, r.renderNode)
|
|
reg.Register(ast.KindHeading, r.renderNode)
|
|
reg.Register(ast.KindBlockquote, r.renderNode)
|
|
reg.Register(ast.KindCodeBlock, r.renderNode)
|
|
reg.Register(ast.KindFencedCodeBlock, r.renderNode)
|
|
reg.Register(ast.KindHTMLBlock, r.renderNode)
|
|
reg.Register(ast.KindList, r.renderNode)
|
|
reg.Register(ast.KindListItem, r.renderNode)
|
|
reg.Register(ast.KindParagraph, r.renderNode)
|
|
reg.Register(ast.KindTextBlock, r.renderNode)
|
|
reg.Register(ast.KindThematicBreak, r.renderNode)
|
|
|
|
// inlines
|
|
reg.Register(ast.KindAutoLink, r.renderNode)
|
|
reg.Register(ast.KindCodeSpan, r.renderNode)
|
|
reg.Register(ast.KindEmphasis, r.renderNode)
|
|
reg.Register(ast.KindImage, r.renderNode)
|
|
reg.Register(ast.KindLink, r.renderNode)
|
|
reg.Register(ast.KindRawHTML, r.renderNode)
|
|
reg.Register(ast.KindText, r.renderNode)
|
|
reg.Register(ast.KindString, r.renderNode)
|
|
|
|
// tables
|
|
reg.Register(astext.KindTable, r.renderNode)
|
|
reg.Register(astext.KindTableHeader, r.renderNode)
|
|
reg.Register(astext.KindTableRow, r.renderNode)
|
|
reg.Register(astext.KindTableCell, r.renderNode)
|
|
|
|
// definitions
|
|
reg.Register(astext.KindDefinitionList, r.renderNode)
|
|
reg.Register(astext.KindDefinitionTerm, r.renderNode)
|
|
reg.Register(astext.KindDefinitionDescription, r.renderNode)
|
|
|
|
// footnotes
|
|
reg.Register(astext.KindFootnote, r.renderNode)
|
|
reg.Register(astext.KindFootnoteList, r.renderNode)
|
|
reg.Register(astext.KindFootnoteLink, r.renderNode)
|
|
reg.Register(astext.KindFootnoteBacklink, r.renderNode)
|
|
|
|
// checkboxes
|
|
reg.Register(astext.KindTaskCheckBox, r.renderNode)
|
|
|
|
// strikethrough
|
|
reg.Register(astext.KindStrikethrough, r.renderNode)
|
|
|
|
// emoji
|
|
reg.Register(east.KindEmoji, r.renderNode)
|
|
}
|
|
|
|
func (r *ANSIRenderer) renderNode(w util.BufWriter, source []byte, node ast.Node, entering bool) (ast.WalkStatus, error) {
|
|
// _, _ = w.Write([]byte(node.Type.String()))
|
|
writeTo := io.Writer(w)
|
|
bs := r.context.blockStack
|
|
|
|
// children get rendered by their parent
|
|
if isChild(node) {
|
|
return ast.WalkContinue, nil
|
|
}
|
|
|
|
e := r.NewElement(node, source)
|
|
if entering {
|
|
// everything below the Document element gets rendered into a block buffer
|
|
if bs.Len() > 0 {
|
|
writeTo = io.Writer(bs.Current().Block)
|
|
}
|
|
|
|
_, _ = writeTo.Write([]byte(e.Entering))
|
|
if e.Renderer != nil {
|
|
err := e.Renderer.Render(writeTo, r.context)
|
|
if err != nil {
|
|
return ast.WalkStop, err
|
|
}
|
|
}
|
|
} else {
|
|
// everything below the Document element gets rendered into a block buffer
|
|
if bs.Len() > 0 {
|
|
writeTo = io.Writer(bs.Parent().Block)
|
|
}
|
|
|
|
// if we're finished rendering the entire document,
|
|
// flush to the real writer
|
|
if node.Type() == ast.TypeDocument {
|
|
writeTo = w
|
|
}
|
|
|
|
if e.Finisher != nil {
|
|
err := e.Finisher.Finish(writeTo, r.context)
|
|
if err != nil {
|
|
return ast.WalkStop, err
|
|
}
|
|
}
|
|
_, _ = bs.Current().Block.Write([]byte(e.Exiting))
|
|
}
|
|
|
|
return ast.WalkContinue, nil
|
|
}
|
|
|
|
func isChild(node ast.Node) bool {
|
|
if node.Parent() != nil && node.Parent().Kind() == ast.KindBlockquote {
|
|
// skip paragraph within blockquote to avoid reflowing text
|
|
return true
|
|
}
|
|
for n := node.Parent(); n != nil; n = n.Parent() {
|
|
// These types are already rendered by their parent
|
|
switch n.Kind() {
|
|
case ast.KindLink, ast.KindImage, ast.KindEmphasis, astext.KindStrikethrough, astext.KindTableCell:
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func resolveURL(baseURL string, rel string) string {
|
|
u, err := url.Parse(rel)
|
|
if err != nil {
|
|
return rel
|
|
}
|
|
if u.IsAbs() {
|
|
return rel
|
|
}
|
|
|
|
base, err := url.Parse(baseURL)
|
|
if err != nil {
|
|
return rel
|
|
}
|
|
return base.ResolveReference(u).String()
|
|
}
|