magic-agent 1.0.1 → 1.0.3

package/internal/cli/repair.go

@@ -0,0 +1,60 @@
+package cli
+
+// repair.go - `--repair-models <engine>`：还原被反向同步写坏的桌面端 models.json。
+//
+// 为什么要有它：2026-09-30 的 Windows 事故里，magic-agent 把桌面端（WorkBuddy）那份
+// **数组格式**的 models.json 覆盖成了 `{availableModels, models}` 对象格式，桌面端
+// 解析不了 → 模型清单整个为空。同步那一侧已经加了「陌生形状一律不动」的硬约束，
+// 但**已经坏掉的现场**需要一条能自己走回来的路 —— 否则用户只能手抄 apiKey。
+//
+// 还原逻辑在 agent.RepairDesktopModels（见 custom_models.go 的事故还原一节）：
+// 只认明确的坏形状才动手，动手前先留 .broken.bak。
+
+import (
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"github.com/darren/magic-agent/internal/agent"
+	"github.com/spf13/cobra"
+)
+
+// runRepairModels 处理 `--repair-models <engine>`。
+func runRepairModels(cmd *cobra.Command, opts *askOptions) error {
+	name := strings.TrimSpace(opts.repairModels)
+	if name == "" {
+		return &usageError{fmt.Errorf(
+			"--repair-models 需要引擎名：magic-agent --repair-models <engine>（可用：%s）",
+			strings.Join(agent.DesktopEngines(), " / "))}
+	}
+
+	// 引擎名先在这里校验：这是**参数错**（exit 2），而不是运行期失败（exit 1）。
+	// agent 层还会再兜一次，但那属于「不该发生」的安全网。
+	if agent.DesktopAppHomeOf(name) == "" {
+		return &usageError{fmt.Errorf("引擎 %q 没有对应的桌面端数据目录，没有可修的 models.json（可用：%s）",
+			name, strings.Join(agent.DesktopEngines(), " / "))}
+	}
+
+	repaired, note, err := agent.RepairDesktopModels(name)
+	if err != nil {
+		return err
+	}
+
+	// 机器可读的一行结果走 stdout（与 --sessions / --stop 同一族）；人看的说明走 stderr。
+	if eerr := json.NewEncoder(cmd.OutOrStdout()).Encode(map[string]any{
+		"engine":   name,
+		"repaired": repaired,
+		"note":     note,
+	}); eerr != nil {
+		return eerr
+	}
+	out := cmd.ErrOrStderr()
+	fmt.Fprintf(out, "magic-agent: %s\n", note)
+	if !repaired {
+		// 没改动不是失败，但必须说清「没动」，不静默（本项目铁律）。
+		fmt.Fprintf(out, "magic-agent: 未改动任何文件。形状对不上就不动，避免二次破坏。\n")
+		return nil
+	}
+	fmt.Fprintf(out, "magic-agent: 已还原。请重开桌面端确认模型清单；官方模型由桌面端自己拉取，不在这个文件里。\n")
+	return nil
+}

package/internal/cli/repair_test.go

@@ -0,0 +1,51 @@
+package cli
+
+// repair_test.go - `--repair-models` 的参数校验。
+//
+// 落地逻辑（形状识别 / 还原 / 备份 / 幂等）在 internal/agent/repair_models_test.go，
+// 这里只锁 CLI 层的契约：引擎名缺失或不支持桌面端时是**参数错**（exit 2），
+// 而不是被当成 prompt 发给引擎。
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestRepairModelsWithoutEngineNameIsUsageError(t *testing.T) {
+	_, _, err := runAskCmd(t, "", "--repair-models")
+	if err == nil {
+		t.Fatal("--repair-models 缺引擎名应报错")
+	}
+	if !strings.Contains(err.Error(), "需要引擎名") {
+		t.Errorf("应是中文用法提示，实际 %q", err.Error())
+	}
+	if _, ok := err.(*usageError); !ok {
+		t.Errorf("应是 usageError（exit 2），实际 %T", err)
+	}
+}
+
+// 显式给空串同样走校验（与 --login 同一取舍：用「传过没有」而不是「值非空」判断）。
+func TestRepairModelsEmptyEngineIsUsageError(t *testing.T) {
+	_, _, err := runAskCmd(t, "", "--repair-models", "")
+	if err == nil {
+		t.Fatal("空引擎名应报错")
+	}
+	if _, ok := err.(*usageError); !ok {
+		t.Errorf("应是 usageError（exit 2），实际 %T", err)
+	}
+}
+
+// 没有桌面端数据目录的引擎：参数错，并点名可用值。
+func TestRepairModelsUnsupportedEngine(t *testing.T) {
+	_, _, err := runAskCmd(t, "", "--repair-models", "claude")
+	if err == nil {
+		t.Fatal("claude 没有桌面端 models.json，应报错")
+	}
+	msg := err.Error()
+	if !strings.Contains(msg, "没有对应的桌面端数据目录") {
+		t.Errorf("错误应说明该引擎没有桌面端目录，实际 %q", msg)
+	}
+	if _, ok := err.(*usageError); !ok {
+		t.Errorf("应是 usageError（exit 2），实际 %T", err)
+	}
+}

