// Copyright 2020-2026 Buf Technologies, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//nolint:errcheck // There is a very noisy, unhelpful lint around WriteString().
package report

import (
	"bytes"
	"fmt"
	"io"
	"math"
	"math/bits"
	"slices"
	"strconv"
	"strings"
	"unicode"
	_ "unsafe" // For go:linkname.

	"google.golang.org/protobuf/encoding/protojson"

	"github.com/bufbuild/protocompile/experimental/source"
	"github.com/bufbuild/protocompile/experimental/source/length"
	"github.com/bufbuild/protocompile/internal/ext/cmpx"
	"github.com/bufbuild/protocompile/internal/ext/iterx"
	"github.com/bufbuild/protocompile/internal/ext/slicesx"
	"github.com/bufbuild/protocompile/internal/ext/stringsx"
	"github.com/bufbuild/protocompile/internal/ext/unicodex"
	"github.com/bufbuild/protocompile/internal/interval"
)

// Renderer configures a diagnostic rendering operation.
type Renderer struct {
	// If set, uses a compact one-line format for each diagnostic.
	Compact bool

	// If set, rendering results are enriched with ANSI color escapes.
	Colorize bool

	// Upgrades all warnings to errors.
	WarningsAreErrors bool

	// If set, remark diagnostics will be printed.
	ShowRemarks bool

	// If set, rendering a diagnostic will show the debug footer.
	ShowDebug bool
}

// renderer contains shared state for a rendering operation, allowing e.g.
// allocations to be re-used and simplifying function signatures.
type renderer struct {
	Renderer
	writer

	ss styleSheet

	// The width, in columns, of the line number margin in the diagnostic
	// currently being rendered.
	margin int
}

// Render renders a diagnostic report.
//
// In addition to returning the rendering result, returns whether the report
// contains any errors.
//
// On the other hand, the actual error-typed return is an error when writing to
// the writer.
func (r Renderer) Render(report *Report, out io.Writer) (errorCount, warningCount int, err error) {
	state := &renderer{
		Renderer: r,
		writer:   writer{out: out},
		ss:       newStyleSheet(r),
	}
	return state.render(report)
}

// RenderString is a helper for calling [Renderer.Render] with a [strings.Builder].
func (r Renderer) RenderString(report *Report) (text string, errorCount, warningCount int) {
	var buf strings.Builder
	e, w, _ := r.Render(report, &buf)
	return buf.String(), e, w
}

func (r *renderer) render(report *Report) (errorCount, warningCount int, err error) {
	for _, diagnostic := range report.Diagnostics {
		if !r.ShowRemarks && diagnostic.level == Remark {
			continue
		}

		r.diagnostic(report, diagnostic)
		if err := r.Flush(); err != nil {
			return errorCount, warningCount, err
		}

		switch {
		case diagnostic.level <= Error:
			errorCount++
		case diagnostic.level <= Warning:
			if r.WarningsAreErrors {
				errorCount++
			} else {
				warningCount++
			}
		}
	}
	if r.Compact {
		return errorCount, warningCount, err
	}

	switch {
	case errorCount > 0 && warningCount > 0:
		fmt.Fprintf(r, "%sencountered %d error%v and %d warning%v\n%s",
			r.ss.bError,
			errorCount, plural(errorCount), warningCount, plural(warningCount),
			r.ss.reset,
		)
	case errorCount > 0:
		fmt.Fprintf(r, "%sencountered %d error%v\n%s",
			r.ss.bError,
			errorCount, plural(errorCount), r.ss.reset,
		)
	case warningCount > 0:
		fmt.Fprintf(r, "%sencountered %d warning%v\n%s",
			r.ss.bWarning,
			warningCount, plural(warningCount), r.ss.reset,
		)
	}
	return errorCount, warningCount, r.Flush()
}

