sq/cli/output/jsonw/errorwriter.go

package jsonw

import (
	"fmt"
	"io"
	"log/slog"

	"github.com/neilotoole/sq/libsq/core/errz"

	"github.com/neilotoole/sq/cli/output"
)

// errorWriter implements output.ErrorWriter.
type errorWriter struct {
	log *slog.Logger
	out io.Writer
	pr  *output.Printing
}

// NewErrorWriter returns an output.ErrorWriter that outputs in JSON.
func NewErrorWriter(log *slog.Logger, out io.Writer, pr *output.Printing) output.ErrorWriter {
	return &errorWriter{log: log, out: out, pr: pr}
}

// Error implements output.ErrorWriter.
func (w *errorWriter) Error(err error) {
	var errMsg string
	var stack []string

	if err == nil {
		errMsg = "nil error"
	} else {
		errMsg = err.Error()
		if w.pr.Verbose {
			for _, st := range errz.Stack(err) {
				s := fmt.Sprintf("%+v", st)
				stack = append(stack, s)
			}
		}
	}

	t := struct {
		Error string   `json:"error"`
		Stack []string `json:"stack,omitempty"`
	}{
		Error: errMsg,
		Stack: stack,
	}

	pr := w.pr.Clone()
	pr.String = pr.Error

	_ = writeJSON(w.out, pr, t)
}