mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-29 06:00:56 +00:00
d9b2f2c81c
原来 CLI 插件只覆盖 WebUI 的一小部分：/memory 只有 query、/knowledge 只有 list、没有 config/tracker/adapters/network，结构化输出还各拼一套文本格式。 按 WebUI 的 REST 面对齐： - /memory query|graph|text [n] （对应 /memory、/memory/graph、/memory/text） - /knowledge | delete <name> | stats（对应 GET/DELETE /knowledge） - /config （对应 GET /config） - /tracker | rollback （对应 GET /tracker、POST /tracker/rollback） - /adapters | remove <name> （对应 GET/DELETE /adapters） - /network （对应 GET /network） - 统一 writeJSONContent：结构化数据一律缩进 JSON，与 WebUI 同口径。 waiter 侧同步：把上述命令在 local（发 CLI 插件）与 remote（走 REST）两条路 都接上，/help 补齐；两条路语义一致。
373 lines
10 KiB
Go
373 lines
10 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
|
||
/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
|
||
/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)
|
||
default:
|
||
fmt.Fprintln(out, "usage: /memory query <text> | /memory graph | /memory text [n]")
|
||
}
|
||
} 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 == "/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))
|
||
}
|