prefix-output.go

package pkg

import (
	"strings"

	"github.com/fatih/color"
)

type PrefixWriter struct {
	prefix    string
	colorName color.Attribute
	ch        chan []byte
}

func (w PrefixWriter) Write(p []byte) (n int, err error) {
	split := strings.Split(string(p), "\n")
	prefix := color.New(w.colorName).Sprintf("%-20s | ", w.prefix)

	withPrefix := ""
	for _, s := range split {
		if len(s) > 0 {
			withPrefix += prefix + s + "\n"
		}
	}

	w.ch <- []byte(withPrefix)

	return len(p), nil
}