add plugins

internal/client/plugin/cache.go

@@ -0,0 +1,114 @@
+package plugin
+
+import (
+	"crypto/sha256"
+	"encoding/hex"
+	"fmt"
+	"os"
+	"path/filepath"
+	"sync"
+	"time"
+
+	"github.com/gotunnel/pkg/plugin"
+)
+
+// CachedPlugin 缓存的 plugin 信息
+type CachedPlugin struct {
+	Metadata plugin.PluginMetadata
+	Path     string
+	LoadedAt time.Time
+}
+
+// Cache 管理本地 plugin 存储
+type Cache struct {
+	dir     string
+	plugins map[string]*CachedPlugin
+	mu      sync.RWMutex
+}
+
+// NewCache 创建 plugin 缓存
+func NewCache(cacheDir string) (*Cache, error) {
+	if err := os.MkdirAll(cacheDir, 0755); err != nil {
+		return nil, err
+	}
+
+	return &Cache{
+		dir:     cacheDir,
+		plugins: make(map[string]*CachedPlugin),
+	}, nil
+}
+
+// Get 返回缓存的 plugin（如果有效）
+func (c *Cache) Get(name, version, checksum string) (*CachedPlugin, error) {
+	c.mu.RLock()
+	defer c.mu.RUnlock()
+
+	cached, ok := c.plugins[name]
+	if !ok {
+		return nil, nil
+	}
+
+	// 验证版本和 checksum
+	if cached.Metadata.Version != version {
+		return nil, nil
+	}
+	if checksum != "" && cached.Metadata.Checksum != checksum {
+		return nil, nil
+	}
+
+	return cached, nil
+}
+
+// Store 保存 plugin 到缓存
+func (c *Cache) Store(meta plugin.PluginMetadata, wasmData []byte) error {
+	c.mu.Lock()
+	defer c.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
+
+	// 写入文件
+	path := filepath.Join(c.dir, meta.Name+".wasm")
+	if err := os.WriteFile(path, wasmData, 0644); err != nil {
+		return err
+	}
+
+	c.plugins[meta.Name] = &CachedPlugin{
+		Metadata: meta,
+		Path:     path,
+		LoadedAt: time.Now(),
+	}
+	return nil
+}
+
+// Remove 删除缓存的 plugin
+func (c *Cache) Remove(name string) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+
+	cached, ok := c.plugins[name]
+	if !ok {
+		return nil
+	}
+
+	os.Remove(cached.Path)
+	delete(c.plugins, name)
+	return nil
+}
+
+// List 返回所有缓存的 plugins
+func (c *Cache) List() []plugin.PluginMetadata {
+	c.mu.RLock()
+	defer c.mu.RUnlock()
+
+	var result []plugin.PluginMetadata
+	for _, cached := range c.plugins {
+		result = append(result, cached.Metadata)
+	}
+	return result
+}