mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 18:08:04 +00:00
1f5af1dccf
核对「CLI 与 WebUI 插件能力对齐」时发现两个此前遗漏的缺口，一并补齐。 1) CLI /memory 缺 context 与 tools（internal/plugins/cli/plugin.go） WebUI 有 GET /api/v1/memory/context 与 /api/v1/memory/tools，CLI 只有 query/graph/text。IndexerAPI 本就对内部插件开放（BuildContext/ FormatContext/GetToolDefinitions/BuildToolPrompt），不是 SDK 缺口。 补上后与 WebUI 同源：context 打印「实际会注入什么上下文」， tools 打印工具定义 + 工具提示词。 waiter 同步接上两条远端路由与 help 文案。 2) 鸿蒙 camerasue 录像（BridgeCaps/BridgeRouter/DeviceBridge.ets） 此前 `camerasue <N秒>` 直接返回「暂不支持录像回传」。查 SDK 后发现 cameraPicker 本身就有 PickerMediaType.VIDEO 与 PickerProfile.videoDuration —— 录像完全可行，只是回传通道没接。 现改为：VIDEO 模式取回 mp4，经 DeviceBridge.sendDataChunked 按 cmd_data_start/分块/cmd_data_end 回传（与 GUI/CLI 录像路径一致）， 网关聚合后落盘成文件、agent 拿路径；照片仍走小体积 base64 内联。 上限 64MB、时长 1~300s，超限明确报错而不是把 WS/上下文撑爆。 依赖方向处理：BridgeCaps 需要「往本请求回传字节」，但 DeviceBridge 为取 CapResult 已 import BridgeCaps，反向 import 会成环。改为 BridgeRouter 注入 DataChunkSender 回调（它同时持有 deviceBridge 与 reqId）， BridgeCaps 不碰 socket。新增 CapResult.chunked 标记「结果已由能力分块 发完」，DeviceBridge 据此不再回 cmd_result，避免网关把已完成请求与后续 分块错配。 验证：hvigor assembleHap BUILD SUCCESSFUL（ArkTS 编译通过，改动文件零告警）； make check-client-versions 一致；go vet 干净；全量 go test ./internal/... ./cmd/... 零失败。
446 lines
13 KiB
Go
446 lines
13 KiB
Go
package main
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"strings"
|
||
)
|
||
|
||
func handleBuiltin(cmd string, cfg *Config, state *State, reconnect func(), out io.Writer) bool {
|
||
switch {
|
||
case cmd == "/help":
|
||
fmt.Fprintln(out, `Built-in commands:
|
||
/help show this help
|
||
/stop [msg] stop generation / send interrupt (alias /interrupt)
|
||
/exit, /quit exit waiter
|
||
/clear clear screen
|
||
/reconnect force reconnection
|
||
/connect <path> switch to a different unix socket
|
||
/remote <url> switch to remote HTTP mode
|
||
/local switch back to local socket mode
|
||
/conn list list saved connections
|
||
/conn save <name> save current connection as <name>
|
||
/conn use <name> switch to saved connection
|
||
/conn del <name> delete saved connection
|
||
|
||
Server commands (local 与 remote 行为一致):
|
||
/status system status
|
||
/kernel kernel status
|
||
/settings [prefix] list settings
|
||
/settings set <k> <v> set a setting
|
||
/plugin list list installed plugins
|
||
/plugin install <url> install plugin
|
||
/plugin remove <name> remove plugin
|
||
/plugin disable <name> disable plugin
|
||
/plugin enable <name> enable plugin
|
||
/plugin info <name> plugin details
|
||
/memory query <text> query graph memory
|
||
/memory graph dump full graph memory snapshot
|
||
/memory text [n] recent text-memory events + stats
|
||
/memory context [q] assembled memory context (what gets injected)
|
||
/memory tools memory tool definitions + tool prompt
|
||
/knowledge list knowledge base (+stats)
|
||
/knowledge delete <name> delete knowledge item
|
||
/config dump kernel config (JSON)
|
||
/tracker change-tracking stats
|
||
/tracker rollback roll back this session's file changes
|
||
/adapters list loaded Lua adapters
|
||
/adapters remove <name> remove a Lua adapter
|
||
/network network status + LLM endpoints
|
||
/runtime scheduler / residents / channel topology
|
||
/terminals list terminal sessions
|
||
/cmd/history command execution history
|
||
/terminal create|write|read|close … (local mode; calls agentcli tools)
|
||
/persona show persona (/persona set default|custom|later [text])
|
||
/agents list agents
|
||
/chat <text> send to agent
|
||
|
||
Any other text is sent to the agent directly.`)
|
||
return true
|
||
|
||
case cmd == "/exit" || cmd == "/quit":
|
||
return true
|
||
|
||
case cmd == "/clear":
|
||
fmt.Fprint(out, "\033[H\033[2J")
|
||
return true
|
||
|
||
case cmd == "/reconnect":
|
||
printlnC(colorYellow, "reconnecting...")
|
||
reconnect()
|
||
return true
|
||
|
||
// /stop 与 /interrupt：取消当前生成（可附带一句新指令）。
|
||
// 之前 /help 里写着这条命令，但 handleBuiltin 根本没有对应 case，
|
||
// 于是它像普通文本一样被发给了 Agent。
|
||
// 本地交给 CLI 插件（内核优先级 L3），远端走 WebUI 的 chat/interrupt
|
||
// （内核优先级 L4）。两条路都是“真中断”，不是发一句话。
|
||
case cmd == "/stop" || cmd == "/interrupt" ||
|
||
strings.HasPrefix(cmd, "/stop ") || strings.HasPrefix(cmd, "/interrupt "):
|
||
msg := stopMessage(cmd)
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
body := fmt.Sprintf(`{"message":%q}`, msg)
|
||
if _, err := rc.DoAPI("POST", "/api/v1/chat/interrupt", body); err != nil {
|
||
fmt.Fprintf(out, "interrupt failed: %v\n", err)
|
||
} else {
|
||
fmt.Fprintln(out, "interrupt sent")
|
||
}
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/connect "):
|
||
cfg.Socket = strings.TrimSpace(cmd[9:])
|
||
cfg.Remote = ""
|
||
reconnect()
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/remote "):
|
||
cfg.Remote = strings.TrimSpace(cmd[8:])
|
||
cfg.Socket = ""
|
||
reconnect()
|
||
return true
|
||
|
||
case cmd == "/local":
|
||
cfg.Remote = ""
|
||
cfg.Socket = discoverSocket("")
|
||
reconnect()
|
||
return true
|
||
|
||
case cmd == "/conn list":
|
||
if len(cfg.Connections) == 0 {
|
||
fmt.Fprintln(out, "no saved connections")
|
||
}
|
||
for _, c := range cfg.Connections {
|
||
mark := " "
|
||
if c.Name == cfg.Default {
|
||
mark = "*"
|
||
}
|
||
addr := c.Remote
|
||
if addr == "" {
|
||
addr = c.Socket
|
||
}
|
||
fmt.Fprintf(out, " %s %-15s %s\n", mark, c.Name, addr)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/conn save "):
|
||
name := strings.TrimSpace(cmd[11:])
|
||
cfg.SaveConnection(name)
|
||
fmt.Fprintf(out, "connection saved as '%s' (default)\n", name)
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/conn use "):
|
||
name := strings.TrimSpace(cmd[10:])
|
||
if cfg.SwitchConnection(name) {
|
||
fmt.Fprintf(out, "switched to '%s'\n", name)
|
||
reconnect()
|
||
} else {
|
||
fmt.Fprintf(out, "connection '%s' not found\n", name)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/conn del "):
|
||
name := strings.TrimSpace(cmd[10:])
|
||
if cfg.DeleteConnection(name) {
|
||
fmt.Fprintf(out, "connection '%s' deleted\n", name)
|
||
} else {
|
||
fmt.Fprintf(out, "connection '%s' not found\n", name)
|
||
}
|
||
return true
|
||
|
||
case cmd == "/status":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/status", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/status")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/kernel":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/kernel", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/kernel")
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/settings set "):
|
||
parts := strings.SplitN(cmd[14:], " ", 2)
|
||
if len(parts) < 2 {
|
||
fmt.Fprintln(out, "usage: /settings set <key> <value>")
|
||
return true
|
||
}
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
body := fmt.Sprintf(`{"%s":%q}`, parts[0], parts[1])
|
||
rc.DoAPI("PUT", "/api/v1/settings", body)
|
||
fmt.Fprintln(out, "ok")
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/settings"):
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/settings", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case cmd == "/plugin list":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/plugins", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/plugin list")
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/plugin install "):
|
||
url := strings.TrimSpace(cmd[16:])
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
body := fmt.Sprintf(`{"url":%q}`, url)
|
||
d, _ := rc.DoAPI("POST", "/api/v1/plugins", body)
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/plugin remove "):
|
||
name := strings.TrimSpace(cmd[15:])
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("DELETE", "/api/v1/plugins/"+name, "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/plugin info "):
|
||
name := strings.TrimSpace(cmd[13:])
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/plugins/"+name, "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
// disable/enable：本地由 CLI 插件处理，远端走插件管理 REST 动作接口。
|
||
case strings.HasPrefix(cmd, "/plugin disable ") || strings.HasPrefix(cmd, "/plugin enable "):
|
||
verb := "disable"
|
||
name := strings.TrimSpace(cmd[16:])
|
||
if strings.HasPrefix(cmd, "/plugin enable ") {
|
||
verb = "enable"
|
||
name = strings.TrimSpace(cmd[15:])
|
||
}
|
||
if name == "" {
|
||
fmt.Fprintln(out, "usage: /plugin disable|enable <name>")
|
||
return true
|
||
}
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("POST", "/api/v1/plugins/"+name+"/"+verb, "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/memory "):
|
||
sub := strings.TrimSpace(cmd[8:])
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
switch {
|
||
case strings.HasPrefix(sub, "query "):
|
||
q := strings.TrimSpace(strings.TrimPrefix(sub, "query "))
|
||
d, _ := rc.DoAPI("GET", "/api/v1/memory?query="+q, "")
|
||
printJSON(out, d)
|
||
case sub == "graph":
|
||
d, _ := rc.DoAPI("GET", "/api/v1/memory/graph", "")
|
||
printJSON(out, d)
|
||
case sub == "text" || strings.HasPrefix(sub, "text "):
|
||
d, _ := rc.DoAPI("GET", "/api/v1/memory/text", "")
|
||
printJSON(out, d)
|
||
case sub == "context" || strings.HasPrefix(sub, "context "):
|
||
q := strings.TrimSpace(strings.TrimPrefix(sub, "context"))
|
||
d, _ := rc.DoAPI("GET", "/api/v1/memory/context?q="+q, "")
|
||
printJSON(out, d)
|
||
case sub == "tools":
|
||
d, _ := rc.DoAPI("GET", "/api/v1/memory/tools", "")
|
||
printJSON(out, d)
|
||
default:
|
||
fmt.Fprintln(out, "usage: /memory query <text> | /memory graph | /memory text [n] | /memory context [q] | /memory tools")
|
||
}
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case strings.HasPrefix(cmd, "/knowledge delete "):
|
||
name := strings.TrimSpace(cmd[18:])
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("DELETE", "/api/v1/knowledge/"+name, "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case cmd == "/knowledge" || cmd == "/knowledge list" || cmd == "/knowledge stats":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/knowledge", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case cmd == "/config":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/config", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/config")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/tracker":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/tracker", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/tracker")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/tracker rollback":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("POST", "/api/v1/tracker/rollback", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/tracker rollback")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/adapters" || strings.HasPrefix(cmd, "/adapters remove "):
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
if strings.HasPrefix(cmd, "/adapters remove ") {
|
||
name := strings.TrimSpace(cmd[17:])
|
||
d, _ := rc.DoAPI("DELETE", "/api/v1/adapters/"+name, "")
|
||
printJSON(out, d)
|
||
} else {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/adapters", "")
|
||
printJSON(out, d)
|
||
}
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case cmd == "/persona" || strings.HasPrefix(cmd, "/persona set "):
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
if strings.HasPrefix(cmd, "/persona set ") {
|
||
rest := strings.TrimSpace(cmd[13:])
|
||
mode := rest
|
||
content := ""
|
||
if idx := strings.IndexByte(rest, ' '); idx > 0 {
|
||
mode = rest[:idx]
|
||
content = strings.TrimSpace(rest[idx+1:])
|
||
}
|
||
body := fmt.Sprintf(`{"mode":%q,"content":%q}`, mode, content)
|
||
d, _ := rc.DoAPI("POST", "/api/v1/persona", body)
|
||
printJSON(out, d)
|
||
} else {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/persona", "")
|
||
printJSON(out, d)
|
||
}
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case cmd == "/terminals":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/terminals", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/terminals")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/cmd/history":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/cmd/history", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/cmd/history")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/terminal" || strings.HasPrefix(cmd, "/terminal "):
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
// 远端 WebUI 没有“开终端”的 REST 端点（终端由 agentcli 工具创建），
|
||
// 不静默当聊天发出去，直接说明。
|
||
fmt.Fprintln(out, "remote 模式暂不支持终端操作；请在 local 模式或让 agent 调 terminal_* 工具")
|
||
} else {
|
||
state.Send(cmd)
|
||
}
|
||
return true
|
||
|
||
case cmd == "/runtime":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/runtime", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/runtime")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/network":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/network", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/network")
|
||
}
|
||
return true
|
||
|
||
case cmd == "/agents":
|
||
if rc := state.RemoteConn(); rc != nil {
|
||
d, _ := rc.DoAPI("GET", "/api/v1/agents", "")
|
||
printJSON(out, d)
|
||
} else {
|
||
state.Send("/agents")
|
||
}
|
||
return true
|
||
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
// stopMessage 从 /stop 或 /interrupt 行里取出可选的中断附带消息（空串=纯取消）。
|
||
func stopMessage(cmd string) string {
|
||
for _, prefix := range []string{"/interrupt", "/stop"} {
|
||
if strings.HasPrefix(cmd, prefix) {
|
||
return strings.TrimSpace(strings.TrimPrefix(cmd, prefix))
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
func printJSON(out io.Writer, d map[string]interface{}) {
|
||
if d == nil {
|
||
fmt.Fprintln(out, "(no data)")
|
||
return
|
||
}
|
||
b, _ := json.MarshalIndent(d, "", " ")
|
||
fmt.Fprintln(out, string(b))
|
||
}
|