magic-agent 2.4.3 → 2.6.3

package/internal/client/health_identity_test.go

@@ -0,0 +1,84 @@
+package client
+
+import (
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"testing"
+	"time"
+)
+
+/* /desk/health 必须交代「我这一代进程是谁」（2026-10-27）。
+ *
+ * 为什么值得一条测试：重启换的是**进程**，会话 / 任务 / 设置全在盘上，
+ * 界面重启前后长得一模一样。用户点了「重启服务」看见列表没变，只会说「没效果」
+ * —— 这正是当天真机反馈的原话。而 `instanceId` 帮不上忙：它是**同一台桌面**的稳定
+ * 身份（多入口归一靠它），跨重启必须不变，所以不能用它判断新旧代。
+ *
+ * 判据只能是 `pid` + `startedAt` 这一对。这里钉住三件事：
+ *   1. 两个字段都得在（客户端靠它们证明换代）；
+ *   2. startedAt 是可解析的 RFC3339（客户端做字符串比较，格式漂了就会误判）；
+ *   3. pid 就是本进程（不是某个常量、也不是状态文件里的旧值）。
+ */
+func TestHealthReportsProcessIdentity(t *testing.T) {
+	svc := newTestService(t, fakeCore(t, goodContract))
+	srv := httptest.NewServer(svc.Handler())
+	defer srv.Close()
+
+	read := func() map[string]any {
+		t.Helper()
+		req, err := http.NewRequest(http.MethodGet, srv.URL+"/desk/health", nil)
+		if err != nil {
+			t.Fatal(err)
+		}
+		req.Header.Set("X-Magic-Token", "tok")
+		res, err := http.DefaultClient.Do(req)
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer res.Body.Close()
+		if res.StatusCode != http.StatusOK {
+			t.Fatalf("期望 200，得到 %d", res.StatusCode)
+		}
+		var got map[string]any
+		if err := json.NewDecoder(res.Body).Decode(&got); err != nil {
+			t.Fatal(err)
+		}
+		return got
+	}
+
+	got := read()
+
+	pid, ok := got["pid"].(float64) // JSON 数字统一解成 float64
+	if !ok {
+		t.Fatalf("health 必须带 pid（客户端靠它证明换代），实际没有或类型不对：%#v", got["pid"])
+	}
+	if int(pid) != os.Getpid() {
+		t.Fatalf("pid 必须是本进程 %d，得到 %v", os.Getpid(), pid)
+	}
+
+	raw, ok := got["startedAt"].(string)
+	if !ok || raw == "" {
+		t.Fatalf("health 必须带 startedAt（RFC3339），实际 %#v", got["startedAt"])
+	}
+	ts, err := time.Parse(time.RFC3339, raw)
+	if err != nil {
+		t.Fatalf("startedAt 必须是可解析的 RFC3339（客户端做字符串比较），得到 %q：%v", raw, err)
+	}
+	/* 启动时刻不该在将来，也不该早得离谱 —— 后者通常意味着拿了包级的零值/编译常量。 */
+	if time.Since(ts) < 0 {
+		t.Fatalf("startedAt 落在将来：%v", ts)
+	}
+	if time.Since(ts) > time.Hour {
+		t.Fatalf("startedAt 早得不合理（%v 之前）：像是没在 New() 里赋值", ts)
+	}
+
+	/* 同一个进程连着问两次，pid 与 startedAt 都必须一模一样 ——
+	   否则客户端会把「还没重启」误判成「换代了」。 */
+	again := read()
+	if again["pid"] != got["pid"] || again["startedAt"] != got["startedAt"] {
+		t.Fatalf("同一进程两次应答的身份必须一致：%v/%v vs %v/%v",
+			got["pid"], got["startedAt"], again["pid"], again["startedAt"])
+	}
+}

package/internal/client/hostenv.go