package/internal/cli/root.go

@@ -129,6 +129,7 @@ func NewRootCommand() *cobra.Command {
   magic-agent --engines                           # 列出引擎、可用性与各引擎支持的模型（JSON 数组）
   magic-agent --engines --no-models               # 只列引擎与可用性（跳过模型探测，不启动 CLI）
   magic-agent --engines --json                    # 等价（--json 为兼容保留）
+  magic-agent --repair-models codebuddy           # 还原被覆盖写坏的桌面端 models.json（事故收尾，先留 .broken.bak）
 
 四档权限模型（--permission，仅 claude / codebuddy / codebuddy-ai 生效）：
   把「哪些动作自动放行 + 放行不了时由谁裁决」收敛成四档，映射到 Claude Code 的参数：
@@ -399,6 +400,9 @@ codebuddy-gateway 引擎（CodeBuddy Code HTTP 网关，走 webhook + SSE；凭
 			case flagChanged(cmd, "login"):
 				// 用「传过没有」而不是「值非空」判断：`--login ""` 也要走参数校验（提示要引擎名）
 				return runLogin(cmd, opts)
+			case flagChanged(cmd, "repair-models"):
+				// 同上：`--repair-models ""` 要走参数校验，而不是掉进「提问」分支报 empty prompt。
+				return runRepairModels(cmd, opts)
 			case opts.listSessions:
 				return runSessions(cmd, opts)
 			case flagChanged(cmd, "session-log"):
@@ -426,6 +430,23 @@ codebuddy-gateway 引擎（CodeBuddy Code HTTP 网关，走 webhook + SSE；凭
 		},
 	}
 	bindAskFlags(root, opts)
+	/* `--login` / `--repair-models` 忘带引擎名时，cobra 在解析阶段就报错，**走不到**
+	   runLogin / runRepairModels 的校验分支，默认文案是英文的
+	   "flag needs an argument: ..."（且退出码 1）。这里翻成与传空串一致的中文
+	   usage（exit 2），免得用户只看到一句看不懂的报错。 */
+	root.SetFlagErrorFunc(func(_ *cobra.Command, err error) error {
+		if err == nil || !strings.Contains(err.Error(), "flag needs an argument") {
+			return err
+		}
+		switch {
+		case strings.Contains(err.Error(), "login"):
+			return &usageError{fmt.Errorf("--login 需要引擎名：magic-agent --login <engine>（可用：%s）", loginCapableEngines())}
+		case strings.Contains(err.Error(), "repair-models"):
+			return &usageError{fmt.Errorf("--repair-models 需要引擎名：magic-agent --repair-models <engine>（可用：%s）",
+				strings.Join(agent.DesktopEngines(), " / "))}
+		}
+		return err
+	})
 	// 无子命令设计：禁用 cobra 自动注入的 completion 命令，
 	// 保证 "magic-agent completion" 也只是被当作 prompt 而非隐藏子命令。
 	root.CompletionOptions.DisableDefaultCmd = true

package/internal/cli/session.go

@@ -72,11 +72,18 @@ func stopActiveSession() {
 // 为什么必须做：引擎 CLI 在**独立进程组**里（Setpgid），调用方对 magic-agent
 // 发 SIGTERM / kill(-magic-agent-pid) 时，引擎子进程不在那一组里 → 会变成孤儿继续跑
 // （观物台「停止」按钮踩的就是这个）。这里在退出前补一刀。
+//
+// 覆盖三处孤儿来源：
+//   - 当前活动会话（stopActiveSession）；
+//   - 交互式登录子进程（stopLoginChild）—— 它必须待在前台组里读 tty，因而
+//     比普通引擎更「自己管不了」，只能在这里记一笔带走（见 login.go）；
+//   - 关终端（SIGHUP）：与 SIGTERM 同路。
 func installSignalStop() {
 	ch := make(chan os.Signal, 1)
-	signal.Notify(ch, os.Interrupt, syscall.SIGTERM)
+	signal.Notify(ch, os.Interrupt, syscall.SIGTERM, syscall.SIGHUP)
 	go func() {
 		<-ch
+		stopLoginChild()
 		stopActiveSession()
 		os.Exit(exitInterrupted)
 	}()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "magic-agent",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "专业的 agent CLI 代理工具 - 统一 claude / codebuddy / trae / llm / codex / openclaw 引擎",
   "license": "MIT",
   "bin": {
@@ -22,6 +22,9 @@
     "internal/",
     "go.mod",
     "go.sum",
+    "client-ui/",
+    "client-ui-mobile/",
+    "CLIENT-CONTRACT.md",
     "README.md",
     "LICENSE"
   ],