mirror of
https://github.com/hanzoai/numscript.git
synced 2026-10-30 18:26:57 +00:00
Alessandro ScandoneandGitHub
5dfc34cd3c
* feat: feature flag syntax * feat: change syntax * feat: reject invalid features * feat: add static check of unknown features * feat: more defensive args handling * fix: fix typo
96 lines
1.5 KiB
Go
96 lines
1.5 KiB
Go
package ansi
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
const resetCol = "\033[0m"
|
|
|
|
func Compose(cols ...func(string) string) func(string) string {
|
|
return func(s string) string {
|
|
for _, mod := range cols {
|
|
s = mod(s)
|
|
}
|
|
return s
|
|
}
|
|
}
|
|
|
|
func replaceLast(s, oldStr, newStr string) string {
|
|
lastIndex := strings.LastIndex(s, oldStr)
|
|
if lastIndex == -1 {
|
|
return s
|
|
}
|
|
return s[:lastIndex] + newStr + s[lastIndex+len(oldStr):]
|
|
}
|
|
|
|
func col(s string, code int) string {
|
|
colorCode := fmt.Sprintf("\033[%dm", code)
|
|
// This trick should allow to stack colors (TODO test)
|
|
s = replaceLast(s, resetCol, resetCol+colorCode)
|
|
return colorCode + s + resetCol
|
|
}
|
|
|
|
func ColorRed(s string) string {
|
|
return col(s, 31)
|
|
}
|
|
|
|
func ColorWhite(s string) string {
|
|
return col(s, 37)
|
|
}
|
|
|
|
func ColorGreen(s string) string {
|
|
return col(s, 32)
|
|
}
|
|
|
|
func ColorYellow(s string) string {
|
|
return col(s, 33)
|
|
}
|
|
|
|
func ColorCyan(s string) string {
|
|
return col(s, 36)
|
|
}
|
|
|
|
func ColorLight(s string) string {
|
|
return col(s, 97) // Bright white → light
|
|
}
|
|
|
|
func ColorBrightBlack(s string) string {
|
|
return col(s, 90)
|
|
}
|
|
|
|
func ColorBrightRed(s string) string {
|
|
return col(s, 91)
|
|
}
|
|
|
|
func ColorBrightGreen(s string) string {
|
|
return col(s, 92)
|
|
}
|
|
|
|
func ColorBrightYellow(s string) string {
|
|
return col(s, 93)
|
|
}
|
|
|
|
// BG
|
|
func BgDark(s string) string {
|
|
return col(s, 100)
|
|
}
|
|
|
|
func BgRed(s string) string {
|
|
return col(s, 41)
|
|
}
|
|
|
|
func BgGreen(s string) string {
|
|
return col(s, 42)
|
|
}
|
|
|
|
// modifiers
|
|
|
|
func Bold(s string) string {
|
|
return col(s, 1)
|
|
}
|
|
|
|
func Underline(s string) string {
|
|
return col(s, 4)
|
|
}
|