fix: do not colorize listing if color is disabled

Author: tdakkota

cmd/ogen/main.go

@@ -92,15 +92,15 @@ func generate(data []byte, packageName, targetDir string, clean bool, opts gen.O
 	return nil
 }
 
-func handleGenerateError(w io.Writer, specPath string, data []byte, err error) (r bool) {
+func handleGenerateError(w io.Writer, color bool, specPath string, data []byte, err error) (r bool) {
 	defer func() {
 		// Add trailing newline to the error message if error is handled.
 		if r {
 			_, _ = fmt.Fprintln(w)
 		}
 	}()
 
-	if location.PrintPrettyError(w, specPath, data, err) {
+	if location.PrintPrettyError(w, color, specPath, data, err) {
 		return true
 	}
 
@@ -287,7 +287,7 @@ func run() error {
 	}
 
 	if err := generate(data, *packageName, *targetDir, *clean, opts); err != nil {
-		if handleGenerateError(os.Stderr, fileName, data, err) {
+		if handleGenerateError(os.Stderr, logOptions.Color, fileName, data, err) {
 			return errors.New("generation failed")
 		}
 		return errors.Wrap(err, "generate")

gen_test.go

@@ -169,7 +169,7 @@ func TestNegative(t *testing.T) {
 		a.Error(err)
 
 		var buf strings.Builder
-		if location.PrintPrettyError(&buf, name, data, err) {
+		if location.PrintPrettyError(&buf, true, name, data, err) {
 			t.Log(buf.String())
 		} else {
 			t.Logf("%+v", err)

internal/location/pretty_error.go

@@ -9,7 +9,7 @@ import (
 )
 
 // PrintPrettyError prints the error in a pretty way and returns true if it was printed successfully.
-func PrintPrettyError(w io.Writer, filename string, data []byte, err error) bool {
+func PrintPrettyError(w io.Writer, color bool, filename string, data []byte, err error) bool {
 	// TODO(tdakkota): make it configurable?
 	const (
 		printLimit   = 5
@@ -24,6 +24,9 @@ func PrintPrettyError(w io.Writer, filename string, data []byte, err error) bool
 			Filename: filename,
 			Context:  context,
 		}
+		if !color {
+			opts = opts.WithoutColor()
+		}
 		_ = lines.PrintListing(w, msg, loc, opts)
 	}
 

internal/location/print.go

@@ -47,6 +47,13 @@ type PrintListingOptions struct {
 	PlainColor *color.Color
 }
 
+// WithoutColor creates a copy of the options with disabled color.
+func (o PrintListingOptions) WithoutColor() PrintListingOptions {
+	o.ErrColor = color.New(color.Reset)
+	o.PlainColor = color.New(color.Reset)
+	return o
+}
+
 func (o PrintListingOptions) contextLines(errLine int) (padNum, top, bottom int) {
 	context := o.Context
 

jsonschema/jsonschema_test.go

@@ -38,7 +38,7 @@ func TestNegative(t *testing.T) {
 		a.Error(err)
 
 		var buf strings.Builder
-		ok := location.PrintPrettyError(&buf, name, data, err)
+		ok := location.PrintPrettyError(&buf, true, name, data, err)
 		// Ensure that the error message is pretty printed.
 		//
 		// There should be a good reason to remove this line.

openapi/parser/parser_test.go

@@ -36,7 +36,7 @@ func TestNegative(t *testing.T) {
 		a.Error(err)
 
 		var buf strings.Builder
-		ok := location.PrintPrettyError(&buf, name, data, err)
+		ok := location.PrintPrettyError(&buf, true, name, data, err)
 		// Ensure that the error message is pretty printed.
 		//
 		// There should be a good reason to remove this line.