优化阅读功能

internal/service/reader/rule/url_binding_test.go

@@ -0,0 +1,149 @@
+package rule
+
+import (
+	"strings"
+	"testing"
+)
+
+// 本文件：URL query 规范化 与 规则 JS 上下文绑定的回归测试。
+//
+// 背景（光遇聚合）：
+//   - 书源把参数拼进 query 后交给 java.ajax，且 POST 也把参数放在 query 里；
+//     MeBox 早期只对 GET 重编码 query，POST 的原生中文/引号会直接写进请求行，
+//     服务端回 400/空响应，书源判定「线路报错」并把全部线路试一遍。
+//   - 书源用 String(baseUrl).startsWith("data:") 判断当前层是不是自搭的参数信封，
+//     所以规则 JS 的 baseUrl 必须是「当前处理的页面地址」，不能固定成书源地址。
+
+// TestNormalizeQueryEncodesNonASCII 非 ASCII 与非法字符必须被百分号编码。
+func TestNormalizeQueryEncodesNonASCII(t *testing.T) {
+	raw := `https://example.com/detail?id=abc&source=番茄&variable={"custom":""}`
+	got := normalizeQuery(raw, "")
+	if strings.Contains(got, "番茄") {
+		t.Fatalf("中文未被编码: %s", got)
+	}
+	if !strings.Contains(got, "source=%E7%95%AA%E8%8C%84") {
+		t.Fatalf("中文编码结果异常: %s", got)
+	}
+	if strings.Contains(got, `{"custom":""}`) {
+		t.Fatalf("引号未被编码: %s", got)
+	}
+	// 分隔符与等号必须保留，否则参数结构会散掉
+	if !strings.Contains(got, "&") || !strings.Contains(got, "=") {
+		t.Fatalf("query 结构被破坏: %s", got)
+	}
+}
+
+// TestNormalizeQueryKeepsEncoded 已编码好的 query 原样保留，不二次编码。
+func TestNormalizeQueryKeepsEncoded(t *testing.T) {
+	raw := "https://example.com/s?q=%E5%AE%A0%E9%AD%85&page=1"
+	if got := normalizeQuery(raw, ""); got != raw {
+		t.Fatalf("已编码 query 被改动:\n got=%s\nwant=%s", got, raw)
+	}
+}
+
+// TestNormalizeQueryLeavesDataURI data: 地址的载荷不是 query，不能动。
+func TestNormalizeQueryLeavesDataURI(t *testing.T) {
+	raw := `data:;base64,eyJhIjoxfQ==`
+	if got := normalizeQuery(raw, ""); got != raw {
+		t.Fatalf("data 地址被改动: %s", got)
+	}
+}
+
+// TestParseAnalyzeUrlEncodesPOSTQuery POST 的 query 也要编码。
+// 回归：之前只有 GET 分支做重编码，POST 的原生中文直接上线。
+func TestParseAnalyzeUrlEncodesPOSTQuery(t *testing.T) {
+	raw := `https://example.com/detail?source=番茄,{"method":"POST","headers":{"Content-Type":"application/json"},"body":"{\"html\":\"\"}"}`
+	req, err := ParseAnalyzeUrlWithJS(raw, "", 0, "https://example.com", NewJSRunner(JSConfig{}))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if req.Method != "POST" {
+		t.Fatalf("method = %q", req.Method)
+	}
+	if strings.Contains(req.URL, "番茄") {
+		t.Fatalf("POST query 未编码: %s", req.URL)
+	}
+	if !strings.Contains(req.URL, "source=%E7%95%AA%E8%8C%84") {
+		t.Fatalf("POST query 编码异常: %s", req.URL)
+	}
+	if req.Body != `{"html":""}` {
+		t.Fatalf("body 被改动: %q", req.Body)
+	}
+}
+
+// TestRuleJSBaseUrlPrefersPageURL 规则 JS 的 baseUrl 应是「当前页面地址」。
+//
+// 书源用 String(baseUrl).startsWith("data:") 判断当前层是不是自搭的参数信封；
+// 若把 baseUrl 固定成书源地址，书源会走 else 分支把 hex 原文当结果返回。
+func TestRuleJSBaseUrlPrefersPageURL(t *testing.T) {
+	const pageURL = `data:;base64,eyJhIjoxfQ==`
+	r := NewJSRunner(JSConfig{BaseURL: "https://source.example.com"})
+	ar := NewAnalyzeRule()
+	ar.SetContent("body", pageURL)
+
+	v, err := r.Run(ar, `baseUrl`, nil, "")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if got := anyToString(v); got != pageURL {
+		t.Fatalf("baseUrl = %q，期望页面地址 %q", got, pageURL)
+	}
+	if !strings.HasPrefix(anyToString(v), "data:") {
+		t.Fatal("书源的 data: 分支判断会失效")
+	}
+}
+
+// TestRuleJSBaseUrlFallsBackToSource 页面地址为空时回退书源地址。
+func TestRuleJSBaseUrlFallsBackToSource(t *testing.T) {
+	r := NewJSRunner(JSConfig{BaseURL: "https://source.example.com"})
+	v, err := r.Run(NewAnalyzeRule(), `baseUrl`, nil, "")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if got := anyToString(v); got != "https://source.example.com" {
+		t.Fatalf("baseUrl = %q", got)
+	}
+}
+
+// TestRuleJSBookObject 规则 JS 的 book 对象要具备书源依赖的成员。
+//
+// 回归：早期只绑了 {"name": ...}，书源一碰 book.setUseReplaceRule(false)
+// 就 TypeError，整段详情/目录规则 JS 直接失败。
+func TestRuleJSBookObject(t *testing.T) {
+	state := NewMemoryState()
+	r := NewJSRunner(JSConfig{State: state})
+	ar := NewAnalyzeRule()
+	ar.SetBookContext("宠魅", nil)
+	ar.SetChapterContext("第1集", nil)
+	ar.SetChapterIndex(0)
+
+	js := `(function(){
+	  book.setUseReplaceRule(false);              // 早期会 TypeError
+	  var v = String(book.getVariable('custom')); // 缺省必须是 ""（不是 "null"）
+	  book.type = 32;                             // 听书源这样声明类型
+	  book.imageStyle = 'TEXT';
+	  return book.name + '|' + v + '|' + book.type + '|' + chapter.title;
+	})()`
+	v, err := r.Run(ar, js, nil, "")
+	if err != nil {
+		t.Fatalf("book 对象成员缺失: %v", err)
+	}
+	if got := anyToString(v); got != "宠魅||32|第1集" {
+		t.Fatalf("book 行为异常: %q", got)
+	}
+	// 书源声明的类型要被服务层读回
+	if bt, ok := ar.BookTypeOverride(); !ok || bt != 32 {
+		t.Fatalf("BookTypeOverride = %v/%v，期望 32", bt, ok)
+	}
+	// putVariable 写回后 getVariable 能读到
+	if _, err := r.Run(ar, `book.putVariable('custom','v1')`, nil, ""); err != nil {
+		t.Fatal(err)
+	}
+	v, err = r.Run(ar, `String(book.getVariable('custom'))`, nil, "")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if got := anyToString(v); got != "v1" {
+		t.Fatalf("putVariable 后 getVariable = %q", got)
+	}
+}