magic-agent 2.4.3 → 2.6.0

package/internal/client/scripts_test.go

@@ -0,0 +1,726 @@
+package client
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/darren/magic-agent/internal/agent"
+)
+
+/* 脚本（发现页的「脚本」）：
+ *   ① 客户端只能登记「脚本文件路径 + 参数」，**不能传任意命令行**；
+ *   ② 扩展名走白名单，.exe / .txt 一律拒绝；
+ *   ③ start 真把脚本跑起来，日志逐行到达，状态如实（done / failed / stopped）；
+ *   ④ stop 杀掉长跑脚本（整组），重复 start / 运行中 delete 都报 409。 */
+
+/* scriptSvc 起一个脚本服务：数据目录指到临时目录，不碰用户真实的 ~/.magic-agent/scripts。 */
+func scriptSvc(t *testing.T) *Service {
+	t.Helper()
+	dir := t.TempDir()
+	t.Setenv(envScriptsDir, dir)
+	return &Service{opts: Options{}}
+}
+
+/* writeScript 造一个能跑的脚本桩，返回路径。
+ *
+ * Windows 上 exec 不了 shebang 脚本（CreateProcess 不认 #!），改发 .cmd
+ * （走 agent.WrapLaunch 的 PowerShell 包装）；unix 上发 .sh。
+ * 两条路的**语义逐条对齐**：同样的行、同样的顺序。
+ *
+ * ⚠️ Windows 上脚本文件**按 GBK 存**（中文 Windows 的记事本默认那样），
+ * 且脚本输出走控制台代码页（936）→ 管道里是 GBK 字节。
+ * 这与服务端 scripts_decode_windows.go 的解码口径正好对上：
+ * 写 GBK、读 GBK，中文能正确还原。
+ * （写成 UTF-8 的话，cmd.exe 会拿代码页去解释脚本里的字节 → 乱码命令。）
+ */
+func writeScript(t *testing.T, name string, lines []string) string {
+	t.Helper()
+	dir := t.TempDir()
+	if runtime.GOOS == "windows" {
+		path := filepath.Join(dir, name+".cmd")
+		out := []string{"@echo off"}
+		for _, l := range lines {
+			// 批处理里 echo 空行要写成 "echo."，否则会被跳过。
+			if strings.TrimSpace(l) == "" {
+				out = append(out, "echo.")
+				continue
+			}
+			out = append(out, "echo "+l)
+		}
+		body := strings.Join(out, "\r\n") + "\r\n"
+		if err := os.WriteFile(path, encodeWindowsScript(body), 0o755); err != nil {
+			t.Fatal(err)
+		}
+		return path
+	}
+	path := filepath.Join(dir, name+".sh")
+	body := "#!/bin/sh\n" + strings.Join(lines, "\n") + "\n"
+	if err := os.WriteFile(path, []byte(body), 0o755); err != nil {
+		t.Fatal(err)
+	}
+	return path
+}
+
+// encodeWindowsScript 把脚本内容按**当前控制台代码页**编码。
+//
+// Windows 上 cmd.exe 按代码页（936=GBK）解释脚本文件里的字节 —— 文件是 UTF-8
+// 的话，「echo 中文」这一行会被拆成乱码命令，直接报「不是内部或外部命令」。
+// 所以测试桩必须按代码页存，才和真实用户脚本一致。
+func encodeWindowsScript(s string) []byte {
+	cp := scriptCodePage()
+	if b, ok := encodeACP(s, cp); ok {
+		return b
+	}
+	return []byte(s)
+}
+
+// writeLongScript 造一个「一直跑」的脚本（等 stop 才退出）。
+func writeLongScript(t *testing.T) string {
+	t.Helper()
+	dir := t.TempDir()
+	if runtime.GOOS == "windows" {
+		path := filepath.Join(dir, "long.cmd")
+		body := "@echo off\r\n" +
+			"echo started\r\n" +
+			":loop\r\n" +
+			"ping -n 2 127.0.0.1 >nul\r\n" +
+			"goto loop\r\n"
+		if err := os.WriteFile(path, []byte(body), 0o755); err != nil {
+			t.Fatal(err)
+		}
+		return path
+	}
+	path := filepath.Join(dir, "long.sh")
+	body := "#!/bin/sh\n" +
+		"echo started\n" +
+		"while true; do sleep 1; done\n"
+	if err := os.WriteFile(path, []byte(body), 0o755); err != nil {
+		t.Fatal(err)
+	}
+	return path
+}
+
+// addScript 登记一个脚本，返回 id。
+func addScript(t *testing.T, s *Service, path string, args ...string) string {
+	t.Helper()
+	rec := httptest.NewRecorder()
+	body, _ := json.Marshal(map[string]any{"name": "测试脚本", "path": path, "args": args})
+	req := httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body)))
+	s.handleScripts(rec, req)
+	if rec.Code != http.StatusOK {
+		t.Fatalf("登记脚本失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	var item struct {
+		ID string `json:"id"`
+	}
+	if err := json.Unmarshal(rec.Body.Bytes(), &item); err != nil {
+		t.Fatalf("响应不是合法 JSON：%v", err)
+	}
+	if item.ID == "" {
+		t.Fatal("登记成功但没给 id")
+	}
+	return item.ID
+}
+
+// addScriptDef 登记一个脚本（完整字段），返回 id。
+//
+// 与 addScript 的区别：能带解释器与环境变量 —— 这两个字段是
+// 「白名单里没有 .exe 时 Python/Node 脚本怎么跑」的落地点。
+func addScriptDef(t *testing.T, s *Service, def map[string]any) string {
+	t.Helper()
+	rec := httptest.NewRecorder()
+	body, _ := json.Marshal(def)
+	req := httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body)))
+	s.handleScripts(rec, req)
+	if rec.Code != http.StatusOK {
+		t.Fatalf("登记脚本失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	var item struct {
+		ID string `json:"id"`
+	}
+	if err := json.Unmarshal(rec.Body.Bytes(), &item); err != nil {
+		t.Fatalf("响应不是合法 JSON：%v", err)
+	}
+	if item.ID == "" {
+		t.Fatal("登记成功但没给 id")
+	}
+	return item.ID
+}
+
+// absInterpreter 找一个**可用的解释器绝对路径**（跨平台）。
+//
+// Windows 上用 PowerShell（.ps1 可以直接当参数传给它），
+// unix 上用 sh（.sh 同理）。
+//
+// ⚠️ 不能用 .cmd 当解释器：Windows 的 CreateProcess **不认批处理**，
+//
+//	exec 直接跑 .cmd 会报「不是有效的 Win32 应用程序」。
+func absInterpreter(t *testing.T) string {
+	t.Helper()
+	name := "sh"
+	if runtime.GOOS == "windows" {
+		name = "powershell.exe"
+	}
+	p, err := exec.LookPath(name)
+	if err != nil {
+		t.Skipf("这台机器上找不到 %s，跳过解释器测试", name)
+	}
+	abs, err := filepath.Abs(p)
+	if err != nil {
+		t.Fatal(err)
+	}
+	return abs
+}
+
+/* ─────────────── 清单 ─────────────── */
+
+func TestScriptsAddListUpdateRemove(t *testing.T) {
+	s := scriptSvc(t)
+	path := writeScript(t, "hello", []string{"hi"})
+	id := addScript(t, s, path)
+
+	// 清单里能看到它，路径与参数原样带出去。
+	rec := httptest.NewRecorder()
+	s.handleScripts(rec, httptest.NewRequest(http.MethodGet, "/desk/scripts", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("读清单失败：HTTP %d", rec.Code)
+	}
+	var body struct {
+		Scripts []ScriptItem `json:"scripts"`
+	}
+	if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
+		t.Fatal(err)
+	}
+	if len(body.Scripts) != 1 {
+		t.Fatalf("脚本数 = %d，期望 1", len(body.Scripts))
+	}
+	if body.Scripts[0].Path != path {
+		t.Fatalf("路径 = %q，期望 %q", body.Scripts[0].Path, path)
+	}
+	if body.Scripts[0].State.State != "stopped" {
+		t.Fatalf("初始状态 = %q，期望 stopped", body.Scripts[0].State.State)
+	}
+
+	// 改：换名字与参数。
+	rec = httptest.NewRecorder()
+	upd, _ := json.Marshal(map[string]any{"name": "改名了", "path": path, "args": []string{"--x"}})
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPut, "/desk/scripts/"+id, strings.NewReader(string(upd))))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("改脚本失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	item, ok := s.scriptMgr().get(id)
+	if !ok {
+		t.Fatal("改完就找不到了")
+	}
+	if item.Name != "改名了" || len(item.Args) != 1 || item.Args[0] != "--x" {
+		t.Fatalf("改完的内容不对：%+v", item.ScriptDef)
+	}
+
+	// 删。
+	rec = httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodDelete, "/desk/scripts/"+id, nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("删脚本失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	if _, ok := s.scriptMgr().get(id); ok {
+		t.Fatal("删完还在")
+	}
+	// 落盘日志也要跟着删 —— 界面上写的是「配置与日志都会被删掉」，
+	// 留着文件就是话说了不算。
+	if p := s.scriptMgr().logPath(id); p != "" {
+		if _, err := os.Stat(p); !os.IsNotExist(err) {
+			t.Fatalf("删完日志文件还在：%s", p)
+		}
+	}
+}
+
+/* ─────────────── 校验：白名单是安全落地点 ─────────────── */
+
+func TestScriptsRejectNonScriptFiles(t *testing.T) {
+	s := scriptSvc(t)
+	dir := t.TempDir()
+
+	// .txt —— 不是脚本
+	txt := filepath.Join(dir, "notes.txt")
+	if err := os.WriteFile(txt, []byte("hello"), 0o644); err != nil {
+		t.Fatal(err)
+	}
+	// .exe —— 可执行文件不是脚本（允许它就等于允许跑任意二进制）
+	exe := filepath.Join(dir, "tool.exe")
+	if err := os.WriteFile(exe, []byte("MZ"), 0o755); err != nil {
+		t.Fatal(err)
+	}
+
+	for _, p := range []string{txt, exe} {
+		rec := httptest.NewRecorder()
+		body, _ := json.Marshal(map[string]any{"name": "x", "path": p})
+		s.handleScripts(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body))))
+		if rec.Code == http.StatusOK {
+			t.Fatalf("扩展名不在白名单内的文件被接受了：%s", p)
+		}
+	}
+
+	// 不存在的文件 → 拒绝（不能登记一个跑不起来的东西然后假装能跑）。
+	missing := filepath.Join(dir, "nope.sh")
+	rec := httptest.NewRecorder()
+	body, _ := json.Marshal(map[string]any{"name": "x", "path": missing})
+	s.handleScripts(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body))))
+	if rec.Code == http.StatusOK {
+		t.Fatal("不存在的文件被接受了")
+	}
+
+	// 相对路径 → 拒绝（服务端不猜工作目录）。
+	rec = httptest.NewRecorder()
+	body, _ = json.Marshal(map[string]any{"name": "x", "path": "rel.sh"})
+	s.handleScripts(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body))))
+	if rec.Code == http.StatusOK {
+		t.Fatal("相对路径被接受了")
+	}
+
+	// 空路径 → 拒绝。
+	rec = httptest.NewRecorder()
+	body, _ = json.Marshal(map[string]any{"name": "x", "path": ""})
+	s.handleScripts(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body))))
+	if rec.Code == http.StatusOK {
+		t.Fatal("空路径被接受了")
+	}
+}
+
+/* ─────────────── 启停与日志 ─────────────── */
+
+// waitState 等脚本进入某个状态（最多 d）。
+func waitState(t *testing.T, s *Service, id string, want string, d time.Duration) ScriptState {
+	t.Helper()
+	deadline := time.Now().Add(d)
+	var last ScriptState
+	for time.Now().Before(deadline) {
+		item, ok := s.scriptMgr().get(id)
+		if !ok {
+			t.Fatalf("脚本 %s 没了", id)
+		}
+		last = item.State
+		if last.State == want {
+			return last
+		}
+		time.Sleep(30 * time.Millisecond)
+	}
+	t.Fatalf("脚本 %s 的状态 = %q，期望 %q（等了 %s）", id, last.State, want, d)
+	return last
+}
+
+func TestScriptsStartRunsAndStreamsLog(t *testing.T) {
+	s := scriptSvc(t)
+	path := writeScript(t, "hello", []string{"第一行", "第二行"})
+	id := addScript(t, s, path)
+
+	rec := httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/start", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("启动失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	var item ScriptItem
+	if err := json.Unmarshal(rec.Body.Bytes(), &item); err != nil {
+		t.Fatal(err)
+	}
+	if item.State.State != "running" {
+		t.Fatalf("刚启动的状态 = %q，期望 running", item.State.State)
+	}
+	if item.State.PID <= 0 {
+		t.Fatal("启动后没有 pid")
+	}
+
+	// 跑完 → done（退出码 0）。
+	waitState(t, s, id, "done", 20*time.Second)
+
+	// 日志里能看到那两行。
+	lines, _, ok := s.scriptMgr().logLines(id, 0)
+	if !ok {
+		t.Fatal("读日志失败")
+	}
+	joined := scriptLinesText(lines)
+	if !strings.Contains(joined, "第一行") || !strings.Contains(joined, "第二行") {
+		t.Fatalf("日志里没有脚本输出：%q", joined)
+	}
+}
+
+func TestScriptsFailedExitIsReportedAsFailed(t *testing.T) {
+	s := scriptSvc(t)
+	// 退出码非 0 的脚本（unix 用 exit 3；Windows 批处理 exit /b 3）。
+	dir := t.TempDir()
+	var path string
+	if runtime.GOOS == "windows" {
+		path = filepath.Join(dir, "bad.cmd")
+		body := "@echo off\r\necho 准备失败\r\nexit /b 3\r\n"
+		if err := os.WriteFile(path, []byte(body), 0o755); err != nil {
+			t.Fatal(err)
+		}
+	} else {
+		path = filepath.Join(dir, "bad.sh")
+		body := "#!/bin/sh\necho 准备失败\nexit 3\n"
+		if err := os.WriteFile(path, []byte(body), 0o755); err != nil {
+			t.Fatal(err)
+		}
+	}
+	id := addScript(t, s, path)
+
+	rec := httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/start", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("启动失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	st := waitState(t, s, id, "failed", 20*time.Second)
+	if st.ExitCode != 3 {
+		t.Fatalf("退出码 = %d，期望 3", st.ExitCode)
+	}
+}
+
+func TestScriptsStopKillsLongRunning(t *testing.T) {
+	s := scriptSvc(t)
+	path := writeLongScript(t)
+	id := addScript(t, s, path)
+
+	rec := httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/start", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("启动失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	item, _ := s.scriptMgr().get(id)
+	pid := item.State.PID
+	if pid <= 0 {
+		t.Fatal("没有 pid")
+	}
+
+	// 等它真的起来了（输出第一行）再停 —— 否则可能停了个还没 spawn 的。
+	waitForLog(t, s, id, "started", 20*time.Second)
+
+	rec = httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/stop", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("停止失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+
+	// 状态必须是 stopped（**不是 failed**）：用户是自己点的停止，不是脚本出错。
+	st := waitState(t, s, id, "stopped", 20*time.Second)
+	if st.ExitCode == 3 {
+		t.Fatalf("主动停止不该记成失败（退出码 %d）", st.ExitCode)
+	}
+
+	// 进程真的没了。
+	deadline := time.Now().Add(10 * time.Second)
+	for time.Now().Before(deadline) {
+		if !pidAlive(pid) {
+			return
+		}
+		time.Sleep(100 * time.Millisecond)
+	}
+	t.Fatalf("pid %d 还活着 —— 停止没把进程带走", pid)
+}
+
+func TestScriptsRejectDoubleStartAndDeleteWhileRunning(t *testing.T) {
+	s := scriptSvc(t)
+	path := writeLongScript(t)
+	id := addScript(t, s, path)
+
+	rec := httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/start", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("启动失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	waitForLog(t, s, id, "started", 20*time.Second)
+
+	// 重复启动 → 409（不能一个脚本跑两份）。
+	rec = httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/start", nil))
+	if rec.Code != http.StatusConflict {
+		t.Fatalf("重复启动的 HTTP 码 = %d，期望 409", rec.Code)
+	}
+
+	// 运行中删除 → 409（先让用户停掉）。
+	rec = httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodDelete, "/desk/scripts/"+id, nil))
+	if rec.Code != http.StatusConflict {
+		t.Fatalf("运行中删除的 HTTP 码 = %d，期望 409", rec.Code)
+	}
+
+	// 停掉之后就能删了。
+	rec = httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/stop", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("停止失败：HTTP %d", rec.Code)
+	}
+	waitState(t, s, id, "stopped", 20*time.Second)
+
+	rec = httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodDelete, "/desk/scripts/"+id, nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("停止后删除的 HTTP 码 = %d，期望 200（%s）", rec.Code, rec.Body.String())
+	}
+}
+
+/* ─────────────── 日志增量 ─────────────── */
+
+func TestScriptsLogAfterReturnsOnlyNewLines(t *testing.T) {
+	s := scriptSvc(t)
+	path := writeScript(t, "hello", []string{"一", "二", "三"})
+	id := addScript(t, s, path)
+
+	rec := httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/start", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("启动失败：HTTP %d", rec.Code)
+	}
+	waitState(t, s, id, "done", 20*time.Second)
+
+	all, _, _ := s.scriptMgr().logLines(id, 0)
+	if len(all) < 3 {
+		t.Fatalf("日志行数 = %d，期望 ≥3", len(all))
+	}
+	// after = 第 2 行的 seq → 只应拿到第 3 行。
+	tail, _, _ := s.scriptMgr().logLines(id, all[1].Seq)
+	if len(tail) != len(all)-2 {
+		t.Fatalf("增量行数 = %d，期望 %d", len(tail), len(all)-2)
+	}
+	for _, l := range tail {
+		if l.Seq <= all[1].Seq {
+			t.Fatalf("增量里混进了旧行：seq %d <= %d", l.Seq, all[1].Seq)
+		}
+	}
+}
+
+/* ─────────────── 解释器与环境变量 ─────────────── */
+
+// writeInterpretedScript 造一个「要靠解释器跑」的脚本桩，返回路径。
+//
+// Windows 用 .ps1（配 powershell.exe），unix 用 .sh（配 sh）。
+//
+// ⚠️ unix 侧**故意不写 shebang、也不给执行位** —— 直接 exec 它必然失败
+//
+//	（格式错 / 权限错），只有「由解释器打开」才跑得起来。
+//	这样「脚本跑起来了」就**证明**走的是解释器那条路，
+//	而不是碰巧直接 exec 也能过。
+//
+// ⚠️ 文件内容只用 ASCII：Windows 上 PowerShell 5.1 读无 BOM 的 .ps1
+//
+//	按 ANSI（系统代码页）解释，而 unix 侧走 UTF-8 —— 掺中文会让
+//	两边行为不一致，测的就不是「解释器有没有生效」而是「编码对不对」。
+func writeInterpretedScript(t *testing.T, body string) string {
+	t.Helper()
+	dir := t.TempDir()
+	name := "interp.sh"
+	mode := os.FileMode(0o644)
+	if runtime.GOOS == "windows" {
+		name = "interp.ps1"
+		mode = 0o755
+	}
+	path := filepath.Join(dir, name)
+	// unix 侧 0644 = 没有执行位 → 直接 exec 会被拒，
+	// 只能由解释器打开 → 「跑起来了」就证明走了解释器那条路。
+	if err := os.WriteFile(path, []byte(body+"\n"), mode); err != nil {
+		t.Fatal(err)
+	}
+	return path
+}
+
+// interpEchoBody 造一段「打印收到的参数、自身路径与某个环境变量」的脚本体。
+//
+// ⚠️ 参数下标两个平台不一样，**这是解释器自己的约定，不是我们的**：
+//
+//   - unix `sh script.sh A` → $0=脚本路径、$1=A
+//   - win  `powershell.exe script.ps1 A` → $args[0]=A（脚本路径被 PowerShell
+//     自己吃掉了，不进 $args）
+//
+// 所以这里按平台取「第一个用户参数」，两边能断言到同一个值。
+func interpEchoBody(envKey string) string {
+	if runtime.GOOS == "windows" {
+		return "Write-Output (\"ARG1=\" + $args[0])\n" +
+			"Write-Output (\"SELF=\" + $PSCommandPath)\n" +
+			"Write-Output (\"ENV=\" + $env:" + envKey + ")"
+	}
+	return "echo \"ARG1=$1\"\n" +
+		"echo \"SELF=$0\"\n" +
+		"echo \"ENV=$" + envKey + "\""
+}
+
+func TestScriptsInterpreterAndEnv(t *testing.T) {
+	s := scriptSvc(t)
+	interp := absInterpreter(t)
+	const envKey = "MAGIC_SCRIPT_TEST_ENV"
+	path := writeInterpretedScript(t, interpEchoBody(envKey))
+
+	id := addScriptDef(t, s, map[string]any{
+		"name":        "解释器脚本",
+		"path":        path,
+		"args":        []string{"arg-one"},
+		"interpreter": interp,
+		"env":         []string{envKey + "=42"},
+	})
+
+	// 定义要原样回显（前端详情页要显示这两项）。
+	item, ok := s.scriptMgr().get(id)
+	if !ok {
+		t.Fatal("登记完就找不到了")
+	}
+	if item.Interpreter != interp {
+		t.Fatalf("解释器没落库：%q", item.Interpreter)
+	}
+	if len(item.Env) != 1 || item.Env[0] != envKey+"=42" {
+		t.Fatalf("环境变量没落库：%v", item.Env)
+	}
+
+	rec := httptest.NewRecorder()
+	s.handleScriptSub(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts/"+id+"/start", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("启动失败：HTTP %d（%s）", rec.Code, rec.Body.String())
+	}
+	waitState(t, s, id, "done", 30*time.Second)
+
+	lines, _, _ := s.scriptMgr().logLines(id, 0)
+	joined := scriptLinesText(lines)
+	/* 用户给的参数要原样到达脚本 → 说明走的是「解释器 + 脚本路径 + 参数」这条路。
+	   ⚠️ 用 ASCII 断言：这一步测的是「参数有没有传对」，
+	   编码对不对是 scripts_decode_test.go 的活儿，别把两件事混在一个断言里。 */
+	if !strings.Contains(joined, "ARG1=arg-one") {
+		t.Fatalf("日志里没有脚本收到的参数（期望 ARG1=arg-one）：\n%s", joined)
+	}
+	// 脚本自己的路径要能被解释器看到（unix $0 / win $PSCommandPath）。
+	//
+	// ⚠️ 只查**末级文件名在不在**：Windows 上 t.TempDir() 给的是 8.3 短路径
+	//（C:\Users\ADMINI~1\...），而 PowerShell 报的是长路径
+	//（C:\Users\Administrator\...）—— 同一处，形态不同。
+	// 比全路径会得到一个假失败。
+	if !strings.Contains(joined, "SELF=") || !strings.Contains(joined, filepath.Base(path)) {
+		t.Fatalf("日志里没有脚本自身路径（期望含 %s）：\n%s", filepath.Base(path), joined)
+	}
+	// 环境变量要真的到了子进程。
+	if !strings.Contains(joined, "ENV=42") {
+		t.Fatalf("子进程没拿到环境变量：\n%s", joined)
+	}
+}
+
+func TestScriptsRejectBadInterpreterAndEnv(t *testing.T) {
+	s := scriptSvc(t)
+	dir := t.TempDir()
+	interp := absInterpreter(t)
+
+	// 一个存在的脚本文件（.sh / .ps1），用来测「只有解释器或 env 出错」的场景。
+	var path string
+	if runtime.GOOS == "windows" {
+		path = filepath.Join(dir, "ok.ps1")
+		if err := os.WriteFile(path, []byte("Write-Output 1\n"), 0o755); err != nil {
+			t.Fatal(err)
+		}
+	} else {
+		path = filepath.Join(dir, "ok.sh")
+		if err := os.WriteFile(path, []byte("#!/bin/sh\necho 1\n"), 0o755); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	cases := []struct {
+		what string
+		def  map[string]any
+	}{
+		{"解释器不存在", map[string]any{
+			"name": "x", "path": path,
+			"interpreter": filepath.Join(dir, "没有这个解释器.exe"),
+		}},
+		{"解释器是相对路径", map[string]any{
+			"name": "x", "path": path, "interpreter": "python.exe",
+		}},
+		{"解释器是目录", map[string]any{
+			"name": "x", "path": path, "interpreter": dir,
+		}},
+		{"环境变量不是 KEY=VALUE", map[string]any{
+			"name": "x", "path": path, "env": []string{"没有等号"},
+		}},
+		{"环境变量名里有空格", map[string]any{
+			"name": "x", "path": path, "env": []string{"BAD KEY=1"},
+		}},
+	}
+	for _, c := range cases {
+		rec := httptest.NewRecorder()
+		body, _ := json.Marshal(c.def)
+		s.handleScripts(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body))))
+		if rec.Code == http.StatusOK {
+			t.Fatalf("%s：被接受了（应该拒绝）", c.what)
+		}
+	}
+
+	// 环境变量条数上限。
+	tooMany := make([]string, scriptMaxEnv+1)
+	for i := range tooMany {
+		tooMany[i] = fmt.Sprintf("K%d=1", i)
+	}
+	rec := httptest.NewRecorder()
+	body, _ := json.Marshal(map[string]any{"name": "x", "path": path, "env": tooMany})
+	s.handleScripts(rec, httptest.NewRequest(http.MethodPost, "/desk/scripts", strings.NewReader(string(body))))
+	if rec.Code == http.StatusOK {
+		t.Fatalf("环境变量超上限（%d 条）：被接受了", len(tooMany))
+	}
+
+	// 正常解释器不该被上面的坏用例带偏（确认 interp 本身合法）。
+	id := addScriptDef(t, s, map[string]any{
+		"name": "正常", "path": path, "interpreter": interp,
+	})
+	if id == "" {
+		t.Fatal("合法解释器没登记上")
+	}
+}
+
+/* ─────────────── 持久化 ─────────────── */
+
+func TestScriptsPersistAcrossRestart(t *testing.T) {
+	dir := t.TempDir()
+	t.Setenv(envScriptsDir, dir)
+
+	path := writeScript(t, "hello", []string{"你好"})
+	// 第一个服务：登记脚本。
+	s1 := &Service{opts: Options{}}
+	id := addScript(t, s1, path)
+
+	// 第二个服务（模拟重启）：清单要还在。
+	s2 := &Service{opts: Options{}}
+	if _, ok := s2.scriptMgr().get(id); !ok {
+		t.Fatal("重启后脚本清单没了 —— 没落盘")
+	}
+}
+
+/* ─────────────── 辅助 ─────────────── */
+
+func scriptLinesText(lines []scriptLine) string {
+	var b strings.Builder
+	for _, l := range lines {
+		b.WriteString(l.Text)
+		b.WriteString("\n")
+	}
+	return b.String()
+}
+
+// waitForLog 等日志里出现某段文字。
+func waitForLog(t *testing.T, s *Service, id, want string, d time.Duration) {
+	t.Helper()
+	deadline := time.Now().Add(d)
+	for time.Now().Before(deadline) {
+		lines, _, _ := s.scriptMgr().logLines(id, 0)
+		if strings.Contains(scriptLinesText(lines), want) {
+			return
+		}
+		time.Sleep(30 * time.Millisecond)
+	}
+	t.Fatalf("等了 %s 也没等到日志里有 %q", d, want)
+}
+
+// pidAlive 探活（用 agent.PIDAlive：Windows 走 OpenProcess 内核调用，
+// unix 走 signal 0 —— 平台差异已经在那一层处理好了）。
+func pidAlive(pid int) bool {
+	return agent.PIDAlive(pid)
+}