mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-14 19:43:37 +00:00
feat: 设备桥共享库 + CLI全能力补齐 + GUI omniparse/computeruse 重构
- 抽取设备桥 WS 协议层为共享库 (internal/devicebridge/client/)
- CLI 补齐 11 项 caps 能力(screensee/screensue/speakeruse/camerasue/...)
- GUI 新增 omniparse 能力(Windows UIA 窗口解析)
- GUI computeruse 改用 koffi 直接调用 user32.dll，不再依赖 PowerShell C# 编译
- GUI computeruse JSON 解析兼容非标准格式 {x:500,y:300}
- 新增 mock-server 用于本地测试设备桥协议
- 新增 GUI DLL 桥接模块 (devicebridge_dll.js)
This commit is contained in:
JianFeeeee
521
cmd/mock-server/main.go
Normal file
521
cmd/mock-server/main.go
Normal file
@ -0,0 +1,521 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// ===== 模拟数据 =====
|
||||
|
||||
type Status struct {
|
||||
Status string `json:"status"`
|
||||
Version string `json:"version"`
|
||||
StartedAt string `json:"startedAt"`
|
||||
Uptime int64 `json:"uptime"`
|
||||
}
|
||||
|
||||
type Kernel struct {
|
||||
Model string `json:"model"`
|
||||
Provider string `json:"provider"`
|
||||
Status string `json:"status"`
|
||||
}
|
||||
|
||||
type Setting struct {
|
||||
Settings map[string]interface{} `json:"settings"`
|
||||
Meta map[string]interface{} `json:"meta"`
|
||||
Plugins []string `json:"plugins"`
|
||||
PluginMeta map[string]interface{} `json:"plugin_meta"`
|
||||
DisabledPlugins []string `json:"disabled_plugins"`
|
||||
}
|
||||
|
||||
type Plugin struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
Version string `json:"version"`
|
||||
Enabled bool `json:"enabled"`
|
||||
Builtin bool `json:"builtin"`
|
||||
}
|
||||
|
||||
type PluginInfo struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
Version string `json:"version"`
|
||||
Enabled bool `json:"enabled"`
|
||||
Builtin bool `json:"builtin"`
|
||||
Tools []PluginTool `json:"tools"`
|
||||
}
|
||||
|
||||
type PluginTool struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
}
|
||||
|
||||
type Adapter struct {
|
||||
Name string `json:"name"`
|
||||
Type string `json:"type"`
|
||||
Enabled bool `json:"enabled"`
|
||||
}
|
||||
|
||||
type ChatMessage struct {
|
||||
Role string `json:"role"`
|
||||
Content string `json:"content"`
|
||||
}
|
||||
|
||||
type MemoryItem struct {
|
||||
ID string `json:"id"`
|
||||
Content string `json:"content"`
|
||||
Time string `json:"time"`
|
||||
}
|
||||
|
||||
type Device struct {
|
||||
DeviceID string `json:"device_id"`
|
||||
Name string `json:"name"`
|
||||
Authorized bool `json:"authorized"`
|
||||
Online bool `json:"online"`
|
||||
Caps []string `json:"caps"`
|
||||
}
|
||||
|
||||
// ===== SSE 管理器 =====
|
||||
|
||||
type SSEManager struct {
|
||||
mu sync.RWMutex
|
||||
clients map[chan string]bool
|
||||
}
|
||||
|
||||
func NewSSEManager() *SSEManager {
|
||||
return &SSEManager{clients: make(map[chan string]bool)}
|
||||
}
|
||||
|
||||
func (m *SSEManager) Add(ch chan string) {
|
||||
m.mu.Lock()
|
||||
m.clients[ch] = true
|
||||
m.mu.Unlock()
|
||||
}
|
||||
|
||||
func (m *SSEManager) Remove(ch chan string) {
|
||||
m.mu.Lock()
|
||||
delete(m.clients, ch)
|
||||
m.mu.Unlock()
|
||||
}
|
||||
|
||||
func (m *SSEManager) Broadcast(eventType, data string) {
|
||||
msg := fmt.Sprintf("event: %s\ndata: %s\n\n", eventType, data)
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
for ch := range m.clients {
|
||||
select {
|
||||
case ch <- msg:
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ===== HTTP 处理器 =====
|
||||
|
||||
type MockServer struct {
|
||||
startedAt time.Time
|
||||
sse *SSEManager
|
||||
mu sync.Mutex
|
||||
plugins []Plugin
|
||||
adapters []Adapter
|
||||
devices []Device
|
||||
settings map[string]interface{}
|
||||
}
|
||||
|
||||
func NewMockServer() *MockServer {
|
||||
now := time.Now()
|
||||
return &MockServer{
|
||||
startedAt: now,
|
||||
sse: NewSSEManager(),
|
||||
plugins: []Plugin{
|
||||
{Name: "core", Description: "核心插件", Version: "1.0.0", Enabled: true, Builtin: true},
|
||||
{Name: "remotedevice", Description: "远程设备管理", Version: "0.9.0", Enabled: true, Builtin: true},
|
||||
{Name: "webui", Description: "Web 用户界面", Version: "0.9.0", Enabled: true, Builtin: true},
|
||||
{Name: "knowledge", Description: "知识库管理", Version: "0.5.0", Enabled: true, Builtin: false},
|
||||
},
|
||||
adapters: []Adapter{
|
||||
{Name: "openai", Type: "llm", Enabled: true},
|
||||
{Name: "siliconflow", Type: "llm", Enabled: true},
|
||||
},
|
||||
devices: []Device{
|
||||
{DeviceID: "gui-test-local", Name: "GUI 测试设备", Authorized: true, Online: true, Caps: []string{"status", "cmdrun", "deviceinfo"}},
|
||||
},
|
||||
settings: map[string]interface{}{
|
||||
"language": "zh-CN",
|
||||
"theme": "dark",
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// 中间件：CORS + API Key 校验
|
||||
func (s *MockServer) middleware(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type, X-API-Key, Authorization, Cookie")
|
||||
|
||||
if r.Method == "OPTIONS" {
|
||||
w.WriteHeader(200)
|
||||
return
|
||||
}
|
||||
|
||||
// API Key 校验（可选）
|
||||
// apiKey := r.Header.Get("X-API-Key")
|
||||
// if apiKey == "" {
|
||||
// http.Error(w, "unauthorized", 401)
|
||||
// return
|
||||
// }
|
||||
|
||||
next.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleStatus(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, Status{
|
||||
Status: "running",
|
||||
Version: "0.9.0",
|
||||
StartedAt: s.startedAt.Format(time.RFC3339),
|
||||
Uptime: int64(time.Since(s.startedAt).Seconds()),
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleKernel(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, Kernel{
|
||||
Model: "sensenova-6.8-flash-lite",
|
||||
Provider: "siliconflow",
|
||||
Status: "ready",
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleSettings(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method == "POST" {
|
||||
var updates map[string]interface{}
|
||||
if err := json.NewDecoder(r.Body).Decode(&updates); err == nil {
|
||||
s.mu.Lock()
|
||||
for k, v := range updates {
|
||||
s.settings[k] = v
|
||||
}
|
||||
s.mu.Unlock()
|
||||
}
|
||||
writeJSON(w, map[string]string{"status": "saved"})
|
||||
return
|
||||
}
|
||||
|
||||
writeJSON(w, Setting{
|
||||
Settings: s.settings,
|
||||
Meta: map[string]interface{}{
|
||||
"version": "0.9.0",
|
||||
"build": "mock-20260823",
|
||||
},
|
||||
Plugins: []string{"core", "remotedevice", "webui", "knowledge"},
|
||||
PluginMeta: map[string]interface{}{
|
||||
"core": map[string]interface{}{"version": "1.0.0"},
|
||||
"remotedevice": map[string]interface{}{"version": "0.9.0"},
|
||||
"webui": map[string]interface{}{"version": "0.9.0"},
|
||||
"knowledge": map[string]interface{}{"version": "0.5.0"},
|
||||
},
|
||||
DisabledPlugins: []string{},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handlePlugins(w http.ResponseWriter, r *http.Request) {
|
||||
// 获取路径中的插件名
|
||||
path := strings.TrimPrefix(r.URL.Path, "/api/v1/plugins")
|
||||
path = strings.TrimSuffix(path, "/")
|
||||
|
||||
if path == "/reload" && r.Method == "POST" {
|
||||
writeJSON(w, map[string]string{"status": "reloaded"})
|
||||
return
|
||||
}
|
||||
|
||||
if path == "" && r.Method == "GET" {
|
||||
writeJSON(w, s.plugins)
|
||||
return
|
||||
}
|
||||
|
||||
if path == "" && r.Method == "POST" {
|
||||
writeJSON(w, map[string]string{"status": "installed"})
|
||||
return
|
||||
}
|
||||
|
||||
// /api/v1/plugins/:name
|
||||
if strings.Contains(path, "/") {
|
||||
parts := strings.Split(strings.TrimPrefix(path, "/"), "/")
|
||||
if len(parts) >= 1 {
|
||||
name := parts[0]
|
||||
if len(parts) >= 2 {
|
||||
action := parts[1]
|
||||
if action == "disable" && r.Method == "POST" {
|
||||
s.mu.Lock()
|
||||
for i := range s.plugins {
|
||||
if s.plugins[i].Name == name {
|
||||
s.plugins[i].Enabled = false
|
||||
}
|
||||
}
|
||||
s.mu.Unlock()
|
||||
writeJSON(w, map[string]string{"status": "disabled"})
|
||||
return
|
||||
}
|
||||
if action == "enable" && r.Method == "POST" {
|
||||
s.mu.Lock()
|
||||
for i := range s.plugins {
|
||||
if s.plugins[i].Name == name {
|
||||
s.plugins[i].Enabled = true
|
||||
}
|
||||
}
|
||||
s.mu.Unlock()
|
||||
writeJSON(w, map[string]string{"status": "enabled"})
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// GET /api/v1/plugins/:name
|
||||
writeJSON(w, PluginInfo{
|
||||
Name: name,
|
||||
Description: name + " 插件描述",
|
||||
Version: "0.9.0",
|
||||
Enabled: true,
|
||||
Builtin: true,
|
||||
Tools: []PluginTool{
|
||||
{Name: name + "_tool1", Description: name + " 工具1"},
|
||||
{Name: name + "_tool2", Description: name + " 工具2"},
|
||||
},
|
||||
})
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
http.NotFound(w, r)
|
||||
}
|
||||
|
||||
func (s *MockServer) handleChatHistory(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, []ChatMessage{
|
||||
{Role: "user", Content: "你好"},
|
||||
{Role: "assistant", Content: "你好！我是 HomeAgent，有什么可以帮你的？"},
|
||||
{Role: "user", Content: "测试消息"},
|
||||
{Role: "assistant", Content: "这是模拟后端的测试回复，GUI 连接正常 ✅"},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleChat(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method == "POST" {
|
||||
// 模拟后端接收消息，通过 SSE 推流
|
||||
go func() {
|
||||
time.Sleep(500 * time.Millisecond)
|
||||
|
||||
// agent_start
|
||||
s.sse.Broadcast("agent_output", `{"type":"agent_start","payload":{"agent":"mock"}}`)
|
||||
|
||||
time.Sleep(300 * time.Millisecond)
|
||||
|
||||
// tool_call
|
||||
s.sse.Broadcast("agent_output", `{"type":"tool_call","payload":{"tool":"mock_tool","args":{},"id":"call_001"}}`)
|
||||
|
||||
time.Sleep(500 * time.Millisecond)
|
||||
|
||||
// channel_output
|
||||
s.sse.Broadcast("agent_output", `{"type":"channel_output","payload":{"kind":"channel_output","channel":"mock","content":"这是一条来自模拟后端的测试回复。\n\n- 模拟后端状态: running\n- 版本: 0.9.0\n- 连接测试: ✅ 成功\n\nGUI 所有功能验证正常！"}}`)
|
||||
|
||||
time.Sleep(300 * time.Millisecond)
|
||||
|
||||
// agent_end
|
||||
s.sse.Broadcast("agent_output", `{"type":"agent_end","payload":{"agent":"mock"}}`)
|
||||
}()
|
||||
|
||||
writeJSON(w, map[string]string{"status": "queued", "id": "mock_" + time.Now().Format("150405")})
|
||||
return
|
||||
}
|
||||
http.Error(w, "method not allowed", 405)
|
||||
}
|
||||
|
||||
func (s *MockServer) handleChatEvents(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "text/event-stream")
|
||||
w.Header().Set("Cache-Control", "no-cache")
|
||||
w.Header().Set("Connection", "keep-alive")
|
||||
|
||||
ch := make(chan string, 100)
|
||||
s.sse.Add(ch)
|
||||
defer s.sse.Remove(ch)
|
||||
|
||||
// 发送初始连接成功事件
|
||||
fmt.Fprintf(w, "event: connected\ndata: {\"status\":\"connected\"}\n\n")
|
||||
w.(http.Flusher).Flush()
|
||||
|
||||
ctx := r.Context()
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case msg := <-ch:
|
||||
fmt.Fprint(w, msg)
|
||||
w.(http.Flusher).Flush()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *MockServer) handleMemoryGraph(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, map[string]interface{}{
|
||||
"nodes": []map[string]interface{}{
|
||||
{"id": "1", "label": "HomeAgent", "group": "system"},
|
||||
{"id": "2", "label": "GUI 测试", "group": "user"},
|
||||
},
|
||||
"edges": []map[string]interface{}{
|
||||
{"from": "1", "to": "2", "label": "connected"},
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleMemory(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, []MemoryItem{
|
||||
{ID: "m1", Content: "这是模拟内存中的测试数据", Time: time.Now().Format(time.RFC3339)},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleMemoryContext(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, map[string]interface{}{
|
||||
"context": "模拟上下文：用户正在测试 GUI 功能",
|
||||
"items": []MemoryItem{},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleKnowledge(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method == "POST" {
|
||||
writeJSON(w, map[string]string{"status": "saved"})
|
||||
return
|
||||
}
|
||||
writeJSON(w, []map[string]interface{}{
|
||||
{"id": "k1", "title": "模拟知识条目1", "content": "这是模拟知识库的测试内容"},
|
||||
{"id": "k2", "title": "模拟知识条目2", "content": "GUI 功能验证测试数据"},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleTerminals(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, []map[string]interface{}{
|
||||
{"id": "t1", "name": "终端 1", "status": "running"},
|
||||
{"id": "t2", "name": "终端 2", "status": "idle"},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleCmdHistory(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, []map[string]interface{}{
|
||||
{"cmd": "echo hello", "time": time.Now().Add(-5 * time.Minute).Format(time.RFC3339)},
|
||||
{"cmd": "ls -la", "time": time.Now().Add(-10 * time.Minute).Format(time.RFC3339)},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MockServer) handleDevices(w http.ResponseWriter, r *http.Request) {
|
||||
path := strings.TrimPrefix(r.URL.Path, "/api/v1/device")
|
||||
path = strings.TrimSuffix(path, "/")
|
||||
|
||||
switch {
|
||||
case path == "/online" || path == "":
|
||||
writeJSON(w, s.devices)
|
||||
case path == "/auth" && r.Method == "POST":
|
||||
var req struct {
|
||||
DeviceID string `json:"device_id"`
|
||||
Authorized bool `json:"authorize"`
|
||||
}
|
||||
json.NewDecoder(r.Body).Decode(&req)
|
||||
s.mu.Lock()
|
||||
for i := range s.devices {
|
||||
if s.devices[i].DeviceID == req.DeviceID {
|
||||
s.devices[i].Authorized = req.Authorized
|
||||
}
|
||||
}
|
||||
s.mu.Unlock()
|
||||
writeJSON(w, map[string]interface{}{
|
||||
"authorized": true,
|
||||
"device_id": req.DeviceID,
|
||||
})
|
||||
case path == "/push" && r.Method == "POST":
|
||||
writeJSON(w, map[string]string{"status": "pushed"})
|
||||
default:
|
||||
http.NotFound(w, r)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *MockServer) handleAdapters(w http.ResponseWriter, r *http.Request) {
|
||||
path := strings.TrimPrefix(r.URL.Path, "/api/v1/adapters")
|
||||
path = strings.TrimSuffix(path, "/")
|
||||
|
||||
switch {
|
||||
case path == "" && r.Method == "GET":
|
||||
writeJSON(w, s.adapters)
|
||||
case path == "" && r.Method == "POST":
|
||||
var a Adapter
|
||||
if err := json.NewDecoder(r.Body).Decode(&a); err == nil {
|
||||
s.mu.Lock()
|
||||
s.adapters = append(s.adapters, a)
|
||||
s.mu.Unlock()
|
||||
}
|
||||
writeJSON(w, map[string]string{"status": "added"})
|
||||
case strings.Count(path, "/") == 1 && r.Method == "DELETE":
|
||||
name := strings.TrimPrefix(path, "/")
|
||||
s.mu.Lock()
|
||||
for i := range s.adapters {
|
||||
if s.adapters[i].Name == name {
|
||||
s.adapters = append(s.adapters[:i], s.adapters[i+1:]...)
|
||||
break
|
||||
}
|
||||
}
|
||||
s.mu.Unlock()
|
||||
writeJSON(w, map[string]string{"status": "deleted"})
|
||||
default:
|
||||
http.NotFound(w, r)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *MockServer) handleWebSocket(w http.ResponseWriter, r *http.Request) {
|
||||
// 简单返回 400，GUI 的 main.js 会尝试连接设备桥 WS
|
||||
// 这里只验证 HTTP 路由可达
|
||||
http.Error(w, "WebSocket upgrade required (mock server)", 400)
|
||||
}
|
||||
|
||||
// ===== 路由注册 =====
|
||||
|
||||
func (s *MockServer) registerRoutes(mux *http.ServeMux) {
|
||||
mux.HandleFunc("/api/v1/status", s.handleStatus)
|
||||
mux.HandleFunc("/api/v1/kernel", s.handleKernel)
|
||||
mux.HandleFunc("/api/v1/settings", s.handleSettings)
|
||||
mux.HandleFunc("/api/v1/plugins", s.handlePlugins)
|
||||
mux.HandleFunc("/api/v1/plugins/", s.handlePlugins)
|
||||
mux.HandleFunc("/api/v1/chat/history", s.handleChatHistory)
|
||||
mux.HandleFunc("/api/v1/chat", s.handleChat)
|
||||
mux.HandleFunc("/api/v1/chat/events", s.handleChatEvents)
|
||||
mux.HandleFunc("/api/v1/memory/graph", s.handleMemoryGraph)
|
||||
mux.HandleFunc("/api/v1/memory", s.handleMemory)
|
||||
mux.HandleFunc("/api/v1/memory/context", s.handleMemoryContext)
|
||||
mux.HandleFunc("/api/v1/knowledge", s.handleKnowledge)
|
||||
mux.HandleFunc("/api/v1/terminals", s.handleTerminals)
|
||||
mux.HandleFunc("/api/v1/cmd/history", s.handleCmdHistory)
|
||||
mux.HandleFunc("/api/v1/device", s.handleDevices)
|
||||
mux.HandleFunc("/api/v1/device/", s.handleDevices)
|
||||
mux.HandleFunc("/api/v1/adapters", s.handleAdapters)
|
||||
mux.HandleFunc("/api/v1/adapters/", s.handleAdapters)
|
||||
mux.HandleFunc("/api/v1/device/ws", s.handleDeviceWS)
|
||||
}
|
||||
|
||||
func writeJSON(w http.ResponseWriter, v interface{}) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(v)
|
||||
}
|
||||
|
||||
func main() {
|
||||
server := NewMockServer()
|
||||
mux := http.NewServeMux()
|
||||
server.registerRoutes(mux)
|
||||
|
||||
addr := ":9099"
|
||||
log.Printf("=== HomeAgent Mock Server ====")
|
||||
log.Printf("监听地址: http://0.0.0.0%s", addr)
|
||||
log.Printf("API 基础路径: http://0.0.0.0%s/api/v1/", addr)
|
||||
log.Printf("SSE 端点: http://0.0.0.0%s/api/v1/chat/events", addr)
|
||||
log.Printf("设备桥 WS: ws://0.0.0.0%s/api/v1/device/ws", addr)
|
||||
log.Printf("==============================")
|
||||
log.Fatal(http.ListenAndServe(addr, server.middleware(mux)))
|
||||
}
|
||||
Reference in New Issue
Block a user