magic-agent 2.6.0 → 2.6.3

package/internal/client/restart_test.go

@@ -8,6 +8,7 @@ import (
 	"net/http/httptest"
 	"os"
 	"path/filepath"
+	"strings"
 	"testing"
 	"time"
 )
@@ -36,7 +37,7 @@ func TestHandleRestartFlagsThenTimer(t *testing.T) {
 	// 假助手：只记录被拉起过几次、带的是什么可执行文件。
 	spawned := make(chan string, 4)
 	oldSpawner := restartSpawner
-	restartSpawner = func(gotExe string, _ *log.Logger) { spawned <- gotExe }
+	restartSpawner = func(gotExe string, _ *log.Logger, _ string) { spawned <- gotExe }
 	defer func() { restartSpawner = oldSpawner }()
 	// 标志是进程级的：用例前后都重置，别把别的用例立过的 pending 带进来。
 	resetRestartPendingForTest()
@@ -149,3 +150,141 @@ func resetRestartPendingForTest() {
 	restartPending = false
 	restartMu.Unlock()
 }
+
+/*
+前台 serve 也要能被手机重启（2026-10-26 用户：「通过 magic-test 启动的无法在移动端
+点击重启服务重启」）。
+
+根因：`POST /desk/restart` 原来只认 `magic-client start` 写的 client.json，
+而 magic-test 的桥接是 `spawn(bin, ['serve', ...])` 起的**前台** serve，什么都不写 ——
+于是手机点重启一律 409。修法是让 serve 监听成功后自登记（<数据目录>/serve.json），
+hanldeRestart 用 selfDaemonState() 通用认领。
+
+这个用例钉住三件事：
+
+	① 前台 serve 自登记后，POST /desk/restart 会给 200（不再是 409）；
+	② 拉起的助手带的是**这份**状态文件的路径（否则它会去重启 client.json 里那个实例）；
+	③ 重启提交给状态文件的令牌是 serve 自己的 —— 手机存的地址靠它继续有效。
+*/
+func TestForegroundServeCanRestartItself(t *testing.T) {
+	dir := t.TempDir()
+	/* 用假 HOME 把**默认**状态目录挪进临时目录，而不是设 EnvState：
+	   EnvState（MAGIC_CLIENT_STATE）的语义是「这是 start 起的实例」，设了它
+	   selfDaemonState 就会优先认它 —— 那是 start 那条路的行为，不是前台 serve 的。 */
+	t.Setenv("HOME", dir)
+	// 默认位置放一份**别人**的登记：证明前台 serve 认的是自己那份 serve.json，
+	// 而不是默认路径那份（认错的话就会去重启别人的服务）。
+	defaultState := DefaultStatePath()
+	if err := os.MkdirAll(filepath.Dir(defaultState), 0o700); err != nil {
+		t.Fatal(err)
+	}
+	other := &DaemonState{PID: os.Getpid() + 100000, Port: 9100, Token: "other-token"}
+	if err := writeDaemonState(defaultState, other); err != nil {
+		t.Fatal(err)
+	}
+
+	exe, err := os.Executable()
+	if err != nil {
+		t.Fatal(err)
+	}
+	selfState := SelfStatePath(dir)
+	if _, err := WriteSelfStateAt("", dir, &DaemonState{
+		PID: os.Getpid(), Port: 8977, Token: "serve-token", Exe: exe, Lan: true,
+	}); err != nil {
+		t.Fatal(err)
+	}
+
+	resetRestartPendingForTest()
+	resetRestartHandoffForTest()
+	defer resetRestartPendingForTest()
+	defer resetRestartHandoffForTest()
+
+	type spawn struct{ exe, statePath string }
+	spawned := make(chan spawn, 1)
+	oldSpawner := restartSpawner
+	restartSpawner = func(gotExe string, _ *log.Logger, gotState string) {
+		spawned <- spawn{gotExe, gotState}
+	}
+	defer func() { restartSpawner = oldSpawner }()
+	oldDelay := restartDelay
+	restartDelay = 50 * time.Millisecond
+	defer func() { restartDelay = oldDelay }()
+
+	// 只给 SelfStatePath，不给 StatePath —— 这正是前台 serve 的形态。
+	s := &Service{
+		log:  log.New(io.Discard, "", 0),
+		opts: Options{SelfStatePath: selfState},
+	}
+	rec := httptest.NewRecorder()
+	s.handleRestart(rec, httptest.NewRequest(http.MethodPost, "/desk/restart", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("前台 serve 点重启 = %d，want 200（body: %s）", rec.Code, rec.Body.String())
+	}
+
+	select {
+	case got := <-spawned:
+		if got.statePath != selfState {
+			t.Errorf("助手拿到的状态文件 = %q，want %q（认错就会去重启别的实例）", got.statePath, selfState)
+		}
+		if got.exe != exe {
+			t.Errorf("助手可执行文件 = %q，want %q", got.exe, exe)
+		}
+	case <-time.After(2 * time.Second):
+		t.Fatal("定时器没有拉起助手")
+	}
+	// 交接里必须是 serve 自己的令牌：手机存的就是这个地址+令牌组合。
+	if tok, _ := restartHandoff()["token"].(string); tok != "serve-token" {
+		t.Errorf("交接令牌 = %q，want serve-token", tok)
+	}
+}
+
+// 没有任何自登记时仍然是 409，而且话要说得能照着做（区分「别人的登记」与「一份都没有」）。
+func TestRestartWithoutSelfStateIsRefused(t *testing.T) {
+	dir := t.TempDir()
+	// 同样只挪 HOME，不设 EnvState（那个代表 start 路线）。
+	t.Setenv("HOME", dir)
+	resetRestartPendingForTest()
+	defer resetRestartPendingForTest()
+
+	s := &Service{log: log.New(io.Discard, "", 0)}
+	rec := httptest.NewRecorder()
+	s.handleRestart(rec, httptest.NewRequest(http.MethodPost, "/desk/restart", nil))
+	if rec.Code != http.StatusConflict {
+		t.Fatalf("没有状态记录时 = %d，want 409", rec.Code)
+	}
+	var body map[string]any
+	if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
+		t.Fatalf("解析响应失败: %v", err)
+	}
+	msg, _ := body["error"].(string)
+	// 必须给出可照做的动作，不能只说「不能自重启」。
+	if !strings.Contains(msg, "magic-client restart") {
+		t.Errorf("拒绝理由应给出可照做的指引，实际：%q", msg)
+	}
+}
+
+// WriteSelfState / RemoveSelfState 的配对语义：新实例接手后，旧实例收尾**不许**删掉它。
+func TestRemoveSelfStateKeepsNewInstance(t *testing.T) {
+	dir := t.TempDir()
+	path, err := WriteSelfStateAt("", dir, &DaemonState{PID: os.Getpid(), Port: 8977, Token: "t"})
+	if err != nil {
+		t.Fatal(err)
+	}
+	// 模拟「新实例已经接管了这份登记」。
+	if err := writeDaemonState(path, &DaemonState{PID: os.Getpid() + 99999, Port: 8977}); err != nil {
+		t.Fatal(err)
+	}
+	RemoveSelfState(path)
+	if _, err := os.Stat(path); err != nil {
+		t.Fatalf("新实例的登记被旧实例删掉了（重启后再点会变 409）：%v", err)
+	}
+
+	// 登记还属于自己时才删。
+	if err := writeDaemonState(path, &DaemonState{PID: os.Getpid(), Port: 8977}); err != nil {
+		t.Fatal(err)
+	}
+	RemoveSelfState(path)
+	if _, err := os.Stat(path); !os.IsNotExist(err) {
+		t.Errorf("自己的登记应被删掉，stat err = %v", err)
+	}
+}

package/internal/client/scripts_decode_implausible.go

@@ -0,0 +1,38 @@
+package client
+
+// scripts_decode_implausible.go - 「解码结果像不像乱码」的判定，**全平台共用一份**。
+//
+// 只保留 !windows 时会怎样：scripts_decode_windows.go 里调用了它却没定义，
+// Windows 目标直接编译不过（`undefined: hasImplausibleRune`）——
+// 2026-10-26 打 2.6.1 时 win32-x64 就是这个错误挂掉的（另外 4 个目标照常出包，
+// 很容易被「完成 4/5」这句糊过去）。
+//
+// 非 Windows 的解码路径本身不走它（UTF-8 解不出就原样保留字节，见
+// scripts_decode_other.go），但 scripts_decode_test.go 不分平台编译、要引用它，
+// 所以两个平台都得有这个符号。判定逻辑只有这一份，不要再往平台文件里抄第二份。
+
+import "unicode/utf8"
+
+// hasImplausibleRune 报告文本里有没有「控制台输出里不会出现」的字符。
+//
+// 这些字符是「字节编码猜错」的典型产物：GBK 字节被当 UTF-8 解会冒出
+// 西里尔文（һ）、希伯来文、中文全角标点残留等。
+func hasImplausibleRune(s string) bool {
+	for _, r := range s {
+		switch {
+		case r == utf8.RuneError: // 替换符：解码失败过
+			return true
+		case r < 0x20 && r != '\t' && r != '\n' && r != '\r':
+			return true // 控制符
+		case r >= 0x80 && r <= 0x9F:
+			return true // C1 控制区
+		case r >= 0xE000 && r <= 0xF8FF:
+			return true // 私用区
+		case r >= 0x0370 && r <= 0x04FF:
+			return true // 希腊文 + 西里尔文
+		case r >= 0x0590 && r <= 0x08FF:
+			return true // 希伯来文 / 阿拉伯文 / 叙利亚文 / 塔安文
+		}
+	}
+	return false
+}

package/internal/client/scripts_decode_other.go

@@ -25,3 +25,7 @@ func decodeScriptBytes(b []byte) string {
 func encodeACP(s string, cp uint32) ([]byte, bool) {
 	return []byte(s), true
 }
+
+// scriptCodePage 非 Windows 平台没有「控制台代码页」：UTF-8（65001）就是它。
+// 单独存在的理由是 scripts_decode_test.go 会引用它（测试文件不分平台编译）。
+func scriptCodePage() uint32 { return 65001 }

package/internal/client/scripts_decode_test.go

@@ -24,18 +24,23 @@ func TestDecodeScriptBytesKeepsUTF8(t *testing.T) {
 }
 
 func TestDecodeScriptBytesRoundTripCodePage(t *testing.T) {
-	if runtime.GOOS != "windows" {
-		t.Skip("只有 Windows 才有控制台代码页这回事")
-	}
 	for _, in := range []string{"第一行", "中文", "hello", "混合 mixed 内容 123"} {
-		enc, ok := encodeACP(in, scriptCodePage())
-		if !ok {
-			t.Fatalf("按代码页编码 %q 失败", in)
-		}
-		got := decodeScriptBytes(enc)
-		if got != in {
-			t.Fatalf("代码页往返不对：%q != %q（字节 % x）", got, in, enc)
-		}
+		in := in
+		t.Run(in, func(t *testing.T) {
+			// 「代码页」只存在于 Windows；unix 上 encodeACP 就是原始字节，往返没有意义。
+			// 按用例跳过而不是整函数跳过：非 Windows 编译时才算这个函数（否则 vet 报未定义）。
+			if runtime.GOOS != "windows" {
+				t.Skip("只有 Windows 才有控制台代码页这回事")
+			}
+			enc, ok := encodeACP(in, scriptCodePage())
+			if !ok {
+				t.Fatalf("按代码页编码 %q 失败", in)
+			}
+			got := decodeScriptBytes(enc)
+			if got != in {
+				t.Fatalf("代码页往返不对：%q != %q（字节 % x）", got, in, enc)
+			}
+		})
 	}
 }
 

package/internal/client/scripts_decode_windows.go

@@ -93,30 +93,6 @@ func decodeScriptBytes(b []byte) string {
 	return string(b)
 }
 
-// hasImplausibleRune 报告文本里有没有「控制台输出里不会出现」的字符。
-//
-// 这些字符是「字节编码猜错」的典型产物：GBK 字节被当 UTF-8 解会冒出
-// 西里尔文（һ）、希伯来文、中文全角标点残留等。
-func hasImplausibleRune(s string) bool {
-	for _, r := range s {
-		switch {
-		case r == utf8.RuneError: // 替换符：解码失败过
-			return true
-		case r < 0x20 && r != '\t' && r != '\n' && r != '\r':
-			return true // 控制符
-		case r >= 0x80 && r <= 0x9F:
-			return true // C1 控制区
-		case r >= 0xE000 && r <= 0xF8FF:
-			return true // 私用区
-		case r >= 0x0370 && r <= 0x04FF:
-			return true // 希腊文 + 西里尔文
-		case r >= 0x0590 && r <= 0x08FF:
-			return true // 希伯来文 / 阿拉伯文 / 叙利亚文 / 塔安文
-		}
-	}
-	return false
-}
-
 // errnoIsZero 判断 syscall 调用的错误码是否为 0（成功）。
 //
 // ⚠️ **不要直接比 err != nil**：Proc.Call 返回的 error **成功时也是非 nil**

package/internal/client/scripts_test.go

@@ -63,7 +63,20 @@ func writeScript(t *testing.T, name string, lines []string) string {
 		return path
 	}
 	path := filepath.Join(dir, name+".sh")
-	body := "#!/bin/sh\n" + strings.Join(lines, "\n") + "\n"
+	/* ⚠️ 每一行都要 `echo ` 前缀 —— 早先这里直接 join(lines) 写进脚本，
+	   在 Windows 分支（下面那支）是 `echo <line>`，到 POSIX 分支却当成了**命令**：
+	   `第一行` 会被 shell 当成命令名，退出码 127 → 脚本状态 failed →
+	   TestScriptsStartRunsAndStreamsLog 在 macOS/Linux 上必然失败。
+	   空行写 `echo` 本身（输出空行），与批处理的 `echo.` 对齐。 */
+	out := make([]string, 0, len(lines))
+	for _, l := range lines {
+		if strings.TrimSpace(l) == "" {
+			out = append(out, "echo")
+			continue
+		}
+		out = append(out, "echo "+l)
+	}
+	body := "#!/bin/sh\n" + strings.Join(out, "\n") + "\n"
 	if err := os.WriteFile(path, []byte(body), 0o755); err != nil {
 		t.Fatal(err)
 	}