block.go

/*
 * Copyright (c) 2021-present Fabien Potencier <fabien@symfony.com>
 *
 * This file is part of Symfony CLI project
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

package terminal

import (
	"bytes"
	"fmt"
	"strings"
)

func FormatBlockMessage(format string, msg string) string {
	var buf bytes.Buffer

	width, _ := GetSize()

	lines, maxLen := splitsBlockLines(msg, width-4) // 2 spaces on the left, 2 on the right
	fullPadding := strings.Repeat(" ", maxLen+4)

	fmt.Fprintf(&buf, "<%s>", format)
	buf.WriteString(fullPadding)
	buf.WriteString("</>\n")
	for _, line := range lines {
		fmt.Fprintf(&buf, "<%s>  ", format)
		lenLine, _ := Stdout.GetFormatter().Format([]byte(line), &buf)
		if n := maxLen - lenLine; n >= 0 {
			buf.WriteString(strings.Repeat(" ", n))
		}
		buf.WriteString("  </>\n")
	}
	fmt.Fprintf(&buf, "<%s>", format)
	buf.WriteString(fullPadding)
	buf.WriteString("</>\n")

	return buf.String()
}

func splitsBlockLines(msg string, width int) ([]string, int) {
	lines := []string{}
	maxLen := 0
	// this can happen in headless mode, like running tests for example
	if width <= 0 {
		width = 80
	}

	for _, line := range strings.Split(msg, "\n") {
		line = strings.ReplaceAll(line, "\t", "        ")
		lastLinePos := 0
		lastOpeningQuotePos := 0
		inAnOpeningTag := false
		inAClosingTag := false
		inATagBody := false
		inQuotes := false
		length := 0
		var lastChar rune
		for pos, char := range line {
			if lastChar != '\\' {
				switch char {
				case '<':
					if len(line) > pos+1 && line[pos+1] == '/' {
						inAClosingTag = true
						inATagBody = false
					} else {
						inAnOpeningTag = true
					}
				case '"':
					inQuotes = !inQuotes
					if inQuotes {
						lastOpeningQuotePos = pos
					}
				}
			}

			if !inAClosingTag && !inAnOpeningTag {
				length += 1
			}

			if char == '>' && lastChar != '\\' {
				if inAnOpeningTag {
					inAnOpeningTag = false
					inATagBody = true
				} else {
					inAClosingTag = false
					inATagBody = false
				}
			}

			if length >= width && !inAClosingTag && !inAnOpeningTag && !inATagBody {
				// if we cross the line boundary but are currently within a
				// quoted text, we jump back just before the opening quote
				// because we don't want to cut the text inside
				if inQuotes && lastOpeningQuotePos > lastLinePos {
					length = pos - lastOpeningQuotePos
					pos = lastOpeningQuotePos - 1
				} else {
					maxLen = width
					length = 0
				}
				lines = append(lines, line[lastLinePos:pos+1])
				lastLinePos = pos + 1
			}

			lastChar = char
		}

		if lastLinePos < len(line) {
			lines = append(lines, line[lastLinePos:])
			if length > maxLen {
				maxLen = length
			}

		}
	}

	return lines, maxLen
}