mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 18:08:04 +00:00
642e1c39b1
拆法：按「资源面」搬家，每个顶层声明（func/type/var/const）整体搬到目标文件， 声明体一字未改，各文件按实际用到的包重新生成 import。文件头加一行说明本文件负责哪一面。 handler.go 骨架：嵌入前端资源、Handler/构造、路由表、鉴权会话日志中间件、静态页 handler_chat.go 对话面：消息模型与内存历史、SSE 事件订阅、对话/历史接口 handler_upload.go 上传面：handleChatFile / handleUploads / 中断对话 handler_memory.go 记忆面：图/文档/文本记忆、知识库、LLM 源、变更追踪 handler_agents.go 内核与代理面：状态、kernel、人格、代理/快照/回滚 handler_settings.go 设置与插件面：配置读写、插件列表详情（含 pluginmgr 反代） handler_terminal.go 终端面：终端会话、终端接口、命令历史 handler_sse.go SSE 环形缓冲（断线重连补发） handler_openai.go OpenAI 兼容面：/v1/chat/completions handler_device.go 设备网关反代（HTTP + WS 升级透传） handler_files.go /files/ 与 /uploads/ 下载 零漂移校验：拿重构前的 handler.go 与新 11 个文件逐行比对（忽略空行、package/import 头）， **丢失行 0**；新增行恰好是 11 个文件头注释（14 行）。 顺带修掉 import 里两处假使用：handler_openai 的 sdk 只作为 Handler 字段名出现（h.sdk.）， handler_settings 的 fmt 只出现在注释里 —— 都从 import 里去掉。 验证：go build ./... / go vet / webui+config+sdk 测试全绿； 起真实实例（沿用已有 data 目录）后 /status /settings /chat/history /plugins /terminals /kernel /memory /config /login 全部 200，设置在注入 5MB 历史的情况下仍是 33,921 字节。 最大文件从 2993 → 706 行（handler_chat.go）。
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package webui
|
||
|
||
import (
|
||
"time"
|
||
|
||
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
||
"net/http"
|
||
)
|
||
|
||
// 终端面：持久终端会话状态、终端接口、命令历史。
|
||
|
||
type CmdExec struct {
|
||
Command string `json:"command"`
|
||
Stdout string `json:"stdout"`
|
||
Stderr string `json:"stderr"`
|
||
ExitCode int `json:"exit_code"`
|
||
Status string `json:"status"`
|
||
Time string `json:"time"`
|
||
}
|
||
|
||
type termState struct {
|
||
ID string `json:"id"`
|
||
Command string `json:"command"`
|
||
Running bool `json:"running"`
|
||
Output string `json:"output"`
|
||
CreatedAt string `json:"created_at"`
|
||
Uptime string `json:"uptime"`
|
||
created time.Time
|
||
}
|
||
|
||
const maxCmdHistory = 100
|
||
|
||
const maxTerminals = 50
|
||
|
||
// subscribeTerminalStream 常驻订阅终端实时画面推流（terminal_output 事件），
|
||
// 维护 termStates 的 Running 状态与全量输出缓冲，供 /api/v1/terminals 与前端轮询使用。
|
||
func (h *Handler) subscribeTerminalStream() {
|
||
if h.sdk == nil {
|
||
return
|
||
}
|
||
h.sdk.Subscribe(sdk.EventTerminalOutput, func(ev *sdk.Event) {
|
||
id, _ := ev.Payload["terminal_id"].(string)
|
||
if id == "" {
|
||
return
|
||
}
|
||
output, _ := ev.Payload["output"].(string)
|
||
running, _ := ev.Payload["running"].(bool)
|
||
h.termMu.Lock()
|
||
ts, ok := h.termStates[id]
|
||
if !ok {
|
||
ts = &termState{ID: id, created: time.Now()}
|
||
h.termStates[id] = ts
|
||
}
|
||
ts.Running = running
|
||
if output != "" {
|
||
const maxTermOutput = 64 * 1024
|
||
if len(ts.Output)+len(output) > maxTermOutput {
|
||
excess := len(ts.Output) + len(output) - maxTermOutput
|
||
if len(ts.Output) > excess {
|
||
ts.Output = ts.Output[excess:]
|
||
} else {
|
||
ts.Output = ""
|
||
}
|
||
}
|
||
ts.Output += output
|
||
}
|
||
h.termMu.Unlock()
|
||
})
|
||
}
|
||
|
||
func (h *Handler) handleTerminals(w http.ResponseWriter, r *http.Request) {
|
||
h.termMu.Lock()
|
||
terms := make([]*termState, 0, len(h.termStates))
|
||
for _, ts := range h.termStates {
|
||
ts.Uptime = time.Since(ts.created).Round(time.Second).String()
|
||
terms = append(terms, ts)
|
||
}
|
||
h.termMu.Unlock()
|
||
writeJSON(w, http.StatusOK, map[string]interface{}{"terminals": terms})
|
||
}
|
||
|
||
func (h *Handler) handleCmdHistory(w http.ResponseWriter, r *http.Request) {
|
||
h.cmdMu.Lock()
|
||
result := make([]CmdExec, len(h.cmdHistory))
|
||
copy(result, h.cmdHistory)
|
||
h.cmdMu.Unlock()
|
||
writeJSON(w, http.StatusOK, map[string]interface{}{"history": result})
|
||
}
|