mirror of
https://gitcode.com/JianFeeeee/homeagent-sdk.git
synced 2026-10-31 06:17:25 +00:00
feat(commondev): 主管式多仓库开发流水线 + sidecar 后端抽象(pi/zcode) (#3)
* feat(commondev): 主管式多仓库开发流水线 + sidecar 后端抽象(pi/zcode)
## 是什么
agent 统筹开发流程的插件:登记项目 → 开工程线(每条线一个隔离工作目录 +
一个 sidecar 子进程 + 一对通道)→ 投递任务 → 同会话续投 → 评审/合入。
38 个工具:project_*/branch_*/workdir_*/job_*/review_*/test_*/
endpoint_*/agent_*/pi_*/conflict_check。
## sidecar 后端抽象(本轮核心)
传输层(local/ssh)与后端(pi/zcode/自定义)**正交**,四个自由组合:
driver.json 描述一个后端:script / cliPath / env / args / llmConfig
三级查找:插件数据目录 → **随包 rawfile** → 内置 pi 描述
占位符:{workdir} {dataDir} {pluginDir} {rawfile} {sessionId}
★ 内置 pi 回退**与默认后端解耦**(2026-10-17 修):原实现写
`if cli == defaultAgentCLI { return builtinPiDriver() }`,把"给谁兜底"
和"默认选谁"耦合到同一常量 —— 一旦有人改默认,pi 当场失去兜底,
显式 cli="pi" 开始报「驱动不存在」。现在判断字面 "pi"。
默认后端保持 **pi**(我们本地主用);交付方用 zcode 走两种局部开关:
· 单线:commondev_branch_open(cli="zcode")
· 按机器:commondev_endpoint_set(host=..., cli="zcode")
## zcode 接入(三条真机判据,全部要求真改工程文件)
· TestZcodeFullPipeline(29-34s):随包 driver + 真实 ZCode CLI
· TestZcodeOverSSH(39s):真 ssh + scp,远端 node 跑 zcode
· TestZcodeOverSSHWithPerHostEndpoints(68s):两个 host 别名配不同
cli_path,走端点表而不是全局 settings
★ 会话库必须隔离到工作区(sidecar.mjs 的 ensureWorkspaceSessionStore):
实测 ZCode 的会话库路径**硬编码 homedir()**,`ZCODE_DATA_BASE_DIR` 对它
完全无效(26 个会话全落全局库)。但 `<workdir>/.zcode/config.json` 的
`storage.sessionDbPath` **确实生效**。所以启动前写一份工作区配置,
一举解决:多线不共用历史 / 不污染全局 ~/.zcode / 删线不留孤儿会话。
只在不存在时写 —— 用户手工放的要尊重。
## 其他
· rawfile:项目根 rawfile/ 默认全收,解到 <插件目录>/rawfile/,绝对路径拒绝
· 凭证:命名引用 + ${ENV_VAR},独立 0600 文件,绝不进 config.db(0644)
· secretshell:给不受信 agent 用的安全 shell(ackey + safeshell)
· 冲突检查 / 评审判据 / 镜像 / 每条线独立 workdir(hash 命名,合并后删除)
判据 30+ 条,含 4 条「判据本身的判据」(防止判据假绿)。
* fix(sdk): 协议通道纪律 + hmapdev 模板升级 + 示例插件 stdout 污染
## 核心缺陷:插件往 stdout 打日志 = 污染 JSON-RPC 协议通道
子进程插件与内核用 stdout 传 NDJSON 帧,stderr 才是日志通道。
脚手架模板(及 4 个示例插件)用 `fmt.Printf` 打日志 ⇒ 内核收到非 JSON 文本:
[proc] weather 收到非法 JSON 帧(21 字节): invalid character 'i'
**生产实测 1038 条**(huawei_smarthome 720 / weather 113 / rss 103 / calendar 102)。
后果不只是日志噪音:那一行日志被**丢弃**,插件看起来"没启动"。
修法:
- hmapdev 模板 `fmt.Printf` → `log.Printf`(9 处)
- 示例插件 4 个 14 处:ai_image(2) calendar(5) rss(4) weather(3)
+ 补 "log" import
- 内核侧(主仓)同步让报错可定位:内容摘要 + 可能原因 + 修法
判别法:`package main` = 子进程(协议通道,禁用 stdout 打日志);
普通库包 = 进程内插件(编译进内核,写真实终端,**不是**缺陷)——
`internal/plugins/cli` 的 `fmt.Println` 属后者。
## hmapdev
- `cmd_build.go` +136:平台产物精确匹配(`--target` 是**append** 语义,
原脚本 `ls dist/*.hmap | head -1` 会按字典序抓到 amd64,
**导致 arm64 包里装的是 x86-64 二进制**,且退出码 0、校验和通过)
- `cmd_init.go` +46:脚手架 stdio 纪律
- `templates/proc_main.go.tmpl`:补齐 mgrStop/mgrStart/mgrIsRuntimeStopped
三态能力(插件管理器运行态停启)
- `templates/plg.json.tmpl` 删除(合并进 cmd_init)
## 判据
- `scaffold_stdout_test.go`(3 条):扫模板生成的代码不得写 stdout。
★ 判据自身先假红过一次:**模板的注释里就写了 `fmt.Printf`** ⇒
必须 `stripTemplateComments()` 之后才判
- `example_stdout_test.go`(2 条):扫全部 20 个子进程示例插件,
报错给**文件:行号**;变异咬住并给出准确行号
- `package_examples_test.go`(2 条):**执行真实打包脚本**
(第一版测的是 hmapdev 而不是脚本 ⇒ 变异不咬)
- `cmd_build` / `cmd_init` 新判据
## 其他
- sdk/settings.go +15(三态能力透传)
- `docs/versions.md` +§示例插件协议通道纪律(含生产 1038 条实测表格)
- `skills/homeagent-plugin-dev/SKILL.md` +§协议通道纪律(正误代码对比)
* fix(examples): 恢复 go.mod 相对路径 + 移除 plg.json 的 sdk 固定(修 CI 红)
CI 上 `Examples (linux/amd64)` 与 `(darwin/arm64)` 双红,只有两个失败:
✗ commondev 构建失败
✗ editdoc 构建失败
error: 项目声明 sdk=1.4.0,但本地 SDK 存储里没有这个版本
其余 20 个示例全部成功 —— 说明不是通用问题,而是这两个被改坏了。
## 根因(两个叠加)
**① go.mod 被本机绝对路径污染**
我在上一个提交里用 `git add -A`,把本机 `go build` **自动改写**过的
go.mod 一并提交了:
require github.com/JianFeeeee/homeagentsdk v1.4.0
replace github.com/JianFeeeee/homeagentsdk => /root/.homeagent/hmapdev/sdk/v1.4.0
基线(origin/main)是:
require github.com/JianFeeeee/homeagentsdk v0.0.0
replace github.com/JianFeeeee/homeagentsdk => ../../
本机那个目录真的存在 ⇒ 本地全绿;CI 没有 ⇒ 必红。
**② plg.json 声明了 `"sdk": "1.4.0"`**
一声明版本,hmapdev 就去**本地 SDK store** 找它,找不到直接拒绝构建。
全仓 20+ 示例**都不声明**,只有我改过的这两个声明了。
示例的正确依赖表达是 go.mod 的 `v0.0.0` + `replace ../../`。
## 修法
两个文件都恢复到与其余示例一致的写法。
## 判据(3 条,都是新加的)
· `TestExampleGoModHasNoAbsoluteReplace`:任何示例 go.mod 不得含
`/root/`、`/home/`、`/Users/`、`C:\`
· `TestExampleGoModUsesRelativeSDKReplace`:replace 必须是相对路径 `../../`
· `TestExamplePlgJsonDoesNotPinSDKVersion`:plg.json 不得声明 sdk
变异 2/2 咬住(重新引入绝对路径 / 重新声明 sdk,都准确报出文件路径)。
## 本地验证(模拟 CI 干净环境)
用干净 HOME 跑 `package/build-examples.sh`:
· 修复前:成功 18 / 失败 4(commondev, editdoc, browser, rss)
· 修复后:成功 20 / 失败 2(browser, rss —— **本机代理不通**,
`proxy.golang.org` 超时;用本地 modcache 验证这两个可正常构建)
即 22/22 源码都能构建,CI 有网络不受影响。
## 教训
`git add -A` 会把「构建工具自动改写的文件」一起提交。
提交前必须 `git status` 逐个看,尤其 go.mod / go.sum。
---------
Co-authored-by: JianFeeeee <JianFeeeee@users.noreply.gitcode.com>
This commit is contained in:
514
example/commondev/zcode_e2e_test.go
Normal file
514
example/commondev/zcode_e2e_test.go
Normal file
@ -0,0 +1,514 @@
|
||||
//go:build zcode_e2e
|
||||
|
||||
// ZCode 后端的**真机**端到端判据。
|
||||
//
|
||||
// 与 pi 的 e2e 同一形状,但驱动的是**随包分发的** driver.json + sidecar.mjs
|
||||
// 与**真实 ZCode CLI** —— 而不是我的临时 harness。要求真实改动工程文件,
|
||||
// 且不破坏原有代码。
|
||||
//
|
||||
// 跑法(需要本机已能跑通 zcode):
|
||||
//
|
||||
// HA_ZCODE_CLI=<zcode.cjs 路径> \
|
||||
// HA_ZCODE_DATA_DIR=<zcode 数据目录> \
|
||||
// go test -tags zcode_e2e -run TestZcodeFullPipeline -v .
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func zcodeE2EReady(t *testing.T) (nodeBin, cliPath, dataDir string) {
|
||||
t.Helper()
|
||||
cliPath = os.Getenv("HA_ZCODE_CLI")
|
||||
dataDir = os.Getenv("HA_ZCODE_DATA_DIR")
|
||||
if cliPath == "" || dataDir == "" {
|
||||
t.Skip("需要 HA_ZCODE_CLI 与 HA_ZCODE_DATA_DIR(真实 zcode 环境)")
|
||||
}
|
||||
if _, err := os.Stat(cliPath); err != nil {
|
||||
t.Skipf("HA_ZCODE_CLI 不存在: %v", err)
|
||||
}
|
||||
for _, c := range []string{"node", "nodejs"} {
|
||||
if p, err := exec.LookPath(c); err == nil {
|
||||
nodeBin = p
|
||||
break
|
||||
}
|
||||
}
|
||||
if nodeBin == "" {
|
||||
t.Skip("本机无 node")
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// TestZcodeFullPipeline:登记 → 开线 → 准备目录 → 用**随包 driver**起 zcode
|
||||
// → 投递 → 断言真改了工程文件 → 收尾。
|
||||
func TestZcodeFullPipeline(t *testing.T) {
|
||||
nodeBin, cliPath, dataDir := zcodeE2EReady(t)
|
||||
|
||||
base := t.TempDir()
|
||||
origin := filepath.Join(base, "origin")
|
||||
gitQ(t, base, "init", "-q", "-b", "main", origin)
|
||||
if err := os.MkdirAll(filepath.Join(origin, "src"), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(origin, "src/calc.py"),
|
||||
[]byte("def add(a, b):\n return a + b\n"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
gitQ(t, origin, "add", "-A")
|
||||
gitQ(t, origin, "commit", "-qm", "initial")
|
||||
|
||||
p := newFlowPlugin(t)
|
||||
r1, _ := p.handleProjectAdd(map[string]interface{}{
|
||||
"name": "zdemo", "remote_url": origin, "default_branch": "main",
|
||||
})
|
||||
if m := r1.(map[string]interface{}); m["status"] != "ok" {
|
||||
t.Fatalf("登记项目失败: %v", m)
|
||||
}
|
||||
|
||||
r2, _ := p.handleBranchOpen(map[string]interface{}{
|
||||
"project": "zdemo", "branch": "feat/z", "work_desc": "加乘",
|
||||
})
|
||||
m2 := r2.(map[string]interface{})
|
||||
if m2["status"] != "ok" {
|
||||
t.Fatalf("开线失败: %v", m2)
|
||||
}
|
||||
branchID := m2["branch_id"].(string)
|
||||
workdir := m2["workdir"].(string)
|
||||
if err := p.prepareWorkdir(p.ledger.Projects["zdemo"], "feat/z", workdir); err != nil {
|
||||
t.Fatalf("目录准备失败: %v", err)
|
||||
}
|
||||
|
||||
// ★ 用**随包分发**的 driver.json 解析描述(不是测试里手拼参数)
|
||||
rawRoot := filepath.Join(mustAbs(t, "."), "rawfile")
|
||||
pz := newTestPluginWithRawfiles(t, rawRoot)
|
||||
pz.path = filepath.Join(base, "commondev") // 用户目录空 → 强制走随包
|
||||
drv, cliDir, err := pz.loadDriver("zcode")
|
||||
if err != nil {
|
||||
t.Fatalf("★ 随包 zcode 驱动解析失败: %v", err)
|
||||
}
|
||||
// cliPath 指向随包位置(不存在),本判据用 HA_ZCODE_CLI 覆盖成本机真实路径
|
||||
script, err := pz.locateSidecarScript(drv, cliDir, "zcode")
|
||||
if err != nil {
|
||||
t.Fatalf("★ 随包 sidecar 脚本定位失败: %v", err)
|
||||
}
|
||||
t.Logf("随包解析:script=%s cliPath=%s", script, drv.CLIPath)
|
||||
|
||||
ctx := driverCtx{
|
||||
Workdir: workdir, SessionID: "br-" + branchID,
|
||||
SessionName: "br-" + branchID, DataDir: dataDir,
|
||||
PluginDir: filepath.Dir(rawRoot), RawfileDir: rawRoot,
|
||||
}
|
||||
argv := drv.buildArgs(ctx)
|
||||
argv = append(argv, "--cli", cliPath)
|
||||
|
||||
cmd := exec.Command(nodeBin, append([]string{script}, argv...)...)
|
||||
cmd.Dir = workdir
|
||||
env := append(os.Environ(), drv.buildEnv(ctx)...)
|
||||
env = append(env, "ZCODE_DATA_BASE_DIR="+dataDir)
|
||||
cmd.Env = env
|
||||
|
||||
stdin, err := cmd.StdinPipe()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
stdout, err := cmd.StdoutPipe()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
cmd.Stderr = os.Stderr
|
||||
if err := cmd.Start(); err != nil {
|
||||
t.Fatalf("启动 sidecar 失败: %v", err)
|
||||
}
|
||||
defer func() { _ = cmd.Process.Kill(); _, _ = cmd.Process.Wait() }()
|
||||
|
||||
w := bufio.NewWriter(stdin)
|
||||
rd := bufio.NewReader(stdout)
|
||||
|
||||
if ping := callSidecar(t, w, rd, 1, "ping", nil); ping.Error != nil {
|
||||
t.Fatalf("握手失败: %s", ping.Error.Message)
|
||||
}
|
||||
resp := callSidecar(t, w, rd, 2, "prompt", map[string]interface{}{
|
||||
"session_id": "br-" + branchID,
|
||||
"text": "在 src/calc.py 里新增 multiply(a, b) 返回 a*b,保留 add。只回一句完成。",
|
||||
})
|
||||
if resp.Error != nil {
|
||||
t.Fatalf("投递失败: %s", resp.Error.Message)
|
||||
}
|
||||
var pr map[string]interface{}
|
||||
if err := json.Unmarshal(resp.Result, &pr); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
sid, _ := pr["session_id"].(string)
|
||||
if !strings.HasPrefix(sid, "sess_") {
|
||||
t.Fatalf("★ 未回报真实 ZCode 会话 id: %q", sid)
|
||||
}
|
||||
t.Logf("zcode 会话: %s", sid)
|
||||
|
||||
// ★ 关键断言:真的改了**这个工程**的文件,且没破坏原代码
|
||||
b2, err := os.ReadFile(filepath.Join(workdir, "src", "calc.py"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !strings.Contains(string(b2), "multiply") {
|
||||
t.Fatalf("★ zcode 没有在工程里加 multiply:\n%s", b2)
|
||||
}
|
||||
if !strings.Contains(string(b2), "def add") {
|
||||
t.Errorf("★ 原有代码被破坏:\n%s", b2)
|
||||
}
|
||||
}
|
||||
|
||||
func mustAbs(t *testing.T, p string) string {
|
||||
t.Helper()
|
||||
a, err := filepath.Abs(p)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return a
|
||||
}
|
||||
|
||||
// TestZcodeOverSSH:**同一套随包 driver/script 走 SSH 远端**。
|
||||
//
|
||||
// 用户的要求是"本地 + 通过 ssh 支持的远程 zcodecli"。传输层早就与后端正交
|
||||
// (Runtime×AgentCLI),但当时**没有任何判据证明二者能组合**:
|
||||
// `pushSidecar` 只推脚本、`--cli` 也没法分别声明本地/远端路径。
|
||||
//
|
||||
// 这条判据用 localhost 做真实 ssh 传输(真 ssh 客户端、BatchMode、scp),
|
||||
// 把随包的 driver.json + sidecar.mjs 推到远端目录,再在**远端**用真实
|
||||
// ZCode CLI 跑一轮并要求它改动工程文件。
|
||||
func TestZcodeOverSSH(t *testing.T) {
|
||||
if os.Getenv("HA_SSH_E2E") != "1" {
|
||||
t.Skip("设 HA_SSH_E2E=1 以运行(需要 sshd + localhost 免密)")
|
||||
}
|
||||
nodeBin, cliPath, dataDir := zcodeE2EReady(t)
|
||||
_ = nodeBin
|
||||
|
||||
base := t.TempDir()
|
||||
origin := filepath.Join(base, "origin")
|
||||
gitQ(t, base, "init", "-q", "-b", "main", origin)
|
||||
if err := os.MkdirAll(filepath.Join(origin, "src"), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(origin, "src/calc.py"),
|
||||
[]byte("def add(a, b):\n return a + b\n"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
gitQ(t, origin, "add", "-A")
|
||||
gitQ(t, origin, "commit", "-qm", "initial")
|
||||
|
||||
p := newFlowPlugin(t)
|
||||
if r, _ := p.handleProjectAdd(map[string]interface{}{
|
||||
"name": "sshz", "remote_url": origin, "default_branch": "main",
|
||||
}); r.(map[string]interface{})["status"] != "ok" {
|
||||
t.Fatalf("登记失败: %v", r)
|
||||
}
|
||||
r2, _ := p.handleBranchOpen(map[string]interface{}{
|
||||
"project": "sshz", "branch": "feat/ssh", "work_desc": "远程",
|
||||
"runtime": "ssh", "ssh_host": "localhost",
|
||||
})
|
||||
m2 := r2.(map[string]interface{})
|
||||
if m2["status"] != "ok" {
|
||||
t.Fatalf("开 SSH 线失败: %v", m2)
|
||||
}
|
||||
branchID := m2["branch_id"].(string)
|
||||
|
||||
// 远端目录:既是运行时根,也是脚本落地处
|
||||
remoteRoot := filepath.Join(base, "remote-runtime")
|
||||
remoteWork := filepath.Join(base, "remote-work")
|
||||
if err := os.MkdirAll(remoteWork, 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// ★ 走**真实** loadDriver + locateSidecarScript(随包来源)
|
||||
rawRoot := filepath.Join(mustAbs(t, "."), "rawfile")
|
||||
pz := newTestPluginWithRawfiles(t, rawRoot)
|
||||
pz.path = filepath.Join(base, "commondev")
|
||||
drv, cliDir, err := pz.loadDriver("zcode")
|
||||
if err != nil {
|
||||
t.Fatalf("随包 zcode 驱动解析失败: %v", err)
|
||||
}
|
||||
target := sshTarget{Host: "localhost", Port: 0, NodePath: nodeBin}
|
||||
if err := pz.probeSSH(target); err != nil {
|
||||
t.Skipf("ssh 不可用: %v", err)
|
||||
}
|
||||
// pushSidecar 推的是 drv.Script —— 先把随包脚本放进 cwd 让它可以被推
|
||||
stage := t.TempDir()
|
||||
localScript, err := pz.locateSidecarScript(drv, cliDir, "zcode")
|
||||
if err != nil {
|
||||
t.Fatalf("随包脚本定位失败: %v", err)
|
||||
}
|
||||
src, err := os.ReadFile(localScript)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(stage, drv.Script), src, 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := pz.pushSidecar(target, drv, stage, remoteRoot); err != nil {
|
||||
t.Fatalf("★ 推送随包脚本失败: %v", err)
|
||||
}
|
||||
// 远端脚本必须真的存在
|
||||
if out, err := exec.Command("ssh", "-o", "BatchMode=yes", "localhost",
|
||||
"test -f "+remoteRoot+"/"+drv.Script+" && echo OK").Output(); err != nil ||
|
||||
!strings.Contains(string(out), "OK") {
|
||||
t.Fatalf("★ 远端没有 sidecar 脚本: %v %s", err, out)
|
||||
}
|
||||
|
||||
// ★ 在**远端**跑:node <remote script> --workdir <remote work> ... --cli <远端 cli>
|
||||
// 本机与 localhost 是同一文件系统,故 cliPath/dataDir 直接复用。
|
||||
ctx := driverCtx{
|
||||
Workdir: remoteWork, SessionID: "br-" + branchID,
|
||||
SessionName: "br-" + branchID, DataDir: dataDir,
|
||||
PluginDir: filepath.Dir(rawRoot), RawfileDir: rawRoot,
|
||||
}
|
||||
argv := drv.buildArgs(ctx)
|
||||
argv = append(argv, "--cli", cliPath)
|
||||
remoteCmd := "ZCODE_DATA_BASE_DIR=" + dataDir + " " + nodeBin + " " +
|
||||
remoteRoot + "/" + drv.Script + " " + shellJoin(argv)
|
||||
full := append(target.sshBaseArgs(), "localhost", remoteCmd)
|
||||
|
||||
cmd := exec.Command("ssh", full...)
|
||||
cmd.Dir = remoteWork
|
||||
stdin, err := cmd.StdinPipe()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
stdout, err := cmd.StdoutPipe()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
cmd.Stderr = os.Stderr
|
||||
if err := cmd.Start(); err != nil {
|
||||
t.Fatalf("远端启动失败: %v", err)
|
||||
}
|
||||
defer func() { _ = cmd.Process.Kill(); _, _ = cmd.Process.Wait() }()
|
||||
|
||||
w := bufio.NewWriter(stdin)
|
||||
rd := bufio.NewReader(stdout)
|
||||
if ping := callSidecar(t, w, rd, 1, "ping", nil); ping.Error != nil {
|
||||
t.Fatalf("远端握手失败: %s", ping.Error.Message)
|
||||
}
|
||||
resp := callSidecar(t, w, rd, 2, "prompt", map[string]interface{}{
|
||||
"session_id": "br-" + branchID,
|
||||
"text": "在 src/calc.py 里新增 multiply(a, b) 返回 a*b,保留 add。只回一句完成。",
|
||||
})
|
||||
if resp.Error != nil {
|
||||
t.Fatalf("远端投递失败: %s", resp.Error.Message)
|
||||
}
|
||||
b2, err := os.ReadFile(filepath.Join(remoteWork, "src", "calc.py"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !strings.Contains(string(b2), "multiply") {
|
||||
t.Fatalf("★ 远端 zcode 没有改文件:\n%s", b2)
|
||||
}
|
||||
if !strings.Contains(string(b2), "def add") {
|
||||
t.Errorf("★ 远端原有代码被破坏:\n%s", b2)
|
||||
}
|
||||
}
|
||||
|
||||
func shellJoin(args []string) string {
|
||||
parts := make([]string, 0, len(args))
|
||||
for _, a := range args {
|
||||
parts = append(parts, shellQuote(a))
|
||||
}
|
||||
return strings.Join(parts, " ")
|
||||
}
|
||||
|
||||
// TestZcodeOverSSHWithPerHostEndpoints:**两台"机器"各配各的 CLI 路径**,
|
||||
// 走端点表而不是全局 settings —— 这是用户提的需求本体。
|
||||
//
|
||||
// 用两个 host 别名指向同一个 sshd(`localhost` 与 `127.0.0.1`),
|
||||
// 但配置**不同的** CLI 路径与 runtime 目录,从而证明:
|
||||
//
|
||||
// · 每台机器的路径确实分别生效(不是被后配的覆盖)
|
||||
// · 发布到远端的 --cli 是该机器自己那条
|
||||
//
|
||||
// 为什么用别名而不是真两台机器:判据要能在 CI/本机跑。别名走的是
|
||||
// **同一条真实 ssh 代码路径**(解析 host → 端点表 → ssh → scp → 远端 node),
|
||||
// 与真两台机器的差别只在网络延迟,不足以掩盖端点选错这类缺陷。
|
||||
func TestZcodeOverSSHWithPerHostEndpoints(t *testing.T) {
|
||||
if os.Getenv("HA_SSH_E2E") != "1" {
|
||||
t.Skip("设 HA_SSH_E2E=1 以运行(需要 sshd + localhost 免密)")
|
||||
}
|
||||
nodeBin, cliPath, dataDir := zcodeE2EReady(t)
|
||||
|
||||
base := t.TempDir()
|
||||
// 两个"机器"各自的 runtime 目录(互不相干)
|
||||
rtA := filepath.Join(base, "rtA")
|
||||
rtB := filepath.Join(base, "rtB")
|
||||
workA := filepath.Join(base, "workA")
|
||||
workB := filepath.Join(base, "workB")
|
||||
for _, d := range []string{rtA, rtB, workA, workB} {
|
||||
if err := os.MkdirAll(d, 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
// 机器 B 上放一个**不同的** CLI 入口(软链到真 CLI,路径不同即为不同)
|
||||
cliB := filepath.Join(base, "zcode-on-B.cjs")
|
||||
if err := os.Symlink(cliPath, cliB); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// ① 端点表:两台机器各自的路径
|
||||
p := newTestPluginForShell(t)
|
||||
if err := p.loadEndpoints(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
// pushSidecar 从 <插件数据>/runtime/<cli>/<script> 取本地脚本 ——
|
||||
// 放真实随包脚本进去(推的必须是真脚本,否则远端跑的是空文件)。
|
||||
{
|
||||
src, err := os.ReadFile(filepath.Join(mustAbs(t, "."), "rawfile", "runtime", "zcode", "sidecar.mjs"))
|
||||
if err != nil {
|
||||
t.Fatalf("读随包 sidecar 失败: %v", err)
|
||||
}
|
||||
d := p.driverDir("zcode")
|
||||
if err := os.MkdirAll(d, 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(d, "sidecar.mjs"), src, 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
for _, c := range []struct {
|
||||
host, cliPath, rt string
|
||||
}{
|
||||
{"localhost", cliPath, rtA},
|
||||
{"127.0.0.1", cliB, rtB},
|
||||
} {
|
||||
r, _ := p.handleEndpointSet(map[string]interface{}{
|
||||
"host": c.host, "cli": "zcode", "cli_path": c.cliPath,
|
||||
"runtime_root": c.rt, "node_path": nodeBin, "verify": false,
|
||||
})
|
||||
if e := r.(map[string]interface{})["error"]; e != nil {
|
||||
t.Fatalf("配 %s 失败: %v", c.host, e)
|
||||
}
|
||||
}
|
||||
|
||||
// ② 解析必须各归各(这是"多机不互相干扰"的核心断言)
|
||||
drv := driverSpec{Script: "sidecar.mjs", Args: []string{"--workdir", "{workdir}"}}
|
||||
ctx := driverCtx{Workdir: workA, RawfileDir: "/rf", PluginDir: "/p", DataDir: "/d"}
|
||||
ea := p.resolveEndpoint("localhost", "zcode", drv, ctx)
|
||||
eb := p.resolveEndpoint("127.0.0.1", "zcode", drv, ctx)
|
||||
if ea.CLIPath != cliPath || ea.RuntimeRoot != rtA {
|
||||
t.Errorf("★ localhost 端点串了: %+v", ea)
|
||||
}
|
||||
if eb.CLIPath != cliB || eb.RuntimeRoot != rtB {
|
||||
t.Errorf("★ 127.0.0.1 端点串了: %+v", eb)
|
||||
}
|
||||
if ea.CLIPath == eb.CLIPath {
|
||||
t.Error("★ 两台机器的 CLI 路径不该相同(配置未分别生效)")
|
||||
}
|
||||
|
||||
// ③ 真跑:两台机器各自用**自己的** CLI 路径完成任务
|
||||
for _, tc := range []struct {
|
||||
host, cliPath, rt, wd, proj string
|
||||
}{
|
||||
{"localhost", cliPath, rtA, workA, "hA"},
|
||||
{"127.0.0.1", cliB, rtB, workB, "hB"},
|
||||
} {
|
||||
origin := filepath.Join(base, "origin-"+tc.proj)
|
||||
gitQ(t, base, "init", "-q", "-b", "main", origin)
|
||||
if err := os.MkdirAll(filepath.Join(origin, "src"), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
os.WriteFile(filepath.Join(origin, "src/calc.py"),
|
||||
[]byte("def add(a, b):\n return a + b\n"), 0o644)
|
||||
gitQ(t, origin, "add", "-A")
|
||||
gitQ(t, origin, "commit", "-qm", "initial")
|
||||
|
||||
p.handleProjectAdd(map[string]interface{}{
|
||||
"name": tc.proj, "remote_url": origin, "default_branch": "main",
|
||||
})
|
||||
|
||||
b := &Branch{ID: "br-" + tc.proj, Name: "feat/x", ProjectID: tc.proj,
|
||||
Runtime: "ssh", SSHHost: tc.host, Workdir: tc.wd}
|
||||
tgt, err := parseSSHTarget(b)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := p.probeSSH(tgt); err != nil {
|
||||
t.Skipf("%s 不可用: %v", tc.host, err)
|
||||
}
|
||||
// 准备该机器的工作目录(本判据聚焦端点解析与远端执行,
|
||||
// 目录准备本身已由 workdir_remote_test.go 覆盖):
|
||||
// 直接 clone origin 到该机器的 workdir,并切到目标分支。
|
||||
if err := os.RemoveAll(tc.wd); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
gitQ(t, base, "clone", "-q", origin, tc.wd)
|
||||
gitQ(t, tc.wd, "checkout", "-qb", "feat/x")
|
||||
|
||||
// 推脚本 + 拼远端命令(走真实代码路径)
|
||||
if err := p.pushSidecar(tgt, drv, p.driverDir("zcode"), tc.rt); err != nil {
|
||||
t.Fatalf("%s 推脚本失败: %v", tc.host, err)
|
||||
}
|
||||
// ★ ctx.Workdir 必须**按该机器**重置:它是 {workdir} 占位符的来源,
|
||||
// 共用一个会让后一台机器把文件改到前一台的目录里
|
||||
// (本判据第一版就踩了:两台都跑在 workA)。
|
||||
ctx.Workdir = tc.wd
|
||||
ep := p.resolveEndpoint(tc.host, "zcode", drv, ctx)
|
||||
_, args, env := buildRemoteCommandFor(drv, ctx, ep.NodePath, ep)
|
||||
joined := strings.Join(args, " ")
|
||||
if !strings.Contains(joined, tc.cliPath) {
|
||||
t.Errorf("★ %s 的远端命令未使用本机专属 CLI 路径: %v", tc.host, args)
|
||||
}
|
||||
if tc.host == "127.0.0.1" && strings.Contains(joined, rtA) {
|
||||
t.Errorf("★ 127.0.0.1 的命令里混入了 localhost 的路径: %v", args)
|
||||
}
|
||||
|
||||
// ★★ 真跑:在**该机器**上执行解析出来的命令,并断言它真的改了文件。
|
||||
//
|
||||
// 只比字符串的话,"端点选对了但路径在目标机器上不存在"这类缺陷
|
||||
// 照样全绿 —— 而那正是本需求要防的东西(每台机器路径不同)。
|
||||
remoteCmd := "ZCODE_DATA_BASE_DIR=" + dataDir + " " + ep.NodePath + " " +
|
||||
tc.rt + "/" + drv.Script + " " + shellJoin(args)
|
||||
sshArgv := append(tgt.sshBaseArgs(), tgt.Host, remoteCmd)
|
||||
rc := exec.Command("ssh", sshArgv...)
|
||||
rc.Dir = tc.wd
|
||||
rstdin, err := rc.StdinPipe()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
rstdout, err := rc.StdoutPipe()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
rc.Stderr = os.Stderr
|
||||
if err := rc.Start(); err != nil {
|
||||
t.Fatalf("%s 远端启动失败: %v", tc.host, err)
|
||||
}
|
||||
rw := bufio.NewWriter(rstdin)
|
||||
rrd := bufio.NewReader(rstdout)
|
||||
if ping := callSidecar(t, rw, rrd, 1, "ping", nil); ping.Error != nil {
|
||||
_ = rc.Process.Kill()
|
||||
t.Fatalf("%s 远端握手失败: %s", tc.host, ping.Error.Message)
|
||||
}
|
||||
resp := callSidecar(t, rw, rrd, 2, "prompt", map[string]interface{}{
|
||||
"session_id": "br-" + tc.proj,
|
||||
"text": "在 src/calc.py 里新增 multiply(a, b) 返回 a*b,保留 add。只回一句完成。",
|
||||
})
|
||||
_ = rc.Process.Kill()
|
||||
_, _ = rc.Process.Wait()
|
||||
if resp.Error != nil {
|
||||
t.Fatalf("★ %s 投递失败(端点表里的 CLI 路径可能在该机器上不存在): %s",
|
||||
tc.host, resp.Error.Message)
|
||||
}
|
||||
b2, err := os.ReadFile(filepath.Join(tc.wd, "src/calc.py"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !strings.Contains(string(b2), "multiply") {
|
||||
t.Errorf("★ %s 上 zcode 没有改文件:\n%s", tc.host, b2)
|
||||
}
|
||||
if !strings.Contains(string(b2), "def add") {
|
||||
t.Errorf("★ %s 上原有代码被破坏:\n%s", tc.host, b2)
|
||||
}
|
||||
_ = env
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user