magic-agent 2.4.3 → 2.6.3

package/internal/client/ab_test.go

@@ -0,0 +1,287 @@
+package client
+
+// ab_test.go - agent-browser 封装与 shot-remote 端点的测试。
+//
+// 不连真 daemon（那要真 Chromium）：用 go build 现场编一个**假 agent-browser exe**
+// 验证探测链、--json 解析与端点行为。go 编译器不在 PATH 时相关用例跳过
+// （本机开发环境一定有；CI 沙箱也总有）。
+
+import (
+	"context"
+	"encoding/json"
+	"io"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"strings"
+	"testing"
+)
+
+// abNativeExeName 当前平台 agent-browser 原生 exe 的文件名（与 ab.go 同规则）。
+func abNativeExeName() string {
+	name := "agent-browser-" + runtime.GOOS + "-" + runtime.GOARCH
+	if runtime.GOOS == "windows" {
+		name += ".exe"
+	}
+	return name
+}
+
+// fakeABBin 现场编一个假 agent-browser exe，返回其路径。
+//
+// 行为由环境变量 AB_FAKE_MODE 控制（abRun 会把调用方环境透传给子进程）：
+//
+//	ok      → 按子命令吐 {"success":true,"data":…}；screenshot 会真写出文件
+//	err     → 吐 {"success":false,"error":"fake boom"}
+//	garbage → 吐非 JSON 乱码
+//	missing → 无输出直接退出 1
+func fakeABBin(t *testing.T) string {
+	t.Helper()
+	src := `package main
+
+import (
+	"fmt"
+	"os"
+	"strings"
+)
+
+func main() {
+	mode := os.Getenv("AB_FAKE_MODE")
+	switch mode {
+	case "err":
+		fmt.Println(` + "`" + `{"success":false,"error":"fake boom"}` + "`" + `)
+		os.Exit(1)
+	case "garbage":
+		fmt.Println("garbage not json at all")
+		os.Exit(0)
+	case "missing":
+		os.Exit(1)
+	}
+	// 按子命令给一点可断言的 data（跳过全局 flag）。
+	sub := ""
+	for i, a := range os.Args[1:] {
+		if strings.HasPrefix(a, "-") || a == "magic-client" {
+			continue
+		}
+		switch a {
+		case "get", "open", "eval", "pdf", "snapshot":
+			continue
+		case "screenshot":
+			// 真的把文件写出来：screenshot 的**最后一个非 flag 参数**是路径
+			// （与真 agent-browser 一致 —— flag 必须放在路径前面）。
+			out := ""
+			for _, b := range os.Args[i+1:] {
+				if !strings.HasPrefix(b, "-") {
+					out = b
+				}
+			}
+			if out != "" {
+				_ = os.WriteFile(out, []byte("png"), 0o600)
+			}
+			fmt.Println(` + "`" + `{"success":true,"data":{"path":"x"}}` + "`" + `)
+			return
+		case "url":
+			fmt.Println(` + "`" + `{"success":true,"data":"https://example.com/page"}` + "`" + `)
+			return
+		case "title":
+			fmt.Println(` + "`" + `{"success":true,"data":"Example"}` + "`" + `)
+			return
+		}
+		if sub == "" {
+			sub = a
+		}
+	}
+	fmt.Println(` + "`" + `{"success":true,"data":{"ok":true}}` + "`" + `)
+}
+`
+	dir := t.TempDir()
+	srcFile := filepath.Join(dir, "main.go")
+	if err := os.WriteFile(srcFile, []byte(src), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	exe := filepath.Join(dir, abNativeExeName())
+	cmd := exec.Command("go", "build", "-o", exe, srcFile)
+	if out, err := cmd.CombinedOutput(); err != nil {
+		t.Skipf("go build 假 exe 失败（跳过，不影响其它用例）: %v: %s", err, out)
+	}
+	return exe
+}
+
+// TestABBin探测链 环境变量 > npm 前缀 > 都没有时空。
+func TestABBin探测链(t *testing.T) {
+	exe := fakeABBin(t)
+	// 1) 显式环境变量最优先。
+	t.Setenv(EnvAgentBrowserBin, exe)
+	if got := abBin(); got != exe {
+		t.Fatalf("环境变量应最优先，实际 %q", got)
+	}
+	// 2) npm 全局前缀（<prefix>/node_modules/agent-browser/bin/<平台名>）。
+	t.Setenv(EnvAgentBrowserBin, "")
+	prefix := t.TempDir()
+	binDir := filepath.Join(prefix, "node_modules", "agent-browser", "bin")
+	if err := os.MkdirAll(binDir, 0o755); err != nil {
+		t.Fatal(err)
+	}
+	b, err := os.ReadFile(exe)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(binDir, abNativeExeName()), b, 0o755); err != nil {
+		t.Fatal(err)
+	}
+	t.Setenv("NPM_CONFIG_PREFIX", prefix)
+	got := abBin()
+	if !strings.Contains(filepath.ToSlash(got), "node_modules/agent-browser/bin/"+abNativeExeName()) {
+		t.Fatalf("应从 npm 前缀探测到，实际 %q", got)
+	}
+}
+
+// TestABRunJSON 解析：--json 成功 / 失败 / 乱码 / 无输出四种形状。
+func TestABRunJSON(t *testing.T) {
+	exe := fakeABBin(t)
+	t.Setenv(EnvAgentBrowserBin, exe)
+
+	t.Setenv("AB_FAKE_MODE", "ok")
+	data, err := abRun(context.Background(), "get", "url")
+	if err != nil {
+		t.Fatalf("ok 模式失败: %v", err)
+	}
+	if string(data) != `"https://example.com/page"` {
+		t.Fatalf("data 不对: %s", data)
+	}
+
+	t.Setenv("AB_FAKE_MODE", "err")
+	if _, err := abRun(context.Background(), "get", "url"); err == nil || !strings.Contains(err.Error(), "fake boom") {
+		t.Fatalf("err 模式应报 fake boom: %v", err)
+	}
+
+	t.Setenv("AB_FAKE_MODE", "garbage")
+	if _, err := abRun(context.Background(), "get", "url"); err == nil {
+		t.Fatal("乱码输出应报错")
+	}
+
+	t.Setenv("AB_FAKE_MODE", "missing")
+	if _, err := abRun(context.Background(), "get", "url"); err == nil {
+		t.Fatal("无输出退出应报错")
+	}
+}
+
+// TestABShot 落盘：假 exe 真写出文件，abShot 返回 nil 且文件在。
+func TestABShot(t *testing.T) {
+	exe := fakeABBin(t)
+	t.Setenv(EnvAgentBrowserBin, exe)
+	t.Setenv("AB_FAKE_MODE", "ok")
+	out := filepath.Join(t.TempDir(), "s.png")
+	if err := abShot(context.Background(), out, true); err != nil {
+		t.Fatalf("截图失败: %v", err)
+	}
+	if b, err := os.ReadFile(out); err != nil || string(b) != "png" {
+		t.Fatalf("截图文件没落盘: %v", err)
+	}
+}
+
+// abHideAll 把 abBin 的全部探测路径指到空目录（测「没装 agent-browser」的分支）。
+// t.Setenv 自动在测试结束时还原。PATH 上可能有真安装（npm 全局 bin），一并清掉 ——
+// 清 PATH 后 go build 假 exe 的用例会受影响，但 hide 只在 501 分支用，不冲突。
+func abHideAll(t *testing.T) {
+	t.Helper()
+	t.Setenv(EnvAgentBrowserBin, "")
+	t.Setenv("NPM_CONFIG_PREFIX", "")
+	t.Setenv("npm_config_prefix", "")
+	t.Setenv("APPDATA", t.TempDir()) // Windows 默认前缀指空
+	t.Setenv("PATH", t.TempDir())    // LookPath 兜底也指空
+}
+
+// TestShotRemoteEndpoint shot-remote 端点：能力位 gate / 参数校验 / 正常路径。
+func TestShotRemoteEndpoint(t *testing.T) {
+	exe := fakeABBin(t)
+	s := &Service{log: log.New(io.Discard, "", 0), contract: Contract{Features: []string{FeatureRequirementShots}}}
+	mux := http.NewServeMux()
+	mux.HandleFunc("/desk/requirements/shot-remote", s.handleRequirementShotRemote)
+	srv := httptest.NewServer(mux)
+	defer srv.Close()
+
+	// 没配 agent-browser → 501 + 安装提示。
+	// ⚠️ 本机可能真装了 agent-browser（npm 默认前缀 / PATH 都能命中）：
+	// 「没装」分支要把**所有**探测路径都指到空目录，否则测的是真安装。
+	abHideAll(t)
+	resp, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"https://example.com"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	body := readAllBody(resp)
+	if resp.StatusCode != http.StatusNotImplemented || !strings.Contains(body, "agent-browser") {
+		t.Fatalf("没装 agent-browser 应 501 + 提示，实际 %d %s", resp.StatusCode, body)
+	}
+
+	// 配上假 exe → 正常路径回 data（截图落 uploads 目录并真实存在）。
+	t.Setenv(EnvAgentBrowserBin, exe)
+	t.Setenv("AB_FAKE_MODE", "ok")
+	t.Setenv("TMP", t.TempDir()) // uploads 目录走 os.TempDir()
+	t.Setenv("TEMP", os.Getenv("TMP"))
+	resp2, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"https://example.com"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	body2 := readAllBody(resp2)
+	if resp2.StatusCode != http.StatusOK {
+		t.Fatalf("正常路径应 200，实际 %d %s", resp2.StatusCode, body2)
+	}
+	var out struct {
+		OK   bool   `json:"ok"`
+		Data string `json:"data"`
+	}
+	if err := json.Unmarshal([]byte(body2), &out); err != nil || !out.OK || !strings.HasPrefix(out.Data, "/desk/uploads/") {
+		t.Fatalf("响应形状不对: %s", body2)
+	}
+	// data 指向的文件真实存在。
+	name, err := url.PathUnescape(strings.TrimPrefix(out.Data, "/desk/uploads/"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, err := os.Stat(filepath.Join(os.TempDir(), "magic-client-uploads", name)); err != nil {
+		t.Fatalf("截图文件不在 uploads: %v", err)
+	}
+
+	// 非法 url → 400。
+	resp3, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"javascript:alert(1)"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	resp3.Body.Close()
+	if resp3.StatusCode != http.StatusBadRequest {
+		t.Fatalf("非法 url 应 400，实际 %d", resp3.StatusCode)
+	}
+}
+
+// TestShotRemote能力位 缺 feature.requirement.shots 必须 501（与 /shots 端点同一杆秤）。
+func TestShotRemote能力位(t *testing.T) {
+	s := &Service{log: log.New(io.Discard, "", 0), contract: Contract{Features: []string{}}}
+	mux := http.NewServeMux()
+	mux.HandleFunc("/desk/requirements/shot-remote", s.handleRequirementShotRemote)
+	srv := httptest.NewServer(mux)
+	defer srv.Close()
+	resp, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"https://example.com"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusNotImplemented {
+		t.Fatalf("缺能力位应 501，实际 %d", resp.StatusCode)
+	}
+}
+
+// readAllBody 把响应体读完转成字符串（测试用，错误一律 fatal）。
+func readAllBody(resp *http.Response) string {
+	b, _ := io.ReadAll(resp.Body)
+	resp.Body.Close()
+	return string(b)
+}

package/internal/client/apk.go

@@ -105,6 +105,11 @@ func (s *Service) handleAppInfo(w http.ResponseWriter, r *http.Request) {
 		Version string `json:"version"`
 		Size    int64  `json:"size"`
 		URL     string `json:"url"`
+		/* Upgrade（2026-10-26 只增字段）：这个客户端**自己能不能**做桌面端 npm 一键升级
+		   （npm 在 PATH 上、找得到目标二进制）。为 true 时界面的「检测更新」会顺手问
+		   /desk/core，一次把「App 有没有新版」「桌面 CLI 有没有新版」都报出来。
+		   旧插件没有这个字段 → 界面只查 APK，不假装能升桌面端。 */
+		Upgrade bool `json:"upgrade"`
 	}{}
 	if s.apkAvailable() {
 		if st, err := os.Stat(s.apkPath); err == nil {
@@ -113,6 +118,9 @@ func (s *Service) handleAppInfo(w http.ResponseWriter, r *http.Request) {
 			out.URL = "/desk/app.apk"
 		}
 	}
+	if s.upgradableCore() {
+		out.Upgrade = true
+	}
 	writeJSON(w, http.StatusOK, out)
 }
 

package/internal/client/apk_test.go

@@ -41,6 +41,7 @@ func TestAppInfoAndAPK(t *testing.T) {
 		Version string `json:"version"`
 		Size    int64  `json:"size"`
 		URL     string `json:"url"`
+		Upgrade bool   `json:"upgrade"`
 	}
 	if err := json.Unmarshal(rec.Body.Bytes(), &got); err != nil {
 		t.Fatal(err)
@@ -48,6 +49,12 @@ func TestAppInfoAndAPK(t *testing.T) {
 	if got.Version != "2.1.2" || got.URL != "/desk/app.apk" || got.Size != int64(len("fake-apk-2.1.2")) {
 		t.Fatalf("app info = %+v", got)
 	}
+	// 2026-10-26：/desk/app 要顺手告诉界面「这份客户端能不能一键升桌面端 npm」，
+	// 界面据此一次检测 APK + npm。值与本机真实探测一致（测试机上有没有 npm 不定），
+	// 不写死。
+	if got.Upgrade != s.upgradableCore() {
+		t.Fatalf("/desk/app.upgrade 与本机探测不一致：%+v", got)
+	}
 
 	rec = httptest.NewRecorder()
 	s.handleAppAPK(rec, httptest.NewRequest(http.MethodGet, "/desk/app.apk", nil))
@@ -87,6 +94,58 @@ func TestAppInfoWithoutAPK(t *testing.T) {
 	}
 }
 
+// 2026-10-26 用户：「版本检测需要同时考虑 apk 和 npm，都要支持升级」。
+// 升级链路（coreupgrade.go）早就写好、单测也一直在跑，但正式路由从没挂上 ——
+// 手机问 /desk/core 拿到的是 404 页而不是版本信息。这条钉住「真的挂上去了」。
+func TestVersionRoutesRegistered(t *testing.T) {
+	s, _ := newUpgradeService(t)
+	h := s.Handler()
+	hdr := map[string]string{"X-Magic-Token": s.opts.Token}
+	for _, path := range []string{"/desk/core", "/desk/core/upgrade"} {
+		req := httptest.NewRequest(http.MethodGet, path, nil)
+		for k, v := range hdr {
+			req.Header.Set(k, v)
+		}
+		rec := httptest.NewRecorder()
+		h.ServeHTTP(rec, req)
+		if rec.Code == http.StatusNotFound {
+			t.Fatalf("%s 没挂到正式路由上（404）", path)
+		}
+	}
+	// GET /desk/core 的响应形状：界面按这些字段决定「能不能一键升」。
+	req := httptest.NewRequest(http.MethodGet, "/desk/core", nil)
+	rec := httptest.NewRecorder()
+	h.ServeHTTP(rec, req)
+	if rec.Code != http.StatusOK {
+		t.Fatalf("/desk/core code = %d: %s", rec.Code, rec.Body.String())
+	}
+	var info coreInfo
+	if err := json.Unmarshal(rec.Body.Bytes(), &info); err != nil {
+		t.Fatal(err)
+	}
+	if !info.CanUpgrade || !info.Upgradable || info.Latest != "2.4.2" {
+		t.Fatalf("/desk/core = %+v", info)
+	}
+}
+
+// 找得到 npm + 目标二进制时，/desk/app 必须明说「能升桌面端」，界面才会去问 /desk/core。
+func TestAppInfoReportsNpmUpgradeAvailability(t *testing.T) {
+	s, _ := newUpgradeService(t)
+	apk := writeTestAPK(t, t.TempDir(), "2.4.1")
+	s.apkPath = apk
+	rec := httptest.NewRecorder()
+	s.handleAppInfo(rec, httptest.NewRequest(http.MethodGet, "/desk/app", nil))
+	var got struct {
+		Upgrade bool `json:"upgrade"`
+	}
+	if err := json.Unmarshal(rec.Body.Bytes(), &got); err != nil {
+		t.Fatal(err)
+	}
+	if !got.Upgrade {
+		t.Fatalf("/desk/app.upgrade = false，但假 npm 与目标二进制都在：%s", rec.Body.String())
+	}
+}
+
 func TestAPKVersionFallsBackToFilename(t *testing.T) {
 	dir := t.TempDir()
 	apk := filepath.Join(dir, "magic-agent-mobile-3.2.1-release.apk")