package toolcall

import (
	"strings"
	"testing"
)

// 4 反引号嵌套 3 反引号
func TestStripFencedCodeBlocks_NestedFourBackticks(t *testing.T) {
	text := "Before\n\x60\x60\x60\x60markdown\nHere is \x60\x60\x60 nested \x60\x60\x60 example\n\x60\x60\x60\x60\nAfter"
	got := stripFencedCodeBlocks(text)
	if !strings.Contains(got, "Before") || !strings.Contains(got, "After") {
		t.Fatalf("expected Before and After preserved, got %q", got)
	}
	if strings.Contains(got, "nested") {
		t.Fatalf("expected nested content stripped, got %q", got)
	}
}

// 波浪线围栏
func TestStripFencedCodeBlocks_TildeFence(t *testing.T) {
	text := "Before\n~~~python\ncode here\n~~~\nAfter"
	got := stripFencedCodeBlocks(text)
	if !strings.Contains(got, "Before") || !strings.Contains(got, "After") {
		t.Fatalf("expected Before/After, got %q", got)
	}
	if strings.Contains(got, "code here") {
		t.Fatalf("expected code stripped, got %q", got)
	}
}

// 未闭合围栏 + 后面跟真正的工具调用：不应返回空字符串
func TestStripFencedCodeBlocks_UnclosedFencePreservesToolCall(t *testing.T) {
	text := "Example:\n\x60\x60\x60xml\n<tool_calls><invoke name=\"read_file\"><parameter name=\"path\">README.md</parameter></invoke></tool_calls>\n\n<tool_calls><invoke name=\"search\"><parameter name=\"q\">go</parameter></invoke></tool_calls>"
	got := stripFencedCodeBlocks(text)
	if got == "" {
		t.Fatalf("unclosed fence should not truncate everything — real tool call after the fence is lost")
	}
}

// CDATA 内的围栏不应被剥离
func TestStripFencedCodeBlocks_FenceInsideCDATA(t *testing.T) {
	text := "<tool_calls><invoke name=\"write\">\n<parameter name=\"content\"><![CDATA[\n\x60\x60\x60python\nprint('hello')\n\x60\x60\x60\n]]></parameter>\n</invoke></tool_calls>"
	got := stripFencedCodeBlocks(text)
	if !strings.Contains(got, "\x60\x60\x60python") {
		t.Fatalf("fenced code inside CDATA should be preserved, got %q", got)
	}
}

// 连续多个围栏
func TestStripFencedCodeBlocks_MultipleFences(t *testing.T) {
	text := "Before\n\x60\x60\x60\nfence1\n\x60\x60\x60\nMiddle\n\x60\x60\x60\nfence2\n\x60\x60\x60\nAfter"
	got := stripFencedCodeBlocks(text)
	if !strings.Contains(got, "Before") || !strings.Contains(got, "Middle") || !strings.Contains(got, "After") {
		t.Fatalf("expected non-fenced content preserved, got %q", got)
	}
}

// 围栏包含内嵌 ``` 行但没有独立成行
func TestStripFencedCodeBlocks_InlineBackticksNotFence(t *testing.T) {
	text := "Before\n\x60\x60\x60go\nfmt.Println(\x60\x60\x60hello\x60\x60\x60)\n\x60\x60\x60\nAfter"
	got := stripFencedCodeBlocks(text)
	if !strings.Contains(got, "Before") || !strings.Contains(got, "After") {
		t.Fatalf("expected Before/After, got %q", got)
	}
}