jim800121chen
3f0175f1a9
從 local-tool 複製出獨立的「visionA Agent」桌面應用（A3 純橋樑： tunnel client + 配對 UI + 設定，不開 HTTP port、不做本機裝置/推論 UI）。 Bundle ID 與 local-tool 不同（com.innovedus.visiona-agent vs visiona-local）， 雙 app 可共存。fork 後不主動 sync，需要時手動 cherry-pick。 Backend / Wails Go（AB1-AB13）： - internal/tunnel：6 狀態機（Idle/Connecting/Connected/Reconnecting/Failed/Stopped） + Pair/Unpair/Reconnect/Disconnect binding + ClientHooks event - internal/auth：encrypted file token store（AES-GCM + scrypt + machineID fallback salt + 13 tests） - internal/config：YAML validation + atomic write + 11 tests - internal/log：ring buffer + ExportLog 升級 zip - visionA-backend /api/pairing/exchange：SessionTokenStore + 17 new tests - 三平台 build 驗證（macOS DMG 160 MB / Windows EXE / Linux AppImage） - end-to-end 5 milestone 全綠（pairing → tunnel → forward → reuse 防護 → tunnel drop failover） Frontend / Next.js（AF1-AF7，沿用 visionA-frontend 基礎）： - AppShell + Header + TabNav（StatusView / PairView / SettingsView 三 tab） - ConnectionStatusBadge 5 種狀態 - TokenInput regex 驗證 + 7 種錯誤 + 0.5s auto-switch 到狀態頁 - 設定頁 4 區塊（含重新配對 AlertDialog） - agent-api.ts 封裝 Wails bindings（mock/real 雙實作）+ 90 tests Phase 0.7 review-driven fix（Round 2）： - A1 Session fixation 防護（RotateSessionID） - A3 mock pairing 預設改 false（必須明確 opt-in）+ startup log - A4 Pair 失敗後 state 清理矩陣（exchange/Save/Start fail 各自終態） - A5 Pair/Unpair/Reconnect lifecycleMu + 50 goroutine race test - F1 重新配對次按鈕 / F2 PairView Esc cancel / F3 Wails BrowserOpenURL / F4 Settings draft 持久 + 未儲存 badge 驗證：agent backend go test -race -count=3 ./... 4 packages 全綠 / agent frontend pnpm test 119 tests 全綠 Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
103 lines
2.4 KiB
Go
103 lines
2.4 KiB
Go
package model
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// ModelStore manages custom model files on disk.
|
|
type ModelStore struct {
|
|
baseDir string
|
|
}
|
|
|
|
func NewModelStore(baseDir string) *ModelStore {
|
|
_ = os.MkdirAll(baseDir, 0755)
|
|
return &ModelStore{baseDir: baseDir}
|
|
}
|
|
|
|
// SaveModel saves a .nef file for the given model ID.
|
|
func (s *ModelStore) SaveModel(id string, file io.Reader) (string, error) {
|
|
dir := filepath.Join(s.baseDir, id)
|
|
if err := os.MkdirAll(dir, 0755); err != nil {
|
|
return "", fmt.Errorf("failed to create model directory: %w", err)
|
|
}
|
|
|
|
nefPath := filepath.Join(dir, "model.nef")
|
|
f, err := os.Create(nefPath)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to create model file: %w", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
written, err := io.Copy(f, file)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to write model file: %w", err)
|
|
}
|
|
|
|
fmt.Printf("[INFO] Saved model file: %s (%d bytes)\n", nefPath, written)
|
|
return nefPath, nil
|
|
}
|
|
|
|
// SaveMetadata saves model metadata as JSON.
|
|
func (s *ModelStore) SaveMetadata(id string, m Model) error {
|
|
dir := filepath.Join(s.baseDir, id)
|
|
if err := os.MkdirAll(dir, 0755); err != nil {
|
|
return fmt.Errorf("failed to create model directory: %w", err)
|
|
}
|
|
|
|
metaPath := filepath.Join(dir, "metadata.json")
|
|
data, err := json.MarshalIndent(m, "", " ")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal metadata: %w", err)
|
|
}
|
|
|
|
return os.WriteFile(metaPath, data, 0644)
|
|
}
|
|
|
|
// GetModelPath returns the .nef file path for a model.
|
|
func (s *ModelStore) GetModelPath(id string) string {
|
|
return filepath.Join(s.baseDir, id, "model.nef")
|
|
}
|
|
|
|
// DeleteModel removes a model's directory and all files.
|
|
func (s *ModelStore) DeleteModel(id string) error {
|
|
dir := filepath.Join(s.baseDir, id)
|
|
return os.RemoveAll(dir)
|
|
}
|
|
|
|
// LoadCustomModels scans the store directory and returns all custom models.
|
|
func (s *ModelStore) LoadCustomModels() ([]Model, error) {
|
|
entries, err := os.ReadDir(s.baseDir)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
var models []Model
|
|
for _, entry := range entries {
|
|
if !entry.IsDir() {
|
|
continue
|
|
}
|
|
|
|
metaPath := filepath.Join(s.baseDir, entry.Name(), "metadata.json")
|
|
data, err := os.ReadFile(metaPath)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
var m Model
|
|
if err := json.Unmarshal(data, &m); err != nil {
|
|
continue
|
|
}
|
|
m.IsCustom = true
|
|
models = append(models, m)
|
|
}
|
|
|
|
return models, nil
|
|
}
|