Render prefix & suffix around chroma code-blocks

This commit is contained in:
Christian Muehlhaeuser
2019-12-09 12:59:35 +01:00
parent 5c62b67e26
commit 00c809072e
+7 -2
View File
@@ -12,6 +12,8 @@ type CodeBlockElement struct {
} }
func (e *CodeBlockElement) Render(w io.Writer, ctx RenderContext) error { func (e *CodeBlockElement) Render(w io.Writer, ctx RenderContext) error {
bs := ctx.blockStack
var indent uint var indent uint
var margin uint var margin uint
rules := ctx.style[CodeBlock] rules := ctx.style[CodeBlock]
@@ -26,7 +28,7 @@ func (e *CodeBlockElement) Render(w io.Writer, ctx RenderContext) error {
iw := &IndentWriter{ iw := &IndentWriter{
Indent: indent + margin, Indent: indent + margin,
IndentFunc: func(wr io.Writer) { IndentFunc: func(wr io.Writer) {
renderText(w, ctx.blockStack.Current().Style, " ") renderText(w, bs.Current().Style, " ")
}, },
Forward: &AnsiWriter{ Forward: &AnsiWriter{
Forward: w, Forward: w,
@@ -34,7 +36,10 @@ func (e *CodeBlockElement) Render(w io.Writer, ctx RenderContext) error {
} }
if len(theme) > 0 { if len(theme) > 0 {
return quick.Highlight(iw, e.Code, e.Language, "terminal16m", theme) renderText(iw, bs.Current().Style, rules.Prefix)
err := quick.Highlight(iw, e.Code, e.Language, "terminal16m", theme)
renderText(iw, bs.Current().Style, rules.Suffix)
return err
} }
// fallback rendering // fallback rendering