fix: disable padding for the codeblock

test: update test snapshots

Author: jahvon

ansi/blockelement.go

@@ -36,7 +36,7 @@ func (e *BlockElement) Finish(w io.Writer, ctx RenderContext) error {
 			" ,.;-+|",
 		)
 
-		mw := NewMarginWriter(ctx, w, bs.Current().Style)
+		mw := NewMarginWriter(ctx, w, bs.Current().Style, true)
 		if _, err := io.WriteString(mw, s); err != nil {
 			return err
 		}

ansi/codeblock.go

@@ -124,7 +124,7 @@ func (e *CodeBlockElement) Finish(w io.Writer, ctx RenderContext) error {
 		mutex.Unlock()
 	}
 
-	mw := NewMarginWriter(ctx, w, bs.Current().Style)
+	mw := NewMarginWriter(ctx, w, bs.Current().Style, false)
 	renderText(mw, ctx.options.ColorProfile, bs.Current().Style.StylePrimitive, rules.BlockPrefix)
 	if len(theme) > 0 {
 		err := quick.Highlight(mw, e.Code, e.Language, "terminal256", theme)

ansi/heading.go

@@ -59,7 +59,7 @@ func (e *HeadingElement) Render(w io.Writer, ctx RenderContext) error {
 func (e *HeadingElement) Finish(w io.Writer, ctx RenderContext) error {
 	bs := ctx.blockStack
 	rules := bs.Current().Style
-	mw := NewMarginWriter(ctx, w, rules)
+	mw := NewMarginWriter(ctx, w, rules, true)
 
 	flow := wordwrap.NewWriter(int(bs.Width(ctx)))
 	_, err := flow.Write(bs.Current().Block.Bytes())

ansi/margin.go

@@ -24,7 +24,7 @@ type MarginWriter struct {
 }
 
 // NewMarginWriter returns a new MarginWriter.
-func NewMarginWriter(ctx RenderContext, w io.Writer, rules StyleBlock) *MarginWriter {
+func NewMarginWriter(ctx RenderContext, w io.Writer, rules StyleBlock, padded bool) *MarginWriter {
 	bs := ctx.blockStack
 	mw := &MarginWriter{
 		w:           w,
@@ -44,11 +44,15 @@ func NewMarginWriter(ctx RenderContext, w io.Writer, rules StyleBlock) *MarginWr
 		mw.margin = *rules.Margin
 	}
 
-	mw.pw = padding.NewWriterPipe(mw.w, bs.Width(ctx), func(wr io.Writer) {
-		renderText(mw.w, mw.profile, mw.rules, " ")
-	})
+	fwd := mw.w
+	if padded {
+		mw.pw = padding.NewWriterPipe(mw.w, bs.Width(ctx), func(wr io.Writer) {
+			renderText(mw.w, mw.profile, mw.rules, "")
+		})
+		fwd = mw.pw
+	}
 
-	mw.iw = indent.NewWriterPipe(mw.pw, mw.indentation+mw.margin, mw.indentFunc)
+	mw.iw = indent.NewWriterPipe(fwd, mw.indentation+(mw.margin*2), mw.indentFunc)
 	return mw
 }
 

ansi/paragraph.go

@@ -35,7 +35,7 @@ func (e *ParagraphElement) Finish(w io.Writer, ctx RenderContext) error {
 	bs := ctx.blockStack
 	rules := bs.Current().Style
 
-	mw := NewMarginWriter(ctx, w, rules)
+	mw := NewMarginWriter(ctx, w, rules, true)
 	if len(strings.TrimSpace(bs.Current().Block.String())) > 0 {
 		flow := wordwrap.NewWriter(int(bs.Width(ctx)))
 		flow.KeepNewlines = ctx.options.PreserveNewLines