KumaChan/standalone/util/richtext/ansi.go

package richtext

import (
	"os"
	"runtime"
)


type AnsiPalette (func(string) string)
var defaultPalette = (func() AnsiPalette {
	if runtime.GOOS == "windows" {
		var _, is_mintty = os.LookupEnv("TERM")
		var _, is_wt = os.LookupEnv("WT_SESSION")
		if is_mintty || is_wt {
			return lightPalette
		} else {
			return nil
		}
	} else {
		return lightPalette
	}
})()

const bold = "\033[1m"
const red = "\033[31m"
const green = "\033[32m"
const orange = "\033[33m"
const blue = "\033[34m"
const magenta = "\033[35m"
const cyan = "\033[36m"
const reset = "\033[0m"

func LightPalette() AnsiPalette { return lightPalette }
var lightPalette AnsiPalette = func(tag string) string {
	switch tag {
	case TAG_EM:         return bold
	case TAG_HIGHLIGHT:  return (bold + red)
	case TAG_ERR_NORMAL: return bold
	case TAG_ERR_INLINE: return (bold + red)
	case TAG_ERR_NOTE:   return (bold + magenta)
	default: return ""
	}
}

func DarkPalette() AnsiPalette { return darkPalette }
var darkPalette AnsiPalette = func(tag string) string {
	switch tag {
	case TAG_EM:         return bold
	case TAG_HIGHLIGHT:  return (bold + orange)
	case TAG_ERR_NORMAL: return bold
	case TAG_ERR_INLINE: return (bold + orange)
	case TAG_ERR_NOTE:   return (bold + cyan)
	default: return ""
	}
}