✨ feat: Add TrimAnsi method to remove possible Ansi encoding in logs

sohaha · Apr 9, 2024 · 31c80f6 · 31c80f6
1 parent ade56e8
commit 31c80f6
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/zlog/color.go b/zlog/color.go
@@ -165,3 +165,8 @@ func init() {
 func isSupportColor() bool {
 	return !DisableColor && IsSupportColor()
 }
+
+func TrimAnsi(str string) string {
+	str, _ = zstring.RegexReplace(`\x1b\[[0-9;]*[a-zA-Z]`, str, "")
+	return str
+}
diff --git a/zlog/color_test.go b/zlog/color_test.go
@@ -26,3 +26,13 @@ func TestColor(t *testing.T) {
 	OutAllColor()
 	T.Equal(testText, ColorTextWrap(ColorGreen, testText))
 }
+
+func TestTrimAnsi(t *testing.T) {
+	tt := zls.NewTest(t)
+
+	testText := "ok\x1b[31m"
+	tt.Equal("ok", TrimAnsi(testText))
+
+	testText = ColorTextWrap(ColorGreen, "ok")
+	tt.Equal("ok", TrimAnsi(testText))
+}