console/output/output.go

package output

import (
	"context"
	"fmt"
	"io"
	"os"

	"gitoa.ru/go-4devs/console/output/label"
	"gitoa.ru/go-4devs/console/output/verbosity"
)

func writeError(_ int, err error) {
	if err != nil {
		fmt.Fprint(os.Stderr, err)
	}
}

type Output func(ctx context.Context, verb verbosity.Verbosity, msg string, args ...label.KeyValue) (int, error)

func (o Output) Print(ctx context.Context, args ...interface{}) {
	writeError(o(ctx, verbosity.Norm, fmt.Sprint(args...)))
}

func (o Output) PrintKV(ctx context.Context, msg string, kv ...label.KeyValue) {
	writeError(o(ctx, verbosity.Norm, msg, kv...))
}

func (o Output) Printf(ctx context.Context, format string, args ...interface{}) {
	writeError(o(ctx, verbosity.Norm, fmt.Sprintf(format, args...)))
}

func (o Output) Println(ctx context.Context, args ...interface{}) {
	writeError(o(ctx, verbosity.Norm, fmt.Sprintln(args...)))
}

func (o Output) Info(ctx context.Context, args ...interface{}) {
	writeError(o(ctx, verbosity.Info, fmt.Sprint(args...)))
}

func (o Output) InfoKV(ctx context.Context, msg string, kv ...label.KeyValue) {
	writeError(o(ctx, verbosity.Info, msg, kv...))
}

func (o Output) Debug(ctx context.Context, args ...interface{}) {
	writeError(o(ctx, verbosity.Debug, fmt.Sprint(args...)))
}

func (o Output) DebugKV(ctx context.Context, msg string, kv ...label.KeyValue) {
	writeError(o(ctx, verbosity.Debug, msg, kv...))
}

func (o Output) Trace(ctx context.Context, args ...interface{}) {
	writeError(o(ctx, verbosity.Trace, fmt.Sprint(args...)))
}

func (o Output) TraceKV(ctx context.Context, msg string, kv ...label.KeyValue) {
	writeError(o(ctx, verbosity.Trace, msg, kv...))
}

func (o Output) Write(b []byte) (int, error) {
	return o(context.Background(), verbosity.Norm, string(b))
}

func (o Output) Writer(ctx context.Context, verb verbosity.Verbosity) io.Writer {
	return verbosityWriter(func(b []byte) (int, error) {
		return o(ctx, verb, string(b))
	})
}

type verbosityWriter func(b []byte) (int, error)

func (w verbosityWriter) Write(b []byte) (int, error) {
	return w(b)
}