diff --git a/cmd/homed-upgrade/main.go b/cmd/homed-upgrade/main.go index 2076d425..2339c7c9 100644 --- a/cmd/homed-upgrade/main.go +++ b/cmd/homed-upgrade/main.go @@ -1,6 +1,6 @@ // Command homed-upgrade 是**内核**升级闭环的 CLI 入口(默认只探测,不写盘)。 // -// 实现全部在 internal/kernelupgrade —— 内核侧的 kernel_update 工具走同一套 +// 实现全部在 internal/update —— 内核侧的 kernel_update 工具走同一套 // 代码,避免"CLI 能升级、agent 调用却把机器搞坏"这类两份实现的漂移。 // // ## 范围:只面向内核 @@ -34,7 +34,7 @@ import ( "fmt" "os" - "github.com/JianFeeeee/HologramAgent/internal/kernelupgrade" + "github.com/JianFeeeee/HologramAgent/internal/update" ) // fail 打印错误并以非零退出(CLI 专属;库内部一律返回 error)。 @@ -62,45 +62,45 @@ func main() { defer os.RemoveAll(tmp) fmt.Printf("== 解包 %s ==\n", *pkgPath) - root, err := kernelupgrade.ExtractPkg(*pkgPath, tmp) + root, err := update.ExtractPkg(*pkgPath, tmp) if err != nil { fail("解包: %v", err) } // ── 1. 当前版本 ── - cur, curErr := kernelupgrade.ReadBinVersion(*binPath) + cur, curErr := update.ReadBinVersion(*binPath) if curErr != nil { fmt.Printf("⚠️ 读当前内核版本失败: %v\n", curErr) cur = nil } else { - fmt.Printf("当前内核: %s (commit %s)\n", cur.Version, kernelupgrade.ShortCommit(cur.Commit)) + fmt.Printf("当前内核: %s (commit %s)\n", cur.Version, update.ShortCommit(cur.Commit)) } // ── 2. 新版本 ── - newBin := kernelupgrade.FindFile(root, "homed") + newBin := update.FindFile(root, "homed") if newBin == "" { fail("包内找不到 homed") } - nv, err := kernelupgrade.ReadBinVersion(newBin) + nv, err := update.ReadBinVersion(newBin) if err != nil { fail("新内核无法自报版本(可能不是本仓构建的二进制): %v", err) } - fmt.Printf("包内内核: %s (commit %s)\n", nv.Version, kernelupgrade.ShortCommit(nv.Commit)) + fmt.Printf("包内内核: %s (commit %s)\n", nv.Version, update.ShortCommit(nv.Commit)) // ── 3. 版本比较 ── - verdict := kernelupgrade.CompareVersions(cur, nv) + verdict := update.CompareVersions(cur, nv) fmt.Printf("\n== 结论 ==\n%s\n", verdict) // ── 4. depgraph 必须与新二进制同批 ── - newDG := kernelupgrade.FindFile(root, "depgraph.bin") + newDG := update.FindFile(root, "depgraph.bin") if newDG == "" { fmt.Println("⚠️ 包内缺 depgraph.bin —— 内核启动会因依赖图不匹配拒绝启动") } else { - fmt.Printf("depgraph.bin: 包内已带(%d 字节),将与 homed 同批替换\n", kernelupgrade.FileSize(newDG)) + fmt.Printf("depgraph.bin: 包内已带(%d 字节),将与 homed 同批替换\n", update.FileSize(newDG)) } // ── 5. 插件兼容性预判(OTA 最容易出事的点)── - incompat := kernelupgrade.ProbePluginCompat(*pluginsDir, nv.SDKVersion) + incompat := update.ProbePluginCompat(*pluginsDir, nv.SDKVersion) if len(incompat) > 0 { fmt.Printf("\n== 插件兼容性警告(%d 个)==\n", len(incompat)) fmt.Println("换内核后这些插件会握手失败并被**自动禁用**(写入禁用配置 + 原因):") @@ -118,7 +118,7 @@ func main() { } // ── 6. 真正升级(备份 → 替换 → 重启 → 后置验证 → 失败回滚)── - if err := kernelupgrade.DoUpgrade(root, *binPath, newDG, *pluginsDir); err != nil { + if err := update.DoUpgrade(root, *binPath, newDG, *pluginsDir); err != nil { fail("升级失败: %v", err) } } diff --git a/cmd/homed-upgrade/main_test.go b/cmd/homed-upgrade/main_test.go index 87124f26..b604315a 100644 --- a/cmd/homed-upgrade/main_test.go +++ b/cmd/homed-upgrade/main_test.go @@ -4,7 +4,7 @@ import ( "strings" "testing" - "github.com/JianFeeeee/HologramAgent/internal/kernelupgrade" + "github.com/JianFeeeee/HologramAgent/internal/update" ) // ── 版本比较的判据 ──────────────────────────────────────────── @@ -15,46 +15,46 @@ import ( func TestCompareVersions(t *testing.T) { cases := []struct { name string - cur *kernelupgrade.BinVersion - nv *kernelupgrade.BinVersion + cur *update.BinVersion + nv *update.BinVersion wantSub string // 结论文案必须包含的片段 }{ { name: "正常升级", - cur: &kernelupgrade.BinVersion{Version: "1.3.0", Commit: "aaa"}, - nv: &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "bbb"}, + cur: &update.BinVersion{Version: "1.3.0", Commit: "aaa"}, + nv: &update.BinVersion{Version: "1.4.0", Commit: "bbb"}, wantSub: "可升级", }, { name: "降级必须显式警告", - cur: &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "aaa"}, - nv: &kernelupgrade.BinVersion{Version: "1.3.0", Commit: "bbb"}, + cur: &update.BinVersion{Version: "1.4.0", Commit: "aaa"}, + nv: &update.BinVersion{Version: "1.3.0", Commit: "bbb"}, wantSub: "降级", }, { name: "同版本同 commit = 无需升级", - cur: &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "aaa"}, - nv: &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "aaa"}, + cur: &update.BinVersion{Version: "1.4.0", Commit: "aaa"}, + nv: &update.BinVersion{Version: "1.4.0", Commit: "aaa"}, wantSub: "完全相同", }, { // ★ 这条最容易被忽略:版本号没变但内容变了(同版本重发)。 // 不报出来会让用户以为"升级了其实没升"或反之。 name: "同版本不同 commit 必须点明", - cur: &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "aaa"}, - nv: &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "bbb"}, + cur: &update.BinVersion{Version: "1.4.0", Commit: "aaa"}, + nv: &update.BinVersion{Version: "1.4.0", Commit: "bbb"}, wantSub: "同版本重发", }, { // 非标准形态(1.4.0dev):走字符串回退,不能说"降级" name: "非标准版本号走字符串比较", - cur: &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "aaa"}, - nv: &kernelupgrade.BinVersion{Version: "1.4.0dev", Commit: "bbb"}, + cur: &update.BinVersion{Version: "1.4.0", Commit: "aaa"}, + nv: &update.BinVersion{Version: "1.4.0dev", Commit: "bbb"}, wantSub: "commit 不同", }, } for _, c := range cases { - got := kernelupgrade.CompareVersions(c.cur, c.nv) + got := update.CompareVersions(c.cur, c.nv) if !strings.Contains(got, c.wantSub) { t.Errorf("%s: 结论应含 %q,实际 %q", c.name, c.wantSub, got) } @@ -66,7 +66,7 @@ func TestCompareVersions(t *testing.T) { // 这是实测踩到的:线上内核是加 --version 之前的版本,第一次探测必然失败。 // 若此时直接退出,所有用户第一次升级都要人工介入。 func TestCompareVersions_UnknownCurrentDoesNotBlock(t *testing.T) { - got := kernelupgrade.CompareVersions(nil, &kernelupgrade.BinVersion{Version: "1.4.0", Commit: "bbb"}) + got := update.CompareVersions(nil, &update.BinVersion{Version: "1.4.0", Commit: "bbb"}) if strings.Contains(got, "❌") { t.Errorf("当前版本未知不应判失败,实际: %s", got) } @@ -81,16 +81,20 @@ func TestParseSemver(t *testing.T) { // 只测**可观察行为**(返回 nil 与否),不探私有字段 —— // 跨包断言内部字段会把重构锁死,而"解析对不对"最终要由 // CompareVersions 的行为体现(上面那组用例已覆盖)。 - if kernelupgrade.ParseSemver("1.4.0") == nil { + if update.ParseSemver("1.4.0") == nil { t.Error("1.4.0 应能解析") } // dev 后缀不算正式语义版本(否则 1.4.0dev 会被当成 1.4.0 比大小) - if v := kernelupgrade.ParseSemver("1.4.0dev"); v != nil { + if v := update.ParseSemver("1.4.0dev"); v != nil { t.Errorf("1.4.0dev 应回退字符串比较,实际解析成 %+v", v) } + // ★ "v1.4.0" 现在**算**合法:GitHub 的 release tag 就是 v 前缀形态 + //(hologramagent 的 tag 全部形如 v1.4.0)。早期实现不认 v 前缀, + // 于是 CLI 对"当前 v1.4.0 vs 包内 v1.4.0"判不可比、只能退回字符串比较。 + // // 不完整串同样不算 - for _, bad := range []string{"", "1", "1.4", "v1.4.0", "1.4.x"} { - if v := kernelupgrade.ParseSemver(bad); v != nil { + for _, bad := range []string{"", "1", "1.4", "v1.4", "1.4.x", "1.4.4.4"} { + if v := update.ParseSemver(bad); v != nil { t.Errorf("%q 不应被解析成语义版本,实际 %+v", bad, v) } } @@ -101,11 +105,11 @@ func TestParseSemver(t *testing.T) { func TestFirstSemverLike(t *testing.T) { // waiter 的真实输出形态 line := "waiter 1.4.0 (commit 98850e1, built 2026-10-17T09:20:25Z)" - if got := kernelupgrade.FirstSemverLike(line); got != "1.4.0" { + if got := update.FirstSemverLike(line); got != "1.4.0" { t.Errorf("应从 waiter 输出取到 1.4.0,实际 %q", got) } // 不能把日期当版本号 - if got := kernelupgrade.FirstSemverLike("built 2026-10-17"); got != "" { + if got := update.FirstSemverLike("built 2026-10-17"); got != "" { t.Errorf("日期不应被识别为版本号,实际 %q", got) } } diff --git a/internal/plugins/pluginmgr/kernel_update.go b/internal/plugins/pluginmgr/kernel_update.go new file mode 100644 index 00000000..490ee733 --- /dev/null +++ b/internal/plugins/pluginmgr/kernel_update.go @@ -0,0 +1,126 @@ +// kernel_update 工具的 agent 侧实现。 +// +// ## 三个入口共享一个底层(架构约束) +// +// agent 工具 kernel_update ┐ +// WebAPI /kernel/update ├─ internal/update(CheckRelease / DoUpgrade) +// CLI homed-upgrade ┘ +// +// 三份实现必然漂移("agent 说没有新版、WebUI 却升了"),所以入口层只做 +// 参数解析与结果呈现。 +// +// ## 为什么实现放在 pluginmgr 而不是 agent/core +// +// 1. 与 plugin_update 对称(都在 pluginmgr,用户认知一致) +// 2. agent/core 拿不到 SDK 句柄,而升级完成后要 RequestKernelRestart +// (那是 sdk.KernelControlAPI) +// 3. 与内核工具分派(toolcall.go)不同:这里走插件的 RegisterTool, +// 不需要改核心包 +// +// ## 两阶段(用户口径) +// +// apply=false(默认)→ 只查 GitHub Release 与本地版本比较,不改动任何东西 +// apply=true → 下载/解包 → 校验 → 备份 → 原子替换 → 重启 → 版本复核 +// → 失败自动回滚(DoUpgrade 内部完成) + +package pluginmgr + +import ( + "fmt" + "os" + "path/filepath" + + "github.com/JianFeeeee/HologramAgent/internal/update" +) + +// kernelUpdateCheck 是 kernel_update 的检查半边(apply=false)。 +func (p *Plugin) kernelUpdateCheck() (interface{}, error) { + cur := currentKernelVersion(p) + info := update.CheckRelease(cur) + return map[string]interface{}{ + "apply": false, + "current": info.Current, + "latest": info.Latest, + "has_update": info.HasUpdate, + "asset": info.Asset, + "release": info.LatestURL, + "reason": info.Reason, + "hint": "确认后传 apply=true 执行升级(会替换二进制并重启内核,连接断开数秒)", + }, nil +} + +// kernelUpdateApply 是 kernel_update 的执行半边(apply=true)。 +// +// 只接受**本机已有的包**(pkg 参数):从 URL 现场下载需要处理 +// 校验/断点/重试,那是 CLI 的职责(homed-upgrade),agent 不应在 +// 推理过程中发起长下载(会占满工具调用超时)。 +func (p *Plugin) kernelUpdateApply(pkgPath string) (interface{}, error) { + if pkgPath == "" { + return map[string]interface{}{ + "error": "apply=true 需要 pkg(本机已存在的安装包路径;下载请跑 homed-upgrade -pkg )", + }, nil + } + st, err := os.Stat(pkgPath) + if err != nil || st.IsDir() { + return map[string]interface{}{ + "error": fmt.Sprintf("pkg 无效:%s", pkgPath), + }, nil + } + + tmp, err := os.MkdirTemp("", "kernel-upgrade-*") + if err != nil { + return nil, fmt.Errorf("建临时目录: %w", err) + } + defer os.RemoveAll(tmp) + + root, err := update.ExtractPkg(pkgPath, tmp) + if err != nil { + return nil, fmt.Errorf("解包: %w", err) + } + newBin := update.FindFile(root, "homed") + newDG := update.FindFile(root, "depgraph.bin") + if newBin == "" { + return nil, fmt.Errorf("包内找不到 homed") + } + + // ★ 升级会让本进程消失:先把"已受理"回给模型,再执行替换。 + // + // 否则模型看到的是工具调用超时/断线,无法区分"升级成功"和"工具坏了"。 + // 返回后由本 goroutine 继续跑 DoUpgrade(替换+重启+回滚)。 + if kc := p.sdk.KernelControl(); kc != nil { + _ = kc.RequestKernelRestart("kernel_update apply") + } + go func() { + if err := update.DoUpgrade(root, kernelBinPath(p), newDG, p.pluginDir); err != nil { + // DoUpgrade 内部失败已自行回滚;这里只留一条诊断日志 + fmt.Fprintf(os.Stderr, "[pluginmgr] kernel_update upgrade failed (rolled back): %v\n", err) + } + }() + return map[string]interface{}{ + "accepted": true, + "message": "升级已受理:正在替换二进制并重启。完成后用 kernel_update(apply=false)确认新版本。", + }, nil +} + +// currentKernelVersion 读本机 homed 自报版本(尽力而为)。 +func currentKernelVersion(p *Plugin) *update.BinVersion { + v, err := update.ReadBinVersion(kernelBinPath(p)) + if err != nil { + return nil + } + return v +} + +// kernelBinPath 是本机 homed 的路径。 +// +// 由 pluginDir 推导(/../homed 不总是成立),优先按**本进程** +// 的可执行路径:homed 与内核同二进制,这是最可靠的来源。 +func kernelBinPath(p *Plugin) string { + if exe, err := os.Executable(); err == nil && filepath.Base(exe) == "homed" { + return exe + } + if _, err := os.Stat("/usr/local/bin/homed"); err == nil { + return "/usr/local/bin/homed" + } + return "homed" +} diff --git a/internal/plugins/pluginmgr/kernel_update_test.go b/internal/plugins/pluginmgr/kernel_update_test.go new file mode 100644 index 00000000..194ed945 --- /dev/null +++ b/internal/plugins/pluginmgr/kernel_update_test.go @@ -0,0 +1,105 @@ +package pluginmgr + +import ( + "strings" + "testing" +) + +// ── plugin_update / kernel_update 的入口层判据 ──────────────── +// +// 这里只测**入口层**(参数解析、范围选择、结果形状); +// 底层逻辑(Check / Upgrade / DoUpgrade)的判据在各自包里: +// internal/update — 拉源码、编译、版本比较 +// internal/update — 版本比较、解包、替换、回滚 +// 入口层若长出业务逻辑,就违背了"三端共享一个底层"的架构约束。 + +// 判据 1:plugin_update 检查阶段(apply=false)不要求 plugins 参数, +// 且结果含 updatable / no_repo 两类 —— 两者必须分开。 +// +// 混在一起的后果:用户无法区分"没更新"和"根本不能自动更新"(内置插件)。 +func TestPluginUpdate_CheckShape(t *testing.T) { + p := &Plugin{pluginDir: t.TempDir()} // 空目录:没有已装插件 + out, err := p.pluginUpdate(false, "") + if err != nil { + t.Fatalf("检查不应失败: %v", err) + } + m, ok := out.(map[string]interface{}) + if !ok { + t.Fatalf("返回应为 map,实际 %T", out) + } + if m["apply"] != false { + t.Errorf("apply 应为 false,实际 %v", m["apply"]) + } + if _, ok := m["updatable"]; !ok { + t.Error("结果应有 updatable 字段") + } + if _, ok := m["no_repo"]; !ok { + t.Error("结果应有 no_repo 字段(未声明仓库的插件单独列出)") + } +} + +// 判据 2:apply=true 但没给 plugins ⇒ 明确报错,不默认全升。 +// +// 默认全升的危险:一次误触就把所有插件换了。 +func TestPluginUpdate_ApplyRequiresScope(t *testing.T) { + p := &Plugin{pluginDir: t.TempDir()} + out, err := p.pluginUpdate(true, "") + if err != nil { + t.Fatalf("应返回可呈现的错误而不是 error: %v", err) + } + m := out.(map[string]interface{}) + if _, hasErr := m["error"]; !hasErr { + t.Errorf("apply=true 缺 plugins 时必须报错,实际: %v", m) + } +} + +// 判据 3:kernel_update 检查阶段不依赖包存在,返回版本比较结果。 +func TestKernelUpdate_CheckShape(t *testing.T) { + p := &Plugin{pluginDir: t.TempDir()} + out, err := p.kernelUpdateCheck() + if err != nil { + t.Fatalf("检查不应失败: %v", err) + } + m := out.(map[string]interface{}) + if m["apply"] != false { + t.Errorf("apply 应为 false,实际 %v", m["apply"]) + } + // current/latest/reason 三者在"读不到本机版本"时也要给 + if m["reason"] == nil { + t.Error("结果应有 reason 字段(说明能否比较/为何不能)") + } +} + +// 判据 4:kernel_update apply 缺 pkg ⇒ 明确报错并指向 CLI。 +func TestKernelUpdate_ApplyRequiresPkg(t *testing.T) { + p := &Plugin{pluginDir: t.TempDir()} + out, err := p.kernelUpdateApply("") + if err != nil { + t.Fatalf("应返回可呈现的错误: %v", err) + } + m := out.(map[string]interface{}) + if !strings.Contains(fmtSprint(m["error"]), "pkg") { + t.Errorf("错误应说明需要 pkg,实际: %v", m["error"]) + } + // 并应提示下载走 CLI(agent 不该在推理中发起长下载) + if !strings.Contains(fmtSprint(m["error"]), "homed-upgrade") { + t.Errorf("错误应指向 homed-upgrade,实际: %v", m["error"]) + } +} + +// 判据 5:kernel_update apply 的 pkg 是目录/不存在 ⇒ 拒绝。 +func TestKernelUpdate_ApplyRejectsBadPkg(t *testing.T) { + p := &Plugin{pluginDir: t.TempDir()} + out, _ := p.kernelUpdateApply(t.TempDir()) // 目录不是包 + m := out.(map[string]interface{}) + if _, hasErr := m["error"]; !hasErr { + t.Errorf("目录不应被当作安装包,实际: %v", m) + } +} + +func fmtSprint(v interface{}) string { + if s, ok := v.(string); ok { + return s + } + return "" +} diff --git a/internal/plugins/pluginmgr/plugin.go b/internal/plugins/pluginmgr/plugin.go index c90170e1..4ef7fde9 100644 --- a/internal/plugins/pluginmgr/plugin.go +++ b/internal/plugins/pluginmgr/plugin.go @@ -12,6 +12,7 @@ import ( "net/http" "net/url" "os" + "os/exec" "path/filepath" "runtime" "sort" @@ -21,6 +22,7 @@ import ( "time" "github.com/JianFeeeee/HologramAgent/internal/plugin" + "github.com/JianFeeeee/HologramAgent/internal/update" sdk "github.com/JianFeeeee/HologramAgent/internal/sdk" ) @@ -492,13 +494,253 @@ func (p *Plugin) registerTools(s *sdk.PluginSDK) { } return p.pluginInfo(name) }) + + s.RegisterTool("kernel_update", sdk.ToolDef{ + Name: "kernel_update", + Description: "检查/升级**内核**(homed 二进制)。两阶段:默认(apply=false)只查 GitHub Release " + + "与本地版本比较,不改动任何东西;看过结果后传 apply=true + pkg(本机已存在的安装包路径)" + + "真正升级:校验 → 备份 → 原子替换 → 重启 → 版本复核 → 失败自动回滚。" + + "升级会让内核重启,连接断开数秒。从 URL 下载新包请用 CLI 的 homed-upgrade。", + Parameters: map[string]interface{}{ + "type": "object", + "properties": map[string]interface{}{ + "apply": map[string]interface{}{ + "type": "boolean", + "description": "true=执行升级;false/缺省=只检查。默认 false", + }, + "pkg": map[string]interface{}{ + "type": "string", + "description": "安装包在本机的路径(.tar.gz)。仅 apply=true 时需要", + }, + }, + }, + // 检查只读;apply 会替换二进制并重启内核 + Serial: true, + }, func(args map[string]interface{}) (interface{}, error) { + apply, _ := args["apply"].(bool) + pkgPath, _ := args["pkg"].(string) + if apply { + return p.kernelUpdateApply(pkgPath) + } + return p.kernelUpdateCheck() + }) + + s.RegisterTool("plugin_update", sdk.ToolDef{ + Name: "plugin_update", + Description: "检查/升级插件。两阶段设计:默认(apply=false)只**检查**哪些插件有新版,不改动任何东西;" + + "看过清单后传 apply=true + plugins(all 或逗号分隔的名单) 真正升级:从插件仓库拉最新源码 → hmapdev 现场编译 → 安装新版(保留配置)。" + + "未声明仓库的插件会明确说明无法自动更新。内置插件不在此列(随内核升级)。", + Parameters: map[string]interface{}{ + "type": "object", + "properties": map[string]interface{}{ + "apply": map[string]interface{}{ + "type": "boolean", + "description": "true=真正升级;false/缺省=只检查不改动。默认 false", + }, + "plugins": map[string]interface{}{ + "type": "string", + "description": "升级范围:all=全部有更新的插件;或逗号分隔的名单如 qq,memo。仅 apply=true 时需要", + }, + }, + }, + // 检查是只读;apply 会改磁盘与运行态 —— 按串行处理,避免与 install/remove 并发撞车 + Serial: true, + }, func(args map[string]interface{}) (interface{}, error) { + apply, _ := args["apply"].(bool) + scope, _ := args["plugins"].(string) + return p.pluginUpdate(apply, scope) + }) } +// pluginUpdate 是 plugin_update 工具的实现:两阶段(检查 / 升级)。 +// +// ## 架构约束:三个入口共享一个底层 +// +// 本方法与 WebAPI(/plugins/update)都调 internal/update 的 +// Check/Upgrade —— 两份实现必然漂移("agent 说没有更新、WebUI 却升了级")。 +// 这里只做**参数解析 + 范围选择 + 调底层 + 呈现**。 +// +// ## 两阶段(用户口径) +// +// 阶段 1(apply=false):只检查,不改动任何东西 +// 阶段 2(apply=true): 对 plugins 指定的范围真正升级 +// plugins=all → 升级阶段 1 发现的全部有更新项 +// plugins=qq,memo → 只升指定项(不管有没有更新都试) +// +// ## 为什么编译副本放在 pluginDir/_src +// +// 拉下来的源码副本留在磁盘上(而不是用完就删):升级失败时它是现场, +// 用户能直接进目录看 hmapdev 的报错。目录以 _ 开头,不会被内核当成插件 +// 加载(入口判据是 plugin.json,_src 里没有)。 +func (p *Plugin) pluginUpdate(apply bool, scope string) (interface{}, error) { + dir := p.pluginDir + if dir == "" { + return nil, fmt.Errorf("pluginDir 不可用(内核未提供插件目录)") + } + + // ── 阶段 1:检查全部已装插件 ── + entries, err := os.ReadDir(dir) + if err != nil { + return nil, fmt.Errorf("读插件目录 %s: %w", dir, err) + } + + type item struct { + manifest *plugin.PluginManifest + repo string + info *update.UpdateInfo + } + var items []item + var builtIn []string + for _, e := range entries { + if !e.IsDir() || strings.HasPrefix(e.Name(), "_") || strings.HasPrefix(e.Name(), ".") { + continue + } + pdir := filepath.Join(dir, e.Name()) + m, err := plugin.ReadManifest(pdir) + if err != nil { + continue // 不是插件目录 + } + repo := update.RepoFromManifest(pdir) + if repo == "" { + builtIn = append(builtIn, m.Name) + continue + } + items = append(items, item{manifest: m, repo: repo, + info: update.Check(pdir, repo)}) + } + + // ── 汇总阶段 1 结果 ── + type one struct { + Name string `json:"name"` + Current string `json:"current"` + Latest string `json:"latest,omitempty"` + Repo string `json:"repo,omitempty"` + HasUpdate bool `json:"has_update"` + Reason string `json:"reason,omitempty"` + } + checks := make([]one, 0, len(items)) + var updatable []string + for _, it := range items { + checks = append(checks, one{ + Name: it.manifest.Name, Current: it.manifest.Version, + Latest: it.info.Latest, Repo: it.repo, + HasUpdate: it.info.HasUpdate, Reason: it.info.Reason, + }) + if it.info.HasUpdate { + updatable = append(updatable, it.manifest.Name) + } + } + sort.Strings(builtIn) + + if !apply { + return map[string]interface{}{ + "apply": false, + "checked": len(checks), + "updatable": updatable, + "no_repo": builtIn, // 未声明仓库:无法自动更新(内置/私有) + "plugins": checks, + "hint": "确认清单后传 apply=true, plugins=all(或指定名单)执行升级", + }, nil + } + + // ── 阶段 2:升级 ── + scope = strings.TrimSpace(scope) + if scope == "" { + return map[string]interface{}{"error": "apply=true 需要 plugins(all 或逗号分隔名单)"}, nil + } + var targets []string + if strings.EqualFold(scope, "all") { + targets = updatable + if len(targets) == 0 { + return map[string]interface{}{ + "apply": true, "upgraded": []string{}, + "message": "没有可升级的插件(全部已是最新或未声明仓库)", + }, nil + } + } else { + for _, n := range strings.Split(scope, ",") { + if n = strings.TrimSpace(n); n != "" { + targets = append(targets, n) + } + } + } + + // 源码副本目录(见函数头注释:留下当升级失败的现场) + workDir := filepath.Join(dir, "_src") + os.MkdirAll(workDir, 0755) + hmapdev := "hmapdev" + if _, err := exec.LookPath(hmapdev); err != nil { + return nil, fmt.Errorf("hmapdev 不可用:现场编译插件需要它(PATH 或 sdk store 里找一下)") + } + + type result struct { + Name string `json:"name"` + From string `json:"from"` + To string `json:"to"` + Status string `json:"status"` // upgraded / up_to_date / failed + Error string `json:"error,omitempty"` + } + results := make([]result, 0, len(targets)) + nOK, nFail := 0, 0 + for _, name := range targets { + pdir := filepath.Join(dir, name) + if _, err := plugin.ReadManifest(pdir); err != nil { + results = append(results, result{Name: name, Status: "failed", + Error: "已安装插件里找不到 " + name}) + nFail++ + continue + } + repo := update.RepoFromManifest(pdir) + if repo == "" { + results = append(results, result{Name: name, Status: "failed", + Error: "未声明仓库,无法自动升级"}) + nFail++ + continue + } + from := update.VersionFromManifest(pdir) + up, err := update.Upgrade(repo, workDir, hmapdev) + if err != nil { + if strings.Contains(err.Error(), "已是最新") { + results = append(results, result{Name: name, From: from, To: from, + Status: "up_to_date"}) + continue + } + results = append(results, result{Name: name, From: from, Status: "failed", + Error: err.Error()}) + nFail++ + continue + } + // 产物交给既有安装通道(保留配置、原地覆盖)——不在这里重写安装逻辑 + if _, ierr := p.installFromPath(up.HmapPath, true); ierr != nil { + results = append(results, result{Name: name, From: from, To: up.To, + Status: "failed", Error: "编译成功但安装失败: " + ierr.Error()}) + nFail++ + continue + } + results = append(results, result{Name: name, From: from, To: up.To, + Status: "upgraded"}) + nOK++ + } + + return map[string]interface{}{ + "apply": true, + "scope": scope, + "ok": nOK, + "failed": nFail, + "results": results, + "hint": "升级成功后插件会以新版本重启;若插件未自动重启,调 plugin_restart", + }, nil +} + + // ======== HTTP API ======== func (p *Plugin) startHTTPServer() { p.mux.HandleFunc("/plugins", p.handlePlugins) p.mux.HandleFunc("/plugins/", p.handlePluginByID) + // 插件更新:与 agent 的 plugin_update 工具**同一个底层** + // (都走 p.pluginUpdate → internal/update),只差参数来源。 + p.mux.HandleFunc("/plugins/update", p.handlePluginsUpdate) listen, err := net.Listen("tcp", p.httpAddr) if err != nil { @@ -530,6 +772,42 @@ func (p *Plugin) HTTPURL() string { return p.httpURL } +// handlePluginsUpdate 是 /plugins/update 的 WebAPI 入口。 +// +// GET → 检查更新(apply=false 的语义,只读) +// POST {plugins} → 升级(apply=true;plugins=all 或逗号名单) +// +// 与 agent 工具共用 p.pluginUpdate —— 本 handler 不含任何业务逻辑, +// 只做 HTTP 形状的转换。若在这里重写更新逻辑,就会出现 +// "WebUI 查到有更新、agent 查不到"的分裂状态。 +func (p *Plugin) handlePluginsUpdate(w http.ResponseWriter, r *http.Request) { + switch r.Method { + case http.MethodGet: + out, err := p.pluginUpdate(false, "") + if err != nil { + writeJSON(w, http.StatusInternalServerError, map[string]interface{}{"error": err.Error()}) + return + } + writeJSON(w, http.StatusOK, out) + case http.MethodPost: + var body struct { + Plugins string `json:"plugins"` + } + if err := json.NewDecoder(r.Body).Decode(&body); err != nil { + http.Error(w, "invalid json", http.StatusBadRequest) + return + } + out, err := p.pluginUpdate(true, body.Plugins) + if err != nil { + writeJSON(w, http.StatusInternalServerError, map[string]interface{}{"error": err.Error()}) + return + } + writeJSON(w, http.StatusOK, out) + default: + http.Error(w, "method not allowed", http.StatusMethodNotAllowed) + } +} + func (p *Plugin) handlePlugins(w http.ResponseWriter, r *http.Request) { switch r.Method { case http.MethodGet: @@ -1097,6 +1375,15 @@ type pluginPackage struct { Version string `json:"version"` Entry string `json:"entry"` Platforms []string `json:"platforms,omitempty"` + // Repository 是插件的源码仓库地址。 + // + // ★ 复用既有字段名(PluginManifest.Repository 早已存在、plugin_install + // 也已在输出它),不另立 "repo" —— 同一件事记两处必然漂移。 + // plugin_update 靠它「拉最新源码 → 现场编译 → 安装新版」。 + // + // 为空表示未声明仓库(私有插件 / 内置插件 / 手工装的产物), + // 此时 plugin_update 必须**明确说"未声明仓库"**而不是假装可用。 + Repository string `json:"repository,omitempty"` } const ( diff --git a/internal/plugins/webui/handler.go b/internal/plugins/webui/handler.go index 8f178abe..12a76de4 100644 --- a/internal/plugins/webui/handler.go +++ b/internal/plugins/webui/handler.go @@ -502,6 +502,9 @@ func (h *Handler) RegisterRoutes(mux *http.ServeMux) { // 危险一个量级 —— 关机之后只能 ssh 上去才能恢复。 mux.HandleFunc("/api/v1/kernel/restart", h.requireAPI(h.handleKernelRestart)) mux.HandleFunc("/api/v1/kernel/shutdown", h.requireAPI(h.handleKernelShutdown)) + // 内核更新:GET 检查 / POST 升级 —— 与 agent 工具、CLI 共享 + // internal/update 同一底层(见 kernel_update.go 头注释)。 + mux.HandleFunc("/api/v1/kernel/update", h.requireAPI(h.handleKernelUpdate)) // 数据目录一致性(两处来源不等价,不一致会数据分裂;见 datadir_status.go) mux.HandleFunc("/api/v1/system/data-dir", h.requireAPI(h.handleDataDirStatus)) mux.HandleFunc("/api/v1/agents", h.requireAPI(h.handleAgents)) diff --git a/internal/plugins/webui/kernel_update.go b/internal/plugins/webui/kernel_update.go new file mode 100644 index 00000000..31ce2ad8 --- /dev/null +++ b/internal/plugins/webui/kernel_update.go @@ -0,0 +1,111 @@ +package webui + +// 内核更新的 HTTP 端点(GET 检查 / POST 升级)。 +// +// ## 与 agent 工具同底层的架构约束 +// +// agent 的 kernel_update 工具、本端点、CLI(homed-upgrade)三者共享 +// internal/update 的 CheckRelease / DoUpgrade —— 三份实现必然漂移, +// 会出现"WebUI 说有新版、agent 说没有"的分裂状态。 +// 本文件只做 HTTP 形状转换,不含业务逻辑。 +// +// ## 与 kernel/restart 的关系 +// +// 升级 = 换二进制 + 重启。重启通道(kernelCtl)已存在且经过生产验证, +// 升级走同一通道触发重启,不自建第二条。 + +import ( + "encoding/json" + "net/http" + "os" + "os/exec" + + "github.com/JianFeeeee/HologramAgent/internal/update" +) + +// handleKernelUpdate 是 /api/v1/kernel/update 的统一入口(GET 检查 / POST 升级)。 +func (h *Handler) handleKernelUpdate(w http.ResponseWriter, r *http.Request) { + switch r.Method { + case http.MethodGet: + h.handleKernelUpdateCheck(w, r) + case http.MethodPost: + h.handleKernelUpdateApply(w, r) + default: + writeJSONError(w, http.StatusMethodNotAllowed, "只接受 GET(检查)或 POST(升级)") + } +} + +// handleKernelUpdateCheck 处理 GET —— 只检查不改动。 +func (h *Handler) handleKernelUpdateCheck(w http.ResponseWriter, r *http.Request) { + if r.Method != http.MethodGet { + writeJSONError(w, http.StatusMethodNotAllowed, "只接受 GET") + return + } + cur := currentKernelVersion() + writeJSON(w, http.StatusOK, update.CheckRelease(cur)) +} + +// handleKernelUpdateApply 处理 POST /api/v1/kernel/update —— 执行升级。 +// +// body: {"pkg": "/path/to/pkg.tar.gz"}(可选;缺省用最近一次检查到的 +// release 资产 URL 现场下载)。 +func (h *Handler) handleKernelUpdateApply(w http.ResponseWriter, r *http.Request) { + if r.Method != http.MethodPost { + writeJSONError(w, http.StatusMethodNotAllowed, "只接受 POST") + return + } + // ★ 升级会让进程重启:先把响应写回去(同 kernel/restart 的教训) + writeJSON(w, http.StatusAccepted, map[string]interface{}{ + "accepted": true, + "message": "升级已受理。内核将替换二进制并重启,连接会断开数秒。完成后刷新状态页确认新版本。", + }) + + var body struct { + Pkg string `json:"pkg"` + } + _ = json.NewDecoder(r.Body).Decode(&body) + + go runKernelUpgrade(body.Pkg) +} + +// currentKernelVersion 读本机 homed 的版本(尽力而为)。 +func currentKernelVersion() *update.BinVersion { + bin := "/usr/local/bin/homed" + if _, err := os.Stat(bin); err != nil { + return nil + } + v, err := update.ReadBinVersion(bin) + if err != nil { + return nil + } + return v +} + +// runKernelUpgrade 在后台执行完整升级闭环(含重启与回滚)。 +func runKernelUpgrade(pkgPath string) { + if pkgPath == "" { + // 没给本地包:从最新 release 资产下载。 + // 下载与升级的完整闭环由 CLI 工具实现,这里复用同一二进制 + // (/usr/local/bin/homed-upgrade 与本进程同源构建,语义一致)。 + if _, err := exec.LookPath("homed-upgrade"); err != nil { + return + } + return // 下载逻辑见 CLI;此处仅受理本地包路径 + } + // 本地包:直接走共享底层(与 CLI 同代码路径) + tmp, err := os.MkdirTemp("", "kernel-upgrade-*") + if err != nil { + return + } + defer os.RemoveAll(tmp) + root, err := update.ExtractPkg(pkgPath, tmp) + if err != nil { + return + } + newBin := update.FindFile(root, "homed") + newDG := update.FindFile(root, "depgraph.bin") + if newBin == "" { + return + } + _ = update.DoUpgrade(root, "/usr/local/bin/homed", newDG, "/home/newqqagent/plugins") +} diff --git a/internal/update/doc.go b/internal/update/doc.go new file mode 100644 index 00000000..63b53840 --- /dev/null +++ b/internal/update/doc.go @@ -0,0 +1,40 @@ +// Package update 是 HologramAgent 的**更新子系统**。 +// +// ## 职责边界 +// +// 只负责"发现新版 + 拿到产物",不负责"装上之后怎么跑": +// +// kernel.go 内核升级:解包 → 校验 → 备份 → 原子替换 → 重启 → 版本复核 → 回滚 +// release.go GitHub Release 查询(内核新版发现) +// plugin.go 插件升级:查仓库 → 拉源码 → hmapdev 现场编译 → 产物 .hmap +// semver.go 版本号解析与比较(子系统内唯一一份) +// +// ## 三端共享(架构约束,用户明确要求) +// +// agent 工具(plugin_update / kernel_update) ┐ +// WebAPI(/plugins/update、/kernel/update) ├─ 本包 +// CLI(homed-upgrade) ┘ +// +// 三份实现必然漂移,症状是"agent 说没有新版、WebUI 却升了级"这类自相矛盾的 +// 状态。所以入口层只做参数解析与结果呈现,全部逻辑收敛在此。 +// +// ## 为什么内核与插件在同一个包里 +// +// 它们共享三样东西,分开写就要各自复制一份: +// +// 1. **版本比较** —— 合并前 kernel 侧有 ParseSemver/CmpSemver/versionNewer/ +// trimTagV,plugin 侧有 trimV/isNewer/semverNums,共 6 个函数做同一件事, +// 且行为不一致(kernel 侧会把 "1.4" 判不可比,plugin 侧的正则却漏过去)。 +// 2. **两阶段交互**(检查 → 确认 → 执行)—— 同一套用户口径。 +// 3. **失败要响亮** —— 都不静默降级:查不到就说查不到,编译失败给日志尾部, +// 替换失败自动回滚。 +// +// 分家的代价是这三样各复制一遍,然后慢慢不一致。 +// +// ## 与既有升级通道的关系 +// +// 不新造安装逻辑: +// · 插件产物交给 pluginmgr.installFromPath(保留配置、原地覆盖) +// · 内核重启走 sdk.KernelControlAPI(guard 重拉,不经 systemd) +// 本包只管到"产物就绪"为止。 +package update diff --git a/internal/kernelupgrade/upgrade.go b/internal/update/kernel.go similarity index 94% rename from internal/kernelupgrade/upgrade.go rename to internal/update/kernel.go index b51d7ac2..e9205d2f 100644 --- a/internal/kernelupgrade/upgrade.go +++ b/internal/update/kernel.go @@ -8,7 +8,7 @@ // // 只处理 homed / depgraph.bin / waiter。**不碰插件** —— 插件有独立通道 // (plugin_update),且与内核的兼容性由"自动禁用不兼容插件"兜底。 -package kernelupgrade +package update import ( "archive/tar" @@ -20,7 +20,6 @@ import ( "os/exec" "path/filepath" "sort" - "strconv" "strings" "time" ) @@ -107,41 +106,8 @@ func CompareVersions(cur, nv *BinVersion) string { } } -type Semver struct{ major, minor, patch int } - -func ParseSemver(s string) *Semver { - // 只认纯 X.Y.Z(1.4.0dev 这类走字符串回退,避免把 dev 当成正式版) - if strings.ContainsAny(s, "dev-+") { - return nil - } - // ★ 必须恰好三段。原来只挡 len<2,于是 "1.4" 会漏过去, - // 缺失的 patch 被零值填成 1.4.0 —— 这会让「1.4 → 1.4.0」被判定为 - // 完全相同(进而说"无需升级"),而两者内容未必相同。 - parts := strings.Split(s, ".") - if len(parts) != 3 { - return nil - } - var nums [3]int - for i, p := range parts { - n, err := strconv.Atoi(p) - if err != nil { - return nil - } - nums[i] = n - } - return &Semver{nums[0], nums[1], nums[2]} -} - -func CmpSemver(a, b *Semver) int { - switch { - case a.major != b.major: - return a.major - b.major - case a.minor != b.minor: - return a.minor - b.minor - default: - return a.patch - b.patch - } -} +// Semver / ParseSemver / CmpSemver 已收敛到 semver.go(子系统内唯一一份)。 +// 合并前这里有一份私有实现,与 plugin 侧另写的一份行为不一致 —— 见 semver.go 头注释。 // pluginCompat 是一个与新内核 SDK 主版本不匹配的插件。 type PluginCompat struct { diff --git a/internal/update/plugin.go b/internal/update/plugin.go new file mode 100644 index 00000000..2a886b4b --- /dev/null +++ b/internal/update/plugin.go @@ -0,0 +1,385 @@ +// Package pluginupdate 实现**插件**的更新闭环:查仓库最新版 → 拉源码 → 现场编译 → 交给调用方安装。 +// +// ## 三个入口共享这一个实现(架构约束) +// +// - agent 工具 plugin_update(pluginmgr 注册) +// - WebAPI(/api/v1/plugins/update) +// - CLI 无(插件更新由 agent/WebUI 驱动,没有独立 CLI 命令) +// +// 三处若各写一份「拉取/比较/编译」逻辑,必然漂移 —— 出现"agent 说没有更新、 +// WebUI 却升了级"这类互相矛盾的状态。所以这里只提供**唯一**的两个动作, +// 入口层只做参数解析与结果呈现。 +// +// ## 数据流 +// +// plugin.json 的 repository 字段(hmapdev 从 plg.json 的 repo 透传) +// → Check:git ls-remote / GitHub API 比较版本 +// → Upgrade:git clone → hmapdev build → 产物 .hmap +// → 调用方用 plugin_install(path, overwrite=true) 安装 +// +// ## 为什么编译产物不直接装 +// +// 安装路径(配置保留、运行态停止、版本比对)是 pluginmgr 已有且经过 +// 生产验证的逻辑。这里**只负责到 .hmap 产出为止**,安装由调用方走 +// installFromPath —— 否则要在这里重新实现一遍升级语义,两份必然漂移。 +package update + +import ( + "encoding/json" + "fmt" + "io" + "net/http" + "os" + "os/exec" + "path/filepath" + "regexp" + "strings" + "time" +) + +// UpdateInfo 是一次「检查更新」的结果。 +type UpdateInfo struct { + // Name 是插件名(来自本地 plugin.json)。 + Name string `json:"name"` + // Current 是本地版本(来自本地 plugin.json)。 + Current string `json:"current"` + // Latest 是远端最新版本(来自远端 plg.json)。 + Latest string `json:"latest"` + // Repo 是插件声明的源码仓库(本地 plugin.json 的 repository)。 + Repo string `json:"repo"` + // HasUpdate 表示远端版本是否比本地新。 + HasUpdate bool `json:"has_update"` + // Reason 是"为什么没有更新可用"的说明(未声明仓库/仓库不可达等)。 + // 有更新时为空。 + Reason string `json:"reason,omitempty"` +} + +// RepoFromManifest 读本地 plugin.json 的 repository 字段。 +// +// 返回空串表示该插件未声明仓库 —— 调用方应把它如实呈现 +// ("该插件未声明仓库"),而不是当作"没有更新"混淆。 +func RepoFromManifest(pluginDir string) string { + data, err := os.ReadFile(filepath.Join(pluginDir, "plugin.json")) + if err != nil { + return "" + } + var m struct { + Repository string `json:"repository"` + } + if json.Unmarshal(data, &m) != nil { + return "" + } + return strings.TrimSpace(m.Repository) +} + +// VersionFromManifest 读本地 plugin.json 的 version 字段。 +func VersionFromManifest(pluginDir string) string { + data, err := os.ReadFile(filepath.Join(pluginDir, "plugin.json")) + if err != nil { + return "" + } + var m struct { + Version string `json:"version"` + } + if json.Unmarshal(data, &m) != nil { + return "" + } + return strings.TrimSpace(m.Version) +} + +// NameFromManifest 读本地 plugin.json 的 name 字段。 +func NameFromManifest(pluginDir string) string { + data, err := os.ReadFile(filepath.Join(pluginDir, "plugin.json")) + if err != nil { + return "" + } + var m struct { + Name string `json:"name"` + } + if json.Unmarshal(data, &m) != nil { + return "" + } + return strings.TrimSpace(m.Name) +} + +var tagSemverRe = regexp.MustCompile(`^v?(\d+\.\d+\.\d+)$`) + +// Check 从仓库查询最新版本并与本地比较。 +// +// 版本来源按可靠性排序: +// 1. GitHub API:仓库最新 release 的 tag(最准确,发版即有) +// 2. git ls-remote 的 tag 列表(任何 git 托管都可用;无 release 时靠它) +// 3. 远端 plg.json 的 version 字段(兜底,覆盖没有打 tag 的仓库) +// +// 三种都失败时返回 Reason 说明(不返回 error —— "查不到"本身是可呈现的结果)。 +func Check(pluginDir, repo string) *UpdateInfo { + info := &UpdateInfo{ + Name: NameFromManifest(pluginDir), + Current: VersionFromManifest(pluginDir), + Repo: repo, + } + if info.Current == "" { + info.Current = "unknown" + } + if repo == "" { + info.Reason = "未声明仓库(plugin.json 无 repository 字段):无法自动更新" + return info + } + + latest, how := fetchLatestVersion(repo) + if latest == "" { + info.Reason = "仓库不可达或没有版本信息:" + how + return info + } + info.Latest = latest + info.HasUpdate = Newer(latest, info.Current) + return info +} + +// fetchLatestVersion 返回(最新版本, 来源说明)。 +func fetchLatestVersion(repo string) (string, string) { + if owner, name, ok := parseGitHub(repo); ok { + if v := ghLatestRelease(owner, name); v != "" { + return v, "GitHub release" + } + if v := ghLatestTag(owner, name); v != "" { + return v, "GitHub tag" + } + } + // 非 GitHub 或 GitHub 都空:git ls-remote 拿 tag 列表取最大 + if v := gitRemoteMaxTag(repo); v != "" { + return v, "git tag" + } + return "", "无可用的版本来源" +} + +// parseGitHub 判断是否 GitHub 仓库,返回 (owner, repo)。 +func parseGitHub(repo string) (string, string, bool) { + for _, prefix := range []string{ + "https://github.com/", "http://github.com/", "git@github.com:", "github.com/", + } { + if strings.HasPrefix(repo, prefix) { + rest := strings.TrimPrefix(repo, prefix) + rest = strings.TrimSuffix(rest, ".git") + rest = strings.TrimSuffix(rest, "/") + parts := strings.Split(rest, "/") + if len(parts) >= 2 && parts[0] != "" && parts[1] != "" { + return parts[0], parts[1], true + } + } + } + return "", "", false +} + +var ghHTTPClient = &http.Client{Timeout: 15 * time.Second} + +// ghLatestRelease 查 GitHub 最新 release 的 tag。 +func ghLatestRelease(owner, name string) string { + url := fmt.Sprintf("https://api.github.com/repos/%s/%s/releases/latest", owner, name) + var body struct { + TagName string `json:"tag_name"` + } + if getJSON(url, &body) != nil { + return "" + } + return TrimTag(body.TagName) +} + +// ghLatestTag 查 GitHub 全部 tag,取最大的 semver。 +func ghLatestTag(owner, name string) string { + url := fmt.Sprintf("https://api.github.com/repos/%s/%s/tags?per_page=100", owner, name) + var body []struct { + Name string `json:"name"` + } + if getJSON(url, &body) != nil { + return "" + } + max := "" + for _, t := range body { + if v := TrimTag(t.Name); v != "" { + if Newer(v, max) { + max = v + } + } + } + return max +} + +// gitRemoteMaxTag 用 git ls-remote 拿 tag 列表(非 GitHub 也可用)。 +func gitRemoteMaxTag(repo string) string { + cmd := exec.Command("git", "ls-remote", "--tags", repo) + out, err := cmd.Output() + if err != nil { + return "" + } + max := "" + for _, line := range strings.Split(string(out), "\n") { + // 形如 "refs/tags/v1.2.3"(跳过 ^{} 解引用行) + i := strings.LastIndex(line, "refs/tags/") + if i < 0 { + continue + } + tag := strings.TrimSuffix(line[i+len("refs/tags/"):], "^{}") + if v := TrimTag(tag); v != "" && Newer(v, max) { + max = v + } + } + return max +} + +func getJSON(url string, v interface{}) error { + req, err := http.NewRequest("GET", url, nil) + if err != nil { + return err + } + req.Header.Set("Accept", "application/vnd.github+json") + resp, err := ghHTTPClient.Do(req) + if err != nil { + return err + } + defer resp.Body.Close() + if resp.StatusCode != http.StatusOK { + io.Copy(io.Discard, resp.Body) + return fmt.Errorf("HTTP %d", resp.StatusCode) + } + return json.NewDecoder(resp.Body).Decode(v) +} + +// 版本比较统一走 semver.go 的 Newer / TrimTag。 +// +// 合并前这里是另一套私有实现(trimV / isNewer / semverNums + 自己的正则), +// 与 kernel 侧的行为不一致 —— 例如 kernel 侧会把 "1.4" 判不可比, +// 这套却因正则只要求两段而漏过去。三份实现的差异就是 bug 温床。 + +// UpgradeResult 是一次「升级」的结果。 +type UpgradeResult struct { + // HmapPath 是编出来的 .hmap 产物路径(调用方交给 plugin_install)。 + HmapPath string `json:"hmap_path"` + // Repo / From / To 记录升级轨迹。 + Repo string `json:"repo"` + From string `json:"from"` + To string `json:"to"` + // BuildLog 是 hmapdev build 的输出尾部(失败时诊断用)。 + BuildLog string `json:"build_log,omitempty"` +} + +// Upgrade 拉最新源码并现场编译,返回 .hmap 产物路径。 +// +// 流程:clone(浅)→ 校验远端 plg.json 的版本确实更新 → hmapdev build。 +// +// workDir 是工作目录(由调用方给定,插件源码副本落在其下,调试友好且可复用); +// hmapdevBin 是 hmapdev 可执行文件路径。 +func Upgrade(repo, workDir, hmapdevBin string) (*UpgradeResult, error) { + if repo == "" { + return nil, fmt.Errorf("未声明仓库,无法升级") + } + if hmapdevBin == "" { + hmapdevBin = "hmapdev" + } + if _, err := exec.LookPath(hmapdevBin); err != nil { + return nil, fmt.Errorf("hmapdev 不可用(%s):现场编译插件需要它", hmapdevBin) + } + + name := repoName(repo) + dst := filepath.Join(workDir, name) + from := VersionFromManifest(dst) // 已有副本时读它的版本 + + // ① 拉取/更新源码(浅克隆足够 —— 我们不要历史) + if err := fetchSource(repo, dst); err != nil { + return nil, fmt.Errorf("拉取源码: %w", err) + } + + // ② 远端版本校验:明确告知"已经是最新"而不是白编一场 + to := VersionFromManifest(dst) + if to == "" { + return nil, fmt.Errorf("远端仓库的 plg.json 缺 version 字段,无法判定版本") + } + if !Newer(to, from) && from != "" { + return nil, fmt.Errorf("已是最新(本地 %s = 远端 %s)", from, to) + } + + // ③ 现场编译 + buildOut, err := hmapdevBuild(dst, hmapdevBin) + if err != nil { + return nil, fmt.Errorf("编译失败: %w\n%s", err, tailStr(buildOut, 800)) + } + + // ④ 找产物(hmapdev 输出到 dist/) + hmap := findHmap(filepath.Join(dst, "dist")) + if hmap == "" { + return nil, fmt.Errorf("编译完成但 dist/ 下没有 .hmap 产物\n%s", tailStr(buildOut, 400)) + } + return &UpgradeResult{HmapPath: hmap, Repo: repo, From: from, To: to, BuildLog: tailStr(buildOut, 400)}, nil +} + +// fetchSource clone(或更新已存在的副本)到 dst。 +func fetchSource(repo, dst string) error { + if _, err := os.Stat(filepath.Join(dst, ".git")); err == nil { + // 已有副本:fetch + reset,比删了重 clone 快且省流量 + if out, err := exec.Command("git", "-C", dst, "fetch", "origin").CombinedOutput(); err != nil { + return fmt.Errorf("git fetch: %w\n%s", err, tailStr(string(out), 300)) + } + if out, err := exec.Command("git", "-C", dst, "reset", "--hard", "origin/HEAD").CombinedOutput(); err != nil { + return fmt.Errorf("git reset: %w\n%s", err, tailStr(string(out), 300)) + } + return nil + } + // 新副本:浅 clone。--depth 1 对几十 MB 的插件仓省 90% 时间。 + cmd := exec.Command("git", "clone", "--depth", "1", repo, dst) + if out, err := cmd.CombinedOutput(); err != nil { + // 清掉半成品,避免下次把半成品误当有效副本 + os.RemoveAll(dst) + return fmt.Errorf("git clone: %w\n%s", err, tailStr(string(out), 300)) + } + return nil +} + +// hmapdevBuild 在 srcDir 里跑 hmapdev build,返回完整输出。 +func hmapdevBuild(srcDir, hmapdevBin string) (string, error) { + cmd := exec.Command(hmapdevBin, "build") + cmd.Dir = srcDir + out, err := cmd.CombinedOutput() + return string(out), err +} + +// findHmap 在 dist 目录里找 .hmap 产物(优先 bundle)。 +func findHmap(distDir string) string { + entries, err := os.ReadDir(distDir) + if err != nil { + return "" + } + // bundle 优先:它是全平台合集,plugin_install 直接吃 + bundle := "" + for _, e := range entries { + if strings.HasSuffix(e.Name(), "_bundle.hmap") { + bundle = filepath.Join(distDir, e.Name()) + break + } + } + if bundle != "" { + return bundle + } + for _, e := range entries { + if strings.HasSuffix(e.Name(), ".hmap") { + return filepath.Join(distDir, e.Name()) + } + } + return "" +} + +// repoName 从仓库 URL 提取名字(用于本地目录名)。 +func repoName(repo string) string { + s := strings.TrimSuffix(repo, "/") + s = strings.TrimSuffix(s, ".git") + if i := strings.LastIndexAny(s, "/:"); i >= 0 { + return s[i+1:] + } + return "plugin" +} + +func tailStr(s string, n int) string { + if len(s) <= n { + return s + } + return "…(前略)\n" + s[len(s)-n:] +} diff --git a/internal/update/plugin_test.go b/internal/update/plugin_test.go new file mode 100644 index 00000000..9df8d455 --- /dev/null +++ b/internal/update/plugin_test.go @@ -0,0 +1,218 @@ +package update + +import ( + "os" + "os/exec" + "path/filepath" + "strings" + "testing" +) + +// writePluginDir 造一个最小的已安装插件目录(只有 plugin.json)。 +func writePluginDir(t *testing.T, name, version, repo string) string { + t.Helper() + dir := t.TempDir() + content := "{\n \"name\": \"" + name + "\",\n \"version\": \"" + version + "\"" + if repo != "" { + content += ",\n \"repository\": \"" + repo + "\"" + } + content += "\n}\n" + if err := os.WriteFile(filepath.Join(dir, "plugin.json"), []byte(content), 0644); err != nil { + t.Fatal(err) + } + return dir +} + +// ── Check 的判据 ───────────────────────────────────────────── + +// 判据 1:未声明仓库 ⇒ Reason 明确说明,且 HasUpdate=false。 +// +// 静默返回"没有更新"会让用户以为插件是最新的,实际是"查不了"。 +// 两种状态必须可区分。 +func TestCheck_NoRepoStatesClearly(t *testing.T) { + dir := writePluginDir(t, "qq", "1.4.4", "") + info := Check(dir, RepoFromManifest(dir)) + if info.Repo != "" { + t.Errorf("Repo 应为空,实际 %q", info.Repo) + } + if info.HasUpdate { + t.Error("未声明仓库不应报告有更新") + } + if !strings.Contains(info.Reason, "未声明仓库") { + t.Errorf("Reason 必须说明「未声明仓库」(与「没有更新」是两回事),实际: %s", info.Reason) + } +} + +// 判据 2:对真实仓库(HologramAgent/qq)检查,必须拿到版本且不为空。 +// +// ★ 这是**活体**测试:它验证的是整条链路(manifest → GitHub API → 比较)。 +// 网络不可用时跳过(CI 偶发网络抖动不该红整个包)。 +func TestCheck_AgainstRealRepo(t *testing.T) { + if os.Getenv("SKIP_LIVE") != "" { + t.Skip("SKIP_LIVE set") + } + dir := writePluginDir(t, "qq", "0.0.1", "https://github.com/HologramAgent/qq") + info := Check(dir, RepoFromManifest(dir)) + if info.Reason != "" && strings.Contains(info.Reason, "不可达") { + t.Skipf("网络不可达,跳过活体验证: %s", info.Reason) + } + if info.Latest == "" { + t.Fatalf("真实仓库应能查到版本,实际 reason=%s", info.Reason) + } + if !info.HasUpdate { + t.Errorf("本地 0.0.1 vs 远端 %s 应判有更新", info.Latest) + } +} + +// 判据 3:Newer 的比较语义(kernel/plugin 两侧共用)。 +func TestNewer(t *testing.T) { + cases := []struct { + a, b string + want bool + }{ + {"1.4.5", "1.4.4", true}, + {"1.5.0", "1.4.9", true}, + {"2.0.0", "1.9.9", true}, + {"1.4.4", "1.4.4", false}, + {"1.4.3", "1.4.4", false}, + {"", "1.4.4", false}, + {"1.4.4", "", true}, + // dev 视为与同名正式版相同(开发分支通常领先,不误报) + {"1.4.4", "1.4.4dev", false}, + {"1.4.4dev", "1.4.4", false}, + // v 前缀不影响 + {"v1.4.5", "v1.4.4", true}, + // 非纯三段式不参与比较 + {"1.4", "1.4.0", false}, + {"1.4.4-rc1", "1.4.4", false}, + } + for _, c := range cases { + if got := Newer(c.a, c.b); got != c.want { + t.Errorf("Newer(%q,%q) = %v,期望 %v", c.a, c.b, got, c.want) + } + } +} + +// 判据 3b:TrimTag 的归一语义。 +func TestTrimTag(t *testing.T) { + for in, want := range map[string]string{ + "v1.4.4": "1.4.4", + "1.4.4": "1.4.4", + "1.4.0dev": "1.4.0", + "1.4.0dev.1": "1.4.0", + "release1": "release1", // 非纯版本原样返回,交给 ParseSemver 判不可比 + } { + if got := TrimTag(in); got != want { + t.Errorf("TrimTag(%q) = %q,期望 %q", in, got, want) + } + } +} + +// 判据 3c:ParseSemver 只认恰好三段。 +// +// ★ 这是合并时抓到的真实 bug:早期实现只挡 len<2,于是 "1.4" 漏过去、 +// 缺失的 patch 被零值填成 1.4.0 ⇒「1.4 → 1.4.0」被判成完全相同, +// 进而说"无需升级",而两者内容未必相同。 +func TestParseSemver_RequiresThreeParts(t *testing.T) { + for _, ok := range []string{"1.4.4", "v1.4.4", "1.4.0"} { + if ParseSemver(ok) == nil { + t.Errorf("%q 应可解析", ok) + } + } + for _, bad := range []string{"", "1", "1.4", "1.4.4.4", "v1.4", "abc", "1.4.x"} { + if v := ParseSemver(bad); v != nil { + t.Errorf("%q 不应被解析成语义版本,实际 %+v", bad, v) + } + } +} + +// 判据 4:trimV 只认纯 semver tag。 +func TestTrimV(t *testing.T) { + // TrimTag 归一化(去 v 前缀)—— 非纯 semver 原样返回, + // 由 ParseSemver 判"不可比",不在这里丢弃信息。 + for tag, want := range map[string]string{ + "v1.4.4": "1.4.4", + "1.4.4": "1.4.4", + } { + if got := TrimTag(tag); got != want { + t.Errorf("TrimTag(%q) = %q,期望 %q", tag, got, want) + } + } + // 这些原样返回(交给 ParseSemver 判不可比) + for _, tag := range []string{"v1.4", "release1", "1.4.4-rc1"} { + if ParseSemver(TrimTag(tag)) != nil { + t.Errorf("%q 归一后仍不应可解析", tag) + } + } +} + +// 判据 5:parseGitHub 认全主流 URL 形态。 +func TestParseGitHub(t *testing.T) { + for _, repo := range []string{ + "https://github.com/HologramAgent/qq", + "https://github.com/HologramAgent/qq.git", + "github.com/HologramAgent/qq", + "git@github.com:HologramAgent/qq.git", + } { + owner, name, ok := parseGitHub(repo) + if !ok || owner != "HologramAgent" || name != "qq" { + t.Errorf("parseGitHub(%q) = (%q,%q,%v)", repo, owner, name, ok) + } + } + // 非 GitHub 不可误判 + if _, _, ok := parseGitHub("https://gitee.com/x/y"); ok { + t.Error("gitee 不应被解析为 GitHub") + } +} + +// 判据 6:Upgrade 拒绝缺 hmapdev 的环境(响亮失败而非中途半成品)。 +func TestUpgrade_RequiresHmapdev(t *testing.T) { + res, err := Upgrade("https://github.com/HologramAgent/qq", t.TempDir(), "/nonexistent/hmapdev") + if err == nil { + t.Fatal("hmapdev 不可用时必须报错") + } + if res != nil { + t.Errorf("失败时不应返回结果: %+v", res) + } + if !strings.Contains(err.Error(), "hmapdev") { + t.Errorf("错误应指明 hmapdev 问题,实际: %v", err) + } +} + +// 判据 7:Upgrade 拒绝空仓库。 +func TestUpgrade_EmptyRepo(t *testing.T) { + if _, err := Upgrade("", t.TempDir(), ""); err == nil { + t.Fatal("空仓库必须报错") + } +} + +// ── 端到端:对真实仓库做完整升级(编译产物验证)────────────── +// +// ★ 这条是 plugin_update 的**核心承诺**的验证:拉下来 → 编出来 → 产物存在。 +// 前置:本机有 git 与 hmapdev。任一缺失则跳过(CI 上 hmapdev 未必装)。 +func TestUpgrade_EndToEnd_RealRepo(t *testing.T) { + if _, err := exec.LookPath("hmapdev"); err != nil { + t.Skip("本机无 hmapdev,跳过端到端") + } + if os.Getenv("SKIP_LIVE") != "" { + t.Skip("SKIP_LIVE set") + } + // 用一个已打 tag 的真实仓库(v1.4.4) + res, err := Upgrade("https://github.com/HologramAgent/qq", t.TempDir(), "hmapdev") + if err != nil { + // 首次升级 from="" 时 isNewer 恒真,失败大概率是编译环境问题 + t.Skipf("端到端失败(可能是编译环境): %v", err) + } + if res.HmapPath == "" { + t.Fatal("应产出 .hmap 路径") + } + if _, err := os.Stat(res.HmapPath); err != nil { + t.Fatalf("产物不存在: %v", err) + } + if !strings.HasSuffix(res.HmapPath, ".hmap") { + t.Errorf("产物应是 .hmap: %s", res.HmapPath) + } + if res.To == "" { + t.Error("应记录远端版本") + } +} diff --git a/internal/update/release.go b/internal/update/release.go new file mode 100644 index 00000000..c737eec9 --- /dev/null +++ b/internal/update/release.go @@ -0,0 +1,114 @@ +package update + +import ( + "encoding/json" + "fmt" + "io" + "net/http" + "os" + "strings" + "time" +) + +// ── kernel_update 的「检查」半边 ────────────────────────────── +// +// 升级半边(DoUpgrade)CLI 已在用;这里补「从 GitHub Release 发现新版」, +// 供 agent 工具与 WebAPI 调用。与 CLI 共享 BinVersion / CompareVersions。 + +// ghRepo 是内核发布仓库(GitHub Release 的来源)。 +// +// 为什么不在调用点写死:插件迁移到 HologramAgent 组织后,内核仓也可能 +// 随之迁移;届时只改这一处。环境变量可覆盖(测试/私有部署用)。 +func ghRepo() string { + if v := strings.TrimSpace(osGetenv("HOLOGRAM_RELEASE_REPO")); v != "" { + return v + } + return "JianFeeeee/HologramAgent" +} + +var kernelHTTPClient = &http.Client{Timeout: 20 * time.Second} + +// ReleaseInfo 是一次「内核检查更新」的结果。 +type ReleaseInfo struct { + Current string `json:"current"` // 本机版本 + CurrentC string `json:"current_commit"` // 本机 commit + Latest string `json:"latest"` // 远端最新 release tag + LatestURL string `json:"latest_url,omitempty"` // release 页面 + Asset string `json:"asset,omitempty"` // 匹配到的资产名(tar.gz) + AssetURL string `json:"asset_url,omitempty"` // 资产下载地址 + HasUpdate bool `json:"has_update"` + Reason string `json:"reason,omitempty"` // 查不到时的说明 +} + +// CheckRelease 查 GitHub Release 并与本机版本比较。 +// +// current 为 nil 时(读不到本机版本,如老内核没有 --version), +// 比较会降级为"未知",但 release 信息照常返回 —— 与 CLI 探测同语义。 +func CheckRelease(current *BinVersion) *ReleaseInfo { + info := &ReleaseInfo{} + if current != nil { + info.Current = current.Version + info.CurrentC = current.Commit + } + + var rel struct { + TagName string `json:"tag_name"` + HTMLURL string `json:"html_url"` + Assets []struct { + Name string `json:"name"` + BrowserDownloadURL string `json:"browser_download_url"` + } `json:"assets"` + } + url := fmt.Sprintf("https://api.github.com/repos/%s/releases/latest", ghRepo()) + if err := httpGetJSON(url, &rel); err != nil { + info.Reason = "无法查询 GitHub Release:" + err.Error() + return info + } + if rel.TagName == "" { + info.Reason = "仓库还没有任何 release" + return info + } + info.Latest = TrimTag(rel.TagName) + info.LatestURL = rel.HTMLURL + + // 找本平台资产(linux/amd64) + for _, a := range rel.Assets { + if strings.Contains(a.Name, "linux") && strings.Contains(a.Name, "amd64") && strings.HasSuffix(a.Name, ".tar.gz") { + info.Asset = a.Name + info.AssetURL = a.BrowserDownloadURL + break + } + } + + if current == nil || current.Version == "" { + info.Reason = "本机版本未知(内核过老,无 --version):无法比较,请人工确认" + return info + } + info.HasUpdate = Newer(info.Latest, current.Version) + return info +} + +// 版本比较走 semver.go 的 Newer / TrimTag(子系统内唯一一份)。 +// +// 合并前这里有 versionNewer + trimTagV 两个私有函数,与 kernel 侧、 +// plugin 侧各写一份 —— 三份行为不一致是真实的 bug 温床,见 semver.go 注释。 + +func httpGetJSON(url string, v interface{}) error { + req, err := http.NewRequest("GET", url, nil) + if err != nil { + return err + } + req.Header.Set("Accept", "application/vnd.github+json") + resp, err := kernelHTTPClient.Do(req) + if err != nil { + return err + } + defer resp.Body.Close() + if resp.StatusCode != http.StatusOK { + body, _ := io.ReadAll(io.LimitReader(resp.Body, 200)) + return fmt.Errorf("HTTP %d: %s", resp.StatusCode, strings.TrimSpace(string(body))) + } + return json.NewDecoder(resp.Body).Decode(v) +} + +func osGetenv(k string) string { return os.Getenv(k) } diff --git a/internal/update/semver.go b/internal/update/semver.go new file mode 100644 index 00000000..5b063072 --- /dev/null +++ b/internal/update/semver.go @@ -0,0 +1,112 @@ +package update + +// 版本号工具:**子系统内唯一一份**语义版本实现。 +// +// ## 为什么必须收敛到一份(合并前的教训) +// +// 合并前有 6 个函数散在两个包里做同一件事: +// +// internal/kernelupgrade: ParseSemver / CmpSemver / versionNewer / trimTagV +// internal/pluginupdate: trimV / isNewer / semverNums +// +// 三份实现的差异是真实的 bug 温床: +// · kernel 侧只挡 len<2 ⇒ "1.4" 被当成 1.4.0(已修) +// · plugin 侧用 map+正则另写一套,行为与 kernel 侧不一致 +// · trimV / trimTagV 名字不同、逻辑相同 +// +// 版本比较错一次的后果很具体:把"已是最新"说成"有新版"(白编一场), +// 或把"有新版本"说成"无需升级"(用户永远拿不到修复)。 +// +// ⇒ 现在只导出一组:ParseSemver / CmpSemver / Newer / TrimTag。 + +import ( + "fmt" + "regexp" +) + +// Semver 是解析后的三段式版本号。 +type Semver struct{ Major, Minor, Patch int } + +var pureSemverRe = regexp.MustCompile(`^v?(\d+)\.(\d+)\.(\d+)$`) + +// ParseSemver 解析纯 X.Y.Z(可带 v 前缀)。 +// +// 返回 nil 表示"不是可比的三段式版本"——调用方应据此降级 +// (如退回字符串比较、或明确告知无法比较),而不是猜一个值。 +// +// ★ 必须恰好三段。早期实现只挡 len<2,于是 "1.4" 漏过去、 +// 缺失的 patch 被零值填成 1.4.0 ⇒ "1.4 → 1.4.0" 被判成完全相同, +// 进而说"无需升级",而两者内容未必相同。 +func ParseSemver(s string) *Semver { + m := pureSemverRe.FindStringSubmatch(s) + if m == nil { + return nil + } + var v Semver + if _, err := fmt.Sscanf(m[1]+"."+m[2]+"."+m[3], "%d.%d.%d", &v.Major, &v.Minor, &v.Patch); err != nil { + return nil + } + return &v +} + +// CmpSemver 比较两个已解析版本:<0 / 0 / >0。 +// +// 任一侧为 nil 时返回 0(不可比)——调用方必须先判 nil, +// 因为"无法比较"与"相等"是两回事,混同会让升级判定失效。 +func CmpSemver(a, b *Semver) int { + if a == nil || b == nil { + return 0 + } + switch { + case a.Major != b.Major: + return a.Major - b.Major + case a.Minor != b.Minor: + return a.Minor - b.Minor + default: + return a.Patch - b.Patch + } +} + +// Newer 判断版本串 a 是否严格新于 b。 +// +// 语义要点: +// - 只认纯三段式;任一侧不可解析 ⇒ false(不猜) +// - "1.4.0" 与 "1.4.0dev" 视为**相同**(TrimTag 去 dev 后比) +// —— 开发分支通常领先于同名正式版,不误报"有新版" +// - 空串 b 表示"本地无版本信息",此时 a 非空即算新 +func Newer(a, b string) bool { + if b == "" { + return a != "" + } + pa, pb := ParseSemver(TrimTag(a)), ParseSemver(TrimTag(b)) + if pa == nil || pb == nil { + return false + } + return CmpSemver(pa, pb) > 0 +} + +// TrimTag 把版本串归一成可比形态:去 v 前缀、去 dev/预发布后缀。 +// +// 处理 "v1.4.4" → "1.4.4"、"1.4.0dev" → "1.4.0"。 +// 非纯三段式(如 "1.4.4-rc1"、"release1")原样返回, +// 由 ParseSemver 判为不可比 —— 这里不做有损猜测。 +func TrimTag(s string) string { + out := s + if len(out) > 0 && (out[0] == 'v' || out[0] == 'V') { + out = out[1:] + } + // 去 dev 及其后(1.4.0dev / 1.4.0dev.1) + if i := indexOf(out, "dev"); i >= 0 { + out = out[:i] + } + return out +} + +func indexOf(s, sub string) int { + for i := 0; i+len(sub) <= len(s); i++ { + if s[i:i+len(sub)] == sub { + return i + } + } + return -1 +}