mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
35 lines
681 B
Go
35 lines
681 B
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
char = "&"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Colors: http://www.lihaoyi.com/post/BuildyourownCommandLinewithANSIescapecodes.html
|
||
|
|
var re = regexp.MustCompile(char + `(?m)([0-9]{1,2};[0-9]{1,2}|[0-9]{1,2})`)
|
||
|
|
|
||
|
|
func Color(str string) string {
|
||
|
|
result := ""
|
||
|
|
lastIndex := 0
|
||
|
|
|
||
|
|
for _, v := range re.FindAllSubmatchIndex([]byte(str), -1) {
|
||
|
|
groups := []string{}
|
||
|
|
for i := 0; i < len(v); i += 2 {
|
||
|
|
groups = append(groups, str[v[i]:v[i+1]])
|
||
|
|
}
|
||
|
|
|
||
|
|
result += str[lastIndex:v[0]] + "\033[" + groups[1] + "m"
|
||
|
|
lastIndex = v[1]
|
||
|
|
}
|
||
|
|
|
||
|
|
return result + str[lastIndex:]
|
||
|
|
}
|
||
|
|
|
||
|
|
func Colorf(format string, a ...interface{}) string {
|
||
|
|
return fmt.Sprintf(Color(format), a...)
|
||
|
|
}
|