INNER CODE UNIT · Go

setWidth

kubecolor/kubecolor · internal/cmd/imagegen/main.go:336

func setWidth(s string, width int) string {
	var buf bytes.Buffer
	scanner := bufio.NewScanner(strings.NewReader(s))
	for scanner.Scan() {
		line := scanner.Text()
		lineLen := len(color.ClearCode(line))
		switch {
		case lineLen > width:
			buf.WriteString(line[:width])
		case lineLen < width:
			buf.WriteString(line)
			buf.WriteString(strings.Repeat(" ", width-lineLen))
		default:
			buf.WriteString(line)
		}
		buf.WriteByte('\n')
	}
	return buf.String()

View source record →