mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-09-19 21:17:31 -07:00
50 lines
1.5 KiB
Go
Executable file
50 lines
1.5 KiB
Go
Executable file
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"gopkg.in/gookit/color.v1"
|
|
)
|
|
|
|
//ColorWriter writes colorized strings
|
|
type ColorWriter struct {
|
|
Buffer *bytes.Buffer
|
|
}
|
|
|
|
func (w *ColorWriter) writeColoredText(background string, foreground string, text string) {
|
|
style := color.HEXStyle(foreground, background)
|
|
text = style.Sprint(text)
|
|
w.Buffer.WriteString(text)
|
|
}
|
|
|
|
func (w *ColorWriter) writeAndRemoveText(background string, foreground string, text string, textToRemove string, parentText string) string {
|
|
w.writeColoredText(background, foreground, text)
|
|
return strings.Replace(parentText, textToRemove, "", 1)
|
|
}
|
|
|
|
func (w *ColorWriter) write(background string, foreground string, text string) {
|
|
style := color.HEXStyle(foreground, background)
|
|
text = style.Sprint(text)
|
|
|
|
r := regexp.MustCompile(`<\s*(?P<color>#[A-Fa-f0-9]{6}|[A-Fa-f0-9]{3})>(?P<text>.*?)<\s*/\s*>`)
|
|
match := r.FindAllStringSubmatch(text, -1)
|
|
for i := range match {
|
|
// get the text before the color override and write that first
|
|
textBeforeColorOverride := strings.Split(text, match[i][0])[0]
|
|
text = w.writeAndRemoveText(background, foreground, textBeforeColorOverride, textBeforeColorOverride, text)
|
|
text = w.writeAndRemoveText(background, match[i][1], match[i][2], match[i][0], text)
|
|
}
|
|
// color the remaining part of text with background and foreground
|
|
w.writeColoredText(background, foreground, text)
|
|
}
|
|
|
|
func (w *ColorWriter) string() string {
|
|
return w.Buffer.String()
|
|
}
|
|
|
|
func (w *ColorWriter) reset() {
|
|
w.Buffer.Reset()
|
|
}
|