// diagnostic renders a single diagnostic to a string.
func (r *renderer) diagnostic(report *Report, d Diagnostic) {
	if report.Tracing > 0 {
		// If we're debugging diagnostic traces, and we panic, show where this
		// particular diagnostic was generated. This is useful for debugging
		// renderer bugs.
		defer func() {
			if panicked := recover(); panicked != nil {
				proto := report.ToProto()
				json, _ := protojson.MarshalOptions{Multiline: true, Indent: " "}.Marshal(proto)

				stack := strings.Join(d.debug[:min(report.Tracing, len(d.debug))], "\n")
				panic(fmt.Sprintf("protocompile/report: panic in renderer: %v\ndiagnosed at:\n%sreport: %s", panicked, stack, json))
			}
		}()
	}

	var level string
	switch d.level {
	case ICE:
		level = "internal compiler error"
	case Error:
		level = "error"
	case Warning:
		if r.WarningsAreErrors {
			level = "error"
		} else {
			level = "warning"
		}
	case Remark:
		level = "remark"
	}

	// For the simple style, we imitate the Go compiler.
	if r.Compact {
		r.WriteString(r.ss.ColorForLevel(d.level))
		primary := d.Primary()
		switch {
		case primary.File != nil:
			start := primary.StartLoc()
			fmt.Fprintf(r, "%s: %s:%d:%d: %s",
				level, primary.Path(),
				start.Line, start.Column,
				d.message,
			)
		case d.inFile != "":
			fmt.Fprintf(r, "%s: %s: %s",
				level, d.inFile, d.message,
			)
		default:
			fmt.Fprintf(r, "%s: %s", level, d.message)
		}
		r.WriteString(r.ss.reset)
		r.WriteString("\n")
		return
	}

	// For the other styles, we imitate the Rust compiler. See
	// https://github.com/rust-lang/rustc-dev-guide/blob/master/src/diagnostics.md

	fmt.Fprint(r, r.ss.BoldForLevel(d.level), level, ": ")
	r.WriteWrapped(d.message, unicodex.MaxMessageWidth)

	locations := make([][2]source.Location, len(d.snippets))
	for i, snip := range d.snippets {
		locations[i][0] = fileLocation(snip.File, snip.Start, length.TermWidth, false)
		if strings.HasSuffix(snip.Text(), "\n") {
			// If the snippet ends in a newline, don't include the newline in the
			// printed span.
			locations[i][1] = fileLocation(snip.File, snip.End-1, length.TermWidth, false)
			locations[i][1].Column++
		} else {
			locations[i][1] = fileLocation(snip.File, snip.End, length.TermWidth, false)
		}
	}

	// Figure out how wide the line bar needs to be. This is given by
	// the width of the largest line value among the snippets.
	var greatestLine int
	for _, loc := range locations {
		greatestLine = max(greatestLine, loc[1].Line)
	}
	r.margin = max(2, len(strconv.Itoa(greatestLine))) // Easier than messing with math.Log10()

	// Render all the diagnostic windows.
	parts := slicesx.PartitionFunc(d.snippets, func(a, b snippet) bool {
		if len(a.edits) > 0 || len(b.edits) > 0 {
			// Suggestions are always rendered in their own windows, so they
			// always split.
			return true
		}

		// If a has a page break, we split.
		if a.pageBreak {
			return true
		}

		// Otherwise, split if the paths are distinct.
		return a.Path() != b.Path()
	})

	var needsTrailingBreak bool
	for i, snippets := range parts {
		if needsTrailingBreak {
			r.WriteString("\n")
			r.WriteString(r.ss.nAccent)
			r.WriteSpaces(r.margin)
			r.WriteString(" | ")
		}

		if i == 0 || d.snippets[i-1].pageBreak || d.snippets[i-1].Path() != d.snippets[i].Path() {
			r.WriteString("\n")
			r.WriteString(r.ss.nAccent)
			r.WriteSpaces(r.margin)

			primary := snippets[0]
			start := locations[i][0]
			sep := ":::"
			if i == 0 {
				sep = "-->"
			}
			fmt.Fprintf(r, "%s %s:%d:%d\n", sep, primary.Path(), start.Line, start.Column)
		} else if len(snippets[0].edits) > 0 {
			r.WriteString("\n")
		}

		if len(snippets[0].edits) > 0 {
			r.suggestion(snippets[0])
			continue
		}

		// Add a blank line after the file. This gives the diagnostic window some
		// visual breathing room.
		r.WriteString(r.ss.nAccent)
		r.WriteSpaces(r.margin)
		r.WriteString(" | ")

		window := buildWindow(d.level, locations[i:i+len(snippets)], snippets)
		needsTrailingBreak = r.window(window)
	}

	// Render a remedial file name for spanless errors.
	if len(d.snippets) == 0 && d.inFile != "" {
		r.WriteString("\n")
		r.WriteString(r.ss.nAccent)
		r.WriteSpaces(r.margin - 1)

		fmt.Fprintf(r, "--> %s", d.inFile)
	}

	if needsTrailingBreak && !(d.notes == nil && d.help == nil && (!r.ShowDebug || d.debug == nil)) {
		r.WriteString("\n")
		r.WriteString(r.ss.nAccent)
		r.WriteSpaces(r.margin)
		r.WriteString(" | ")
	}

	type footer struct {
		color, label, text string
	}
	footers := iterx.Chain(
		slicesx.Map(d.notes, func(s string) footer { return footer{r.ss.bRemark, "note", s} }),
		slicesx.Map(d.help, func(s string) footer { return footer{r.ss.bRemark, "help", s} }),
		slicesx.Map(d.debug, func(s string) footer { return footer{r.ss.bError, "debug", s} }),
	)

	var haveFooter bool
	for f := range footers {
		haveFooter = true

		isDebug := f.label == "debug"
		if isDebug && !r.ShowDebug {
			continue
		}

		r.WriteString("\n")
		r.WriteSpaces(r.margin)
		fmt.Fprintf(r, "%s = %s%s: %s", r.ss.nAccent, f.color, f.label, r.ss.reset)

		if isDebug {
			r.WriteWrapped(f.text, math.MaxInt)
		} else {
			r.WriteWrapped(f.text, unicodex.MaxMessageWidth)
		}
	}

	if !haveFooter && bytes.Equal(bytes.TrimSpace(r.buf), []byte("|")) {
		r.buf = r.buf[:0]
		r.WriteString(r.ss.reset)
		r.WriteString("\n")
		return
	}

	r.WriteString("\n")
	r.WriteString(r.ss.reset)
	r.WriteString("\n")
}

