chore: tidy and fix some lint issues

charmbracelet · Oct 31, 2024 · 68e6666 · 68e6666
1 parent 5de2f41
commit 68e6666
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 7 deletions.
diff --git a/env.go b/env.go
@@ -161,7 +161,8 @@ func envColorProfile(env map[string]string) (p Profile) {
 		p = ANSI
 	}
 
-	if ti, err := terminfo.Load(term); err == nil {
+	ti, err := terminfo.Load(term)
+	if err == nil {
 		extbools := ti.ExtBoolCapsShort()
 		if _, ok := extbools["RGB"]; ok {
 			p = TrueColor

diff --git a/profile.go b/profile.go
@@ -12,15 +12,15 @@ import (
 type Profile byte
 
 const (
-	// TrueColor, 24-bit color profile
+	// TrueColor, 24-bit color profile.
 	TrueColor Profile = iota
-	// ANSI256, 8-bit color profile
+	// ANSI256, 8-bit color profile.
 	ANSI256
-	// ANSI, 4-bit color profile
+	// ANSI, 4-bit color profile.
 	ANSI
-	// Ascii, uncolored profile
-	Ascii // nolint: revive
-	// NoTTY, not a terminal profile
+	// Ascii, uncolored profile.
+	Ascii //nolint:revive
+	// NoTTY, not a terminal profile.
 	NoTTY
 )
 

diff --git a/writer.go b/writer.go
@@ -40,8 +40,13 @@ func (w *Writer) Write(p []byte) (int, error) {
 		return w.Forward.Write(p)
 	case NoTTY:
 		return io.WriteString(w.Forward, ansi.Strip(string(p)))
+	default:
+		return w.downsample(p)
 	}
+}
 
+// downsample downgrades the given text to the appropriate color profile.
+func (w *Writer) downsample(p []byte) (int, error) {
 	var buf bytes.Buffer
 	var state byte