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")
	}
}