fix: deps sync optimize

Author: Code2Life

.gitignore

@@ -53,4 +53,5 @@ cmd/ggo/ggo
 # .idea/
 # .vscode/
 .vscode-test/
-*.log
+*.log
+bin/

bin/ggo

@@ -82,16 +82,23 @@ func newSyncCmd() *cobra.Command {
 			if manifest != nil {
 				fmt.Printf("Synced %d libraries (manifest version: %s)\n", len(manifest.Libraries), manifest.Version)
 
-				if verbose {
-					fmt.Println("\nSynced libraries:")
-					for _, lib := range manifest.Libraries {
-						fmt.Printf("  Name: %s\n", lib.Name)
-						fmt.Printf("    Version: %s\n", lib.Version)
-						fmt.Printf("    Platform: %s/%s\n", lib.Platform, lib.Arch)
-						fmt.Printf("    Size: %d bytes\n", lib.Size)
-						fmt.Printf("    SHA256: %s\n", lib.SHA256)
-						fmt.Printf("    URL: %s\n", lib.URL)
-						fmt.Println()
+				if len(manifest.Libraries) > 0 {
+					if verbose {
+						fmt.Println("\nSynced libraries:")
+						for _, lib := range manifest.Libraries {
+							fmt.Printf("  Name: %s\n", lib.Name)
+							fmt.Printf("    Version: %s\n", lib.Version)
+							fmt.Printf("    Platform: %s/%s\n", lib.Platform, lib.Arch)
+							fmt.Printf("    Size: %d bytes\n", lib.Size)
+							fmt.Printf("    SHA256: %s\n", lib.SHA256)
+							fmt.Printf("    URL: %s\n", lib.URL)
+							fmt.Println()
+						}
+					} else {
+						fmt.Println("\nSynced libraries:")
+						for _, lib := range manifest.Libraries {
+							fmt.Printf("  %s (version: %s, platform: %s/%s)\n", lib.Name, lib.Version, lib.Platform, lib.Arch)
+						}
 					}
 				}
 			}

cmd/ggo/deps/deps.go

@@ -59,48 +59,33 @@ func DownloadOrFindAccelerator() (string, error) {
 	// Step 1: Detect vendor (config has highest priority)
 	vendor, version := detectVendor()
 	if vendor == "" {
-		vendor = "STUB" // Fallback to stub if detection fails
+		vendor = "stub" // Fallback to stub if detection fails (use lowercase for slug matching)
 	}
+	// Normalize vendor to lowercase for slug matching
+	vendorSlug := strings.ToLower(vendor)
 
-	// Step 2: Try to find library locally
-	suffix := getLibSuffix()
-	libName := fmt.Sprintf("libaccelerator_%s%s", vendor, suffix)
-
-	searchPaths := buildSearchPaths()
-	for _, dir := range searchPaths {
-		if path := filepath.Join(dir, libName); fileExists(path) {
-			return path, nil
-		}
-	}
-
-	// Step 3: Library not found locally, try to download from CDN
+	// Step 2: Initialize deps manager and fetch manifest
+	// This will auto-sync on first use if manifest doesn't exist
 	paths := platform.DefaultPaths()
-
-	// Check config for accelerator library version (version already detected from env/config in detectVendor)
-	// TODO: Add AcceleratorVersion field to config.Config struct for persistent config
-
-	// Use deps manager to download
 	depsMgr := deps.NewManager(deps.WithPaths(paths))
 	ctx := context.Background()
 
-	// Fetch manifest to get latest version or use configured version
+	// Fetch manifest (auto-syncs if not cached)
 	manifest, err := depsMgr.FetchManifest(ctx)
 	if err != nil {
-		return "", fmt.Errorf("failed to fetch manifest from CDN: %w", err)
+		return "", fmt.Errorf("failed to fetch manifest: %w", err)
 	}
 
-	// Find library in manifest matching platform and vendor
+	// Step 3: Find library in manifest matching platform and vendor slug
 	var targetLib *deps.Library
 	var candidateLibs []deps.Library
 	platformLibs := depsMgr.GetLibrariesForPlatform(manifest, "", "")
 
-	// Match library name pattern: contains "accelerator_{vendor}"
-	vendorPattern := "accelerator_" + vendor
+	// Match by vendor slug from manifest
 	for i := range platformLibs {
 		lib := platformLibs[i]
 		libNameLower := strings.ToLower(lib.Name)
-		// Match if library name contains the vendor pattern
-		if strings.Contains(libNameLower, vendorPattern) {
+		if strings.Contains(libNameLower, "accelerator") && lib.VendorSlug == vendorSlug {
 			candidateLibs = append(candidateLibs, lib)
 		}
 	}
@@ -121,13 +106,47 @@ func DownloadOrFindAccelerator() (string, error) {
 	}
 
 	if targetLib == nil {
-		// Library not in manifest, return error (could fallback to stub)
+		// Library not in manifest, return error with available libraries
 		availableNames := getLibraryNames(platformLibs)
-		return "", fmt.Errorf("accelerator library for vendor %s (name: %s) not found in CDN manifest. Available libraries: %v",
-			vendor, libName, availableNames)
+		return "", fmt.Errorf("accelerator library for vendor %s (slug: %s) not found in CDN manifest. Available libraries: %v",
+			vendor, vendorSlug, availableNames)
 	}
 
-	// Download the library
+	// Step 4: Check if library is already downloaded in cache directory
+	// Search cache directory based on manifest entries
+	cacheDir := paths.CacheDir()
+	cachedPath := filepath.Join(cacheDir, targetLib.Name)
+	if fileExists(cachedPath) {
+		// Verify the cached file matches the expected hash
+		if targetLib.SHA256 != "" {
+			// Use deps manager's VerifyLibrary method
+			if depsMgr.VerifyLibrary(cachedPath, targetLib.SHA256) {
+				// Library found in cache, check if installed
+				installedPath := depsMgr.GetLibraryPath(targetLib.Name)
+				if fileExists(installedPath) {
+					return installedPath, nil
+				}
+				// Install from cache
+				if err := depsMgr.InstallLibrary(*targetLib); err != nil {
+					return "", fmt.Errorf("failed to install library from cache: %w", err)
+				}
+				return installedPath, nil
+			}
+		} else {
+			// No hash to verify, assume cached file is valid
+			installedPath := depsMgr.GetLibraryPath(targetLib.Name)
+			if fileExists(installedPath) {
+				return installedPath, nil
+			}
+			// Install from cache
+			if err := depsMgr.InstallLibrary(*targetLib); err != nil {
+				return "", fmt.Errorf("failed to install library from cache: %w", err)
+			}
+			return installedPath, nil
+		}
+	}
+
+	// Step 5: Library not found locally, download from CDN
 	progressFn := func(downloaded, total int64) {
 		// Silent progress for library download
 	}
@@ -290,50 +309,6 @@ func CreateMockGPUs(count int) []api.GPUInfo {
 	return gpus
 }
 
-func getLibSuffix() string {
-	switch runtime.GOOS {
-	case "darwin":
-		// In provider build, we only build .so libraries, even for MacOS
-		return ".so"
-	case "windows":
-		return ".dll"
-	default:
-		return ".so"
-	}
-}
-
-func buildSearchPaths() []string {
-	paths := []string{
-		"/usr/lib/tensor-fusion",
-		"/usr/local/lib/tensor-fusion",
-		"/opt/tensor-fusion/lib",
-	}
-
-	// Add platform lib directory (where deps manager installs libraries)
-	platformPaths := platform.DefaultPaths()
-	paths = append(paths, platformPaths.LibDir())
-
-	// Environment variable takes priority
-	if tfLibPath := os.Getenv("TENSOR_FUSION_LIB_PATH"); tfLibPath != "" {
-		paths = append([]string{tfLibPath}, paths...)
-	}
-
-	// Add home directory paths
-	if home, err := os.UserHomeDir(); err == nil {
-		paths = append(paths,
-			filepath.Join(home, ".tensor-fusion", "libs"),
-			filepath.Join(home, ".local", "lib", "tensor-fusion"),
-		)
-	}
-
-	// Add cwd for development
-	if cwd, err := os.Getwd(); err == nil {
-		paths = append(paths, cwd)
-	}
-
-	return paths
-}
-
 func fileExists(path string) bool {
 	_, err := os.Stat(path)
 	return err == nil

internal/agent/device.go

@@ -42,6 +42,9 @@ type Library struct {
 	URL      string `json:"url"`
 	SHA256   string `json:"sha256"`
 	Size     int64  `json:"size"`
+	// Vendor information from release
+	VendorSlug string `json:"vendorSlug,omitempty"` // e.g., "stub", "nvidia", "amd"
+	VendorName string `json:"vendorName,omitempty"` // e.g., "STUB", "NVIDIA", "AMD"
 }
 
 // Manifest represents the version manifest from CDN
@@ -186,13 +189,15 @@ func (m *Manager) SyncReleases(ctx context.Context, os, arch string) (*Manifest,
 				}
 
 				lib := Library{
-					Name:     libName,
-					Version:  release.Version,
-					Platform: artifactOS,
-					Arch:     artifactArch,
-					URL:      artifact.URL,
-					SHA256:   artifact.SHA256,
-					Size:     size,
+					Name:       libName,
+					Version:    release.Version,
+					Platform:   artifactOS,
+					Arch:       artifactArch,
+					URL:        artifact.URL,
+					SHA256:     artifact.SHA256,
+					Size:       size,
+					VendorSlug: strings.ToLower(release.Vendor.Slug),
+					VendorName: release.Vendor.Name,
 				}
 				manifest.Libraries = append(manifest.Libraries, lib)
 			}
@@ -363,7 +368,7 @@ func (m *Manager) DownloadLibrary(ctx context.Context, lib Library, progressFn f
 	tmpPath := destPath + ".tmp"
 
 	// Check if already downloaded with correct hash (skip if SHA256 is empty)
-	if lib.SHA256 != "" && m.verifyLibrary(destPath, lib.SHA256) {
+	if lib.SHA256 != "" && m.VerifyLibrary(destPath, lib.SHA256) {
 		return nil // Already downloaded
 	}
 
@@ -539,9 +544,9 @@ func (m *Manager) CleanCache() error {
 	return nil
 }
 
-// verifyLibrary checks if a file exists and has the expected hash
+// VerifyLibrary checks if a file exists and has the expected hash
 // Returns true if expectedHash is empty (verification skipped)
-func (m *Manager) verifyLibrary(path, expectedHash string) bool {
+func (m *Manager) VerifyLibrary(path, expectedHash string) bool {
 	// Skip verification if hash is empty
 	if expectedHash == "" {
 		// Just check if file exists