diff --git a/cmd/homed/main.go b/cmd/homed/main.go index bab8949..cca9973 100644 --- a/cmd/homed/main.go +++ b/cmd/homed/main.go @@ -358,6 +358,7 @@ func main() { pluginReg.SetProviderManager(providerMgr) pluginReg.SetConfigRegistry(cfgReg) pluginReg.SetPluginDir(cfg.Plugin.Dir) + pluginReg.SetDataDir(*dataDir) // 插件 SettingsAPI.DataDir() 的数据根目录 // Wire registration callbacks: plugins' RegisterTool/RegisterStage → StageHost pluginReg.SetToolRegistrar(func(name string, def sdk.ToolDef, handler sdk.ToolHandler) error { diff --git a/internal/plugin/cabi/loader.go b/internal/plugin/cabi/loader.go index f42b16a..505602d 100644 --- a/internal/plugin/cabi/loader.go +++ b/internal/plugin/cabi/loader.go @@ -877,6 +877,12 @@ func go_core_dispatch(methodID C.int, ctx unsafe.Pointer, s1, s2, s3 *C.char, i1 } return 0 + case 51: // CORE_SETTINGS_DATA_DIR：插件专属数据目录（内核保证存在） + if sett := s.Settings(); sett != nil { + setResult(result, sett.DataDir()) + } + return 0 + case 46: // CORE_REGISTER_INPUT_CH chDef := sdk.ChannelDef{} if a2 != "" { diff --git a/internal/plugin/registry.go b/internal/plugin/registry.go index 39c2c90..2903270 100644 --- a/internal/plugin/registry.go +++ b/internal/plugin/registry.go @@ -81,6 +81,7 @@ type Registry struct { mgr *agentAPI.ProviderManager cfgReg *internalConfig.ConfigRegistry plgDir string + dataDir string // 守护进程数据目录（注入给插件 SettingsAPI.DataDir） lua *luaVM.VM baseKey string @@ -125,6 +126,7 @@ func (r *Registry) SetKnowledge(ks *knowledge.Store) { r. func (r *Registry) SetProviderManager(mgr *agentAPI.ProviderManager) { r.mgr = mgr } func (r *Registry) SetConfigRegistry(cfgReg *internalConfig.ConfigRegistry) { r.cfgReg = cfgReg } func (r *Registry) SetPluginDir(dir string) { r.plgDir = dir } +func (r *Registry) SetDataDir(dir string) { r.dataDir = dir } func (r *Registry) SetLuaVM(vm *luaVM.VM) { r.lua = vm } func (r *Registry) SetBaseAPIKey(key string) { r.baseKey = key } func (r *Registry) SetToolRegistrar(fn sdk.ToolRegistrar) { r.regTool = fn } @@ -191,6 +193,12 @@ func (d *channelDevice) ChannelDef() agentIO.ChannelDef { return d.chDef } func (r *Registry) buildSDK(name string) *sdk.PluginSDK { sett := sdk.NewSettings(name, r.cfgReg) + if sd, ok := sett.(interface{ SetDataDir(string) }); ok { + // 插件专属数据目录：/plugin_data/，保证存在 + dir := filepath.Join(r.dataDir, "plugin_data", name) + os.MkdirAll(dir, 0755) + sd.SetDataDir(dir) + } regTool := r.regTool if regTool == nil { diff --git a/internal/plugins/ai_image/plugin.go b/internal/plugins/ai_image/plugin.go index 42b955d..52024a3 100644 --- a/internal/plugins/ai_image/plugin.go +++ b/internal/plugins/ai_image/plugin.go @@ -6,6 +6,8 @@ import ( "fmt" "io" "net/http" + "os" + "path/filepath" "strings" "time" @@ -27,6 +29,7 @@ type Plugin struct { baseURL string model string size string + dataDir string } func New(name string) *Plugin { @@ -55,6 +58,14 @@ func getSettingString(s sdk.SettingsAPI, key, def string) string { func (p *Plugin) Start(s *sdk.PluginSDK) error { s.SetAutoRestart(true) + // 插件专属数据目录（SDK DataDir API，内核保证存在） + if dd := s.Settings().DataDir(); dd != "" { + p.dataDir = dd + } + if p.dataDir != "" { + os.MkdirAll(p.dataDir, 0755) + } + s.Settings().RegisterDef(sdk.ConfigDef{ Key: "base_url", Default: "http://127.0.0.1:8081/v1", Type: "string", DisplayName: "Base URL", Description: "OpenAI 兼容生图网关地址（默认本机 llmsproxy）", @@ -193,8 +204,7 @@ func (p *Plugin) handleGenerate(args map[string]interface{}) (interface{}, error } var urls []string - var saved []string - for i, d := range result.Data { + for _, d := range result.Data { imgURL := d.URL if imgURL == "" && d.B64JSON != "" { imgURL = "data:image/png;base64," + d.B64JSON @@ -202,18 +212,87 @@ func (p *Plugin) handleGenerate(args map[string]interface{}) (interface{}, error if imgURL != "" { urls = append(urls, imgURL) } - if i == 0 { - saved = append(saved, imgURL) - } } if len(urls) == 0 { return map[string]interface{}{"isError": true, "content": "生图响应中没有可用图片"}, nil } + // 下载到插件数据目录，返回本地文件路径（而非临时 S3 URL）： + // - S3 临时 URL 约 1 小时过期，且对无浏览器 UA 客户端拒绝访问（agent 裸 curl 验证必败） + // - 本地路径经 webui /files/ 永久下发，支持 output_send type=image + var localPaths []string + var dlErrs []string + for i, u := range urls { + if strings.HasPrefix(u, "data:") { + continue // base64 内联图不落盘 + } + path, err := p.downloadImage(u, fmt.Sprintf("ai_%d_%d", time.Now().UnixNano(), i)) + if err != nil { + dlErrs = append(dlErrs, fmt.Sprintf("第%d张保存失败: %v", i+1, err)) + continue + } + localPaths = append(localPaths, path) + } + + content := fmt.Sprintf("Generated %d image(s) with model %s", len(urls), model) + if len(localPaths) > 0 { + content += "\n本地文件：\n" + strings.Join(localPaths, "\n") + content += "\n\n图片已保存到本地（不会过期）。如需展示请用 output_send__webui(payload=本地路径, type=image)。" + } + if len(dlErrs) > 0 { + content += "\n\n" + strings.Join(dlErrs, "\n") + } + if len(localPaths) < len(urls) { + content += "\n原始 URL（1小时内有效）：\n" + strings.Join(urls, "\n") + } + return map[string]interface{}{ - "content": fmt.Sprintf("Generated %d image(s) with model %s:\n%s", len(urls), model, strings.Join(urls, "\n")), - "images": urls, - "prompt": prompt, - "model": model, + "content": content, + "images": urls, + "local_paths": localPaths, + "prompt": prompt, + "model": model, }, nil } + +// downloadImage 把生图返回的临时 URL 下载为本地文件，返回路径。 +// 带浏览器 UA 规避图床对无 UA 客户端的拦截。 +func (p *Plugin) downloadImage(imgURL, baseName string) (string, error) { + if p.dataDir == "" { + return "", fmt.Errorf("data dir unavailable") + } + dl := &http.Client{Timeout: 60 * time.Second} + req, err := http.NewRequest(http.MethodGet, imgURL, nil) + if err != nil { + return "", err + } + req.Header.Set("User-Agent", "Mozilla/5.0 (compatible; HomeAgent/1.0)") + resp, err := dl.Do(req) + if err != nil { + return "", err + } + defer resp.Body.Close() + if resp.StatusCode != http.StatusOK { + b, _ := io.ReadAll(resp.Body) + if len(b) > 200 { + b = b[:200] + } + return "", fmt.Errorf("HTTP %d: %s", resp.StatusCode, strings.TrimSpace(string(b))) + } + data, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + ext := ".png" + switch ct := resp.Header.Get("Content-Type"); { + case strings.Contains(ct, "jpeg"), strings.Contains(ct, "jpg"): + ext = ".jpg" + case strings.Contains(ct, "webp"): + ext = ".webp" + } + path := filepath.Join(p.dataDir, baseName+ext) + if err := os.WriteFile(path, data, 0644); err != nil { + return "", err + } + return path, nil +} diff --git a/internal/plugins/clawhubadapter/sidecar_test.go b/internal/plugins/clawhubadapter/sidecar_test.go index c8b4467..7b6fc9a 100644 --- a/internal/plugins/clawhubadapter/sidecar_test.go +++ b/internal/plugins/clawhubadapter/sidecar_test.go @@ -13,6 +13,8 @@ import ( // mockSettings implements pubsdk.SettingsAPI for tests type mockSettings struct{} +func (m *mockSettings) DataDir() string { return "/tmp/mock_data" } + func (m *mockSettings) Get(key string) (interface{}, error) { return nil, nil } func (m *mockSettings) Set(key string, value interface{}) error { return nil } func (m *mockSettings) List(prefix string) ([]string, error) { return nil, nil } diff --git a/internal/plugins/pluginmgr/upgrade_test.go b/internal/plugins/pluginmgr/upgrade_test.go index 6d4f467..e5ce59b 100644 --- a/internal/plugins/pluginmgr/upgrade_test.go +++ b/internal/plugins/pluginmgr/upgrade_test.go @@ -35,6 +35,8 @@ func TestCmpVersion(t *testing.T) { type pmSettings struct{} +func (m *pmSettings) DataDir() string { return "/tmp/mock_data" } + func (m *pmSettings) Get(string) (interface{}, error) { return nil, nil } func (m *pmSettings) Set(string, interface{}) error { return nil } func (m *pmSettings) List(string) ([]string, error) { return nil, nil } diff --git a/internal/plugins/skillmgr/skillmgr_test.go b/internal/plugins/skillmgr/skillmgr_test.go index 6a7bbb9..0c9b842 100644 --- a/internal/plugins/skillmgr/skillmgr_test.go +++ b/internal/plugins/skillmgr/skillmgr_test.go @@ -19,6 +19,8 @@ import ( type nilSettings struct{} +func (m *nilSettings) DataDir() string { return "/tmp/mock_data" } + func (m *nilSettings) Get(string) (interface{}, error) { return nil, nil } func (m *nilSettings) Set(string, interface{}) error { return nil } func (m *nilSettings) List(string) ([]string, error) { return nil, nil } diff --git a/internal/sdk/settings_impl.go b/internal/sdk/settings_impl.go index 6bb8850..42c8269 100644 --- a/internal/sdk/settings_impl.go +++ b/internal/sdk/settings_impl.go @@ -10,12 +10,19 @@ import ( type settingsImpl struct { pluginName string reg *internalConfig.ConfigRegistry + dataDir string } func NewSettings(name string, reg *internalConfig.ConfigRegistry) SettingsAPI { return &settingsImpl{pluginName: name, reg: reg} } +// SetDataDir 注入本插件的数据目录（内核装配时调用）。 +func (s *settingsImpl) SetDataDir(dir string) { s.dataDir = dir } + +// DataDir 返回插件专属数据目录 /plugin_data/，保证目录存在。 +func (s *settingsImpl) DataDir() string { return s.dataDir } + func (s *settingsImpl) Get(key string) (interface{}, error) { if s.reg == nil { return nil, nil diff --git a/third_party/homeagent-sdk/example/ai_image/plg.json b/third_party/homeagent-sdk/example/ai_image/plg.json index 3d15c16..cf36da2 100644 --- a/third_party/homeagent-sdk/example/ai_image/plg.json +++ b/third_party/homeagent-sdk/example/ai_image/plg.json @@ -2,7 +2,7 @@ "name": "ai_image", "name_zh": "AI绘图", "name_en": "AI Image", - "version": "1.1.0", + "version": "1.3.0", "description": "AI 图像生成插件，支持 OpenAI DALL·E / Stable Diffusion", "author": "HomeAgent", "entry": "plugin.so", diff --git a/third_party/homeagent-sdk/example/ai_image/plugin.go b/third_party/homeagent-sdk/example/ai_image/plugin.go index f1c421a..612be0c 100644 --- a/third_party/homeagent-sdk/example/ai_image/plugin.go +++ b/third_party/homeagent-sdk/example/ai_image/plugin.go @@ -5,7 +5,10 @@ import ( "encoding/json" "fmt" "io" + "log" "net/http" + "os" + "path/filepath" "strconv" "strings" "time" @@ -22,6 +25,7 @@ type Plugin struct { model string size string baseURL string + dataDir string // /ai_images：生成本地图片存放目录 } func NewPluginFactory(name string, config map[string]interface{}) (sdk.Plugin, error) { @@ -139,9 +143,21 @@ func (p *Plugin) Start(s *sdk.PluginSDK) error { p.size = getSetting(s.Settings(), "size", "1024x1024") p.baseURL = strings.TrimRight(strings.TrimSpace(getSetting(s.Settings(), "base_url", "")), "/") +// 生图本地存放目录：插件专属数据目录（SDK DataDir API，内核保证存在）。 +if p.sdk != nil { + if dd := s.Settings().DataDir(); dd != "" { + p.dataDir = dd + } +} +if p.dataDir == "" { + // 旧版内核无 DataDir API 时退到 /tmp + p.dataDir = filepath.Join(os.TempDir(), "homeagent_ai_images") +} +os.MkdirAll(p.dataDir, 0755) + tp := p.name + "_" s.RegisterTool(tp+"generate", sdk.ToolDef{ - Name: tp + "generate", Description: "Generate image from text prompt using AI. Returns image URL.", + Name: tp + "generate", Description: "Generate image from text prompt using AI. Downloads the result locally and returns a local file path (permanent, no expiry). To show the user, send it via output_send with type=image and payload=the returned path.", Parameters: map[string]interface{}{ "type": "object", "properties": map[string]interface{}{ @@ -232,6 +248,7 @@ func (p *Plugin) generateOpenAI(prompt, model, size string, n int, apiKey string ResponseFormat: "url", } + log.Printf("[ai_image] endpoint=%s baseURL=%q model=%q", endpoint, p.baseURL, model) b, _ := json.Marshal(body) req, _ := http.NewRequest("POST", endpoint, bytes.NewReader(b)) req.Header.Set("Content-Type", "application/json") @@ -262,14 +279,74 @@ func (p *Plugin) generateOpenAI(prompt, model, size string, n int, apiKey string urls[i] = d.URL } + // 下载到本地 data 目录，返回本地文件路径（而非临时 S3 URL）： + // - S3 临时 URL 约 1 小时过期，且对无浏览器 UA 的客户端拒绝访问 + // - 本地路径可经 webui /files/ 永久下发给所有客户端（含 API key 客户端） + localPaths := make([]string, len(urls)) + var errs []string + for i, u := range urls { + path, err := p.downloadImage(u, fmt.Sprintf("ai_%s_%d", model, time.Now().UnixNano())) + if err != nil { + errs = append(errs, fmt.Sprintf("第%d张下载失败: %v", i+1, err)) + continue + } + localPaths[i] = path + } + + content := fmt.Sprintf("Generated %d image(s) with model %s:", len(urls), model) + for _, pth := range localPaths { + if pth != "" { + content += "\n" + pth + } + } + if len(errs) > 0 { + content += "\n\n" + strings.Join(errs, "\n") + } + content += "\n\n已将图片保存到本地（不会过期）。如需展示请用 output_send__webui(payload=本地路径, type=image)。" return map[string]interface{}{ - "content": fmt.Sprintf("Generated %d image(s) with model %s:\n%s", len(urls), model, strings.Join(urls, "\n")), - "images": urls, - "prompt": prompt, - "model": model, + "content": content, + "images": localPaths, + "prompt": prompt, + "model": model, + "local_paths": localPaths, }, nil } +// downloadImage 把生图返回的临时 URL 下载为本地文件，返回本地路径。 +// 带浏览器 UA 以规避图床对无 UA 客户端的拦截。 +func (p *Plugin) downloadImage(url, baseName string) (string, error) { + dl := &http.Client{Timeout: 60 * time.Second} + req, err := http.NewRequest("GET", url, nil) + if err != nil { + return "", err + } + req.Header.Set("User-Agent", "Mozilla/5.0 (compatible; HomeAgent/1.0)") + resp, err := dl.Do(req) + if err != nil { + return "", err + } + defer resp.Body.Close() + if resp.StatusCode != http.StatusOK { + b, _ := io.ReadAll(resp.Body) + return "", fmt.Errorf("HTTP %d: %s", resp.StatusCode, strings.TrimSpace(string(b))[:200]) + } + data, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + ext := ".png" + if ct := resp.Header.Get("Content-Type"); strings.Contains(ct, "jpeg") || strings.Contains(ct, "jpg") { + ext = ".jpg" + } else if strings.Contains(ct, "webp") { + ext = ".webp" + } + path := filepath.Join(p.dataDir, baseName+ext) + if err := os.WriteFile(path, data, 0644); err != nil { + return "", err + } + return path, nil +} + type stabilityReq struct { TextPrompts []stabilityPrompt `json:"text_prompts"` Width int `json:"width"` @@ -349,7 +426,7 @@ func (p *Plugin) generateStability(prompt, model, size string, n int, apiKey str } return map[string]interface{}{ - "content": fmt.Sprintf("Generated %d image(s) via Stability AI:\n%s", len(urls), strings.Join(urls, "\n")), + "content": fmt.Sprintf("Generated %d image(s) via Stability AI:\n%s\n\n图片已保存到本地，如需展示请用 output_send(type=image)。", len(urls), strings.Join(urls, "\n")), "images": urls, "prompt": prompt, "model": model, diff --git a/third_party/homeagent-sdk/sdk/settings.go b/third_party/homeagent-sdk/sdk/settings.go index 61bfdc6..75f553d 100644 --- a/third_party/homeagent-sdk/sdk/settings.go +++ b/third_party/homeagent-sdk/sdk/settings.go @@ -19,6 +19,11 @@ type SettingsAPI interface { // ListCore lists core config keys matching the prefix. ListCore(prefix string) ([]string, error) + // DataDir returns the plugin-specific data directory (guaranteed to exist): + // /plugin_data/. Plugins should persist any + // runtime files (generated images, caches, downloads) here. + DataDir() string + // GetPlugin reads another plugin's config table. GetPlugin(plugin, key string) (interface{}, error) diff --git a/third_party/homeagent-sdk/tools/plugindev/templates.go b/third_party/homeagent-sdk/tools/plugindev/templates.go new file mode 100644 index 0000000..4669342 --- /dev/null +++ b/third_party/homeagent-sdk/tools/plugindev/templates.go @@ -0,0 +1,1227 @@ +package main + +// tmplPlgJSON is the plg.json template +const tmplPlgJSON = `{ + "name": "{{.Plg.Name}}", + "name_zh": "{{.Plg.NameZh}}", + "name_en": "{{.Plg.NameEn}}", + "version": "{{.Plg.Version}}", + "description": "{{.Plg.Description}}", + "author": "{{.Plg.Author}}", + "entry": "{{.Plg.Entry}}", + "tags": [{{range $i, $t := .Plg.Tags}}{{if $i}}, {{end}}"{{$t}}"{{end}}], + "targets": "{{.Plg.Targets}}" +} +` + +const tmplGoMod = `module {{.ModulePath}} + +go {{.GoVersion}} + +require {{.SDKModule}} {{.SDKVersion}} +` + +const tmplPluginGo = `package main + +import ( + "fmt" + "gitcode.com/JianFeeeee/homeagent-sdk/sdk" +) + +type Plugin struct { + name string + sdk *sdk.PluginSDK +} + +func (p *Plugin) Name() string { return p.name } + +func (p *Plugin) Start(s *sdk.PluginSDK) error { + p.sdk = s + s.RegisterStopHandler(func() { fmt.Printf("[%s] stop handler running\n", p.name) }) + s.Settings().RegisterDef(sdk.ConfigDef{ + Key: "plugin.{{.Plg.Name}}.example", Default: "hello", Type: "string", + DisplayName: "示例配置", Description: "An example configuration key", + Category: "{{.Plg.Name}}", + }) + tp := p.name + "_" + s.RegisterTool(tp+"hello", sdk.ToolDef{ + Name: tp + "hello", + Description: "A hello world tool", + Parameters: map[string]interface{}{"type": "object", "properties": map[string]interface{}{}}, + NoMemory: false, // 工具输出对 LLM 注意力有信号价值时为 false，纯操作工具为 true + // Cleaner: func(output string) string { + // // 工具输出参与向量化/jieba/蒸馏前，在此过滤噪音 + // return output + // }, + }, p.handleHello) + fmt.Printf("[%s] started\n", p.name) + return nil +} + +func (p *Plugin) Stop() error { fmt.Printf("[%s] stopped\n", p.name); return nil } + +func (p *Plugin) handleHello(args map[string]interface{}) (interface{}, error) { + return map[string]interface{}{"content": "Hello from {{.Plg.Name}} plugin!"}, nil +} + +func NewPluginFactory(name string, config map[string]interface{}) (sdk.Plugin, error) { + return &Plugin{name: name}, nil +} +` + +const tmplSDKLua = `-- HomeAgent Lua Plugin SDK (standalone mock) +sdk = {} +function sdk.log(level, msg) print("[lua-plugin] " .. tostring(level) .. ": " .. tostring(msg)) end +function sdk.register_tool(name, def, handler) print("[lua-plugin] register_tool: " .. tostring(name)) end +function sdk.register_stage(stage, handler, scope) print("[lua-plugin] register_stage: " .. tostring(stage) .. " scope=" .. tostring(scope)) end +function sdk.register_api(name) print("[lua-plugin] register_api: " .. tostring(name)) end +function sdk.register_output_channel(name, caps, desc, def, handler) print("[lua-plugin] register_output_channel: " .. tostring(name)) end +function sdk.register_input_channel(name, def) print("[lua-plugin] register_input_channel: " .. tostring(name)) end +function sdk.get_setting(key) return nil end +function sdk.set_setting(key, value) print("[lua-plugin] set_setting: " .. tostring(key)) end +function sdk.inject_text(source, channel, text) print("[lua-plugin] inject_text: " .. tostring(source)) end +function sdk.inject_interrupt(source, channel, text) print("[lua-plugin] inject_interrupt: " .. tostring(source)) end +function sdk.inject_text_no_memory(source, channel, text) print("[lua-plugin] inject_text_no_memory: " .. tostring(source)) end +function sdk.set_auto_restart(enabled) print("[lua-plugin] set_auto_restart: " .. tostring(enabled)) end +sdk.memory = {} +function sdk.memory.recall(query, depth) return {entities={}, relations={}} end +function sdk.memory.commit(triples) return nil end +function sdk.memory.introspect() return {} end +function sdk.memory.merge(source, target) return 0 end +function sdk.memory.purge(criteria, hard) return 0 end +sdk.doc = {} +function sdk.doc.query(text, top_k) return {} end +function sdk.doc.insert(doc) return nil end +function sdk.doc.remove(id) return nil end +function sdk.doc.stats() return {} end +sdk.knowledge = {} +function sdk.knowledge.search(query, limit) return {} end +function sdk.knowledge.add(tag, content) return nil end +function sdk.knowledge.list() return {} end +sdk.text_memory = {} +function sdk.text_memory.append(evt) return nil end +sdk.llm = {} +function sdk.llm.list_sources() return {} end +function sdk.llm.set_source(name) return nil end +function sdk.llm.current_source() return nil end +sdk.social = {} +function sdk.social.get_person(name) return {} end +function sdk.social.get_network(name, depth) return {} end +function sdk.social.get_trait(name, trait) return {value=nil, found=false} end +function sdk.social.get_relations(name) return {} end +function sdk.social.list_persons() return {} end +sdk.settings = {} +function sdk.settings.get_core(key) return nil end +function sdk.settings.set_core(key, value) return nil end +function sdk.settings.list_core(prefix) return {} end +function sdk.settings.get_plugin(plugin, key) return nil end +function sdk.settings.set_plugin(plugin, key, value) return nil end +function sdk.settings.list_plugin(plugin, prefix) return {} end +function sdk.settings.list(prefix) return {} end +function sdk.settings.register_def(def) return nil end +function sdk.settings.defs(prefix) return {} end +function sdk.settings.dump() return {} end +function sdk.settings.plugins() return {} end +sdk.json = {} +function sdk.json.encode(val) + if type(val) == "string" then return '"' .. val:gsub('"', '\\"'):gsub('\n', '\\n') .. '"' + elseif type(val) == "number" or type(val) == "boolean" then return tostring(val) + elseif type(val) == "table" then local parts, i = {}, 1 + for k, v in pairs(val) do parts[i] = sdk.json.encode(k) .. ":" .. sdk.json.encode(v); i = i + 1 end + return "{" .. table.concat(parts, ",") .. "}" end + return "null" +end +function sdk.json.decode(str) local ok, fn = pcall(load, "return " .. str); if ok then return fn() end; return nil end +sdk.http = {} +function sdk.http.get(url) print("[lua-plugin] http.get: " .. tostring(url)); return {status=200, body='{"mock":true}', headers={}} end +function sdk.http.post(url, body, ct) print("[lua-plugin] http.post: " .. tostring(url)); return {status=200, body='{"mock":true}', headers={}} end +return sdk +` + +const tmplMainLua = `-- {{.Plg.Name}} plugin +local plugin = { name = "{{.Plg.Name}}" } +function plugin.start(sdk) + sdk.log("info", "{{.Plg.Name}} starting...") + sdk.register_tool("{{.Plg.Name}}_hello", { + description = "A hello world tool", + parameters = { type = "object", properties = {} } + }, function(args) return { content = "Hello from {{.Plg.Name}} plugin!" } end) + sdk.log("info", "{{.Plg.Name}} started") +end +function plugin.stop() sdk.log("info", "{{.Plg.Name}} stopped") end +return plugin +` + +// tmplBridge — Windows DLL C ABI bridge (unchanged) +const tmplBridge = `//go:build windows && cgo + +package main + +/* +#include +*/ +import "C" +import ( + "encoding/json" + "sync" + "unsafe" + sdk "gitcode.com/JianFeeeee/homeagent-sdk/sdk" +) + +var ( + mu sync.Mutex + handleMap = map[unsafe.Pointer]*bridgeState{} +) + +type bridgeState struct { + plugin sdk.Plugin + toolDefs map[string]sdk.ToolDef + handlers map[string]sdk.ToolHandler + stages map[string]sdk.StageHandler + settings map[string]interface{} + sdk *sdk.PluginSDK +} + +func newHandle(plg sdk.Plugin) unsafe.Pointer { + mu.Lock(); defer mu.Unlock() + h := C.malloc(C.size_t(1)) + handleMap[h] = &bridgeState{ + plugin: plg, toolDefs: make(map[string]sdk.ToolDef), + handlers: make(map[string]sdk.ToolHandler), stages: make(map[string]sdk.StageHandler), + settings: make(map[string]interface{}), + } + return h +} +func getState(h unsafe.Pointer) *bridgeState { mu.Lock(); defer mu.Unlock(); return handleMap[h] } +func delState(h unsafe.Pointer) { mu.Lock(); defer mu.Unlock(); delete(handleMap, h); C.free(h) } + +//export NewPlugin +func NewPlugin(name *C.char, configJSON *C.char) unsafe.Pointer { + goName := C.GoString(name) + var config map[string]interface{} + if configJSON != nil { + var wrapper map[string]interface{} + if err := json.Unmarshal([]byte(C.GoString(configJSON)), &wrapper); err == nil { + if c, ok := wrapper["config"].(map[string]interface{}); ok { config = c } + } + } + plg, err := NewPluginFactory(goName, config) + if err != nil { return nil } + return newHandle(plg) +} + +//export StartPlugin +func StartPlugin(handle unsafe.Pointer) C.int { + bs := getState(handle) + if bs == nil { return 1 } + mockSett := &bridgeSettings{data: bs.settings} + mockSDK := sdk.New(bs.plugin.Name(), mockSett, + func(name string, def sdk.ToolDef, handler sdk.ToolHandler) error { + bs.toolDefs[name] = def; bs.handlers[name] = handler; return nil + }, + func(stage sdk.Stage, handler sdk.StageHandler) { bs.stages[string(stage)] = handler }, + func(name string) error { return nil }, + func(name string, caps int, desc string, def sdk.ChannelDef, handler sdk.ToolHandler) error { return nil }, + ) + mockSDK.SetInputChannelRegistrar(func(name string, def sdk.ChannelDef) error { return nil }) + bs.sdk = mockSDK + if err := bs.plugin.Start(mockSDK); err != nil { return 1 } + return 0 +} + +//export StopPlugin +func StopPlugin(handle unsafe.Pointer) C.int { + bs := getState(handle) + if bs == nil { return 1 } + if bs.sdk != nil { + bs.sdk.RunStopHandlers() + } + if err := bs.plugin.Stop(); err != nil { return 1 } + return 0 +} + +//export DestroyPlugin +func DestroyPlugin(handle unsafe.Pointer) { + if bs := getState(handle); bs != nil { delState(handle) } +} + +//export GetToolDefsJSON +func GetToolDefsJSON(handle unsafe.Pointer) *C.char { + bs := getState(handle) + if bs == nil { return nil } + defs := make([]sdk.ToolDef, 0, len(bs.toolDefs)) + for _, def := range bs.toolDefs { defs = append(defs, def) } + b, _ := json.Marshal(defs) + return C.CString(string(b)) +} + +//export InvokeToolJSON +func InvokeToolJSON(handle unsafe.Pointer, toolName *C.char, argsJSON *C.char) *C.char { + bs := getState(handle) + if bs == nil || toolName == nil { return nil } + goName := C.GoString(toolName) + handler, ok := bs.handlers[goName] + if !ok { errMsg, _ := json.Marshal(map[string]interface{}{"error": "tool not found: " + goName}); return C.CString(string(errMsg)) } + var args map[string]interface{} + if argsJSON != nil { json.Unmarshal([]byte(C.GoString(argsJSON)), &args) } + r, err := handler(args) + if err != nil { errMsg, _ := json.Marshal(map[string]interface{}{"error": err.Error()}); return C.CString(string(errMsg)) } + b, _ := json.Marshal(r) + return C.CString(string(b)) +} + +//export GetStagesJSON +func GetStagesJSON(handle unsafe.Pointer) *C.char { + bs := getState(handle) + if bs == nil { return nil } + type se struct { Stage string ` + "`" + `json:"stage"` + "`" + ` } + var entries []se + for s := range bs.stages { entries = append(entries, se{s}) } + b, _ := json.Marshal(entries) + return C.CString(string(b)) +} + +//export InvokeStage +func InvokeStage(handle unsafe.Pointer, stage *C.char, contextJSON *C.char) C.int { + bs := getState(handle) + if bs == nil || stage == nil { return 1 } + goStage := C.GoString(stage) + handler, ok := bs.stages[goStage] + if !ok { return 1 } + var ctx map[string]interface{} + if contextJSON != nil { json.Unmarshal([]byte(C.GoString(contextJSON)), &ctx) } + sc := &sdk.StageContext{} + if ctx != nil { + if v, ok := ctx["raw_message"].(string); ok { sc.RawMessage = v } + if v, ok := ctx["user_id"].(string); ok { sc.UserID = v } + if v, ok := ctx["phase"].(string); ok { sc.Phase = sdk.Stage(v) } + } + if err := handler(sc); err != nil { return 1 } + return 0 +} + +//export FreeCString +func FreeCString(s *C.char) { C.free(unsafe.Pointer(s)) } + +type bridgeSettings struct{ data map[string]interface{} } +func (s *bridgeSettings) Get(key string) (interface{}, error) { v, ok := s.data[key]; if !ok { return nil, nil }; return v, nil } +func (s *bridgeSettings) Set(key string, value interface{}) error { s.data[key] = value; return nil } +func (s *bridgeSettings) List(prefix string) ([]string, error) { + var keys []string + for k := range s.data { if len(k) >= len(prefix) && k[:len(prefix)] == prefix { keys = append(keys, k) } } + return keys, nil +} +func (s *bridgeSettings) GetCore(key string) (interface{}, error) { return nil, nil } +func (s *bridgeSettings) SetCore(key string, value interface{}) error { return nil } +func (s *bridgeSettings) ListCore(prefix string) ([]string, error) { return nil, nil } +func (s *bridgeSettings) GetPlugin(plugin, key string) (interface{}, error) { return nil, nil } +func (s *bridgeSettings) SetPlugin(plugin, key string, value interface{}) error { return nil } +func (s *bridgeSettings) ListPlugin(plugin, prefix string) ([]string, error) { return nil, nil } +func (s *bridgeSettings) RegisterDef(def sdk.ConfigDef) {} +func (s *bridgeSettings) Defs(prefix string) []*sdk.ConfigDef { return nil } +func (s *bridgeSettings) Dump() map[string]interface{} { return s.data } +func (s *bridgeSettings) Plugins() []string { return nil } + +func main() {} +` + +// tmplCABIHeader — shared C ABI type definitions for both core and plugin +// 此模板中的常量应与 core/internal/meta/meta.go 保持一致（ABI 版本、dispatch method IDs）。 +const tmplCABIHeader = ` +#ifndef HOMEAGENT_CABI_H +#define HOMEAGENT_CABI_H +// HOMEAGENT_ABI_VERSION 与 sdk/meta/meta.go CABINum 同步（major*100+minor，v0.9.x→900） +#define HOMEAGENT_ABI_VERSION 900 +#ifdef __cplusplus +extern "C" { +#endif + +// PluginAPI — implemented by the plugin, called by the core +typedef struct { + int version; int version_min; + int (*init_plugin)(char*, char*, char**); + int (*start_plugin)(void*, int, char**); + int (*stop_plugin)(char**); + int (*invoke_tool)(char*, char*, char**, char**); + int (*invoke_stage)(char*, char*, char**, char**); + int (*invoke_output)(char*, char*, char*, char**); + void (*free_string)(char*); +} PluginAPI; + +// CoreAPI — implemented by the core, passed to plugin via start_plugin +// Uses single dispatch function to avoid function pointer ABI issues +typedef struct { + int version; int version_min; + int (*dispatch)(int method_id, void* ctx, char* s1, char* s2, char* s3, int i1, int i2, char** result, char** error); + void* ctx; +} CoreAPI; + +// Dispatch method IDs (plugin→core SDK calls) +enum { + CORE_REGISTER_TOOL = 1, + CORE_REGISTER_STAGE = 2, + CORE_REGISTER_OUTPUT_CH = 3, + CORE_REGISTER_PLUGIN_API = 4, + CORE_INJECT_TEXT = 5, + CORE_INJECT_INTERRUPT_TEXT = 6, + CORE_INJECT_TEXT_NO_MEMORY = 7, + CORE_INJECT_INPUT_SYNC = 47, + CORE_SET_AUTO_RESTART = 8, + CORE_MEMORY_RECALL = 9, + CORE_MEMORY_COMMIT = 10, + CORE_MEMORY_INTROSPECT = 11, + CORE_MEMORY_MERGE = 12, + CORE_MEMORY_PURGE = 13, + CORE_DOC_QUERY = 14, + CORE_KNOWLEDGE_SEARCH = 15, + CORE_SETTINGS_GET = 16, + CORE_SETTINGS_SET = 17, + CORE_SETTINGS_REGISTER_DEF = 18, + CORE_LLM_LIST_SOURCES = 19, + CORE_LLM_SET_SOURCE = 20, + CORE_SOCIAL_GET_PERSON = 21, + CORE_SOCIAL_GET_NETWORK = 22, + CORE_SUBSCRIBE = 23, + CORE_UNSUBSCRIBE = 24, + CORE_FREE_STRING = 25, + CORE_SETTINGS_GET_CORE = 26, + CORE_SETTINGS_SET_CORE = 27, + CORE_SETTINGS_LIST_CORE = 28, + CORE_SETTINGS_GET_PLUGIN = 29, + CORE_SETTINGS_SET_PLUGIN = 30, + CORE_SETTINGS_LIST_PLUGIN = 31, + CORE_DOC_INSERT = 32, + CORE_DOC_REMOVE = 33, + CORE_DOC_STATS = 34, + CORE_KNOWLEDGE_ADD = 35, + CORE_KNOWLEDGE_LIST = 36, + CORE_LLM_CURRENT_SOURCE = 37, + CORE_SOCIAL_GET_TRAIT = 38, + CORE_SOCIAL_GET_RELATIONS = 39, + CORE_SOCIAL_LIST_PERSONS = 40, + CORE_TEXT_MEMORY_APPEND = 41, + CORE_SETTINGS_LIST = 42, + CORE_SETTINGS_DEFS = 43, + CORE_SETTINGS_DUMP = 44, + CORE_SETTINGS_PLUGINS = 45, + CORE_REGISTER_INPUT_CH = 46, + CORE_INJECT_INPUT_SYNC = 47, + CORE_PLUGIN_RELOAD_ONE = 48, + CORE_PLUGIN_LIST_LOADED = 49, + CORE_PLUGIN_IS_DISABLED = 50, +}; + +#ifdef __cplusplus +} +#endif +#endif +` + +// tmplLinuxBridge — auto-generated Go bridge for Linux c-shared builds. +// Called by plugin's Start() with a PluginSDK that wraps CoreAPI dispatch. +// PluginSDK calls go through C ABI → CoreAPI dispatch → core's Go PluginSDK. +const tmplLinuxBridge = `package main + +/* +#include +int ha_dispatch(int method_id, void* core_api, char* s1, char* s2, char* s3, int i1, int i2, char** result, char** error); +*/ +import "C" +import ( + "encoding/json" + "fmt" + "sync" + "unsafe" + sdk "gitcode.com/JianFeeeee/homeagent-sdk/sdk" +) + +// ---- global state ---- + +var ( + mu sync.Mutex + currentPlg sdk.Plugin + currentSDK *sdk.PluginSDK + coreAPI unsafe.Pointer + + handlerMu sync.RWMutex + coreAPIMu sync.RWMutex + toolHandlers = map[string]sdk.ToolHandler{} + stageHandlers = map[string]sdk.StageHandler{} + outputHandlers = map[string]sdk.ToolHandler{} +) + +// ---- CoreAPI dispatch helpers ---- + +func callVoid(methodID int, s1, s2, s3 string, i1, i2 int) error { + coreAPIMu.RLock() + api := coreAPI + coreAPIMu.RUnlock() + var c1, c2, c3 *C.char + if s1 != "" { c1 = C.CString(s1); defer C.free(unsafe.Pointer(c1)) } + if s2 != "" { c2 = C.CString(s2); defer C.free(unsafe.Pointer(c2)) } + if s3 != "" { c3 = C.CString(s3); defer C.free(unsafe.Pointer(c3)) } + var cErr *C.char + if C.ha_dispatch(C.int(methodID), api, c1, c2, c3, C.int(i1), C.int(i2), nil, &cErr) != 0 && cErr != nil { + return fmt.Errorf("%s", C.GoString(cErr)) + } + return nil +} + +func callString(methodID int, s1, s2, s3 string, i1, i2 int) (string, error) { + coreAPIMu.RLock() + api := coreAPI + coreAPIMu.RUnlock() + var c1, c2, c3 *C.char + if s1 != "" { c1 = C.CString(s1); defer C.free(unsafe.Pointer(c1)) } + if s2 != "" { c2 = C.CString(s2); defer C.free(unsafe.Pointer(c2)) } + if s3 != "" { c3 = C.CString(s3); defer C.free(unsafe.Pointer(c3)) } + var strResult, cErr *C.char + if C.ha_dispatch(C.int(methodID), api, c1, c2, c3, C.int(i1), C.int(i2), &strResult, &cErr) != 0 && cErr != nil { + return "", fmt.Errorf("%s", C.GoString(cErr)) + } + if strResult != nil { + result := C.GoString(strResult) + C.ha_dispatch(C.int(25), api, strResult, nil, nil, 0, 0, nil, nil) + return result, nil + } + return "", nil +} + +// ---- buildPluginSDK: PluginSDK backed by CoreAPI dispatch ---- +// - ALL SDK methods route through C ABI → CoreAPI → core's PluginSDK +// - Handlers for tools/stages/output are stored locally AND registered via dispatch + +func buildPluginSDK(name string) *sdk.PluginSDK { + sett := &dispatchSettings{} + base := sdk.New(name, sett, + func(toolName string, def sdk.ToolDef, handler sdk.ToolHandler) error { + handlerMu.Lock() + toolHandlers[toolName] = handler + handlerMu.Unlock() + b, _ := json.Marshal(def) + return callVoid(1, toolName, string(b), "", 0, 0) + }, + func(stage sdk.Stage, handler sdk.StageHandler) { + handlerMu.Lock() + stageHandlers[string(stage)] = handler + handlerMu.Unlock() + callVoid(2, string(stage), "", "", 0, 0) + }, + func(name string) error { return callVoid(4, name, "", "", 0, 0) }, + func(name string, caps int, desc string, def sdk.ChannelDef, handler sdk.ToolHandler) error { + handlerMu.Lock() + outputHandlers[name] = handler + handlerMu.Unlock() + defJSON, _ := json.Marshal(def) + return callVoid(3, name, desc, string(defJSON), caps, 0) + }, + ) + base.SetIOInjector(dispatchIO{}) + base.SetMemoryAPI(dispatchMemory{}) + base.SetDocMemoryAPI(dispatchDocMemory{}) + base.SetKnowledgeAPI(dispatchKnowledge{}) + base.SetLLMAPI(dispatchLLM{}) + base.SetSocialAPI(dispatchSocial{}) + base.SetTextMemoryAPI(dispatchTextMemory{}) + base.SetPluginMgrAPI(dispatchPluginMgr{}) + base.SetInputChannelRegistrar( + func(name string, def sdk.ChannelDef) error { + defJSON, _ := json.Marshal(def) + return callVoid(46, name, string(defJSON), "", 0, 0) + }, + ) + return base +} + +// ---- dispatch IO (inline definitions) ---- + +type dispatchIO struct{} +func (dispatchIO) InjectInterruptText(s, c, t string) { callVoid(6, s, c, t, 0, 0) } +func (dispatchIO) InjectText(s, c, t string) { callVoid(5, s, c, t, 0, 0) } +func (dispatchIO) InjectTextNoMemory(s, c, t string) { callVoid(7, s, c, t, 0, 0) } +func (dispatchIO) InjectInputSync(s, c, t string) string { r, _ := callString(47, s, c, t, 0, 0); return r } + +type dispatchMemory struct{} +func (dispatchMemory) Recall(q []string, d int) ([]sdk.Entity, []sdk.Relation, error) { + b, _ := json.Marshal(q); r, e := callString(9, string(b), "", "", d, 0) + if e != nil || r == "" { return nil, nil, e } + var v struct{ Entities []sdk.Entity; Relations []sdk.Relation } + if e = json.Unmarshal([]byte(r), &v); e != nil { return nil, nil, e } + if v.Entities == nil { v.Entities = []sdk.Entity{} } + if v.Relations == nil { v.Relations = []sdk.Relation{} } + return v.Entities, v.Relations, nil +} +func (dispatchMemory) Commit(t []sdk.Triple) error { b, _ := json.Marshal(t); return callVoid(10, string(b), "", "", 0, 0) } +func (dispatchMemory) Introspect() (map[string]interface{}, error) { r, e := callString(11, "", "", "", 0, 0); if e != nil || r == "" { return nil, e }; var m map[string]interface{}; return m, json.Unmarshal([]byte(r), &m) } +func (dispatchMemory) MergeEntities(s, t string) (int, error) { return 1, callVoid(12, s, t, "", 0, 0) } +func (dispatchMemory) Purge(c map[string]string, m string) (int, error) { b, _ := json.Marshal(c); i := 0; if m == "hard" { i = 1 }; return 1, callVoid(13, string(b), "", "", i, 0) } + +type dispatchDocMemory struct{} +func (dispatchDocMemory) Query(t string, k int) []*sdk.Doc { r, e := callString(14, t, "", "", k, 0); if e != nil || r == "" { return nil }; var d []*sdk.Doc; json.Unmarshal([]byte(r), &d); return d } +func (dispatchDocMemory) Insert(doc *sdk.Doc) error { b, _ := json.Marshal(doc); return callVoid(32, string(b), "", "", 0, 0) } +func (dispatchDocMemory) Remove(id string) { callVoid(33, id, "", "", 0, 0) } +func (dispatchDocMemory) Stats() map[string]interface{} { r, e := callString(34, "", "", "", 0, 0); if e != nil || r == "" { return nil }; var m map[string]interface{}; json.Unmarshal([]byte(r), &m); return m } + +type dispatchKnowledge struct{} +func (dispatchKnowledge) Search(q string, k int) ([]*sdk.Knowledge, error) { r, e := callString(15, q, "", "", k, 0); if e != nil || r == "" { return nil, e }; var v []*sdk.Knowledge; return v, json.Unmarshal([]byte(r), &v) } +func (dispatchKnowledge) Add(n, c string) error { return callVoid(35, n, c, "", 0, 0) } +func (dispatchKnowledge) List() ([]string, error) { r, e := callString(36, "", "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v []string; return v, json.Unmarshal([]byte(r), &v) } + +type dispatchLLM struct{} +func (dispatchLLM) ListSources() []string { r, e := callString(19, "", "", "", 0, 0); if e != nil || r == "" { return nil }; var v []string; json.Unmarshal([]byte(r), &v); return v } +func (dispatchLLM) SetSource(n string) error { return callVoid(20, n, "", "", 0, 0) } +func (dispatchLLM) CurrentSource() string { r, e := callString(37, "", "", "", 0, 0); if e != nil || r == "" { return "" }; return r } + +type dispatchSocial struct{} +func (dispatchSocial) GetPerson(n string) (*sdk.PersonProfile, error) { r, e := callString(21, n, "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v sdk.PersonProfile; return &v, json.Unmarshal([]byte(r), &v) } +func (dispatchSocial) GetTrait(n, t string) (string, bool) { r, e := callString(38, n, t, "", 0, 0); if e != nil || r == "" { return "", false }; var m map[string]interface{}; json.Unmarshal([]byte(r), &m); v, _ := m["value"].(string); ok, _ := m["found"].(bool); return v, ok } +func (dispatchSocial) GetRelations(name string) ([]sdk.SocialRelation, error) { r, e := callString(39, name, "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v []sdk.SocialRelation; return v, json.Unmarshal([]byte(r), &v) } +func (dispatchSocial) GetNetwork(n string, d int) ([]*sdk.PersonProfile, error) { r, e := callString(22, n, "", "", d, 0); if e != nil || r == "" { return nil, e }; var v []*sdk.PersonProfile; return v, json.Unmarshal([]byte(r), &v) } +func (dispatchSocial) ListPersons() ([]string, error) { r, e := callString(40, "", "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v []string; return v, json.Unmarshal([]byte(r), &v) } + +type dispatchTextMemory struct{} +func (dispatchTextMemory) Append(evt sdk.TextEvent) error { b, _ := json.Marshal(evt); return callVoid(41, string(b), "", "", 0, 0) } + +// ---- dispatchPluginMgr (CORE_PLUGIN_RELOAD_ONE = 48) ---- + +type dispatchPluginMgr struct{} + +func (dispatchPluginMgr) ReloadOne(name string) error { + return callVoid(48, name, "", "", 0, 0) +} + +func (dispatchPluginMgr) ListLoadedPlugins() []string { + r, e := callString(49, "", "", "", 0, 0) + if e != nil || r == "" { + return nil + } + var list []string + if json.Unmarshal([]byte(r), &list) != nil { + return nil + } + return list +} + +func (dispatchPluginMgr) IsPluginDisabled(name string) bool { + r, e := callString(50, name, "", "", 0, 0) + return e == nil && r == "1" +} + +// ---- dispatchSettings (inline) ---- + +type dispatchSettings struct{} +func (d *dispatchSettings) Get(key string) (interface{}, error) { + r, e := callString(16, key, "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v interface{}; return v, json.Unmarshal([]byte(r), &v) +} +func (d *dispatchSettings) Set(key string, value interface{}) error { + b, _ := json.Marshal(value); return callVoid(17, key, string(b), "", 0, 0) +} +func (d *dispatchSettings) RegisterDef(def sdk.ConfigDef) { b, _ := json.Marshal(def); callVoid(18, string(b), "", "", 0, 0) } +func (d *dispatchSettings) List(prefix string) ([]string, error) { + r, e := callString(42, prefix, "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v []string; return v, json.Unmarshal([]byte(r), &v) +} +func (d *dispatchSettings) GetCore(key string) (interface{}, error) { + r, e := callString(26, key, "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v interface{}; return v, json.Unmarshal([]byte(r), &v) +} +func (d *dispatchSettings) SetCore(key string, value interface{}) error { + b, _ := json.Marshal(value); return callVoid(27, key, string(b), "", 0, 0) +} +func (d *dispatchSettings) ListCore(prefix string) ([]string, error) { + r, e := callString(28, prefix, "", "", 0, 0); if e != nil || r == "" { return nil, e }; var v []string; return v, json.Unmarshal([]byte(r), &v) +} +func (d *dispatchSettings) GetPlugin(plugin, key string) (interface{}, error) { + r, e := callString(29, plugin, key, "", 0, 0); if e != nil || r == "" { return nil, e }; var v interface{}; return v, json.Unmarshal([]byte(r), &v) +} +func (d *dispatchSettings) SetPlugin(plugin, key string, value interface{}) error { + b, _ := json.Marshal(value); return callVoid(30, plugin, key, string(b), 0, 0) +} +func (d *dispatchSettings) ListPlugin(plugin, prefix string) ([]string, error) { + r, e := callString(31, plugin, prefix, "", 0, 0); if e != nil || r == "" { return nil, e }; var v []string; return v, json.Unmarshal([]byte(r), &v) +} +func (d *dispatchSettings) Defs(prefix string) []*sdk.ConfigDef { + r, e := callString(43, prefix, "", "", 0, 0); if e != nil || r == "" { return nil }; var v []*sdk.ConfigDef; json.Unmarshal([]byte(r), &v); return v +} +func (d *dispatchSettings) Dump() map[string]interface{} { + r, e := callString(44, "", "", "", 0, 0); if e != nil || r == "" { return nil }; var m map[string]interface{}; json.Unmarshal([]byte(r), &m); return m +} +func (d *dispatchSettings) Plugins() []string { + r, e := callString(45, "", "", "", 0, 0); if e != nil || r == "" { return nil }; var v []string; json.Unmarshal([]byte(r), &v); return v +} +func (d *dispatchSettings) DataDir() string { + r, e := callString(51, "", "", "", 0, 0); if e != nil { return "" }; return r +} + +// ---- Go callbacks (called from z_entry.c via C) ---- + +//export go_init_plugin +func go_init_plugin(name *C.char, configJSON *C.char, errorOut **C.char) C.int { + plg, err := NewPluginFactory(C.GoString(name), nil) + if err != nil || plg == nil { + if err != nil { *errorOut = C.CString(err.Error()) } else { *errorOut = C.CString("NewPluginFactory returned nil") } + return 1 + } + mu.Lock(); currentPlg = plg; mu.Unlock() + _ = configJSON + return 0 +} + +//export go_start_plugin +func go_start_plugin(coreAPIptr unsafe.Pointer, coreVersion C.int, errorOut **C.char) C.int { + mu.Lock() + plg := currentPlg + coreAPIMu.Lock() + coreAPI = coreAPIptr + coreAPIMu.Unlock() + mu.Unlock() + _ = coreVersion + if plg == nil { *errorOut = C.CString("not initialized"); return 1 } + sdk := buildPluginSDK(plg.Name()) + mu.Lock(); currentSDK = sdk; mu.Unlock() + if err := plg.Start(sdk); err != nil { *errorOut = C.CString(err.Error()); return 1 } + return 0 +} + +//export go_stop_plugin +func go_stop_plugin(errorOut **C.char) C.int { + mu.Lock() + plg := currentPlg + sdk := currentSDK + currentPlg = nil + currentSDK = nil + coreAPIMu.Lock() + coreAPI = nil + coreAPIMu.Unlock() + mu.Unlock() + if sdk != nil { + sdk.RunStopHandlers() + } + if plg != nil { + if err := plg.Stop(); err != nil { *errorOut = C.CString(err.Error()); return 1 } + } + return 0 +} + +//export go_invoke_tool +func go_invoke_tool(name *C.char, argsJSON *C.char, resultOut **C.char, errorOut **C.char) C.int { + goName := C.GoString(name) + handlerMu.RLock() + h, ok := toolHandlers[goName] + handlerMu.RUnlock() + if !ok { *errorOut = C.CString("tool not found"); return 1 } + var args map[string]interface{} + if argsJSON != nil { json.Unmarshal([]byte(C.GoString(argsJSON)), &args) } + r, err := h(args) + if err != nil { *errorOut = C.CString(err.Error()); return 1 } + b, _ := json.Marshal(r) + *resultOut = C.CString(string(b)) + return 0 +} + +// fillStageContext 将内核传来的 ctx JSON 填充到插件侧 StageContext。 +func fillStageContext(sc *sdk.StageContext, ctxJSON string) { + var m map[string]interface{} + if err := json.Unmarshal([]byte(ctxJSON), &m); err != nil { + return + } + if v, _ := m["raw_message"].(string); v != "" { sc.RawMessage = v } + if v, _ := m["user_id"].(string); v != "" { sc.UserID = v } + if v, _ := m["group_id"].(string); v != "" { sc.GroupID = v } + if v, _ := m["phase"].(string); v != "" { sc.Phase = sdk.Stage(v) } + if v, _ := m["llm_text"].(string); v != "" { sc.LLMText = v } + if v, _ := m["final_text"].(string); v != "" { sc.FinalText = v } + if v, _ := m["no_memory"].(bool); v { sc.NoMemory = true } + if v, _ := m["response"].(string); v != "" { sc.Response = &v } + if v, _ := m["tool_calls"].([]interface{}); len(v) > 0 { + b, _ := json.Marshal(v); json.Unmarshal(b, &sc.ToolCalls) + } + if v, _ := m["tool_results"].([]interface{}); len(v) > 0 { + b, _ := json.Marshal(v); json.Unmarshal(b, &sc.ToolResults) + } +} + +// stageContextWritable 提取插件可写且内核会同步回去的字段。 +func stageContextWritable(sc *sdk.StageContext) map[string]interface{} { + m := map[string]interface{}{ + "raw_message": sc.RawMessage, + "user_id": sc.UserID, + "group_id": sc.GroupID, + "phase": string(sc.Phase), + "llm_text": sc.LLMText, + "final_text": sc.FinalText, + "no_memory": sc.NoMemory, + } + if sc.Response != nil { + m["response"] = *sc.Response + } + if len(sc.ToolCalls) > 0 { + m["tool_calls"] = sc.ToolCalls + } + if len(sc.ToolResults) > 0 { + m["tool_results"] = sc.ToolResults + } + return m +} + +//export go_invoke_stage +func go_invoke_stage(stage *C.char, ctxJSON *C.char, resultOut **C.char, errorOut **C.char) C.int { + goStage := C.GoString(stage) + handlerMu.RLock() + h, ok := stageHandlers[goStage] + handlerMu.RUnlock() + if !ok { return 0 } + sc := &sdk.StageContext{} + if ctxJSON != nil { + fillStageContext(sc, C.GoString(ctxJSON)) + } + if err := h(sc); err != nil { *errorOut = C.CString(err.Error()); return 1 } + // ABI v2: 回传插件修改后的上下文（若调用方要求） + if resultOut != nil { + if b, err := json.Marshal(stageContextWritable(sc)); err == nil { + *resultOut = C.CString(string(b)) + } + } + return 0 +} + +//export go_invoke_output +func go_invoke_output(channel *C.char, msgType *C.char, payloadJSON *C.char, errorOut **C.char) C.int { + goChan := C.GoString(channel) + handlerMu.RLock() + h, ok := outputHandlers[goChan] + handlerMu.RUnlock() + if !ok { return 0 } + // payloadJSON contains the full args JSON from output_send (e.g. {"content":"...","user_id":123}) + var args map[string]interface{} + if payloadJSON != nil { + json.Unmarshal([]byte(C.GoString(payloadJSON)), &args) + } + if _, err := h(args); err != nil { *errorOut = C.CString(err.Error()); return 1 } + return 0 +} + +//export go_free_string +func go_free_string(ptr *C.char) { C.free(unsafe.Pointer(ptr)) } + +func main() {} +` + +// tmplPluginInitC — C entry point for the plugin .so file. +// Contains PluginAPI, CoreAPI (single dispatch), and ha_dispatch bridge. +const tmplPluginInitC = `#include +#include + +// HOMEAGENT_ABI_VERSION 与 sdk/meta/meta.go CABINum 同步（major*100+minor，v0.9.x→900） +#define HOMEAGENT_ABI_VERSION 900 + +typedef struct { + int version; int version_min; + int (*init_plugin)(char*, char*, char**); + int (*start_plugin)(void*, int, char**); + int (*stop_plugin)(char**); + int (*invoke_tool)(char*, char*, char**, char**); + int (*invoke_stage)(char*, char*, char**, char**); + int (*invoke_output)(char*, char*, char*, char**); + void (*free_string)(char*); +} PluginAPI; + +typedef struct { + int version; int version_min; + int (*dispatch)(int, void*, char*, char*, char*, int, int, char**, char**); + void* ctx; +} CoreAPI; + +extern int go_init_plugin(char*, char*, char**); +extern int go_start_plugin(void*, int, char**); +extern int go_stop_plugin(char**); +extern int go_invoke_tool(char*, char*, char**, char**); +extern int go_invoke_stage(char*, char*, char**, char**); +extern int go_invoke_output(char*, char*, char*, char**); +extern void go_free_string(char*); + +int c_init_plugin(char* n, char* c, char** e) { return go_init_plugin(n, c, e); } +int c_start_plugin(void* a, int v, char** e) { return go_start_plugin(a, v, e); } +int c_stop_plugin(char** e) { return go_stop_plugin(e); } +int c_invoke_tool(char* n, char* a, char** r, char** e) { return go_invoke_tool(n, a, r, e); } +int c_invoke_stage(char* s, char* c, char** r, char** e) { return go_invoke_stage(s, c, r, e); } +int c_invoke_output(char* c, char* m, char* p, char** e) { return go_invoke_output(c, m, p, e); } +void c_free_string(char* p) { go_free_string(p); } + +// ha_dispatch — called by Go bridge, passes through to CoreAPI dispatch +int ha_dispatch(int id, void* api, char* s1, char* s2, char* s3, int i1, int i2, char** r, char** e) { + CoreAPI* a = (CoreAPI*)api; + if (!a || !a->dispatch) return 1; + return a->dispatch(id, a->ctx, s1, s2, s3, i1, i2, r, e); +} + +PluginAPI* plugin_init(void) { + static PluginAPI api; + memset(&api, 0, sizeof(api)); + api.version = HOMEAGENT_ABI_VERSION; api.version_min = HOMEAGENT_ABI_VERSION; + api.init_plugin = c_init_plugin; api.start_plugin = c_start_plugin; api.stop_plugin = c_stop_plugin; + api.invoke_tool = c_invoke_tool; api.invoke_stage = c_invoke_stage; api.invoke_output = c_invoke_output; + api.free_string = c_free_string; + return &api; +} +` + +// ============================================================ +// Remote Device Adapter Templates +// ============================================================ + +const tmplRemoteDeviceMain = `#include +#include +#include + +#include "ha_remotedevice.h" + +/* ============================================================ + * {{.Plg.Name}} — Remote Device Adapter + * + * 声明式远程设备接入示例。 + * 用户只需实现： + * 1. ha_transport_t 的 4 个函数 + * 2. 声明 handlers 表（设备支持哪些命令 + 对应的处理函数） + * 其余协议细节（WS 握手、hello/bind、心跳、重连、命令分发、结果回执）由 SDK 自动处理。 + * ============================================================ */ + +/* ====================== 传输层实现 ====================== + * + * 请为你的平台实现以下 4 个函数： + * connect(ctx, host, port) — 建立 TCP 连接 + * send(ctx, data, len) — 发送数据 + * recv(ctx, buf, len) — 接收数据（阻塞，返回实际接收字节数） + * close(ctx) — 关闭连接 + * + * 示例：POSIX socket 实现 + */ + +#if defined(_WIN32) || defined(_WIN64) +/* Windows 平台需包含 winsock2.h */ +#error "Please implement transport for your platform (see example below)" +#else +/* POSIX (Linux, macOS, ESP-IDF, Zephyr, etc.) */ +#include +#include +#include +#include +#include + +struct transport_ctx { + int sock; +}; + +static int transport_connect(void *ctx, const char *host, uint16_t port) { + struct transport_ctx *tc = (struct transport_ctx *)ctx; + struct hostent *he = gethostbyname(host); + if (!he) return -1; + tc->sock = socket(AF_INET, SOCK_STREAM, 0); + if (tc->sock < 0) return -1; + struct sockaddr_in addr; + memset(&addr, 0, sizeof(addr)); + addr.sin_family = AF_INET; + addr.sin_port = htons(port); + memcpy(&addr.sin_addr, he->h_addr_list[0], he->h_length); + if (connect(tc->sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) { + close(tc->sock); + tc->sock = -1; + return -1; + } + return 0; +} + +static int transport_send(void *ctx, const uint8_t *data, int len) { + struct transport_ctx *tc = (struct transport_ctx *)ctx; + int sent = 0; + while (sent < len) { + int n = (int)send(tc->sock, data + sent, len - sent, 0); + if (n <= 0) return -1; + sent += n; + } + return sent; +} + +static int transport_recv(void *ctx, uint8_t *buf, int len) { + struct transport_ctx *tc = (struct transport_ctx *)ctx; + int n = (int)recv(tc->sock, buf, len, 0); + return n; +} + +static void transport_close(void *ctx) { + struct transport_ctx *tc = (struct transport_ctx *)ctx; + if (tc->sock >= 0) { + close(tc->sock); + tc->sock = -1; + } +} +#endif + +/* ====================== 声明式命令处理 ====================== + * + * 每个命令对应一个处理函数，通过填写 ha_cmd_result_t 返回数据。 + * SDK 自动回执结果，无需手动调用 send_result。 + * + * 返回方式： + * 1. 文本输出：填写 result->output + * 2. 二进制数据：设置 result->has_binary=1 并填写 binary_data/len/mime + * 3. 错误：设置 result->status=1 并填写 result->error + * 4. 返回 HA_OK 表示处理成功，其他值表示处理失败 + */ + +/* ESP32-CAM 摄像头处理 */ +static ha_status_t handle_camerasue(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)userdata; + int duration = 0; + if (args && args[0]) duration = atoi(args); + printf("[camera] %s (duration=%ds)\n", duration ? "record" : "snapshot", duration); + + /* 返回文本结果（base64 图片） */ + result->status = 0; + result->output = "data:image/jpeg;base64,/9j/4AAQ..."; + return HA_OK; +} + +/* 屏幕截图处理 */ +static ha_status_t handle_screensee(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)args; (void)userdata; + printf("[screen] screenshot\n"); + result->status = 0; + result->output = "data:image/png;base64,iVBORw0KGgo..."; + return HA_OK; +} + +/* 语音播报处理 */ +static ha_status_t handle_speakeruse(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)userdata; + printf("[speaker] TTS: %s\n", args ? args : ""); + result->status = 0; + result->output = "speakeruse done"; + return HA_OK; +} + +/* 远程操控处理（computeruse） */ +static ha_status_t handle_computeruse(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)userdata; + const char *action = NULL; + const char *json_str = NULL; + ha_cmd_parse_json(args, &action, &json_str); + printf("[computeruse] action=%s\n", action ? action : "unknown"); + result->status = 0; + result->output = "computeruse done"; + return HA_OK; +} + +/* 剪贴板读取 */ +static ha_status_t handle_clipboardsee(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)args; (void)userdata; + result->status = 0; + result->output = "clipboard content"; + return HA_OK; +} + +/* 剪贴板写入 */ +static ha_status_t handle_clipboardsue(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)userdata; + printf("[clipboard] write: %s\n", args ? args : ""); + result->status = 0; + result->output = "clipboard written"; + return HA_OK; +} + +/* 屏幕显示 */ +static ha_status_t handle_screensue(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)userdata; + printf("[screensue] show: %s\n", args ? args : ""); + result->status = 0; + result->output = "screensue shown"; + return HA_OK; +} + +/* Shell 命令处理 */ +static ha_status_t handle_shell(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)userdata; + printf("[shell] cmd: %s\n", args ? args : ""); + result->status = 0; + result->output = "shell output"; + return HA_OK; +} + +/* 设备信息查询 */ +static ha_status_t handle_deviceinfo(const char *req_id, const char *args, + ha_cmd_result_t *result, void *userdata) { + (void)req_id; (void)args; (void)userdata; + result->status = 0; + result->output = "{\"platform\":\"linux\",\"arch\":\"x86_64\"}"; + return HA_OK; +} + +/* ====================== 连接状态回调 ====================== */ + +static void on_state(int connected, void *userdata) { + (void)userdata; + printf("[devicelink] state: %s\n", connected ? "connected" : "disconnected"); +} + +/* ====================== 主函数 ====================== */ + +int main(int argc, char *argv[]) { + /* 传输层上下文 */ + struct transport_ctx tctx; + tctx.sock = -1; + + ha_transport_t transport = { + .connect = transport_connect, + .send = transport_send, + .recv = transport_recv, + .close = transport_close, + .ctx = &tctx, + }; + + /* ===== 声明式设备配置 ===== */ + + /* 声明设备能力 */ + const char *caps[] = { + "status", "cmdrun", "deviceinfo", + "camerasue", "screensee", "speakeruse", + "computeruse", "clipboardsee", "clipboardsue", + "screensue", + NULL + }; + + /* 声明命令处理表：设备支持哪些命令，以及对应的处理函数 */ + ha_cmd_handler_def_t handlers[] = { + {.command = "shell", .handler = handle_shell}, + {.command = "camerasue", .handler = handle_camerasue}, + {.command = "screensee", .handler = handle_screensee}, + {.command = "speakeruse", .handler = handle_speakeruse}, + {.command = "computeruse", .handler = handle_computeruse}, + {.command = "clipboardsee", .handler = handle_clipboardsee}, + {.command = "clipboardsue", .handler = handle_clipboardsue}, + {.command = "screensue", .handler = handle_screensue}, + {.command = "deviceinfo", .handler = handle_deviceinfo}, + {.command = NULL}, /* 标记结束 */ + }; + + ha_config_t config = { + .transport = transport, + .server = "127.0.0.1:9890", + .token = "your-token-here", + .device = { + .device_id = "{{.Plg.Name}}", + .name = "{{.Plg.NameEn}}", + .kind = "computer", + .caps = caps, + .info_json = "{\"platform\":\"linux\",\"arch\":\"x86_64\"}", + }, + .handlers = handlers, /* 声明式命令处理表 */ + .on_state = on_state, + .ping_interval = 30, + }; + + ha_client_t *client = ha_client_new(&config); + if (!client) { + fprintf(stderr, "Failed to create client\n"); + return 1; + } + + printf("Starting remote device adapter: {{.Plg.Name}}\n"); + printf(" Server: %s\n", config.server); + printf(" Device ID: %s\n", config.device.device_id); + printf(" Kind: %s\n", config.device.kind); + printf(" Caps: "); + for (const char **p = caps; *p; p++) printf("%s ", *p); + printf("\n"); + + ha_status_t st = ha_client_start(client); + if (st != HA_OK) { + fprintf(stderr, "Failed to connect: %d\n", st); + ha_client_destroy(client); + return 1; + } + + printf("Connected! Entering main loop...\n"); + + /* 主循环 */ + while (1) { + ha_status_t st = ha_client_process(client); + if (st == HA_ERR_DISCONNECTED) { + printf("Disconnected, exiting.\n"); + break; + } +#if defined(_WIN32) || defined(_WIN64) + Sleep(10); +#else + usleep(10000); +#endif + } + + ha_client_stop(client); + ha_client_destroy(client); + return 0; +} +` + +const tmplRemoteDeviceCMake = `cmake_minimum_required(VERSION 3.10) +project({{.Plg.Name}} VERSION 0.1.0 LANGUAGES C) + +# ============================================================ +# {{.Plg.Name}} — Remote Device Adapter +# ============================================================ + +# 设置 SDK 路径（默认使用内置 SDK，也可通过 -DSDK_PATH=... 指定） +set(SDK_PATH "${CMAKE_CURRENT_SOURCE_DIR}/ha_remotedevice" + CACHE PATH "Path to ha_remotedevice SDK") + +# 添加 SDK 子目录 +if(EXISTS "${SDK_PATH}/CMakeLists.txt") + add_subdirectory(${SDK_PATH} ha_remotedevice) +else() + message(FATAL_ERROR "ha_remotedevice SDK not found at ${SDK_PATH}") +endif() + +# 创建设备适配器可执行文件 +add_executable(${PROJECT_NAME} + main.c +) + +# 链接 SDK +target_link_libraries(${PROJECT_NAME} PRIVATE ha_remotedevice) + +# 包含 SDK 头文件 +target_include_directories(${PROJECT_NAME} PRIVATE + ${HA_REMOTEDEVICE_INCLUDE_DIR} +) + +# 编译选项 +if(CMAKE_C_COMPILER_ID MATCHES "GNU|Clang") + target_compile_options(${PROJECT_NAME} PRIVATE + -Wall -Wextra -Wpedantic + -Wno-unused-parameter + ) +endif() + +# 安装 +install(TARGETS ${PROJECT_NAME} RUNTIME DESTINATION bin) +` + +const tmplReadme = `# {{.Plg.Name}} + +{{.Plg.Description}} + +## Build + +` + "```bash" + ` +plugindev build +` + "```" + ` + +## Install + +Upload the .hmap file through the Plugin Manager API. +`