hugo/markup/goldmark/codeblocks/transform.go

package codeblocks

import (
	"github.com/yuin/goldmark/ast"
	"github.com/yuin/goldmark/parser"
	"github.com/yuin/goldmark/text"
)

// KindCodeBlock is the kind of an Hugo code block.
var KindCodeBlock = ast.NewNodeKind("HugoCodeBlock")

// Its raw contents are the plain text of the code block.
type codeBlock struct {
	ast.BaseBlock
	ordinal int
	b       *ast.FencedCodeBlock
}

func (*codeBlock) Kind() ast.NodeKind { return KindCodeBlock }

func (*codeBlock) IsRaw() bool { return true }

func (b *codeBlock) Dump(src []byte, level int) {
}

type Transformer struct{}

// Transform transforms the provided Markdown AST.
func (*Transformer) Transform(doc *ast.Document, reader text.Reader, pctx parser.Context) {
	var codeBlocks []*ast.FencedCodeBlock

	ast.Walk(doc, func(node ast.Node, enter bool) (ast.WalkStatus, error) {
		if !enter {
			return ast.WalkContinue, nil
		}

		cb, ok := node.(*ast.FencedCodeBlock)
		if !ok {
			return ast.WalkContinue, nil
		}

		codeBlocks = append(codeBlocks, cb)

		return ast.WalkContinue, nil
	})

	for i, cb := range codeBlocks {
		b := &codeBlock{b: cb, ordinal: i}
		parent := cb.Parent()
		if parent != nil {
			parent.ReplaceChild(parent, cb, b)
		}
	}
}