tee_printer.go

package terminal

import (
	"fmt"
)

type Printer interface {
	Print(a ...interface{}) (n int, err error)
	Printf(format string, a ...interface{}) (n int, err error)
	Println(a ...interface{}) (n int, err error)
	ForcePrint(a ...interface{}) (n int, err error)
	ForcePrintf(format string, a ...interface{}) (n int, err error)
	ForcePrintln(a ...interface{}) (n int, err error)
}

type OutputCapture interface {
	SetOutputBucket(*[]string)
}

type TerminalOutputSwitch interface {
	DisableTerminalOutput(bool)
}

type TeePrinter struct {
	disableTerminalOutput bool
	outputBucket          *[]string
}

func NewTeePrinter() *TeePrinter {
	return &TeePrinter{}
}

func (t *TeePrinter) SetOutputBucket(bucket *[]string) {
	t.outputBucket = bucket
}

func (t *TeePrinter) Print(values ...interface{}) (n int, err error) {
	str := fmt.Sprint(values...)
	t.saveOutputToBucket(str)
	if !t.disableTerminalOutput {
		return fmt.Print(str)
	}
	return
}

func (t *TeePrinter) Printf(format string, a ...interface{}) (n int, err error) {
	str := fmt.Sprintf(format, a...)
	t.saveOutputToBucket(str)
	if !t.disableTerminalOutput {
		return fmt.Print(str)
	}
	return
}

func (t *TeePrinter) Println(values ...interface{}) (n int, err error) {
	str := fmt.Sprint(values...)
	t.saveOutputToBucket(str)
	if !t.disableTerminalOutput {
		return fmt.Println(str)
	}
	return
}

func (t *TeePrinter) ForcePrint(values ...interface{}) (n int, err error) {
	str := fmt.Sprint(values...)
	t.saveOutputToBucket(str)
	return fmt.Print(str)
}

func (t *TeePrinter) ForcePrintf(format string, a ...interface{}) (n int, err error) {
	str := fmt.Sprintf(format, a...)
	t.saveOutputToBucket(str)
	return fmt.Print(str)
}

func (t *TeePrinter) ForcePrintln(values ...interface{}) (n int, err error) {
	str := fmt.Sprint(values...)
	t.saveOutputToBucket(str)
	return fmt.Println(str)
}

func (t *TeePrinter) DisableTerminalOutput(disable bool) {
	t.disableTerminalOutput = disable
}

func (t *TeePrinter) saveOutputToBucket(output string) {
	if t.outputBucket == nil {
		return
	}

	*t.outputBucket = append(*t.outputBucket, Decolorize(output))
}