Mining/pkg/mining/profile_manager.go

package mining

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
	"sync"

	"github.com/adrg/xdg"
	"github.com/google/uuid"
)

const profileConfigFileName = "mining_profiles.json"

// ProfileManager handles CRUD operations for MiningProfiles.
type ProfileManager struct {
	mu         sync.RWMutex
	profiles   map[string]*MiningProfile
	configPath string
}

// NewProfileManager creates and initializes a new ProfileManager.
func NewProfileManager() (*ProfileManager, error) {
	configPath, err := xdg.ConfigFile(filepath.Join("lethean-desktop", profileConfigFileName))
	if err != nil {
		return nil, fmt.Errorf("could not resolve config path: %w", err)
	}

	pm := &ProfileManager{
		profiles:   make(map[string]*MiningProfile),
		configPath: configPath,
	}

	if err := pm.loadProfiles(); err != nil {
		// If the file doesn't exist, that's fine, but any other error is a problem.
		if !os.IsNotExist(err) {
			return nil, fmt.Errorf("could not load profiles: %w", err)
		}
	}

	return pm, nil
}

// loadProfiles reads the profiles from the JSON file into memory.
func (pm *ProfileManager) loadProfiles() error {
	pm.mu.Lock()
	defer pm.mu.Unlock()

	data, err := os.ReadFile(pm.configPath)
	if err != nil {
		return err
	}

	var profiles []*MiningProfile
	if err := json.Unmarshal(data, &profiles); err != nil {
		return err
	}

	pm.profiles = make(map[string]*MiningProfile)
	for _, p := range profiles {
		pm.profiles[p.ID] = p
	}

	return nil
}

// saveProfiles writes the current profiles from memory to the JSON file.
// This is an internal method and assumes the caller holds the appropriate lock.
// Uses atomic write pattern: write to temp file, sync, then rename.
func (pm *ProfileManager) saveProfiles() error {
	profileList := make([]*MiningProfile, 0, len(pm.profiles))
	for _, p := range pm.profiles {
		profileList = append(profileList, p)
	}

	data, err := json.MarshalIndent(profileList, "", "  ")
	if err != nil {
		return err
	}

	// Atomic write: write to temp file in same directory, then rename
	dir := filepath.Dir(pm.configPath)
	tmpFile, err := os.CreateTemp(dir, "profiles-*.tmp")
	if err != nil {
		return fmt.Errorf("failed to create temp file: %w", err)
	}
	tmpPath := tmpFile.Name()

	// Clean up temp file on any error
	success := false
	defer func() {
		if !success {
			os.Remove(tmpPath)
		}
	}()

	if _, err := tmpFile.Write(data); err != nil {
		tmpFile.Close()
		return fmt.Errorf("failed to write temp file: %w", err)
	}

	// Sync to ensure data is flushed to disk before rename
	if err := tmpFile.Sync(); err != nil {
		tmpFile.Close()
		return fmt.Errorf("failed to sync temp file: %w", err)
	}

	if err := tmpFile.Close(); err != nil {
		return fmt.Errorf("failed to close temp file: %w", err)
	}

	// Set permissions before rename
	if err := os.Chmod(tmpPath, 0600); err != nil {
		return fmt.Errorf("failed to set temp file permissions: %w", err)
	}

	// Atomic rename (on POSIX systems)
	if err := os.Rename(tmpPath, pm.configPath); err != nil {
		return fmt.Errorf("failed to rename temp file: %w", err)
	}

	success = true
	return nil
}

// CreateProfile adds a new profile and saves it.
func (pm *ProfileManager) CreateProfile(profile *MiningProfile) (*MiningProfile, error) {
	pm.mu.Lock()
	defer pm.mu.Unlock()

	profile.ID = uuid.New().String()
	pm.profiles[profile.ID] = profile

	if err := pm.saveProfiles(); err != nil {
		// Rollback
		delete(pm.profiles, profile.ID)
		return nil, err
	}

	return profile, nil
}

// GetProfile retrieves a profile by its ID.
func (pm *ProfileManager) GetProfile(id string) (*MiningProfile, bool) {
	pm.mu.RLock()
	defer pm.mu.RUnlock()
	profile, exists := pm.profiles[id]
	return profile, exists
}

// GetAllProfiles returns a list of all profiles.
func (pm *ProfileManager) GetAllProfiles() []*MiningProfile {
	pm.mu.RLock()
	defer pm.mu.RUnlock()

	profileList := make([]*MiningProfile, 0, len(pm.profiles))
	for _, p := range pm.profiles {
		profileList = append(profileList, p)
	}
	return profileList
}

// UpdateProfile modifies an existing profile.
func (pm *ProfileManager) UpdateProfile(profile *MiningProfile) error {
	pm.mu.Lock()
	defer pm.mu.Unlock()

	if _, exists := pm.profiles[profile.ID]; !exists {
		return fmt.Errorf("profile with ID %s not found", profile.ID)
	}
	pm.profiles[profile.ID] = profile

	return pm.saveProfiles()
}

// DeleteProfile removes a profile by its ID.
func (pm *ProfileManager) DeleteProfile(id string) error {
	pm.mu.Lock()
	defer pm.mu.Unlock()

	if _, exists := pm.profiles[id]; !exists {
		return fmt.Errorf("profile with ID %s not found", id)
	}
	delete(pm.profiles, id)

	return pm.saveProfiles()
}