update

internal/server/plugin/manager.go

@@ -2,28 +2,26 @@ package plugin
 
 import (
 	"context"
-	"crypto/sha256"
-	"encoding/hex"
 	"fmt"
 	"log"
 	"sync"
 
+	"github.com/gotunnel/internal/server/db"
 	"github.com/gotunnel/pkg/plugin"
 	"github.com/gotunnel/pkg/plugin/builtin"
-	"github.com/gotunnel/pkg/plugin/store"
 	"github.com/gotunnel/pkg/plugin/wasm"
 )
 
 // Manager 服务端 plugin 管理器
 type Manager struct {
 	registry *plugin.Registry
-	store    store.PluginStore
+	store    db.PluginStore
 	runtime  *wasm.Runtime
 	mu       sync.RWMutex
 }
 
 // NewManager 创建 plugin 管理器
-func NewManager(pluginStore store.PluginStore) (*Manager, error) {
+func NewManager(pluginStore db.PluginStore) (*Manager, error) {
 	ctx := context.Background()
 
 	runtime, err := wasm.NewRuntime(ctx)
@@ -51,12 +49,11 @@ func NewManager(pluginStore store.PluginStore) (*Manager, error) {
 // 注意: tcp, udp, http, https 是内置类型，直接在 tunnel 中处理
 // 这里只注册需要通过 plugin 系统提供的协议
 func (m *Manager) registerBuiltins() error {
-	// 注册 SOCKS5 plugin
-	if err := m.registry.RegisterBuiltin(builtin.NewSOCKS5Plugin()); err != nil {
-		return fmt.Errorf("register socks5: %w", err)
+	// 使用统一的插件注册入口
+	if err := m.registry.RegisterAll(builtin.GetAll()); err != nil {
+		return err
 	}
-
-	log.Println("[Plugin] Builtin plugins registered: socks5")
+	log.Printf("[Plugin] Registered %d builtin plugins", len(builtin.GetAll()))
 	return nil
 }
 
@@ -71,15 +68,15 @@ func (m *Manager) LoadStoredPlugins(ctx context.Context) error {
 		return err
 	}
 
-	for _, meta := range plugins {
-		data, err := m.store.GetPluginData(meta.Name)
+	for _, p := range plugins {
+		data, err := m.store.GetPluginWASM(p.Name)
 		if err != nil {
-			log.Printf("[Plugin] Failed to load %s: %v", meta.Name, err)
+			log.Printf("[Plugin] Failed to load %s: %v", p.Name, err)
 			continue
 		}
 
-		if err := m.loadWASMPlugin(ctx, meta.Name, data); err != nil {
-			log.Printf("[Plugin] Failed to init %s: %v", meta.Name, err)
+		if err := m.loadWASMPlugin(ctx, p.Name, data); err != nil {
+			log.Printf("[Plugin] Failed to init %s: %v", p.Name, err)
 		}
 	}
 
@@ -97,28 +94,19 @@ func (m *Manager) loadWASMPlugin(ctx context.Context, name string, data []byte)
 }
 
 // InstallPlugin 安装新的 WASM plugin
-func (m *Manager) InstallPlugin(ctx context.Context, meta plugin.PluginMetadata, wasmData []byte) error {
+func (m *Manager) InstallPlugin(ctx context.Context, p *db.PluginData) error {
 	m.mu.Lock()
 	defer m.mu.Unlock()
 
-	// 验证 checksum
-	hash := sha256.Sum256(wasmData)
-	checksum := hex.EncodeToString(hash[:])
-	if meta.Checksum != "" && meta.Checksum != checksum {
-		return fmt.Errorf("checksum mismatch")
-	}
-	meta.Checksum = checksum
-	meta.Size = int64(len(wasmData))
-
 	// 存储到数据库
 	if m.store != nil {
-		if err := m.store.SavePlugin(meta, wasmData); err != nil {
+		if err := m.store.SavePlugin(p); err != nil {
 			return err
 		}
 	}
 
 	// 加载到运行时
-	return m.loadWASMPlugin(ctx, meta.Name, wasmData)
+	return m.loadWASMPlugin(ctx, p.Name, p.WASMData)
 }
 
 // GetHandler 返回指定代理类型的 handler