@@ -0,0 +1,68 @@
+package client
+
+// hostenv.go - 给 core 子进程的准备环境。
+//
+// 为什么在客户端做而不是 core 里做：core 是**被客户端 exec 起来的**，它继承什么
+// 完全由这里决定。宿主 IDE（WorkBuddy）的终端会给客户端塞三样东西：
+//
+//	NODE_OPTIONS=--require=".../app.asar.unpacked/.../node-language-shim.cjs"
+//	ELECTRON_RUN_AS_NODE=1
+//	PATH=<...>/shim/brokered-bin:<...>      ← 假 cat/cp/dd，require 时去连宿主 broker
+//
+// 后果（2026-10-26 真机实测）：引擎的 node 子进程加载宿主 shim → 连不上 broker →
+// `connect failed: Connection refused` → 每条任务 failed，错误看着像引擎坏了。
+// 在 core 一启动就把这三样摘干净，比让每个引擎各自防御可靠。
+//
+// 判据只认 `.asar` / `workbuddy` / `brokered-bin` 这几个特征：用户自己设的
+// NODE_OPTIONS（--max-old-space-size 之类）与正常 PATH 段原样保留。
+
+import (
+	"os"
+	"strings"
+)
+
+// hostShimValue 判断一个值是否指向宿主 Electron 应用的垫片。
+func hostShimValue(v string) bool {
+	l := strings.ToLower(v)
+	return strings.Contains(l, ".asar") || strings.Contains(l, "workbuddy") || strings.Contains(l, "brokered-bin")
+}
+
+// cleanHostEnv 返回一份可以交给 core 的环境：摘掉宿主垫片变量与假命令目录。
+func cleanHostEnv(env []string) []string {
+	out := make([]string, 0, len(env))
+	for _, kv := range env {
+		name, val, ok := strings.Cut(kv, "=")
+		if !ok {
+			out = append(out, kv)
+			continue
+		}
+		switch name {
+		case "ELECTRON_RUN_AS_NODE":
+			continue
+		case "NODE_OPTIONS":
+			if hostShimValue(val) {
+				continue
+			}
+		case "PATH":
+			if cleaned := stripHostShimPath(val); cleaned != val {
+				out = append(out, "PATH="+cleaned)
+				continue
+			}
+		}
+		out = append(out, kv)
+	}
+	return out
+}
+
+// stripHostShimPath 从 PATH 值里摘掉宿主垫片目录，其余段与顺序不动。
+func stripHostShimPath(val string) string {
+	parts := strings.Split(val, string(os.PathListSeparator))
+	kept := parts[:0]
+	for _, p := range parts {
+		if p == "" || hostShimValue(p) {
+			continue
+		}
+		kept = append(kept, p)
+	}
+	return strings.Join(kept, string(os.PathListSeparator))
+}

package/internal/client/hostenv_test.go

@@ -0,0 +1,37 @@
+package client
+
+import (
+	"strings"
+	"testing"
+)
+
+// 2026-10-26 真机：宿主 IDE 的终端把这三个变量塞进客户端，引擎子进程全崩在 require。
+func TestCleanHostEnvDropsShimVars(t *testing.T) {
+	in := []string{
+		`NODE_OPTIONS=--require="/Applications/WorkBuddy.app/Contents/Resources/app.asar.unpacked/cli/vendor/shim/node-language-shim.cjs"`,
+		"ELECTRON_RUN_AS_NODE=1",
+		"PATH=/Applications/WorkBuddy.app/Contents/Resources/app.asar.unpacked/cli/vendor/shim/brokered-bin:/opt/homebrew/bin:/usr/bin",
+		"HOME=/Users/x",
+	}
+	out := cleanHostEnv(in)
+	joined := strings.Join(out, "\n")
+	for _, bad := range []string{"ELECTRON_RUN_AS_NODE", ".asar", "brokered-bin"} {
+		if strings.Contains(joined, bad) {
+			t.Fatalf("还留着 %s：\n%s", bad, joined)
+		}
+	}
+	if !strings.Contains(joined, "PATH=/opt/homebrew/bin:/usr/bin") {
+		t.Fatalf("PATH 被误改：\n%s", joined)
+	}
+	if !strings.Contains(joined, "HOME=/Users/x") {
+		t.Fatal("无关变量被删了")
+	}
+}
+
+// 用户自己设的 NODE_OPTIONS 要留着。
+func TestCleanHostEnvKeepsUserNodeOptions(t *testing.T) {
+	out := cleanHostEnv([]string{"NODE_OPTIONS=--max-old-space-size=4096"})
+	if len(out) != 1 || out[0] != "NODE_OPTIONS=--max-old-space-size=4096" {
+		t.Fatalf("误删：%v", out)
+	}
+}