const maxMultilinesPerWindow = 8

// window is an intermediate structure for rendering an annotated code snippet
// consisting of multiple spans in the same file.
type window struct {
	file *source.File
	// The line number at which the text starts in the overall source file.
	start int
	// The byte offset range this window's text occupies in the containing
	// source File.
	offsets [2]int
	// A list of all underline elements in this window.
	underlines []underline
	multilines []multiline
}

// buildWindow builds a diagnostic window for the given snippets, which must all have
// the same file.
//
// This is separate from [window.Render] because it performs certain layout
// decisions that cannot happen in the middle of actually rendering the source
// code (well, they could, but the resulting code would be far more complicated).
func buildWindow(level Level, locations [][2]source.Location, snippets []snippet) *window {
	w := new(window)
	w.file = snippets[0].File

	// Calculate the range of the file we will be printing. This is given
	// by every line that has a piece of diagnostic in it. To find this, we
	// calculate the join of all of the spans in the window, and find the
	// nearest \n runes in the text.
	w.start = locations[0][0].Line
	w.offsets[0] = snippets[0].Start
	for i := range snippets {
		w.start = min(w.start, locations[i][0].Line)
		w.offsets[0] = min(w.offsets[0], locations[i][0].Offset)
		w.offsets[1] = max(w.offsets[1], locations[i][1].Offset)
	}
	w.offsets[0], w.offsets[1] = adjustLineOffsets(w.file.Text(), w.offsets[0], w.offsets[1])

	// Now, convert each span into an underline or multiline.
	for i, snippet := range snippets {
		isMulti := locations[i][0].Line != locations[i][1].Line

		if isMulti && len(w.multilines) < maxMultilinesPerWindow {
			w.multilines = append(w.multilines, multiline{
				start:      locations[i][0].Line,
				end:        locations[i][1].Line,
				startWidth: locations[i][0].Column,
				endWidth:   locations[i][1].Column,
				level:      noteLevel, message: snippet.message,
			})
			ml := &w.multilines[len(w.multilines)-1]

			if ml.startWidth == ml.endWidth {
				ml.endWidth++
			}

			// Calculate whether this snippet starts on the first non-space rune of
			// the line.
			if snippet.Start != 0 {
				firstLineStart := strings.LastIndexByte(w.file.Text()[:snippet.Start], '\n')
				if !strings.ContainsFunc(
					w.file.Text()[firstLineStart+1:snippet.Start],
					func(r rune) bool { return !unicode.IsSpace(r) },
				) {
					ml.startWidth = 0
				}
			}

			if snippet.primary {
				ml.level = level
			}
			continue
		}

		w.underlines = append(w.underlines, underline{
			line:    locations[i][0].Line,
			start:   locations[i][0].Column,
			end:     locations[i][1].Column,
			level:   noteLevel,
			message: snippet.message,
			subline: -1,
		})

		ul := &w.underlines[len(w.underlines)-1]
		if snippet.primary {
			ul.level = level
		}
		if ul.start == ul.end {
			ul.end++
		}

		if isMulti {
			// This is an "overflow multiline" for diagnostics with too
			// many multilines. In this case, we want to end the underline at
			// the end of the first line.
			lineEnd := strings.Index(w.file.Text()[snippet.Start:], "\n")
			if lineEnd == -1 {
				lineEnd = len(w.file.Text())
			} else {
				lineEnd += snippet.Start
			}
			uw := unicodex.Width{Column: ul.start, EscapeNonPrint: true}
			uw.WriteString(w.file.Text()[snippet.Start:lineEnd])
			ul.end = uw.Column
		}

		// Make sure no empty underlines exist.
		if ul.Len() == 0 {
			ul.start++
		}
	}

	slices.SortFunc(w.underlines, cmpx.Join(
		cmpx.Key(func(u underline) int { return u.line }),
		cmpx.Key(func(u underline) int { return u.start }),
		cmpx.Key(func(u underline) Level { return u.level }),
	))
	slices.SortFunc(w.multilines, cmpx.Join(
		cmpx.Key(func(m multiline) int { return m.start }),
		cmpx.Key(func(m multiline) int { return -m.end }), // Descending order.
	))
	return w
}

