如何在Go语言中编程检测当前终端是否支持16色或256色

检测逻辑核心思路 #

终端色彩支持的检测不需要依赖色彩渲染包自身逻辑，可按以下优先级判断，天然兼容Windows、Linux、macOS全平台：

• 先判断当前输出目标是否为真实终端：如果输出被重定向到文件、管道，默认不启用彩色输出，可借助go-isatty包的isatty.IsTerminal()、isatty.IsCygwinTerminal()方法实现
• 遵循行业通用NO_COLOR约定：只要环境变量中存在NO_COLOR，无论终端能力如何都返回不支持彩色
• 通过环境变量判断色彩等级：
  • 若COLORTERM环境变量值为24bit或truecolor，说明终端支持真彩色，自然向下兼容256色、16色
  • 若TERM环境变量后缀为256color（如xterm-256color、screen-256color），判定支持256色
  • 若TERM为xterm、vt100、linux、screen等基础终端类型，或Windows系统下未显式设置TERM变量，默认至少支持16色
• Windows系统额外适配：Windows 10 1511及以上版本的cmd、PowerShell默认支持256色，可通过调用Win32 API检测控制台是否支持VT虚拟终端序列做更精确判断

简化版实现代码 #

package main

import (
	"os"
	"strings"

	"github.com/mattn/go-isatty"
)

// 色彩支持等级枚举
const (
	ColorNone = iota // 不支持彩色
	Color16          // 支持16色
	Color256         // 支持256色
	ColorTrueColor   // 支持真彩色
)

// DetectColorSupport 检测当前终端的色彩支持能力
func DetectColorSupport() int {
	// 非终端输出直接返回无色彩
	if !isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd()) {
		return ColorNone
	}

	// 遵循NO_COLOR约定
	if _, hasNoColor := os.LookupEnv("NO_COLOR"); hasNoColor {
		return ColorNone
	}

	// 检测真彩色支持
	colorTermVal := os.Getenv("COLORTERM")
	if colorTermVal == "truecolor" || colorTermVal == "24bit" {
		return ColorTrueColor
	}

	// 检测TERM环境变量
	termVal := os.Getenv("TERM")
	if termVal == "" {
		// Windows环境无TERM变量默认至少支持16色
		if os.Getenv("WINDIR") != "" {
			return Color16
		}
		return ColorNone
	}

	// 判断256色支持
	if strings.HasSuffix(termVal, "256color") {
		return Color256
	}

	// 判断16色支持
	baseTermSet := map[string]bool{
		"xterm":  true,
		"vt100":  true,
		"linux":  true,
		"screen": true,
		"rxvt":   true,
	}
	if baseTermSet[termVal] {
		return Color16
	}

	return ColorNone
}

现成工具复用 #

你提到的几款主流色彩渲染包都已经内置了成熟的跨平台检测逻辑，不需要自己手动实现：

• fatih/color提供了color.SupportColors()、color.Support256Colors()方法直接返回16色、256色的支持结果
• gookit/color提供了color.SupportColor()、color.Support256Color()、color.SupportTrueColor()三个方法对应三个色彩等级的检测结果

内容的提问来源于stack exchange，提问作者Wu Shanhe