feat: Phase 3 插件与工具系统 — Plugin SDK + Plugin Manager + 13内置插件 (40文件, 3293行)
- Plugin SDK: Plugin/Tool/ComplexTool/HostAPI 标准化接口 - Plugin Manager: 插件生命周期管理 (Install/Enable/Disable/Uninstall/Reload) - Tool Registry: 聚合工具注册表 (Register/Execute/Dispatch) - 13 个内置插件: 将原有硬编码工具迁移为标准插件格式 - REST API: 11 个端点 (net/http, 零外部依赖) - ai-core 集成: PluginManagerClient 替代本地工具调用 - plugin.json 元数据: 每个插件含完整 author/version/category/permissions Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/internal/manager"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/internal/sdk"
|
||||
)
|
||||
|
||||
type hostAPI struct {
|
||||
registry *manager.ToolRegistry
|
||||
}
|
||||
|
||||
func newHostAPI(registry *manager.ToolRegistry) *hostAPI {
|
||||
return &hostAPI{registry: registry}
|
||||
}
|
||||
|
||||
func (h *hostAPI) CallLLM(_ context.Context, _ []sdk.LLMMessage) (*sdk.LLMResponse, error) {
|
||||
return nil, fmt.Errorf("LLM call not available in plugin host")
|
||||
}
|
||||
|
||||
func (h *hostAPI) SearchMemory(_ context.Context, _, _ string, _ int) ([]sdk.MemoryEntry, error) {
|
||||
return nil, fmt.Errorf("memory search not available in plugin host")
|
||||
}
|
||||
|
||||
func (h *hostAPI) StoreMemory(_ context.Context, _ sdk.MemoryEntry) error {
|
||||
return fmt.Errorf("memory store not available in plugin host")
|
||||
}
|
||||
|
||||
func (h *hostAPI) Logger() sdk.Logger {
|
||||
return log.Default()
|
||||
}
|
||||
|
||||
func (h *hostAPI) GetConfig(key string) (string, error) {
|
||||
return "", fmt.Errorf("config key not found: %s", key)
|
||||
}
|
||||
|
||||
func (h *hostAPI) SetConfig(_, _ string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (h *hostAPI) PublishEvent(_ context.Context, _ map[string]interface{}) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (h *hostAPI) HTTPClient() *http.Client {
|
||||
return http.DefaultClient
|
||||
}
|
||||
@@ -0,0 +1,112 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/internal/sdk"
|
||||
iotquery "github.com/yourname/cyrene-ai/plugin-manager/plugins/iot_query"
|
||||
)
|
||||
|
||||
type iotClient struct {
|
||||
baseURL string
|
||||
httpClient *http.Client
|
||||
}
|
||||
|
||||
func newIoTClient(baseURL string) *iotClient {
|
||||
return &iotClient{
|
||||
baseURL: baseURL,
|
||||
httpClient: &http.Client{Timeout: 5 * time.Second},
|
||||
}
|
||||
}
|
||||
|
||||
func (c *iotClient) GetAllDevices(ctx context.Context) ([]sdk.IoTDeviceState, error) {
|
||||
url := c.baseURL + "/api/v1/devices"
|
||||
req, _ := http.NewRequestWithContext(ctx, "GET", url, nil)
|
||||
resp, err := c.httpClient.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var result struct {
|
||||
Devices []sdk.IoTDeviceState `json:"devices"`
|
||||
}
|
||||
if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return result.Devices, nil
|
||||
}
|
||||
|
||||
func (c *iotClient) GetDevice(ctx context.Context, deviceID string) (*sdk.IoTDeviceState, error) {
|
||||
url := fmt.Sprintf("%s/api/v1/devices/%s", c.baseURL, deviceID)
|
||||
req, _ := http.NewRequestWithContext(ctx, "GET", url, nil)
|
||||
resp, err := c.httpClient.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var dev sdk.IoTDeviceState
|
||||
if err := json.NewDecoder(resp.Body).Decode(&dev); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &dev, nil
|
||||
}
|
||||
|
||||
// iotControllerAdapter adapts IoTClient to iotcontrol.IoTController.
|
||||
type iotControllerAdapter struct {
|
||||
query iotquery.IoTClient
|
||||
client *http.Client
|
||||
baseURL string
|
||||
}
|
||||
|
||||
func newIoTControllerAdapter(query iotquery.IoTClient, baseURL string) *iotControllerAdapter {
|
||||
return &iotControllerAdapter{
|
||||
query: query,
|
||||
client: &http.Client{Timeout: 5 * time.Second},
|
||||
baseURL: baseURL,
|
||||
}
|
||||
}
|
||||
|
||||
func (a *iotControllerAdapter) GetDevice(ctx context.Context, deviceID string) (*sdk.IoTDeviceState, error) {
|
||||
return a.query.GetDevice(ctx, deviceID)
|
||||
}
|
||||
|
||||
func (a *iotControllerAdapter) SetDeviceProperty(ctx context.Context, deviceID, property string, value interface{}) error {
|
||||
url := fmt.Sprintf("%s/api/v1/devices/%s/property", a.baseURL, deviceID)
|
||||
body, _ := json.Marshal(map[string]interface{}{"property": property, "value": value})
|
||||
req, _ := http.NewRequestWithContext(ctx, "POST", url, strings.NewReader(string(body)))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
resp, err := a.client.Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode >= 400 {
|
||||
msg, _ := io.ReadAll(resp.Body)
|
||||
return fmt.Errorf("set property failed: HTTP %d - %s", resp.StatusCode, string(msg))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *iotControllerAdapter) ToggleDevice(ctx context.Context, deviceID string) (*sdk.IoTDeviceState, error) {
|
||||
url := fmt.Sprintf("%s/api/v1/devices/%s/toggle", a.baseURL, deviceID)
|
||||
req, _ := http.NewRequestWithContext(ctx, "POST", url, nil)
|
||||
resp, err := a.client.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var dev sdk.IoTDeviceState
|
||||
if err := json.NewDecoder(resp.Body).Decode(&dev); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &dev, nil
|
||||
}
|
||||
@@ -0,0 +1,99 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/internal/config"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/internal/handler"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/internal/manager"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/internal/sdk"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/plugins/calculator"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/plugins/crypto"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/plugins/datetime"
|
||||
fileplugin "github.com/yourname/cyrene-ai/plugin-manager/plugins/file"
|
||||
httpplugin "github.com/yourname/cyrene-ai/plugin-manager/plugins/http"
|
||||
iotcontrol "github.com/yourname/cyrene-ai/plugin-manager/plugins/iot_control"
|
||||
iotquery "github.com/yourname/cyrene-ai/plugin-manager/plugins/iot_query"
|
||||
jsonplugin "github.com/yourname/cyrene-ai/plugin-manager/plugins/json"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/plugins/markdown"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/plugins/random"
|
||||
"github.com/yourname/cyrene-ai/plugin-manager/plugins/text"
|
||||
webfetch "github.com/yourname/cyrene-ai/plugin-manager/plugins/web_fetch"
|
||||
websearch "github.com/yourname/cyrene-ai/plugin-manager/plugins/web_search"
|
||||
)
|
||||
|
||||
func main() {
|
||||
cfg := config.Load()
|
||||
|
||||
var iotAPI iotquery.IoTClient
|
||||
if cfg.IoTSvcURL != "" {
|
||||
iotAPI = newIoTClient(cfg.IoTSvcURL)
|
||||
}
|
||||
|
||||
registry := manager.NewToolRegistry()
|
||||
host := newHostAPI(registry)
|
||||
mgr := manager.NewPluginManager(registry, host)
|
||||
|
||||
builtins := []sdk.Plugin{
|
||||
&calculator.CalculatorPlugin{},
|
||||
&datetime.DatetimePlugin{},
|
||||
&text.TextPlugin{},
|
||||
&crypto.CryptoPlugin{},
|
||||
&random.RandomPlugin{},
|
||||
&markdown.MarkdownPlugin{},
|
||||
&jsonplugin.JSONPlugin{},
|
||||
fileplugin.NewFilePlugin(cfg.DataDir),
|
||||
httpplugin.NewHTTPPlugin(),
|
||||
websearch.NewWebSearchPlugin(),
|
||||
webfetch.NewWebFetchPlugin(),
|
||||
iotquery.NewIoTQueryPlugin(iotAPI),
|
||||
}
|
||||
for _, p := range builtins {
|
||||
if err := mgr.Install(p); err != nil {
|
||||
println("WARN: install plugin failed:", err.Error())
|
||||
}
|
||||
}
|
||||
if iotAPI != nil {
|
||||
ctrlPlugin := iotcontrol.NewIoTControlPlugin(newIoTControllerAdapter(iotAPI, cfg.IoTSvcURL))
|
||||
if err := mgr.Install(ctrlPlugin); err != nil {
|
||||
println("WARN: install plugin failed:", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
errs := mgr.EnableAll(ctx)
|
||||
for _, e := range errs {
|
||||
println("WARN: enable plugin failed:", e.Error())
|
||||
}
|
||||
println("Plugin Manager: all built-in plugins enabled")
|
||||
|
||||
mux := http.NewServeMux()
|
||||
ph := handler.NewPluginHandler(mgr)
|
||||
ph.RegisterRoutes(mux)
|
||||
|
||||
println("Plugin Manager listening on port", cfg.Port)
|
||||
srv := &http.Server{Addr: ":" + cfg.Port, Handler: mux}
|
||||
|
||||
go func() {
|
||||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
println("FATAL:", err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
}()
|
||||
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-quit
|
||||
println("Shutting down Plugin Manager...")
|
||||
|
||||
shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
mgr.Shutdown(shutdownCtx)
|
||||
srv.Shutdown(shutdownCtx)
|
||||
println("Plugin Manager stopped")
|
||||
}
|
||||
Reference in New Issue
Block a user