literal_blocks.go

package html5

import (
	"bytes"
	"math"
	"strings"
	texttemplate "text/template"

	"github.com/bytesparadise/libasciidoc/pkg/renderer"
	"github.com/bytesparadise/libasciidoc/pkg/types"
	"github.com/pkg/errors"
	log "github.com/sirupsen/logrus"
)

var literalBlockTmpl texttemplate.Template

// initializes the templates
func init() {
	literalBlockTmpl = newTextTemplate("literal block", `{{ $ctx := .Context }}{{ with .Data }}<div {{ if .ID }}id="{{ .ID }}" {{ end }}class="literalblock">
{{ if .Title }}<div class="title">{{ escape .Title }}</div>
{{ end }}<div class="content">
<pre>{{ $lines := .Lines }}{{ range $index, $line := $lines}}{{ $line }}{{ includeNewline $ctx $index $lines }}{{ end }}</pre>
</div>
</div>{{ end }}`, texttemplate.FuncMap{
		"includeNewline": includeNewline,
		"escape":         EscapeString,
	})
}

func renderLiteralBlock(ctx renderer.Context, b types.LiteralBlock) ([]byte, error) {
	log.Debugf("rendering delimited block with content: %s", b.Lines)
	var lines []string
	switch b.Attributes.GetAsString(types.AttrLiteralBlockType) {
	case types.LiteralBlockWithSpacesOnFirstLine:
		if len(b.Lines) == 1 {
			lines = []string{strings.TrimLeft(b.Lines[0], " ")}
		} else {
			lines = make([]string, len(b.Lines))
			// remove as many spaces as needed on each line
			spaceCount := float64(0)
			// first pass to detemine the minimum number of spaces to remove
			for i, line := range b.Lines {
				l := strings.TrimLeft(line, " ")
				if i == 0 {
					spaceCount = float64(len(line) - len(l))
				} else {
					spaceCount = math.Min(spaceCount, float64(len(line)-len(l)))
				}
			}
			log.Debugf("trimming %d space(s) on each line", int(spaceCount))
			// then remove the same number of spaces on each line
			spaces := strings.Repeat(" ", int(spaceCount))
			for i, line := range b.Lines {
				lines[i] = strings.TrimPrefix(line, spaces)
			}
		}
	default:
		lines = b.Lines
	}
	result := bytes.NewBuffer(nil)
	err := literalBlockTmpl.Execute(result, ContextualPipeline{
		Context: ctx,
		Data: struct {
			ID    string
			Title string
			Lines []string
		}{
			ID:    renderElementID(b.Attributes),
			Title: renderElementTitle(b.Attributes),
			Lines: lines,
		}})
	if err != nil {
		return nil, errors.Wrapf(err, "unable to render delimited block")
	}
	return result.Bytes(), nil
}