4623a7f031
All checks were successful
Build Multi-Platform Binaries / build (push) Successful in 11m9s
115 lines
2.1 KiB
Go
115 lines
2.1 KiB
Go
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
|
||
}
|