fix(parse): strip OSC escape sequences in StripAnsi for backwards compat
This commit is contained in:
@@ -0,0 +1,68 @@
|
||||
package parse
|
||||
|
||||
import (
|
||||
"io"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestStripAnsi_CSI(t *testing.T) {
|
||||
input := "Hello \033[31mred\033[0m world\n"
|
||||
r := StripAnsi(strings.NewReader(input))
|
||||
out, _ := io.ReadAll(r)
|
||||
if string(out) != "Hello red world\n" {
|
||||
t.Errorf("got %q", string(out))
|
||||
}
|
||||
}
|
||||
|
||||
func TestStripAnsi_OSC9ImageMarker(t *testing.T) {
|
||||
input := "Before\n\033]9;image:path=/tmp/img.png;alt=photo\007\nAfter\n"
|
||||
r := StripAnsi(strings.NewReader(input))
|
||||
out, _ := io.ReadAll(r)
|
||||
expected := "Before\n\nAfter\n"
|
||||
if string(out) != expected {
|
||||
t.Errorf("got %q, want %q", string(out), expected)
|
||||
}
|
||||
}
|
||||
|
||||
func TestStripAnsi_OSCWithEscBackslash(t *testing.T) {
|
||||
input := "Text \033]9;image:path=/tmp/x.png\033\\ more\n"
|
||||
r := StripAnsi(strings.NewReader(input))
|
||||
out, _ := io.ReadAll(r)
|
||||
if strings.Contains(string(out), "\033") {
|
||||
t.Errorf("OSC not stripped: %q", string(out))
|
||||
}
|
||||
}
|
||||
|
||||
func TestStripAnsi_OSC8Hyperlink(t *testing.T) {
|
||||
input := "\033]8;;https://example.com\033\\click\033]8;;\033\\\n"
|
||||
r := StripAnsi(strings.NewReader(input))
|
||||
out, _ := io.ReadAll(r)
|
||||
if strings.Contains(string(out), "\033") {
|
||||
t.Errorf("OSC 8 not stripped: %q", string(out))
|
||||
}
|
||||
if !strings.Contains(string(out), "click") {
|
||||
t.Error("visible text lost")
|
||||
}
|
||||
}
|
||||
|
||||
func TestStripAnsi_PlainText(t *testing.T) {
|
||||
input := "No escape sequences here\n"
|
||||
r := StripAnsi(strings.NewReader(input))
|
||||
out, _ := io.ReadAll(r)
|
||||
if string(out) != input {
|
||||
t.Errorf("got %q", string(out))
|
||||
}
|
||||
}
|
||||
|
||||
func TestStripAnsi_MixedCSIAndOSC(t *testing.T) {
|
||||
input := "\033[1mBold\033[0m \033]9;image:path=/tmp/a.png\007 text\n"
|
||||
r := StripAnsi(strings.NewReader(input))
|
||||
out, _ := io.ReadAll(r)
|
||||
if strings.Contains(string(out), "\033") {
|
||||
t.Errorf("escape sequences remain: %q", string(out))
|
||||
}
|
||||
if !strings.Contains(string(out), "Bold") || !strings.Contains(string(out), "text") {
|
||||
t.Error("visible text lost")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user