func (r *renderer) window(w *window) (needsTrailingBreak bool) {
	// lineInfo is layout information for a single line of this window. There
	// is one lineInfo for each line of w.file.Text we intend to render, as
	// given by w.offsets.
	type lineInfo struct {
		// This is the multilines whose pipes intersect with this line.
		sidebar []*multiline
		// This is a set of strings to render verbatim under the actual source
		// code line. This makes it possible to lay out all of the complex
		// underlines ahead of time instead of interleaved with rendering the
		// source code lines.
		underlines []string
		// This is whether this line should be printed in the window. This is
		// used to avoid emitting e.g. lines between the start and end of a
		// 100-line multi.
		shouldEmit bool
	}

	lines := strings.Split(w.file.Text()[w.offsets[0]:w.offsets[1]], "\n")
	// Populate ancillary info for each line.
	info := make([]lineInfo, len(lines))

	nesting := new(interval.Nesting[int, *underline])

	// First, lay out the multilines, and compute how wide the sidebar is.
	for i := range w.multilines {
		multi := &w.multilines[i]
		// Find the smallest unused index by every line in the range.
		//
		// We want to assign to each multiline a "sidebar index", which is which
		// column its connecting pipes | are placed on. For each multiline, we
		// want to allocate the leftmost index such that it does not conflict
		// with any previously allocated sidebar pipes that are in the same
		// range as this multiline. We cannot simply take the max of their
		// indices, because it might happen that this multiline only intersects
		// with multis on lines that only use indices 0 and 2. This can happen
		// if the multi on index 2 intersects a *different* range that already
		// has two other multis in it.
		//
		// We achieve this by looking at all already-laid-out multis in this
		// multi's range and using a bitset to detect the least unused index.
		// Note that we artificially limit the number of rendered multis to
		// 8 in the code that builds the window itself.
		var multilineBitset uint
		for i := multi.start; i <= multi.end; i++ {
			for col, ml := range info[i-w.start].sidebar {
				if ml != nil {
					multilineBitset |= 1 << col
				}
			}
		}
		idx := bits.TrailingZeros(^multilineBitset)

		// Apply the index to every element of sidebar.
		for i := multi.start; i <= multi.end; i++ {
			line := &info[i-w.start].sidebar
			for len(*line) < idx+1 {
				*line = append(*line, nil)
			}
			(*line)[idx] = multi
		}

		// Mark the start and end as must-emit.
		info[multi.start-w.start].shouldEmit = true
		info[multi.end-w.start].shouldEmit = true
	}
	var sidebarLen int
	for _, info := range info {
		sidebarLen = max(sidebarLen, len(info.sidebar))
	}

	// Next, we can render the underline parts. This aggregates all underlines
	// for the same line into rendered chunks
	parts := slicesx.PartitionKey(w.underlines, func(u underline) int { return u.line })
	for _, part := range parts {
		cur := &info[part[0].line-w.start]
		cur.shouldEmit = true

		// Arrange for a "sidebar prefix" for this line. This is determined by any sidebars that are
		// active on this line, even if they end on it.
		sidebar := r.sidebar(sidebarLen, -1, -1, cur.sidebar)

		// Lay out the physical underlines. We use a cubic (oops) algorithm
		// to pack the underlines in such a way that none of them overlap.
		// To do so, we loop over the underlines several times until all of them
		// are laid out.
		//
		// This algorithm is O(n^3) where n is the number of underlines in a
		// line, meaning that n can't really get out of hand for normal
		// inputs.

		slices.SortStableFunc(part, cmpx.Key(func(ul underline) int { return -ul.Len() }))
		nesting.Clear()
		for _, ul := range slicesx.Pointers(part) {
			nesting.Insert(ul.start, ul.end-1, ul)
		}

		var sublines [][]*underline
		for set := range nesting.Sets() {
			var subline []*underline
			for entry := range set {
				subline = append(subline, entry.Value)
			}
			slices.SortStableFunc(subline, cmpx.Key(func(ul *underline) int { return ul.start }))
			sublines = append(sublines, subline)
		}

		// Convert the underlines into strings. Collect the rightmost underlines
		// in a slice.
		rightmost := slicesx.Transform(sublines, func(subline []*underline) *underline {
			var rightmost *underline
			for _, ul := range subline {
				if rightmost == nil {
					rightmost = ul
					continue
				}

				if ul.end > rightmost.end {
					rightmost = ul
				}
			}
			return rightmost
		})

		sublineLens := slicesx.Transform(rightmost, func(ul *underline) int {
			return ul.end
		})

		// Clear rightmost messages that will result in wrapping.
		for i, ul := range rightmost {
			if ul == nil {
				continue
			}

			startCol := 4 +
				int(math.Log10(float64(w.start+len(lines)))) + // Approximation.
				len(sidebar) + ul.end

			uw := unicodex.Width{Column: startCol}
			uw.WriteString(ul.message)
			if uw.Column > unicodex.MaxMessageWidth {
				// Move rightmost into the normal underlines, because it causes wrapping.
				rightmost[i] = nil
			}
		}

		// *Now*, lay out the underlines, so that we can stick pipes on them.
		// We will also need to add color later.
		sublineBufs := make([][]byte, len(sublines))
		for i, sub := range sublines {
			buf := bytes.Repeat([]byte(" "), sublineLens[i])
			for _, ul := range sub {
				b := byte('^')
				if ul.level == noteLevel {
					b = '-'
				}
				slicesx.Fill(buf[ul.start-1:ul.end-1], b)
			}
			sublineBufs[i] = buf
		}

		// Reserve space for the above, which still need further processing.
		cur.underlines = make([]string, len(sublines))

		// Now, do all the other messages. We're going to split them such that
		// no two messages overlap, using an interval nesting collection.
		//
		// For each message, we also need to draw pipes (|) above each one to
		// connect it to its underline.
		//
		// This is slightly complicated, because there are two layers: the
		// pipes, and whatever message goes on the pipes.
		var rest []*underline
		for _, ul := range slicesx.Pointers(part) {
			if slices.Contains(rightmost, ul) || ul.message == "" {
				continue
			}
			rest = append(rest, ul)
		}

		slices.SortStableFunc(rest, cmpx.Key(func(ul *underline) int {
			return len(ul.message)
		}))
		nesting.Clear()
		for _, ul := range rest {
			start := ul.start - 1
			if ul.Len() > 3 {
				start++ // Move the pipe a little forward for long underlines.
			}

			nesting.Insert(start, start+len(ul.message)-1, ul)
		}

		var messages [][]*underline
		for set := range nesting.Sets() {
			var s []*underline
			for entry := range set {
				s = append(s, entry.Value)
			}
			slices.SortStableFunc(s, cmpx.Key(func(ul *underline) int {
				return ul.start
			}))
			messages = append(messages, s)
		}
		slices.SortStableFunc(messages, cmpx.Key(func(uls []*underline) int {
			return uls[0].start
		}))

		var buf1, buf2 []byte
		for i, line := range messages {
			// Clear the temp buffers.
			buf1 = buf1[:0]
			buf2 = buf2[:0]

			// First, lay out the pipes for this and all following lines.
			var nonColorLen int
			for _, line := range messages[i:] {
				for _, ul := range line {
					col := ul.start - 1
					if ul.Len() > 3 {
						col++ // Move the pipe a little forward for long underlines.
					}

					for nonColorLen < col {
						buf1 = append(buf1, ' ')
						nonColorLen++
					}

					if nonColorLen == col {
						// Two pipes may appear on the same column!
						// This is why this is in a conditional. To record the
						// level, rather than actually using a pipe, we use
						// the value of level here.
						buf1 = append(buf1, ^byte(ul.level))
						nonColorLen++

						if i == 0 {
							// Apply this pipe to all of the sublines.
							for _, sub := range slicesx.Pointers(sublineBufs) {
								for len(*sub) < col+1 {
									*sub = append(*sub, ' ')
								}

								if (*sub)[col] == ' ' {
									(*sub)[col] = ^byte(ul.level)
								}
							}
						}
					}
				}
			}

			// Insert the colors for the pipes.
			buf2 = append(buf2, buf1...)
			for i := 0; i < len(buf1); i++ {
				b := buf1[i]
				if b&0x80 == 0 {
					continue
				}

				color := r.ss.BoldForLevel(Level(^b))
				buf1[i] = '|'
				buf1 = slices.Insert(buf1, i, []byte(color)...)
				i += len(color)
			}

			// Splat in the one with all the pipes in it as-is. Having two rows like
			// this ensures that each message has one pipe directly above it.
			cur.underlines = append(cur.underlines, strings.TrimRight(sidebar+string(buf1), " "))

			// Then, splat in the messages for this line.
			offset := 0
			for _, ul := range line {
				start := ul.start - 1 + offset
				if ul.Len() > 3 {
					start++ // Move the pipe a little forward for long underlines.
				}

				for len(buf2) < start+len(ul.message)+1 {
					buf2 = append(buf2, ' ')
				}

				color := r.ss.BoldForLevel(ul.level)
				copy(buf2[start:], ul.message)
				buf2 = slices.Insert(buf2, start, []byte(color)...)
				offset += len(color)
			}

			// Insert the colors for the pipes, again...
			for i := 0; i < len(buf2); i++ {
				b := buf2[i]
				if b&0x80 == 0 {
					continue
				}

				color := r.ss.BoldForLevel(Level(^b))
				buf2[i] = '|'
				buf2 = slices.Insert(buf2, i, []byte(color)...)
				i += len(color)
			}

			cur.underlines = append(cur.underlines, strings.TrimRight(sidebar+string(buf2), " "))
		}

		// Finally, finalize the underlines by coloring them!
		for i, sub := range sublines {
			buf := sublineBufs[i]
			ul := rightmost[i]

			// Count the number of pipes that come before the end of the
			// underlines.
			pipes := 0
			for i, b := range buf {
				if ul != nil && ul.end <= i {
					break
				}
				if b&0x80 != 0 {
					pipes++
				}
			}

			offset := 0
			// First, just the underlines.
			for _, ul := range sub {
				color := r.ss.BoldForLevel(ul.level)
				buf = slices.Insert(buf, ul.start-1+offset, []byte(color)...)
				offset += len(color)
			}

			// Now, color the pipes.
			for i := 0; i < len(buf); i++ {
				b := buf[i]
				if b&0x80 == 0 {
					continue
				}

				color := r.ss.BoldForLevel(Level(^b))
				buf[i] = '|'
				buf = slices.Insert(buf, i, []byte(color)...)
				i += len(color)

				if pipes > 0 {
					offset += len(color)
				}
				pipes--
			}

			// Append the message for this subline, if any.
			if ul != nil && ul.message != "" {
				color := r.ss.BoldForLevel(ul.level)
				end := ul.end + offset + len(color) + len(ul.message)
				for len(buf) < end {
					buf = append(buf, ' ')
				}
				copy(buf[ul.end+offset:], color)
				copy(buf[ul.end+offset+len(color):], ul.message)
			}

			cur.underlines[i] = sidebar + string(buf)
		}
	}

	//nolint:dupword
	// Now that we've laid out the underlines, we can add the starts and ends of all
	// of the multilines, which go after the underlines.
	//
	// The result is that a multiline will look like this:
	//
	//   code
	//  ____^
	// | code code code
	// \______________^ message
	var line strings.Builder
	for lineIdx := range info {
		cur := &info[lineIdx]
		prevStart := -1
		for mlIdx, ml := range cur.sidebar {
			if ml == nil {
				continue
			}

			line.Reset()
			var isStart bool
			switch w.start + lineIdx {
			case ml.start:
				if ml.startWidth == 0 {
					continue
				}

				isStart = true
				fallthrough
			case ml.end:
				// We need to be flush with the sidebar here, so we trim the trailing space.
				sidebar := []byte(strings.TrimRight(r.sidebar(0, -1, prevStart, cur.sidebar[:mlIdx+1]), " "))

				// We also need to erase the bars of any multis that are before this multi
				// and start/end on the same line.
				if !isStart {
					for mlIdx, otherML := range cur.sidebar[:mlIdx+1] {
						if otherML != nil && otherML.end == ml.end {
							// All the color escapes have the same byte length, so we can use the length of
							// any of them to measure how far we need to adjust the offset to get to the
							// pipe. We need to account for one escape per multiline, and also need to skip
							// past the color escape on the pipe we want to erase.
							codeLen := len(r.ss.bAccent)
							idx := mlIdx*(2+codeLen) + codeLen
							if idx < len(sidebar) {
								sidebar[idx] = ' '
							}
						}
					}
				}

				// Delete the last pipe and replace it with a slash or space, depending.
				// on orientation.
				line.Write(sidebar[:len(sidebar)-1])
				if isStart {
					line.WriteByte(' ')
				} else {
					line.WriteByte('\\')
				}

				// Pad out to the gutter of the code block.
				remaining := sidebarLen - (mlIdx + 1)
				padByRune(&line, remaining*2, '_')

				// Pad to right before we need to insert a ^ or -
				if isStart {
					padByRune(&line, ml.startWidth-1, '_')
				} else {
					padByRune(&line, ml.endWidth-1, '_')
				}

				if ml.level == noteLevel {
					line.WriteByte('-')
				} else {
					line.WriteByte('^')
				}

				// TODO: If the source code has extremely long lines, this will cause
				// the message to wind up wrapped crazy far. It may be worth doing
				// wrapping ourselves in some cases (beyond a threshold of, say, 120
				// columns). It is unlikely users will hit this problem with "realistic"
				// inputs, though, and e.g. rustc and clang do not bother to handle this
				// case nicely.
				if !isStart && ml.message != "" {
					line.WriteByte(' ')
					line.WriteString(ml.message)
				}
				cur.underlines = append(cur.underlines, line.String())
			}

			if isStart {
				prevStart = mlIdx
			} else {
				prevStart = -1
			}
		}
	}

	// Make sure to emit any lines adjacent to another line we want to emit, so long as that
	// line contains printable characters.
	//
	// We copy a set of all the lines we plan to emit before this transformation;
	// otherwise, doing it in-place will cause every nonempty line after a must-emit line
	// to be shown, which we don't want.
	mustEmit := make(map[int]bool)
	for i := range info {
		if info[i].shouldEmit {
			mustEmit[i] = true
		}
	}
	for i := range info {
		printable := func(r rune) bool { return !unicode.IsSpace(r) }

		// At least two of the below conditions must be true for
		// this line to be shown. Annoyingly, go does not have a conversion
		// from bool to int...
		var score int
		if strings.IndexFunc(lines[i], printable) != -1 {
			score++
		}

		sameIndent := func(a, b string) bool {
			if a == "" || b == "" {
				return true
			}
			d1 := strings.IndexFunc(a, printable)
			if d1 == -1 {
				d1 = len(a)
			}
			d2 := strings.IndexFunc(b, printable)
			if d2 == -1 {
				d2 = len(b)
			}
			return a[:d1] == b[:d2]
		}

		if mustEmit[i-1] && sameIndent(lines[i-1], lines[i]) {
			score++
		}
		if mustEmit[i+1] && sameIndent(lines[i+1], lines[i]) {
			score++
		}
		if score >= 2 {
			info[i].shouldEmit = true
		}
	}
	// Ensure that there are no single-line elided chunks.
	// This necessarily results in a fixed point after one iteration.
	for i := range info {
		mustEmit[i] = info[i].shouldEmit
	}
	for i := range info {
		if mustEmit[i-1] && mustEmit[i+1] {
			info[i].shouldEmit = true
		}
	}

	lastEmit := w.start
	for i, line := range lines {
		cur := &info[i]
		lineno := i + w.start

		if !cur.shouldEmit {
			continue
		}

		// If the last multi of the previous line starts on that line, make its
		// pipe here a slash so that it connects properly.
		slashAt := -1
		if i > 0 {
			prevSidebar := info[i-1].sidebar
			if len(prevSidebar) > 0 &&
				prevSidebar[len(prevSidebar)-1].start == lineno-1 &&
				prevSidebar[len(prevSidebar)-1].startWidth > 0 {
				slashAt = len(prevSidebar) - 1
			}
		}
		sidebar := r.sidebar(sidebarLen, lineno, slashAt, cur.sidebar)

		if i > 0 && !info[i-1].shouldEmit {
			// Generate a visual break if this is right after a real line.
			r.WriteString("\n")
			r.WriteString(r.ss.nAccent)
			r.WriteSpaces(r.margin - 2)
			r.WriteString("...  ")

			// Generate a sidebar as before but this time we want to look at the
			// last line that was actually emitted.
			slashAt := -1
			prevSidebar := info[lastEmit-w.start].sidebar
			if len(prevSidebar) > 0 &&
				prevSidebar[len(prevSidebar)-1].start == lastEmit &&
				prevSidebar[len(prevSidebar)-1].startWidth > 0 {
				slashAt = len(prevSidebar) - 1
			}

			r.WriteString(r.sidebar(sidebarLen, lastEmit+1, slashAt, info[lastEmit-w.start].sidebar))
		}

		// Ok, we are definitely printing this line out.
		//
		// Note that sidebar already includes a trailing ss.reset for us.
		fmt.Fprintf(r, "\n%s%*d | %s%s", r.ss.nAccent, r.margin, lineno, sidebar, r.ss.reset)
		lastEmit = lineno

		// Re-use the logic from width calculation to correctly format a line for
		// showing in a terminal.
		uw := &unicodex.Width{EscapeNonPrint: true, Out: &r.writer}
		uw.WriteString(line)
		needsTrailingBreak = true

		// If this happens to be an annotated line, this is when it gets annotated.
		for _, line := range cur.underlines {
			r.WriteString("\n")
			r.WriteString(r.ss.nAccent)
			r.WriteSpaces(r.margin)
			r.WriteString(" | ")
			r.WriteString(line)

			// Gross hack to pick up whether a trailing break is necessary; we
			// only add one if the underline contains text.
			needsTrailingBreak = strings.ContainsFunc(line, unicode.IsLetter)
		}
	}

	return needsTrailingBreak
}

