cops: Display.Set: support nil colors

borkshop · Dec 18, 2017 · 6559a46 · 6559a46
1 parent 80e26fb
commit 6559a46
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/internal/cops/display/display.go b/internal/cops/display/display.go
@@ -81,8 +81,12 @@ func (d *Display) Clear(r image.Rectangle) {
 // the given position.
 func (d *Display) Set(x, y int, t string, f, b color.Color) {
 	d.Text.Set(x, y, t)
-	d.Foreground.Set(x, y, f)
-	d.Background.Set(x, y, b)
+	if f != nil {
+		d.Foreground.Set(x, y, f)
+	}
+	if b != nil {
+		d.Background.Set(x, y, b)
+	}
 }
 
 // SetRGBA is a faster Set.