add --color override (always/never/auto) (#38)

- auto: current behavior (don't show color of TERM=dumb or not a tty)
- always: always output colors
- never: never outputs colors

Author: oyarsa

config.go

@@ -62,8 +62,9 @@ type Config struct {
 	Format     string
 	FileOutput string
 	Verbose    bool
-	NoSyntax   bool // disable syntax highlighting
-	Dedup      bool // collapse byte-identical matches
+	NoSyntax   bool   // disable syntax highlighting
+	Dedup      bool   // collapse byte-identical matches
+	Color      string // color mode: auto, always, never
 
 	// MCP
 	MCPServer bool

console.go

@@ -58,15 +58,23 @@ func ConsoleSearch(cfg *Config) {
 }
 
 func formatDefault(cfg *Config, results []*common.FileJob) {
-	noColor := os.Getenv("TERM") == "dumb" ||
-		(!isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd()))
+	var noColor bool
+	switch cfg.Color {
+	case "always":
+		noColor = false
+	case "never":
+		noColor = true
+	default: // "auto"
+		noColor = os.Getenv("TERM") == "dumb" ||
+			(!isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd()))
+	}
+	color.NoColor = noColor
 
 	fmtBegin := "\033[1;31m"
 	fmtEnd := "\033[0m"
 	if noColor {
 		fmtBegin = ""
 		fmtEnd = ""
-		color.NoColor = true
 	}
 
 	documentFrequency := ranker.CalculateDocumentTermFrequency(results)

main.go

@@ -313,6 +313,12 @@ func main() {
 		false,
 		"disable syntax highlighting in output",
 	)
+	flags.StringVar(
+		&cfg.Color,
+		"color",
+		"auto",
+		"color output mode [auto, always, never]",
+	)
 	flags.Float64Var(
 		&cfg.WeightCode,
 		"weight-code",

tui.go

@@ -15,6 +15,7 @@ import (
 	"github.com/charmbracelet/bubbles/textinput"
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
+	"github.com/muesli/termenv"
 )
 
 // searchResult represents a single search result
@@ -127,6 +128,13 @@ type model struct {
 }
 
 func initialModel(cfg *Config) model {
+	switch cfg.Color {
+	case "always":
+		lipgloss.SetColorProfile(termenv.ANSI256)
+	case "never":
+		lipgloss.SetColorProfile(termenv.Ascii)
+	}
+
 	si := textinput.New()
 	si.Placeholder = "search query..."
 	si.Prompt = "> "