type underline struct {
	line       int
	start, end int
	level      Level
	message    string
	subline    int
}

func (u underline) Len() int {
	return u.end - u.start
}

type multiline struct {
	start, end           int
	startWidth, endWidth int
	level                Level
	message              string
}

func (r *renderer) sidebar(bars, lineno, slashAt int, multis []*multiline) string {
	var sidebar strings.Builder
	for i, ml := range multis {
		if ml == nil {
			sidebar.WriteString("  ")
			continue
		}

		sidebar.WriteString(r.ss.BoldForLevel(ml.level))

		switch {
		case slashAt == i:
			sidebar.WriteByte('/')
		case lineno != ml.start:
			sidebar.WriteByte('|')
		case ml.startWidth == 0:
			sidebar.WriteByte('/')
		default:
			sidebar.WriteByte(' ')
		}
		sidebar.WriteByte(' ')
	}
	for sidebar.Len() < bars*2 {
		sidebar.WriteByte(' ')
	}
	return sidebar.String()
}

// suggestion renders a single suggestion window.
func (r *renderer) suggestion(snip snippet) {
	r.WriteString(r.ss.nAccent)
	r.WriteSpaces(r.margin)
	r.WriteString("help: ")
	r.WriteWrapped(snip.message, unicodex.MaxMessageWidth)

	// Add a blank line after the file. This gives the diagnostic window some
	// visual breathing room.
	r.WriteString("\n")
	r.WriteSpaces(r.margin)
	r.WriteString(r.ss.nAccent)
	r.WriteString(" | ")

	// When the suggestion spans multiple lines, we don't bother doing a by-the-rune
	// diff, because the result can be hard for users to understand how to apply
	// to their code. Also, if the suggestion contains deletions, use
	multiline := slices.ContainsFunc(snip.edits, func(e Edit) bool {
		// Prefer multiline suggestions in the case of deletions.
		return e.IsDeletion() || strings.Contains(e.Replace, "\n")
	}) ||
		strings.Contains(snip.Span.Text(), "\n")

	if multiline {
		span, hunks := unifiedDiff(snip.Span, snip.edits)
		startLine := span.StartLoc().Line

		aLine := startLine
		bLine := startLine
		for i, hunk := range hunks {
			if hunk.content == "" {
				continue
			}

			// Skip addition lines that only contain whitespace, if the previous
			// hunk was a deletion. This helps avoid cases where a whole line
			// was deleted and some indentation was left over.
			if prev, _ := slicesx.Get(hunks, i-1); prev.kind == hunkDelete &&
				hunk.kind == hunkAdd &&
				stringsx.EveryFunc(hunk.content, unicode.IsSpace) {
				continue
			}

			for line := range strings.SplitSeq(hunk.content, "\n") {
				lineno := aLine
				if hunk.kind == '+' {
					lineno = bLine
				}

				// Draw the line as we would for an ordinary window, but prefix
				// each line with a the hunk's kind and color.
				fmt.Fprintf(r, "\n%s%*d | %s%c%s ",
					r.ss.nAccent, r.margin, lineno,
					hunk.bold(&r.ss), hunk.kind, hunk.color(&r.ss),
				)
				uw := &unicodex.Width{EscapeNonPrint: true, Out: &r.writer}
				uw.WriteString(line)

				switch hunk.kind {
				case hunkUnchanged:
					aLine++
					bLine++
				case hunkDelete:
					aLine++
				case hunkAdd:
					bLine++
				}
			}
		}

		r.WriteString("\n")
		r.WriteString(r.ss.nAccent)
		r.WriteSpaces(r.margin)
		r.WriteString(" | ")
		return
	}

	span, hunks := hunkDiff(snip.Span, snip.edits)
	fmt.Fprintf(r, "\n%s%*d | ", r.ss.nAccent, r.margin, span.StartLoc().Line)
	uw := &unicodex.Width{EscapeNonPrint: true, Out: &r.writer}
	for _, hunk := range hunks {
		if hunk.content == "" {
			continue
		}

		r.WriteString(hunk.color(&r.ss))
		// Re-use the logic from width calculation to correctly format a line for
		// showing in a terminal.
		uw.WriteString(hunk.content)
	}

	// Draw underlines for each modified segment, using + and - as the
	// underline characters.
	r.WriteString("\n")
	r.WriteString(r.ss.nAccent)
	r.WriteSpaces(r.margin)
	r.WriteString(" | ")
	uw.Column = 0
	uw.Out = nil
	for _, hunk := range hunks {
		if hunk.content == "" {
			continue
		}

		prev := uw.Column
		uw.WriteString(hunk.content)
		r.WriteString(hunk.bold(&r.ss))
		for range uw.Column - prev {
			r.WriteString(string(hunk.kind))
		}
	}
}

func adjustLineOffsets(text string, start, end int) (int, int) {
	// Find the newlines before and after the given ranges, respectively.
	// This snaps the range to start immediately after a newline (or SOF) and
	// end immediately before a newline (or EOF).
	start = strings.LastIndexByte(text[:start], '\n') + 1 // +1 gives the byte *after* the newline.
	if offset := strings.IndexByte(text[end:], '\n'); offset != -1 {
		end += offset
	} else {
		end = len(text)
	}
	return start, end
}

func padByRune(out *strings.Builder, spaces int, r rune) {
	for range spaces {
		out.WriteRune(r)
	}
}

// This is an unexported method of source.File which we need in this file. It's
// simpler to just linkname it than to expose allowNonPrint, which should really
// not be a public API.
//
// The linkname-ing actually happens in source.File.
//
//go:linkname fileLocation
func fileLocation(f *source.File, offset int, units length.Unit, allowNonPrint bool) source.Location