开发

internal/service/reader/rule/jsobjects.go

@@ -3,8 +3,8 @@ package rule
 import (
 	"encoding/base64"
 	"encoding/hex"
+	"encoding/json"
 	"fmt"
-	"net/url"
 	"strings"
 	"sync"
 	"time"
@@ -61,88 +61,240 @@ func hexEncodeBytes(b []byte) string          { return hex.EncodeToString(b) }
 
 // ─── cookie 对象（对应 legado CookieStore 注入的 `cookie`） ─────────────────
 
-var cookieJar = struct {
-	mu sync.Mutex
-	m  map[string]map[string]string // host → name → value
-}{m: map[string]map[string]string{}}
+// 会话状态由服务层注入（落库）；未注入时用进程内 MemoryState。
+// 见 state.go —— 旧的全局 cookieJar 因无法按书源隔离且不落库已废弃。
 
-// CookieJarRecord 供服务层在执行 HTTP 请求时记录 Set-Cookie。
-func CookieJarRecord(rawURL string, cookies []string) {
-	u, err := url.Parse(rawURL)
-	if err != nil {
-		return
-	}
-	host := u.Host
-	cookieJar.mu.Lock()
-	defer cookieJar.mu.Unlock()
-	jar, ok := cookieJar.m[host]
-	if !ok {
-		jar = map[string]string{}
-		cookieJar.m[host] = jar
-	}
-	for _, c := range cookies {
-		pair := strings.SplitN(c, ";", 2)[0]
-		kv := strings.SplitN(pair, "=", 2)
-		if len(kv) != 2 {
-			continue
-		}
-		name := strings.TrimSpace(kv[0])
-		if name == "" {
-			continue
-		}
-		jar[name] = strings.TrimSpace(kv[1])
-	}
-}
-
-// CookieJarHeader 供书源 JS 查询（对应 CookieStore.getCookie(tag[, key])）。
-func CookieJarHeader(rawURL, key string) string {
-	u, err := url.Parse(rawURL)
-	if err != nil {
-		return ""
-	}
-	cookieJar.mu.Lock()
-	defer cookieJar.mu.Unlock()
-	jar, ok := cookieJar.m[u.Host]
-	if !ok {
-		return ""
-	}
-	if key != "" {
-		return jar[key]
-	}
-	var parts []string
-	for k, v := range jar {
-		parts = append(parts, k+"="+v)
-	}
-	return strings.Join(parts, "; ")
-}
-
-func newCookieObject(vm *goja.Runtime) *goja.Object {
+// newCookieObject 构造 JS 的 `cookie` 对象：
+// getCookie(url[,key]) / setCookie(url,cookie) / replaceCookie / removeCookie / getKey。
+func newCookieObject(vm *goja.Runtime, state SourceState) *goja.Object {
 	o := vm.NewObject()
 	set := func(k string, v any) {
 		if err := o.Set(k, v); err != nil {
 			panic(vm.ToValue(err.Error()))
 		}
 	}
-	// getCookie(key) 或 getCookie(url, key)
 	set("getCookie", func(call goja.FunctionCall) goja.Value {
 		tag := stringArg(call, 0)
-		key := ""
 		if len(call.Arguments) > 1 {
-			key = stringArg(call, 1)
+			return vm.ToValue(state.GetCookieKey(tag, stringArg(call, 1)))
 		}
-		return vm.ToValue(CookieJarHeader(tag, key))
+		return vm.ToValue(state.GetCookie(tag))
 	})
 	set("setCookie", func(call goja.FunctionCall) goja.Value {
 		rawURL := stringArg(call, 0)
 		cookie := stringArgOr(call, 1, "")
 		if cookie != "" {
-			CookieJarRecord(rawURL, []string{cookie})
+			state.SetCookie(rawURL, cookie)
 		}
 		return vm.ToValue(cookie)
 	})
+	// replaceCookie 与 setCookie 在服务端实现中同为覆盖式合并（对应 legado 的语义）
+	set("replaceCookie", func(call goja.FunctionCall) goja.Value {
+		rawURL := stringArg(call, 0)
+		cookie := stringArgOr(call, 1, "")
+		if cookie != "" {
+			state.SetCookie(rawURL, cookie)
+		}
+		return vm.ToValue(cookie)
+	})
+	set("removeCookie", func(call goja.FunctionCall) goja.Value {
+		state.RemoveCookie(stringArg(call, 0))
+		return goja.Null()
+	})
+	set("getKey", func(call goja.FunctionCall) goja.Value {
+		return vm.ToValue(state.GetCookieKey(stringArg(call, 0), stringArg(call, 1)))
+	})
 	return o
 }
 
+// newSourceObject 构造 JS 的 `source` 对象。
+// 对应 legado BaseSource 的变量与登录信息读写。
+func newSourceObject(vm *goja.Runtime, state SourceState, props map[string]any) *goja.Object {
+	o := vm.NewObject()
+	set := func(k string, v any) {
+		if err := o.Set(k, v); err != nil {
+			panic(vm.ToValue(err.Error()))
+		}
+	}
+	bindSourceState(vm, set, state, props)
+	// source.get/put：书源级键值缓存（legado 中即 source 的方法）。
+	// 注意 java.get/put 另有语义（网络 / 解析器变量），在 bridge.go 中定义，
+	// 不在这里覆盖。
+	set("get", func(call goja.FunctionCall) goja.Value {
+		return vm.ToValue(sourceKeyGet(state, stringArg(call, 0)))
+	})
+	set("put", func(call goja.FunctionCall) goja.Value {
+		val := stringArgOr(call, 1, "")
+		sourceKeyPut(state, stringArg(call, 0), val)
+		return vm.ToValue(val)
+	})
+	// 书源自身属性（bookSourceUrl 等）原样可读，存量 JS 会读 source.loginUi 等做能力探测。
+	for k, v := range props {
+		if o.Get(k) != nil {
+			continue
+		}
+		_ = o.Set(k, v)
+	}
+	return o
+}
+
+// bindSourceState 把书源会话状态方法绑定到目标 JS 对象。
+// legado 中 book source 的 evalJS 把 `java` 绑成书源对象自身，因此 `java` 与
+// `source` 都能读到 getVariable/getLoginInfo 等方法——这里保持一致。
+func bindSourceState(vm *goja.Runtime, set func(k string, v any), state SourceState, props map[string]any) {
+	set("getVariable", func(call goja.FunctionCall) goja.Value {
+		return vm.ToValue(state.GetVariable())
+	})
+	setVariable := func(call goja.FunctionCall) goja.Value {
+		if len(call.Arguments) == 0 || goja.IsUndefined(call.Arguments[0]) || goja.IsNull(call.Arguments[0]) {
+			state.SetVariable("")
+			return goja.Null()
+		}
+		state.SetVariable(call.Arguments[0].String())
+		return goja.Null()
+	}
+	set("setVariable", setVariable)
+	set("putVariable", setVariable)
+
+	set("getLoginInfo", func(call goja.FunctionCall) goja.Value {
+		if v := state.GetLoginInfo(); v != "" {
+			return vm.ToValue(v)
+		}
+		return goja.Null()
+	})
+	set("putLoginInfo", func(call goja.FunctionCall) goja.Value {
+		v := ""
+		if len(call.Arguments) > 0 && !goja.IsUndefined(call.Arguments[0]) && !goja.IsNull(call.Arguments[0]) {
+			v = call.Arguments[0].String()
+		}
+		state.SetLoginInfo(v)
+		return vm.ToValue(true)
+	})
+	set("removeLoginInfo", func(call goja.FunctionCall) goja.Value {
+		state.SetLoginInfo("")
+		return goja.Null()
+	})
+	// getLoginInfoMap 对应 Kotlin：解析登录信息 JSON；
+	// 未设置且 loginUi 非空时，用 loginUi 里各字段的 default 初始化。
+	set("getLoginInfoMap", func(call goja.FunctionCall) goja.Value {
+		if m := parseLoginInfoMap(state.GetLoginInfo()); m != nil {
+			return vm.ToValue(m)
+		}
+		return vm.ToValue(initLoginInfoFromUI(props))
+	})
+
+	set("getLoginHeader", func(call goja.FunctionCall) goja.Value {
+		if v := state.GetLoginHeader(); v != "" {
+			return vm.ToValue(v)
+		}
+		return goja.Null()
+	})
+	set("putLoginHeader", func(call goja.FunctionCall) goja.Value {
+		header := stringArgOr(call, 0, "")
+		state.SetLoginHeader(header)
+		// 请求头里的 Cookie 同步进 cookie 存储（对应 legado putLoginHeader）
+		if header != "" {
+			var m map[string]any
+			if json.Unmarshal([]byte(header), &m) == nil {
+				for k, v := range m {
+					if strings.EqualFold(k, "cookie") {
+						state.SetCookie(headerCookieURL(props), fmt.Sprintf("%v", v))
+					}
+				}
+			}
+		}
+		return goja.Null()
+	})
+	set("removeLoginHeader", func(call goja.FunctionCall) goja.Value {
+		state.SetLoginHeader("")
+		state.RemoveCookie(headerCookieURL(props))
+		return goja.Null()
+	})
+	set("getKey", func(call goja.FunctionCall) goja.Value {
+		return vm.ToValue(headerCookieURL(props))
+	})
+}
+
+// sourceKeyGet / sourceKeyPut：source.get/put 的独立键值槽，
+// 以 "__kv__" 前缀编码进同一份变量 JSON，从而与 getVariable 一起持久化。
+const sourceKVVar = "__kv__"
+
+func sourceKeyGet(state SourceState, key string) string {
+	if key == "" {
+		return ""
+	}
+	var m map[string]string
+	_ = json.Unmarshal([]byte(state.GetVariable()), &m)
+	return m[sourceKVVar+key]
+}
+
+func sourceKeyPut(state SourceState, key, val string) {
+	if key == "" {
+		return
+	}
+	m := map[string]string{}
+	_ = json.Unmarshal([]byte(state.GetVariable()), &m)
+	if m == nil {
+		m = map[string]string{}
+	}
+	m[sourceKVVar+key] = val
+	b, err := json.Marshal(m)
+	if err != nil {
+		return
+	}
+	state.SetVariable(string(b))
+}
+
+// headerCookieURL 取 loginHeader 里 Cookie 归属的 URL。
+func headerCookieURL(props map[string]any) string {
+	if props != nil {
+		if u, ok := props["bookSourceUrl"].(string); ok && u != "" {
+			return u
+		}
+	}
+	return ""
+}
+
+// parseLoginInfoMap 解析登录信息 JSON；空串或非法 JSON 返回 nil。
+func parseLoginInfoMap(s string) map[string]string {
+	if strings.TrimSpace(s) == "" {
+		return nil
+	}
+	var raw map[string]any
+	if json.Unmarshal([]byte(s), &raw) != nil {
+		return nil
+	}
+	out := make(map[string]string, len(raw))
+	for k, v := range raw {
+		out[k] = fmt.Sprintf("%v", v)
+	}
+	return out
+}
+
+// initLoginInfoFromUI 用 loginUi 各字段 default 生成初始登录信息
+// （对应 legado getLoginInfoMap：跳过 button 类型）。
+func initLoginInfoFromUI(props map[string]any) map[string]string {
+	out := map[string]string{}
+	raw, _ := props["loginUi"].(string)
+	if strings.TrimSpace(raw) != "" {
+		var rows []map[string]any
+		if json.Unmarshal([]byte(raw), &rows) == nil {
+			for _, row := range rows {
+				if t, _ := row["type"].(string); t == "button" {
+					continue
+				}
+				name, _ := row["name"].(string)
+				if name == "" {
+					continue
+				}
+				def, _ := row["default"].(string)
+				out[name] = def
+			}
+		}
+	}
+	return out
+}
+
 // ─── cache 对象（对应 legado CacheManager 注入的 `cache`） ──────────────────
 
 var jsCache = struct {