Pulse/internal/config/persistence.go

package config

import (
	"bufio"
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"os"
	"path/filepath"
	"sort"
	"strings"
	"sync"
	"time"

	"github.com/rcourtman/pulse-go-rewrite/internal/alerts"
	"github.com/rcourtman/pulse-go-rewrite/internal/crypto"
	"github.com/rcourtman/pulse-go-rewrite/internal/mock"
	"github.com/rcourtman/pulse-go-rewrite/internal/models"
	"github.com/rcourtman/pulse-go-rewrite/internal/notifications"
	"github.com/rcourtman/pulse-go-rewrite/internal/securityutil"
	"github.com/rs/zerolog/log"
)

// ConfigPersistence handles saving and loading configuration
type ConfigPersistence struct {
	mu                   sync.RWMutex
	tx                   *importTransaction
	configDir            string
	alertFile            string
	emailFile            string
	webhookFile          string
	appriseFile          string
	nodesFile            string
	trueNASFile          string
	vmwareFile           string
	systemFile           string
	ssoFile              string
	apiTokensFile        string
	aiFile               string
	quickstartFile       string
	aiFindingsFile       string
	aiPatrolRunsFile     string
	aiUsageHistoryFile   string
	agentProfilesFile    string
	agentAssignmentsFile string
	aiChatSessionsFile   string
	orgFile              string
	relayFile            string
	crypto               *crypto.CryptoManager
	fs                   FileSystem

	// Lazy loaded metadata stores
	guestMetadataStore  *GuestMetadataStore
	dockerMetadataStore *DockerMetadataStore
	hostMetadataStore   *HostMetadataStore
	metadataMu          sync.Mutex
}

// FileSystem interface for mocking file operations
type FileSystem interface {
	ReadFile(name string) ([]byte, error)
	WriteFile(name string, data []byte, perm os.FileMode) error
	Rename(oldpath, newpath string) error
	Remove(name string) error
	Stat(name string) (os.FileInfo, error)
	MkdirAll(path string, perm os.FileMode) error
}

type defaultFileSystem struct{}

func (dfs defaultFileSystem) ReadFile(name string) ([]byte, error) { return os.ReadFile(name) }
func (dfs defaultFileSystem) WriteFile(name string, data []byte, perm os.FileMode) error {
	return os.WriteFile(name, data, perm)
}
func (dfs defaultFileSystem) Rename(oldpath, newpath string) error {
	return os.Rename(oldpath, newpath)
}
func (dfs defaultFileSystem) Remove(name string) error              { return os.Remove(name) }
func (dfs defaultFileSystem) Stat(name string) (os.FileInfo, error) { return os.Stat(name) }
func (dfs defaultFileSystem) MkdirAll(path string, perm os.FileMode) error {
	return os.MkdirAll(path, perm)
}

type alertSchedulePresence struct {
	Schedule *alertScheduleFieldPresence `json:"schedule"`
}

type alertScheduleFieldPresence struct {
	NotifyOnResolve *bool `json:"notifyOnResolve"`
}

type resolvedConfigPersistencePaths struct {
	alertFile            string
	emailFile            string
	webhookFile          string
	appriseFile          string
	nodesFile            string
	trueNASFile          string
	vmwareFile           string
	systemFile           string
	ssoFile              string
	apiTokensFile        string
	aiFile               string
	quickstartFile       string
	aiFindingsFile       string
	aiPatrolRunsFile     string
	aiUsageHistoryFile   string
	agentProfilesFile    string
	agentAssignmentsFile string
	aiChatSessionsFile   string
	orgFile              string
	relayFile            string
}

func resolveConfigPersistencePaths(configDir string) (string, resolvedConfigPersistencePaths, error) {
	normalizedConfigDir, err := securityutil.NormalizeStorageDir(ResolveRuntimeDataDir(configDir))
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve config directory: %w", err)
	}

	resolveLeaf := func(leaf string) (string, error) {
		return securityutil.JoinStorageLeaf(normalizedConfigDir, leaf)
	}

	alertFile, err := resolveLeaf("alerts.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve alerts.json: %w", err)
	}
	emailFile, err := resolveLeaf("email.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve email.enc: %w", err)
	}
	webhookFile, err := resolveLeaf("webhooks.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve webhooks.enc: %w", err)
	}
	appriseFile, err := resolveLeaf("apprise.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve apprise.enc: %w", err)
	}
	nodesFile, err := resolveLeaf("nodes.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve nodes.enc: %w", err)
	}
	trueNASFile, err := resolveLeaf("truenas.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve truenas.enc: %w", err)
	}
	vmwareFile, err := resolveLeaf("vmware.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve vmware.enc: %w", err)
	}
	systemFile, err := resolveLeaf("system.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve system.json: %w", err)
	}
	ssoFile, err := resolveLeaf("sso.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve sso.enc: %w", err)
	}
	apiTokensFile, err := resolveLeaf("api_tokens.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve api_tokens.json: %w", err)
	}
	aiFile, err := resolveLeaf("ai.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve ai.enc: %w", err)
	}
	quickstartFile, err := resolveLeaf("quickstart.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve quickstart.enc: %w", err)
	}
	aiFindingsFile, err := resolveLeaf("ai_findings.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve ai_findings.json: %w", err)
	}
	aiPatrolRunsFile, err := resolveLeaf("ai_patrol_runs.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve ai_patrol_runs.json: %w", err)
	}
	aiUsageHistoryFile, err := resolveLeaf("ai_usage_history.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve ai_usage_history.json: %w", err)
	}
	agentProfilesFile, err := resolveLeaf("agent_profiles.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve agent_profiles.json: %w", err)
	}
	agentAssignmentsFile, err := resolveLeaf("agent_profile_assignments.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve agent_profile_assignments.json: %w", err)
	}
	aiChatSessionsFile, err := resolveLeaf("ai_chat_sessions.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve ai_chat_sessions.json: %w", err)
	}
	orgFile, err := resolveLeaf("org.json")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve org.json: %w", err)
	}
	relayFile, err := resolveLeaf("relay.enc")
	if err != nil {
		return "", resolvedConfigPersistencePaths{}, fmt.Errorf("resolve relay.enc: %w", err)
	}

	return normalizedConfigDir, resolvedConfigPersistencePaths{
		alertFile:            alertFile,
		emailFile:            emailFile,
		webhookFile:          webhookFile,
		appriseFile:          appriseFile,
		nodesFile:            nodesFile,
		trueNASFile:          trueNASFile,
		vmwareFile:           vmwareFile,
		systemFile:           systemFile,
		ssoFile:              ssoFile,
		apiTokensFile:        apiTokensFile,
		aiFile:               aiFile,
		quickstartFile:       quickstartFile,
		aiFindingsFile:       aiFindingsFile,
		aiPatrolRunsFile:     aiPatrolRunsFile,
		aiUsageHistoryFile:   aiUsageHistoryFile,
		agentProfilesFile:    agentProfilesFile,
		agentAssignmentsFile: agentAssignmentsFile,
		aiChatSessionsFile:   aiChatSessionsFile,
		orgFile:              orgFile,
		relayFile:            relayFile,
	}, nil
}

// NewConfigPersistence creates a new config persistence manager.
// The process terminates if encryption cannot be initialized to avoid
// writing secrets to disk in plaintext.
func NewConfigPersistence(configDir string) *ConfigPersistence {
	cp, err := newConfigPersistence(configDir)
	if err != nil {
		log.Fatal().
			Str("configDir", configDir).
			Err(err).
			Msg("Failed to initialize config persistence")
	}
	return cp
}

func newConfigPersistence(configDir string) (*ConfigPersistence, error) {
	resolvedConfigDir, resolvedPaths, err := resolveConfigPersistencePaths(configDir)
	if err != nil {
		return nil, err
	}

	// Initialize crypto manager
	cryptoMgr, err := crypto.NewCryptoManagerAt(resolvedConfigDir)
	if err != nil {
		return nil, err
	}

	cp := &ConfigPersistence{
		configDir:            resolvedConfigDir,
		alertFile:            resolvedPaths.alertFile,
		emailFile:            resolvedPaths.emailFile,
		webhookFile:          resolvedPaths.webhookFile,
		appriseFile:          resolvedPaths.appriseFile,
		nodesFile:            resolvedPaths.nodesFile,
		trueNASFile:          resolvedPaths.trueNASFile,
		vmwareFile:           resolvedPaths.vmwareFile,
		systemFile:           resolvedPaths.systemFile,
		ssoFile:              resolvedPaths.ssoFile,
		apiTokensFile:        resolvedPaths.apiTokensFile,
		aiFile:               resolvedPaths.aiFile,
		quickstartFile:       resolvedPaths.quickstartFile,
		aiFindingsFile:       resolvedPaths.aiFindingsFile,
		aiPatrolRunsFile:     resolvedPaths.aiPatrolRunsFile,
		aiUsageHistoryFile:   resolvedPaths.aiUsageHistoryFile,
		agentProfilesFile:    resolvedPaths.agentProfilesFile,
		agentAssignmentsFile: resolvedPaths.agentAssignmentsFile,
		aiChatSessionsFile:   resolvedPaths.aiChatSessionsFile,
		orgFile:              resolvedPaths.orgFile,
		relayFile:            resolvedPaths.relayFile,
		crypto:               cryptoMgr,
		fs:                   defaultFileSystem{},
	}

	log.Debug().
		Str("configDir", resolvedConfigDir).
		Str("systemFile", cp.systemFile).
		Str("nodesFile", cp.nodesFile).
		Bool("encryptionEnabled", cryptoMgr != nil).
		Msg("Config persistence initialized")

	return cp, nil
}

// DataDir returns the configuration directory path
func (c *ConfigPersistence) DataDir() string {
	return c.configDir
}

// SharedInstallationDataDir returns the installation-scoped config directory
// that owns activation and licensing identity. Tenant config directories live
// under <base>/orgs/<org>, but activation remains rooted at <base>.
func (c *ConfigPersistence) SharedInstallationDataDir() string {
	if c == nil {
		return ""
	}
	sharedDir, err := resolveSharedInstallationDataDir(c.configDir)
	if err != nil {
		return c.configDir
	}
	return sharedDir
}

// GetConfigDir returns the configuration directory path (alias for DataDir to match interface expectations)
func (c *ConfigPersistence) GetConfigDir() string {
	return c.configDir
}

// LoadOrganization loads the organization metadata from org.json.
func (c *ConfigPersistence) LoadOrganization() (*models.Organization, error) {
	c.mu.RLock()
	defer c.mu.RUnlock()

	data, err := c.fs.ReadFile(c.orgFile)
	if err != nil {
		if os.IsNotExist(err) {
			return nil, err
		}
		return nil, fmt.Errorf("failed to read org file: %w", err)
	}

	var org models.Organization
	if err := json.Unmarshal(data, &org); err != nil {
		return nil, fmt.Errorf("failed to parse org file: %w", err)
	}

	return &org, nil
}

// SaveOrganization saves the organization metadata to org.json.
func (c *ConfigPersistence) SaveOrganization(org *models.Organization) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	data, err := json.MarshalIndent(org, "", "  ")
	if err != nil {
		return fmt.Errorf("failed to serialize org: %w", err)
	}

	if err := c.fs.WriteFile(c.orgFile, data, 0600); err != nil {
		return fmt.Errorf("failed to write org file: %w", err)
	}

	log.Debug().Str("org_id", org.ID).Msg("Saved organization metadata")
	return nil
}

// EnsureConfigDir ensures the configuration directory exists
func (c *ConfigPersistence) EnsureConfigDir() error {
	if err := c.fs.MkdirAll(c.configDir, 0700); err != nil {
		return fmt.Errorf("ensure config directory %s: %w", c.configDir, err)
	}
	return nil
}

func (c *ConfigPersistence) beginTransaction(tx *importTransaction) {
	c.mu.Lock()
	c.tx = tx
	c.mu.Unlock()
}

func (c *ConfigPersistence) endTransaction(tx *importTransaction) {
	c.mu.Lock()
	if c.tx == tx {
		c.tx = nil
	}
	c.mu.Unlock()
}

// writeConfigFileLocked writes a config file, staging it in a transaction if one is active.
// NOTE: Caller MUST hold c.mu lock (despite reading c.tx, which has separate synchronization)
func (c *ConfigPersistence) writeConfigFileLocked(path string, data []byte, perm os.FileMode) error {
	// Read transaction pointer - safe because caller holds c.mu
	// Transaction field is only modified while holding c.mu in begin/endTransaction
	tx := c.tx

	if tx != nil {
		if err := tx.StageFile(path, data, perm); err != nil {
			return fmt.Errorf("stage config file %s: %w", path, err)
		}
		return nil
	}

	tmp := path + ".tmp"
	if err := c.fs.WriteFile(tmp, data, perm); err != nil {
		return fmt.Errorf("write temp config file %s: %w", tmp, err)
	}
	if err := c.fs.Rename(tmp, path); err != nil {
		renameErr := fmt.Errorf("rename temp config file %s to %s: %w", tmp, path, err)
		if removeErr := c.fs.Remove(tmp); removeErr != nil && !os.IsNotExist(removeErr) {
			return errors.Join(renameErr, fmt.Errorf("cleanup temp config file %s: %w", tmp, removeErr))
		}
		return renameErr
	}
	return nil
}

// saveJSON is a generic helper that marshals data to JSON, optionally encrypts,
// and writes to file. Caller must NOT already hold c.mu.
func saveJSON[T any](c *ConfigPersistence, filePath string, data T, encrypt bool) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	jsonData, err := json.MarshalIndent(data, "", "  ")
	if err != nil {
		return fmt.Errorf("marshal config data for %s: %w", filePath, err)
	}

	if encrypt && c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(jsonData)
		if err != nil {
			return fmt.Errorf("encrypt config data for %s: %w", filePath, err)
		}
		jsonData = encrypted
	}

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for %s: %w", filePath, err)
	}

	if err := c.writeConfigFileLocked(filePath, jsonData, 0600); err != nil {
		return fmt.Errorf("persist config file %s: %w", filePath, err)
	}
	return nil
}

func rewriteEncryptedJSONLocked(c *ConfigPersistence, filePath string, jsonData []byte, context string) error {
	if c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(jsonData)
		if err != nil {
			return fmt.Errorf("encrypt %s: %w", context, err)
		}
		jsonData = encrypted
	}

	if err := c.writeConfigFileLocked(filePath, jsonData, 0600); err != nil {
		return fmt.Errorf("persist %s: %w", context, err)
	}
	return nil
}

func rewriteEncryptedSliceLocked[T any](c *ConfigPersistence, filePath string, data []T, context string) error {
	jsonData, err := json.Marshal(data)
	if err != nil {
		return fmt.Errorf("marshal %s: %w", context, err)
	}
	return rewriteEncryptedJSONLocked(c, filePath, jsonData, context)
}

func loadEncryptedJSONLocked[T any](c *ConfigPersistence, filePath string, target *T, context string) (bool, []byte, error) {
	data, err := c.fs.ReadFile(filePath)
	if err != nil {
		return false, nil, err
	}

	migratedPlaintext := false
	if c.crypto != nil {
		if decrypted, decErr := c.crypto.Decrypt(data); decErr == nil {
			data = decrypted
		} else {
			log.Warn().Err(decErr).Str("file", filePath).Msg("Failed to decrypt config - falling back to plaintext")
			migratedPlaintext = true
		}
	}

	if err := json.Unmarshal(data, target); err != nil {
		return false, nil, fmt.Errorf("decode %s %s: %w", context, filePath, err)
	}

	return migratedPlaintext, data, nil
}

// loadSlice is a generic helper that reads a JSON file, optionally decrypts,
// and unmarshals into a slice of T. Returns an empty slice if the file doesn't
// exist or is empty. Caller must NOT already hold c.mu.
func loadSlice[T any](c *ConfigPersistence, filePath string, decrypt bool) ([]T, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	data, err := c.fs.ReadFile(filePath)
	if err != nil {
		if os.IsNotExist(err) {
			return []T{}, nil
		}
		return nil, fmt.Errorf("read config file %s: %w", filePath, err)
	}

	if len(data) == 0 {
		return []T{}, nil
	}

	migratedPlaintext := false
	if decrypt && c.crypto != nil {
		if decrypted, err := c.crypto.Decrypt(data); err == nil {
			data = decrypted
		} else {
			log.Warn().Err(err).Str("file", filePath).Msg("Failed to decrypt config - falling back to plaintext")
			migratedPlaintext = true
		}
	}

	var result []T
	if err := json.Unmarshal(data, &result); err != nil {
		return nil, fmt.Errorf("decode config file %s: %w", filePath, err)
	}

	if migratedPlaintext {
		if err := rewriteEncryptedSliceLocked(c, filePath, result, "config slice migration rewrite"); err != nil {
			return nil, err
		}
	}

	return result, nil
}

// loadJSON is a generic helper that reads a JSON file, optionally decrypts,
// and unmarshals into the provided target value. Caller must NOT already hold c.mu.
func loadJSON[T any](c *ConfigPersistence, filePath string, decrypt bool, target *T) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	data, err := c.fs.ReadFile(filePath)
	if err != nil {
		return fmt.Errorf("read config file %s: %w", filePath, err)
	}

	migratedPlaintext := false
	if decrypt && c.crypto != nil {
		if decrypted, decErr := c.crypto.Decrypt(data); decErr == nil {
			data = decrypted
		} else {
			log.Warn().Err(decErr).Str("file", filePath).Msg("Failed to decrypt config - falling back to plaintext")
			migratedPlaintext = true
		}
	}

	if err := json.Unmarshal(data, target); err != nil {
		return fmt.Errorf("decode config file %s: %w", filePath, err)
	}

	if migratedPlaintext {
		if err := rewriteEncryptedJSONLocked(c, filePath, data, "config object migration rewrite"); err != nil {
			return err
		}
	}
	return nil
}

// persistMetadata writes metadata to disk atomically (temp file + rename).
// Caller must hold the lock. The data should already be marshaled to JSON.
// This is used by both GuestMetadataStore and HostMetadataStore to avoid code duplication.
func persistMetadata(fs FileSystem, dataPath, fileName string, data []byte) error {
	filePath := filepath.Join(dataPath, fileName)

	log.Debug().Str("path", filePath).Msg("Saving metadata to disk")

	if err := fs.MkdirAll(dataPath, 0o700); err != nil {
		return fmt.Errorf("failed to create data directory: %w", err)
	}

	tempFile := filePath + ".tmp"
	if err := fs.WriteFile(tempFile, data, 0o600); err != nil {
		return fmt.Errorf("failed to write metadata file: %w", err)
	}

	if err := fs.Rename(tempFile, filePath); err != nil {
		return fmt.Errorf("failed to rename metadata file: %w", err)
	}

	return nil
}

// LoadAPITokens loads API token metadata from disk.
func (c *ConfigPersistence) LoadAPITokens() ([]APITokenRecord, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	data, err := c.fs.ReadFile(c.apiTokensFile)
	if err != nil {
		if os.IsNotExist(err) {
			return []APITokenRecord{}, nil
		}
		return nil, fmt.Errorf("read api tokens file %s: %w", c.apiTokensFile, err)
	}

	if len(data) == 0 {
		return []APITokenRecord{}, nil
	}

	migratedPlaintext := false
	// Attempt decryption; fall back to plaintext for migration from unencrypted files.
	if c.crypto != nil {
		if decrypted, decErr := c.crypto.Decrypt(data); decErr == nil {
			data = decrypted
		} else {
			migratedPlaintext = true
		}
	}

	var tokens []APITokenRecord
	if err := json.Unmarshal(data, &tokens); err != nil {
		return nil, fmt.Errorf("decode api tokens file %s: %w", c.apiTokensFile, err)
	}

	for i := range tokens {
		tokens[i].ensureID()
		tokens[i].ensureScopes()
	}

	if migratedPlaintext && c.crypto != nil {
		if err := c.persistAPITokensLocked(tokens); err != nil {
			return nil, fmt.Errorf("rewrite plaintext api tokens file %s: %w", c.apiTokensFile, err)
		}
	}

	return tokens, nil
}

// SaveTrueNASConfig persists TrueNAS instance configuration to encrypted storage.
func (c *ConfigPersistence) SaveTrueNASConfig(instances []TrueNASInstance) error {
	return saveJSON(c, c.trueNASFile, instances, true)
}

// LoadTrueNASConfig loads TrueNAS instance configuration from encrypted storage.
func (c *ConfigPersistence) LoadTrueNASConfig() ([]TrueNASInstance, error) {
	return loadSlice[TrueNASInstance](c, c.trueNASFile, true)
}

// SaveVMwareConfig persists VMware vCenter instance configuration to encrypted
// storage.
func (c *ConfigPersistence) SaveVMwareConfig(instances []VMwareVCenterInstance) error {
	return saveJSON(c, c.vmwareFile, instances, true)
}

// LoadVMwareConfig loads VMware vCenter instance configuration from encrypted
// storage.
func (c *ConfigPersistence) LoadVMwareConfig() ([]VMwareVCenterInstance, error) {
	return loadSlice[VMwareVCenterInstance](c, c.vmwareFile, true)
}

// SaveAPITokens persists API token metadata to disk.
func (c *ConfigPersistence) SaveAPITokens(tokens []APITokenRecord) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	return c.persistAPITokensLocked(tokens)
}

func (c *ConfigPersistence) persistAPITokensLocked(tokens []APITokenRecord) error {

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for api tokens: %w", err)
	}

	// Backup previous state (best effort).
	if existing, err := c.fs.ReadFile(c.apiTokensFile); err == nil && len(existing) > 0 {
		if err := c.fs.WriteFile(c.apiTokensFile+".backup", existing, 0600); err != nil {
			log.Warn().Err(err).Msg("Failed to create API token backup file")
		}
	}

	sanitized := make([]APITokenRecord, len(tokens))
	for i := range tokens {
		record := tokens[i]
		record.ensureID()
		record.ensureScopes()
		sanitized[i] = record
	}

	data, err := json.Marshal(sanitized)
	if err != nil {
		return fmt.Errorf("marshal api token records: %w", err)
	}

	if c.crypto != nil {
		if encrypted, encErr := c.crypto.Encrypt(data); encErr == nil {
			data = encrypted
		} else {
			return fmt.Errorf("encrypt api tokens: %w", encErr)
		}
	}

	if err := c.writeConfigFileLocked(c.apiTokensFile, data, 0600); err != nil {
		return fmt.Errorf("persist api tokens file %s: %w", c.apiTokensFile, err)
	}
	return nil
}

// normalizeHysteresisThreshold ensures a hysteresis threshold pointer has valid
// trigger/clear values. If the pointer is nil or trigger is negative, it is set
// to the given defaults. Trigger==0 means "disabled" (clear forced to 0).
// Otherwise a non-positive clear is derived from trigger-5 with defaultClear as floor.
func normalizeHysteresisThreshold(t **alerts.HysteresisThreshold, defaultTrigger, defaultClear float64) {
	if *t == nil || (*t).Trigger < 0 {
		*t = &alerts.HysteresisThreshold{Trigger: defaultTrigger, Clear: defaultClear}
	} else if (*t).Trigger == 0 {
		(*t).Clear = 0
	} else if (*t).Clear <= 0 {
		(*t).Clear = (*t).Trigger - 5
		if (*t).Clear <= 0 {
			(*t).Clear = defaultClear
		}
	}
}

// normalizeStorageDefault normalizes a non-pointer HysteresisThreshold used for
// storage defaults. Same logic as normalizeHysteresisThreshold but operates on
// a value (not pointer) and uses 0 as the clear floor.
func normalizeStorageDefault(t *alerts.HysteresisThreshold) {
	if t.Trigger < 0 {
		t.Trigger = 85
		t.Clear = 80
	} else if t.Trigger == 0 {
		t.Clear = 0
	} else if t.Clear <= 0 {
		t.Clear = t.Trigger - 5
		if t.Clear < 0 {
			t.Clear = 0
		}
	}
}

// normalizeAlertDefaults applies shared normalization logic to an AlertConfig.
// Called by both SaveAlertConfig and LoadAlertConfig to avoid duplicating the
// same validation and defaulting code.
func normalizeAlertDefaults(config *alerts.AlertConfig) {
	alerts.NormalizeAlertConfigAliases(config)

	// Storage threshold
	normalizeStorageDefault(&config.StorageDefault)

	if config.MinimumDelta <= 0 {
		config.MinimumDelta = 2.0
	}
	if config.SuppressionWindow <= 0 {
		config.SuppressionWindow = 5
	}
	if config.HysteresisMargin <= 0 {
		config.HysteresisMargin = 5.0
	}

	// Agent defaults
	normalizeHysteresisThreshold(&config.AgentDefaults.CPU, 80, 75)
	normalizeHysteresisThreshold(&config.AgentDefaults.Memory, 85, 80)
	normalizeHysteresisThreshold(&config.AgentDefaults.Disk, 90, 85)

	// Time thresholds
	config.MetricTimeThresholds = alerts.NormalizeMetricTimeThresholds(config.MetricTimeThresholds)
	if config.TimeThresholds == nil {
		config.TimeThresholds = make(map[string]int)
	}
	defaultDelay := 5
	ensureDelay := func(key string) {
		if delay, ok := config.TimeThresholds[key]; !ok || delay <= 0 {
			config.TimeThresholds[key] = defaultDelay
		}
	}
	ensureDelay("guest")
	ensureDelay("node")
	ensureDelay("agent")
	ensureDelay("storage")
	ensureDelay("pbs")
	if delay, ok := config.TimeThresholds["all"]; ok && delay <= 0 {
		config.TimeThresholds["all"] = defaultDelay
	}

	// Snapshot defaults
	if config.SnapshotDefaults.WarningDays < 0 {
		config.SnapshotDefaults.WarningDays = 0
	}
	if config.SnapshotDefaults.CriticalDays < 0 {
		config.SnapshotDefaults.CriticalDays = 0
	}
	if config.SnapshotDefaults.CriticalDays > 0 && config.SnapshotDefaults.WarningDays > config.SnapshotDefaults.CriticalDays {
		config.SnapshotDefaults.WarningDays = config.SnapshotDefaults.CriticalDays
	}
	if config.SnapshotDefaults.CriticalSizeGiB < 0 {
		config.SnapshotDefaults.CriticalSizeGiB = 0
	}
	if config.SnapshotDefaults.WarningSizeGiB < 0 {
		config.SnapshotDefaults.WarningSizeGiB = 0
	}
	if config.SnapshotDefaults.CriticalSizeGiB > 0 && config.SnapshotDefaults.WarningSizeGiB > config.SnapshotDefaults.CriticalSizeGiB {
		config.SnapshotDefaults.WarningSizeGiB = config.SnapshotDefaults.CriticalSizeGiB
	}
	if config.SnapshotDefaults.CriticalSizeGiB == 0 && config.SnapshotDefaults.WarningSizeGiB > 0 {
		config.SnapshotDefaults.CriticalSizeGiB = config.SnapshotDefaults.WarningSizeGiB
	}

	// Backup defaults
	if config.BackupDefaults.WarningDays < 0 {
		config.BackupDefaults.WarningDays = 0
	}
	if config.BackupDefaults.CriticalDays < 0 {
		config.BackupDefaults.CriticalDays = 0
	}
	if config.BackupDefaults.CriticalDays > 0 && config.BackupDefaults.WarningDays > config.BackupDefaults.CriticalDays {
		config.BackupDefaults.WarningDays = config.BackupDefaults.CriticalDays
	}
	if config.BackupDefaults.AlertOrphaned == nil {
		alertOrphaned := true
		config.BackupDefaults.AlertOrphaned = &alertOrphaned
	}

	// Deduplicate IgnoreVMIDs
	if len(config.BackupDefaults.IgnoreVMIDs) > 0 {
		seen := make(map[string]struct{}, len(config.BackupDefaults.IgnoreVMIDs))
		normalized := make([]string, 0, len(config.BackupDefaults.IgnoreVMIDs))
		for _, entry := range config.BackupDefaults.IgnoreVMIDs {
			value := strings.TrimSpace(entry)
			if value == "" {
				continue
			}
			if _, exists := seen[value]; exists {
				continue
			}
			seen[value] = struct{}{}
			normalized = append(normalized, value)
		}
		config.BackupDefaults.IgnoreVMIDs = normalized
	}

	config.DockerIgnoredContainerPrefixes = alerts.NormalizeDockerIgnoredPrefixes(config.DockerIgnoredContainerPrefixes)
}

// SaveAlertConfig saves alert configuration to file
func (c *ConfigPersistence) SaveAlertConfig(config alerts.AlertConfig) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	// Save-specific: normalize override clear values before shared defaults run
	if config.MinimumDelta <= 0 {
		margin := config.HysteresisMargin
		if margin <= 0 {
			margin = 5.0
		}
		for id, override := range config.Overrides {
			if override.Usage != nil {
				if override.Usage.Clear <= 0 {
					override.Usage.Clear = override.Usage.Trigger - margin
					if override.Usage.Clear < 0 {
						override.Usage.Clear = 0
					}
				}
				config.Overrides[id] = override
			}
		}
	}

	normalizeAlertDefaults(&config)

	data, err := json.Marshal(config)
	if err != nil {
		return fmt.Errorf("marshal alert config: %w", err)
	}

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for alert config: %w", err)
	}

	if err := c.writeConfigFileLocked(c.alertFile, data, 0600); err != nil {
		return fmt.Errorf("persist alert config: %w", err)
	}

	log.Info().Str("file", c.alertFile).Msg("Alert configuration saved")
	return nil
}

// LoadAlertConfig loads alert configuration from file
func (c *ConfigPersistence) LoadAlertConfig() (*alerts.AlertConfig, error) {
	c.mu.RLock()
	defer c.mu.RUnlock()

	alertOrphaned := true
	data, err := c.fs.ReadFile(c.alertFile)
	if err != nil {
		if os.IsNotExist(err) {
			// Return default config if file doesn't exist
			return &alerts.AlertConfig{
				Enabled: true,
				GuestDefaults: alerts.ThresholdConfig{
					CPU:    &alerts.HysteresisThreshold{Trigger: 80, Clear: 75},
					Memory: &alerts.HysteresisThreshold{Trigger: 85, Clear: 80},
					Disk:   &alerts.HysteresisThreshold{Trigger: 90, Clear: 85},
				},
				NodeDefaults: alerts.ThresholdConfig{
					CPU:         &alerts.HysteresisThreshold{Trigger: 80, Clear: 75},
					Memory:      &alerts.HysteresisThreshold{Trigger: 85, Clear: 80},
					Disk:        &alerts.HysteresisThreshold{Trigger: 90, Clear: 85},
					Temperature: &alerts.HysteresisThreshold{Trigger: 80, Clear: 75},
				},
				AgentDefaults: alerts.ThresholdConfig{
					CPU:    &alerts.HysteresisThreshold{Trigger: 80, Clear: 75},
					Memory: &alerts.HysteresisThreshold{Trigger: 85, Clear: 80},
					Disk:   &alerts.HysteresisThreshold{Trigger: 90, Clear: 85},
				},
				Schedule: alerts.ScheduleConfig{
					NotifyOnResolve: true,
				},
				StorageDefault: alerts.HysteresisThreshold{Trigger: 85, Clear: 80},
				TimeThresholds: map[string]int{
					"guest":   5,
					"node":    5,
					"agent":   5,
					"storage": 5,
					"pbs":     5,
				},
				MinimumDelta:      2.0,
				SuppressionWindow: 5,
				HysteresisMargin:  5.0,
				SnapshotDefaults: alerts.SnapshotAlertConfig{
					Enabled:         false,
					WarningDays:     30,
					CriticalDays:    45,
					WarningSizeGiB:  0,
					CriticalSizeGiB: 0,
				},
				BackupDefaults: alerts.BackupAlertConfig{
					Enabled:       false,
					WarningDays:   7,
					CriticalDays:  14,
					FreshHours:    24,
					StaleHours:    72,
					AlertOrphaned: &alertOrphaned,
					IgnoreVMIDs:   []string{},
				},
				Overrides: make(map[string]alerts.ThresholdConfig),
			}, nil
		}
		return nil, err
	}

	var config alerts.AlertConfig
	if err := json.Unmarshal(data, &config); err != nil {
		return nil, err
	}

	// For empty config files ({}), enable alerts by default
	// This handles the case where the file exists but is empty
	if string(data) == "{}" {
		config.Enabled = true
	}

	// Load-specific: NotifyOnResolve defaults to true (send recovery notifications).
	// Since bool zero-value is false, we check raw JSON to distinguish
	// "missing field" (old configs) from "explicitly set to false".
	if !config.Schedule.NotifyOnResolve {
		var presence alertSchedulePresence
		// Ignore error here as we've already unmarshaled successfully above.
		if json.Unmarshal(data, &presence) == nil {
			if presence.Schedule == nil || presence.Schedule.NotifyOnResolve == nil {
				config.Schedule.NotifyOnResolve = true
			}
		}
	}

	// Load-specific: NodeDefaults.Temperature normalization
	normalizeHysteresisThreshold(&config.NodeDefaults.Temperature, 80, 75)

	// Shared normalization (storage, host defaults, time thresholds, snapshots, backups, etc.)
	normalizeAlertDefaults(&config)

	// Load-specific: Default indicator thresholds for dashboard
	if config.BackupDefaults.FreshHours <= 0 {
		config.BackupDefaults.FreshHours = 24
	}
	if config.BackupDefaults.StaleHours <= 0 {
		config.BackupDefaults.StaleHours = 72
	}
	if config.BackupDefaults.StaleHours < config.BackupDefaults.FreshHours {
		config.BackupDefaults.StaleHours = config.BackupDefaults.FreshHours
	}

	// Load-specific migration: Set I/O metrics to Off (0) if they have old default values
	if config.GuestDefaults.DiskRead != nil && config.GuestDefaults.DiskRead.Trigger == 150 {
		config.GuestDefaults.DiskRead = &alerts.HysteresisThreshold{Trigger: 0, Clear: 0}
	}
	if config.GuestDefaults.DiskWrite != nil && config.GuestDefaults.DiskWrite.Trigger == 150 {
		config.GuestDefaults.DiskWrite = &alerts.HysteresisThreshold{Trigger: 0, Clear: 0}
	}
	if config.GuestDefaults.NetworkIn != nil && config.GuestDefaults.NetworkIn.Trigger == 200 {
		config.GuestDefaults.NetworkIn = &alerts.HysteresisThreshold{Trigger: 0, Clear: 0}
	}
	if config.GuestDefaults.NetworkOut != nil && config.GuestDefaults.NetworkOut.Trigger == 200 {
		config.GuestDefaults.NetworkOut = &alerts.HysteresisThreshold{Trigger: 0, Clear: 0}
	}

	log.Info().
		Str("file", c.alertFile).
		Bool("enabled", config.Enabled).
		Msg("Alert configuration loaded")
	return &config, nil
}

// SaveEmailConfig saves email configuration to file (encrypted)
func (c *ConfigPersistence) SaveEmailConfig(config notifications.EmailConfig) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	// Marshal to JSON first
	data, err := json.Marshal(config)
	if err != nil {
		return fmt.Errorf("marshal email config: %w", err)
	}

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for email config: %w", err)
	}

	// Encrypt if crypto manager is available
	if c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(data)
		if err != nil {
			return fmt.Errorf("encrypt email config: %w", err)
		}
		data = encrypted
	}

	// Save with restricted permissions (owner read/write only)
	if err := c.writeConfigFileLocked(c.emailFile, data, 0600); err != nil {
		return fmt.Errorf("persist email config: %w", err)
	}

	log.Info().
		Str("file", c.emailFile).
		Bool("encrypted", c.crypto != nil).
		Msg("Email configuration saved")
	return nil
}

// LoadEmailConfig loads email configuration from file (decrypts if encrypted)
func (c *ConfigPersistence) LoadEmailConfig() (*notifications.EmailConfig, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	var config notifications.EmailConfig
	migratedPlaintext, _, err := loadEncryptedJSONLocked(c, c.emailFile, &config, "email config")
	if err != nil {
		if os.IsNotExist(err) {
			// Return empty config if encrypted file doesn't exist
			return &notifications.EmailConfig{
				Enabled:  false,
				SMTPPort: 587,
				TLS:      true,
				To:       []string{},
			}, nil
		}
		return nil, err
	}

	if migratedPlaintext {
		jsonData, err := json.Marshal(config)
		if err != nil {
			return nil, fmt.Errorf("marshal email config migration rewrite: %w", err)
		}
		if err := rewriteEncryptedJSONLocked(c, c.emailFile, jsonData, "email config migration rewrite"); err != nil {
			return nil, err
		}
	}

	log.Info().
		Str("file", c.emailFile).
		Bool("encrypted", c.crypto != nil).
		Msg("Email configuration loaded")
	return &config, nil
}

// SaveAppriseConfig saves Apprise configuration to file (encrypted if available)
func (c *ConfigPersistence) SaveAppriseConfig(config notifications.AppriseConfig) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	config = notifications.NormalizeAppriseConfig(config)

	data, err := json.Marshal(config)
	if err != nil {
		return fmt.Errorf("marshal apprise config: %w", err)
	}

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for apprise config: %w", err)
	}

	if c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(data)
		if err != nil {
			return fmt.Errorf("encrypt apprise config: %w", err)
		}
		data = encrypted
	}

	if err := c.writeConfigFileLocked(c.appriseFile, data, 0600); err != nil {
		return fmt.Errorf("persist apprise config: %w", err)
	}

	log.Info().
		Str("file", c.appriseFile).
		Bool("encrypted", c.crypto != nil).
		Msg("Apprise configuration saved")
	return nil
}

// LoadAppriseConfig loads Apprise configuration from file (decrypts if encrypted)
func (c *ConfigPersistence) LoadAppriseConfig() (*notifications.AppriseConfig, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	var config notifications.AppriseConfig
	migratedPlaintext, _, err := loadEncryptedJSONLocked(c, c.appriseFile, &config, "apprise config")
	if err != nil {
		if os.IsNotExist(err) {
			defaultCfg := notifications.AppriseConfig{
				Enabled:        false,
				Mode:           notifications.AppriseModeCLI,
				Targets:        []string{},
				CLIPath:        "apprise",
				TimeoutSeconds: 15,
				APIKeyHeader:   "X-API-KEY",
			}
			return &defaultCfg, nil
		}
		return nil, err
	}

	if migratedPlaintext {
		jsonData, err := json.Marshal(config)
		if err != nil {
			return nil, fmt.Errorf("marshal apprise config migration rewrite: %w", err)
		}
		if err := rewriteEncryptedJSONLocked(c, c.appriseFile, jsonData, "apprise config migration rewrite"); err != nil {
			return nil, err
		}
	}

	normalized := notifications.NormalizeAppriseConfig(config)

	log.Info().
		Str("file", c.appriseFile).
		Bool("encrypted", c.crypto != nil).
		Msg("Apprise configuration loaded")
	return &normalized, nil
}

// SaveWebhooks saves webhook configurations to file
func (c *ConfigPersistence) SaveWebhooks(webhooks []notifications.WebhookConfig) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	webhooks = normalizeWebhookConfigs(webhooks)

	data, err := json.Marshal(webhooks)
	if err != nil {
		return fmt.Errorf("marshal webhooks config: %w", err)
	}

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for webhooks: %w", err)
	}

	// Encrypt if crypto manager is available
	if c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(data)
		if err != nil {
			return fmt.Errorf("encrypt webhooks config: %w", err)
		}
		data = encrypted
	}

	if err := c.writeConfigFileLocked(c.webhookFile, data, 0600); err != nil {
		return fmt.Errorf("persist webhooks config: %w", err)
	}

	log.Info().Str("file", c.webhookFile).
		Int("count", len(webhooks)).
		Bool("encrypted", c.crypto != nil).
		Msg("Webhooks saved")
	return nil
}

// LoadWebhooks loads webhook configurations from file (decrypts if encrypted)
func (c *ConfigPersistence) LoadWebhooks() ([]notifications.WebhookConfig, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	// First try to load from encrypted file
	data, err := c.fs.ReadFile(c.webhookFile)
	if err != nil {
		if os.IsNotExist(err) {
			// Check for legacy unencrypted file
			legacyFile := filepath.Join(c.configDir, "webhooks.json")
			legacyData, legacyErr := c.fs.ReadFile(legacyFile)
			if legacyErr == nil {
				// Legacy file exists, parse it
				var webhooks []notifications.WebhookConfig
				if err := json.Unmarshal(legacyData, &webhooks); err == nil {
					webhooks = normalizeWebhookConfigs(webhooks)
					log.Info().
						Str("file", legacyFile).
						Int("count", len(webhooks)).
						Msg("Found unencrypted webhooks - migrating to encrypted storage")

					if err := rewriteEncryptedSliceLocked(c, c.webhookFile, webhooks, "legacy webhook migration rewrite"); err != nil {
						return nil, err
					}
					backupFile := legacyFile + ".backup"
					if err := c.fs.Rename(legacyFile, backupFile); err != nil {
						log.Warn().Err(err).Str("file", legacyFile).Msg("Failed to back up legacy webhooks file after migration")
					}
					return webhooks, nil
				}
			}
			// No webhooks file exists
			return []notifications.WebhookConfig{}, nil
		}
		return nil, err
	}

	// Decrypt if crypto manager is available
	if c.crypto != nil {
		decrypted, err := c.crypto.Decrypt(data)
		if err != nil {
			// Try parsing as plain JSON (migration case)
			var webhooks []notifications.WebhookConfig
			if jsonErr := json.Unmarshal(data, &webhooks); jsonErr == nil {
				webhooks = normalizeWebhookConfigs(webhooks)
				log.Info().
					Str("file", c.webhookFile).
					Int("count", len(webhooks)).
					Msg("Loaded unencrypted webhooks - rewriting encrypted storage")
				if err := rewriteEncryptedSliceLocked(c, c.webhookFile, webhooks, "webhook migration rewrite"); err != nil {
					return nil, err
				}
				return webhooks, nil
			}
			return nil, fmt.Errorf("failed to decrypt webhooks: %w", err)
		}
		data = decrypted
	}

	var webhooks []notifications.WebhookConfig
	if err := json.Unmarshal(data, &webhooks); err != nil {
		return nil, err
	}
	webhooks = normalizeWebhookConfigs(webhooks)

	log.Info().
		Str("file", c.webhookFile).
		Int("count", len(webhooks)).
		Bool("encrypted", c.crypto != nil).
		Msg("Webhooks loaded")
	return webhooks, nil
}

func normalizeWebhookConfigs(webhooks []notifications.WebhookConfig) []notifications.WebhookConfig {
	if len(webhooks) == 0 {
		return []notifications.WebhookConfig{}
	}

	normalized := make([]notifications.WebhookConfig, 0, len(webhooks))
	for _, webhook := range webhooks {
		normalized = append(normalized, notifications.NormalizeWebhookConfig(webhook))
	}
	return normalized
}

// MigrateWebhooksIfNeeded checks for legacy webhooks.json and migrates to encrypted format
func (c *ConfigPersistence) MigrateWebhooksIfNeeded() error {
	// Check if encrypted file already exists
	if _, err := c.fs.Stat(c.webhookFile); err == nil {
		// Encrypted file exists, no migration needed
		return nil
	}

	// Check for legacy unencrypted file
	legacyFile := filepath.Join(c.configDir, "webhooks.json")
	legacyData, err := c.fs.ReadFile(legacyFile)
	if err != nil {
		if os.IsNotExist(err) {
			// No legacy file, nothing to migrate
			return nil
		}
		return fmt.Errorf("failed to read legacy webhooks file: %w", err)
	}

	// Parse legacy webhooks
	var webhooks []notifications.WebhookConfig
	if err := json.Unmarshal(legacyData, &webhooks); err != nil {
		return fmt.Errorf("failed to parse legacy webhooks: %w", err)
	}

	log.Info().
		Str("from", legacyFile).
		Str("to", c.webhookFile).
		Int("count", len(webhooks)).
		Msg("Migrating webhooks to encrypted format")

	// Save to encrypted file
	if err := c.SaveWebhooks(webhooks); err != nil {
		return fmt.Errorf("failed to save encrypted webhooks: %w", err)
	}

	// Create backup of original file
	backupFile := legacyFile + ".backup"
	if err := c.fs.Rename(legacyFile, backupFile); err != nil {
		log.Warn().Err(err).Msg("Failed to rename legacy webhooks file to backup")
	} else {
		log.Info().Str("backup", backupFile).Msg("Legacy webhooks file backed up")
	}

	return nil
}

// NodesConfig represents the saved nodes configuration
type NodesConfig struct {
	PVEInstances []PVEInstance `json:"pveInstances"`
	PBSInstances []PBSInstance `json:"pbsInstances"`
	PMGInstances []PMGInstance `json:"pmgInstances"`
}

// SystemSettings represents system configuration settings
type SystemSettings struct {
	PVEPollingInterval           int             `json:"pvePollingInterval"` // PVE polling interval in seconds
	PBSPollingInterval           int             `json:"pbsPollingInterval"` // PBS polling interval in seconds
	PMGPollingInterval           int             `json:"pmgPollingInterval"` // PMG polling interval in seconds
	BackupPollingInterval        int             `json:"backupPollingInterval,omitempty"`
	BackupPollingEnabled         *bool           `json:"backupPollingEnabled,omitempty"`
	AdaptivePollingEnabled       *bool           `json:"adaptivePollingEnabled,omitempty"`
	AdaptivePollingBaseInterval  int             `json:"adaptivePollingBaseInterval,omitempty"`
	AdaptivePollingMinInterval   int             `json:"adaptivePollingMinInterval,omitempty"`
	AdaptivePollingMaxInterval   int             `json:"adaptivePollingMaxInterval,omitempty"`
	FrontendPort                 int             `json:"frontendPort,omitempty"`
	AllowedOrigins               string          `json:"allowedOrigins,omitempty"`
	ConnectionTimeout            int             `json:"connectionTimeout,omitempty"`
	UpdateChannel                string          `json:"updateChannel,omitempty"`
	AutoUpdateEnabled            bool            `json:"autoUpdateEnabled"` // Removed omitempty so false is saved
	AutoUpdateCheckInterval      int             `json:"autoUpdateCheckInterval,omitempty"`
	AutoUpdateTime               string          `json:"autoUpdateTime,omitempty"`
	LogLevel                     string          `json:"logLevel,omitempty"`
	DiscoveryEnabled             bool            `json:"discoveryEnabled"`
	DiscoverySubnet              string          `json:"discoverySubnet,omitempty"`
	DiscoveryConfig              DiscoveryConfig `json:"discoveryConfig"`
	Theme                        string          `json:"theme,omitempty"`               // User theme preference: "light", "dark", or empty for system default
	FullWidthMode                bool            `json:"fullWidthMode"`                 // Full-width layout mode preference
	AllowEmbedding               bool            `json:"allowEmbedding"`                // Allow iframe embedding
	AllowedEmbedOrigins          string          `json:"allowedEmbedOrigins,omitempty"` // Comma-separated list of allowed origins for embedding
	TemperatureMonitoringEnabled bool            `json:"temperatureMonitoringEnabled"`
	DNSCacheTimeout              int             `json:"dnsCacheTimeout,omitempty"`            // DNS cache timeout in seconds (0 = default 5 minutes)
	SSHPort                      int             `json:"sshPort,omitempty"`                    // Default SSH port for temperature monitoring (0 = use 22)
	WebhookAllowedPrivateCIDRs   string          `json:"webhookAllowedPrivateCIDRs,omitempty"` // Comma-separated list of private CIDR ranges allowed for webhooks (e.g., "192.168.1.0/24,10.0.0.0/8")
	HideLocalLogin               bool            `json:"hideLocalLogin"`                       // Hide local login form (username/password)
	PublicURL                    string          `json:"publicURL,omitempty"`                  // Public URL for email notifications (e.g., http://198.51.100.100:8080)

	// Metrics retention configuration (in hours)
	// These control how long historical metrics are stored at each aggregation tier.
	// Longer retention enables trend analysis but increases storage usage.
	MetricsRetentionRawHours    int `json:"metricsRetentionRawHours,omitempty"`    // Raw data (~5s intervals), default: 2 hours
	MetricsRetentionMinuteHours int `json:"metricsRetentionMinuteHours,omitempty"` // Minute averages, default: 24 hours
	MetricsRetentionHourlyDays  int `json:"metricsRetentionHourlyDays,omitempty"`  // Hourly averages, default: 7 days
	MetricsRetentionDailyDays   int `json:"metricsRetentionDailyDays,omitempty"`   // Daily averages, default: 90 days

	// Docker update control - server-wide settings
	// These allow admins to control Docker update features globally, addressing concerns
	// about Pulse being a "monitoring-first" tool vs an orchestration tool.
	DisableDockerUpdateActions bool `json:"disableDockerUpdateActions"` // Hide update buttons while still detecting updates

	// UX + privacy knobs (server-wide)
	ReduceProUpsellNoise       bool `json:"reduceProUpsellNoise,omitempty"`       // Hide proactive Pro prompts; paywalls still appear when accessing gated features
	DisableLocalUpgradeMetrics bool `json:"disableLocalUpgradeMetrics,omitempty"` // Disable local-only upgrade UX metrics collection (no network calls)

	// Telemetry (enabled by default, opt-out)
	TelemetryEnabled *bool `json:"telemetryEnabled,omitempty"` // Send anonymous usage telemetry (install ID, version, resource counts, feature flags — no PII)

	// APIToken is not persisted in system settings; API tokens are managed in api_tokens.json.
}

// DefaultSystemSettings returns a SystemSettings struct populated with sane defaults.
func DefaultSystemSettings() *SystemSettings {
	defaultDiscovery := DefaultDiscoveryConfig()
	return &SystemSettings{
		PVEPollingInterval:           10,
		PBSPollingInterval:           60,
		PMGPollingInterval:           60,
		AutoUpdateEnabled:            false,
		DiscoveryEnabled:             false,
		DiscoverySubnet:              "auto",
		DiscoveryConfig:              defaultDiscovery,
		AllowEmbedding:               false,
		TemperatureMonitoringEnabled: true,
		DNSCacheTimeout:              300, // Default: 5 minutes (300 seconds)
	}
}

// SaveNodesConfig saves nodes configuration to file (encrypted)
func (c *ConfigPersistence) SaveNodesConfig(pveInstances []PVEInstance, pbsInstances []PBSInstance, pmgInstances []PMGInstance) error {
	return c.saveNodesConfig(pveInstances, pbsInstances, pmgInstances, false)
}

// SaveNodesConfigAllowEmpty saves nodes configuration even when all nodes are removed.
// Use sparingly for explicit administrative actions (e.g. deleting the final node).
func (c *ConfigPersistence) SaveNodesConfigAllowEmpty(pveInstances []PVEInstance, pbsInstances []PBSInstance, pmgInstances []PMGInstance) error {
	return c.saveNodesConfig(pveInstances, pbsInstances, pmgInstances, true)
}

func (c *ConfigPersistence) saveNodesConfig(pveInstances []PVEInstance, pbsInstances []PBSInstance, pmgInstances []PMGInstance, allowEmpty bool) error {
	// CRITICAL: Prevent saving empty nodes when in mock mode
	// Mock mode should NEVER modify real node configuration
	if mock.IsMockEnabled() {
		log.Warn().Msg("Skipping nodes save - mock mode is enabled")
		return nil // Silently succeed to prevent errors but don't save
	}

	c.mu.Lock()
	defer c.mu.Unlock()

	// CRITICAL: Never save empty nodes configuration
	// This prevents data loss from accidental wipes
	if !allowEmpty && len(pveInstances) == 0 && len(pbsInstances) == 0 && len(pmgInstances) == 0 {
		// If we're replacing an existing non-empty config, block the wipe.
		// We must not call LoadNodesConfig here because it acquires c.mu again.
		if _, err := c.fs.Stat(c.nodesFile); err == nil {
			data, err := c.fs.ReadFile(c.nodesFile)
			if err != nil {
				return fmt.Errorf("refusing to save empty nodes config: failed to read existing nodes config: %w", err)
			}
			if c.crypto != nil {
				decrypted, err := c.crypto.Decrypt(data)
				if err != nil {
					return fmt.Errorf("refusing to save empty nodes config: existing nodes config is not decryptable: %w", err)
				}
				data = decrypted
			}

			var existing NodesConfig
			if err := json.Unmarshal(data, &existing); err != nil {
				return fmt.Errorf("refusing to save empty nodes config: existing nodes config is not parseable: %w", err)
			}

			if len(existing.PVEInstances) > 0 || len(existing.PBSInstances) > 0 || len(existing.PMGInstances) > 0 {
				log.Error().
					Int("existing_pve", len(existing.PVEInstances)).
					Int("existing_pbs", len(existing.PBSInstances)).
					Int("existing_pmg", len(existing.PMGInstances)).
					Msg("BLOCKED attempt to save empty nodes config - would delete existing nodes!")
				return fmt.Errorf("refusing to save empty nodes config when %d nodes exist",
					len(existing.PVEInstances)+len(existing.PBSInstances)+len(existing.PMGInstances))
			}
		}
	}

	config := NodesConfig{
		PVEInstances: pveInstances,
		PBSInstances: pbsInstances,
		PMGInstances: pmgInstances,
	}

	data, err := json.Marshal(config)
	if err != nil {
		return fmt.Errorf("marshal nodes config: %w", err)
	}

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for nodes config: %w", err)
	}

	// Create TIMESTAMPED backup of existing file before overwriting (if it exists and has content)
	// This ensures we keep multiple backups and can recover from disasters
	if info, err := c.fs.Stat(c.nodesFile); err == nil && info.Size() > 0 {
		// Create timestamped backup
		timestampedBackup := fmt.Sprintf("%s.backup-%s", c.nodesFile, time.Now().Format("20060102-150405"))
		if backupData, err := c.fs.ReadFile(c.nodesFile); err == nil {
			if err := c.fs.WriteFile(timestampedBackup, backupData, 0600); err != nil {
				log.Warn().Err(err).Msg("Failed to create timestamped backup of nodes config")
			} else {
				log.Info().Str("backup", timestampedBackup).Msg("Created timestamped backup of nodes config")
			}
		}

		// Also maintain a "latest" backup for quick recovery
		latestBackup := c.nodesFile + ".backup"
		if backupData, err := c.fs.ReadFile(c.nodesFile); err == nil {
			if err := c.fs.WriteFile(latestBackup, backupData, 0600); err != nil {
				log.Warn().Err(err).Msg("Failed to create latest backup of nodes config")
			}
		}

		// Clean up old timestamped backups (keep last 10)
		c.cleanupOldBackups(c.nodesFile + ".backup-*")
	}

	// Encrypt if crypto manager is available
	if c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(data)
		if err != nil {
			return fmt.Errorf("encrypt nodes config: %w", err)
		}
		data = encrypted
	}

	if err := c.writeConfigFileLocked(c.nodesFile, data, 0600); err != nil {
		return fmt.Errorf("persist nodes config: %w", err)
	}

	log.Info().Str("file", c.nodesFile).
		Int("pve", len(pveInstances)).
		Int("pbs", len(pbsInstances)).
		Int("pmg", len(pmgInstances)).
		Bool("encrypted", c.crypto != nil).
		Msg("Nodes configuration saved")
	return nil
}

// persistEmptyRecoveryConfig writes an empty NodesConfig to the nodes file.
// Used during recovery scenarios when the existing config is corrupted.
func (c *ConfigPersistence) persistEmptyRecoveryConfig(config *NodesConfig) {
	emptyData, marshalErr := json.Marshal(config)
	if marshalErr != nil {
		log.Error().Err(marshalErr).Msg("Failed to marshal empty recovery nodes config")
		return
	}

	if c.crypto != nil {
		encryptedData, encryptErr := c.crypto.Encrypt(emptyData)
		if encryptErr != nil {
			log.Error().Err(encryptErr).Msg("Failed to encrypt empty recovery nodes config")
			return
		}
		emptyData = encryptedData
	}

	if writeErr := c.fs.WriteFile(c.nodesFile, emptyData, 0600); writeErr != nil {
		log.Error().Err(writeErr).Str("file", c.nodesFile).Msg("Failed to persist empty recovery nodes config")
	}
}

// LoadNodesConfig loads nodes configuration from file (decrypts if encrypted)
func (c *ConfigPersistence) LoadNodesConfig() (*NodesConfig, error) {
	c.mu.RLock()
	unlocked := false
	defer func() {
		if !unlocked {
			c.mu.RUnlock()
		}
	}()

	data, err := c.fs.ReadFile(c.nodesFile)
	if err != nil {
		if os.IsNotExist(err) {
			// Return empty config if encrypted file doesn't exist
			log.Info().Msg("No encrypted nodes configuration found, returning empty config")
			return &NodesConfig{
				PVEInstances: []PVEInstance{},
				PBSInstances: []PBSInstance{},
				PMGInstances: []PMGInstance{},
			}, nil
		}
		return nil, fmt.Errorf("read nodes config: %w", err)
	}

	plaintextMigrated := false
	// Decrypt if crypto manager is available
	if c.crypto != nil {
		decrypted, err := c.crypto.Decrypt(data)
		if err != nil {
			// Accept plaintext nodes.enc only as migration input. If the payload still
			// parses, keep it in memory and let the canonical save path rewrite
			// encrypted storage below. Otherwise continue into corruption recovery.
			var plaintext NodesConfig
			if jsonErr := json.Unmarshal(data, &plaintext); jsonErr == nil {
				log.Warn().
					Err(err).
					Str("file", c.nodesFile).
					Msg("Failed to decrypt nodes config - treating plaintext file as migration input")
				plaintextMigrated = true
			} else {
				// Decryption failed - file may be corrupted
				log.Error().Err(err).Str("file", c.nodesFile).Msg("Failed to decrypt nodes config - file may be corrupted")

				// Try to restore from backup
				backupFile := c.nodesFile + ".backup"
				if backupData, backupErr := c.fs.ReadFile(backupFile); backupErr == nil {
					log.Info().Str("backup", backupFile).Msg("Attempting to restore nodes config from backup")
					if decryptedBackup, decryptErr := c.crypto.Decrypt(backupData); decryptErr == nil {
						log.Info().Msg("Successfully decrypted backup file")
						data = decryptedBackup

						// Move corrupted file out of the way with timestamp
						corruptedFile := fmt.Sprintf("%s.corrupted-%s", c.nodesFile, time.Now().Format("20060102-150405"))
						if renameErr := c.fs.Rename(c.nodesFile, corruptedFile); renameErr != nil {
							log.Warn().Err(renameErr).Msg("Failed to rename corrupted file")
						} else {
							log.Warn().Str("corruptedFile", corruptedFile).Msg("Moved corrupted nodes config")
						}

						// Restore backup as current file
						if writeErr := c.fs.WriteFile(c.nodesFile, backupData, 0600); writeErr != nil {
							log.Error().Err(writeErr).Msg("Failed to restore backup as current file")
						} else {
							log.Info().Msg("Successfully restored nodes config from backup")
						}
					} else {
						log.Error().Err(decryptErr).Msg("Backup file is also corrupted or encrypted with different key")

						// CRITICAL: Don't delete the corrupted file - leave it for manual recovery
						// Create an empty config so startup can continue, but log prominently
						log.Error().
							Str("corruptedFile", c.nodesFile).
							Str("backupFile", backupFile).
							Msg("CRITICAL: Both nodes.enc and backup are corrupted/unreadable. Encryption key may have been regenerated. Manual recovery required. Starting with empty config.")

						// Move corrupted file with timestamp for forensics
						corruptedFile := fmt.Sprintf("%s.corrupted-%s", c.nodesFile, time.Now().Format("20060102-150405"))
						if renameErr := c.fs.Rename(c.nodesFile, corruptedFile); renameErr != nil {
							log.Warn().Err(renameErr).Msg("Failed to rename corrupted nodes config during recovery")
						} else {
							log.Warn().Str("corruptedFile", corruptedFile).Msg("Moved corrupted nodes config")
						}

						// Create empty but valid config so system can start
						emptyConfig := NodesConfig{PVEInstances: []PVEInstance{}, PBSInstances: []PBSInstance{}, PMGInstances: []PMGInstance{}}
						c.persistEmptyRecoveryConfig(&emptyConfig)

						return &emptyConfig, nil
					}
				} else {
					log.Error().Err(backupErr).Msg("No backup file available for recovery")

					// CRITICAL: Don't delete the corrupted file - leave it for manual recovery
					log.Error().
						Str("corruptedFile", c.nodesFile).
						Msg("CRITICAL: nodes.enc is corrupted and no backup exists. Encryption key may have been regenerated. Manual recovery required. Starting with empty config.")

					// Move corrupted file with timestamp for forensics
					corruptedFile := fmt.Sprintf("%s.corrupted-%s", c.nodesFile, time.Now().Format("20060102-150405"))
					if renameErr := c.fs.Rename(c.nodesFile, corruptedFile); renameErr != nil {
						log.Warn().Err(renameErr).Msg("Failed to rename corrupted nodes config during recovery")
					} else {
						log.Warn().Str("corruptedFile", corruptedFile).Msg("Moved corrupted nodes config")
					}

					// Create empty but valid config so system can start
					emptyConfig := NodesConfig{PVEInstances: []PVEInstance{}, PBSInstances: []PBSInstance{}, PMGInstances: []PMGInstance{}}
					c.persistEmptyRecoveryConfig(&emptyConfig)

					return &emptyConfig, nil
				}
			}
		} else {
			data = decrypted
		}
	}

	var config NodesConfig
	if err := json.Unmarshal(data, &config); err != nil {
		return nil, fmt.Errorf("parse nodes config: %w", err)
	}

	if config.PVEInstances == nil {
		config.PVEInstances = []PVEInstance{}
	}
	if config.PBSInstances == nil {
		config.PBSInstances = []PBSInstance{}
	}
	if config.PMGInstances == nil {
		config.PMGInstances = []PMGInstance{}
	}

	// Track if any migrations were applied
	migrationApplied := plaintextMigrated

	normalizeHostWithDefault := func(host, defaultPort string) (string, bool) {
		normalized := normalizeHostPort(host, defaultPort)
		if normalized == "" || normalized == host {
			return host, false
		}
		return normalized, true
	}

	// Normalize PVE hosts (and cluster endpoints) to restore default ports for configs
	// saved during the 4.32.0 regression that skipped default port injection.
	for i := range config.PVEInstances {
		if normalized, changed := normalizeHostWithDefault(config.PVEInstances[i].Host, defaultPVEPort); changed {
			log.Info().
				Str("instance", config.PVEInstances[i].Name).
				Str("oldHost", config.PVEInstances[i].Host).
				Str("newHost", normalized).
				Msg("Normalized PVE host to include default port")
			config.PVEInstances[i].Host = normalized
			migrationApplied = true
		}

		for j := range config.PVEInstances[i].ClusterEndpoints {
			if normalized, changed := normalizeHostWithDefault(config.PVEInstances[i].ClusterEndpoints[j].Host, defaultPVEPort); changed {
				log.Info().
					Str("instance", config.PVEInstances[i].Name).
					Str("node", config.PVEInstances[i].ClusterEndpoints[j].NodeName).
					Str("oldHost", config.PVEInstances[i].ClusterEndpoints[j].Host).
					Str("newHost", normalized).
					Msg("Normalized PVE cluster endpoint host to include default port")
				config.PVEInstances[i].ClusterEndpoints[j].Host = normalized
				migrationApplied = true
			}
		}

		// Migration: Ensure MonitorBackups is enabled for PVE instances
		// This fixes issue #1139 where PVE backups weren't showing
		if !config.PVEInstances[i].MonitorBackups {
			log.Info().
				Str("instance", config.PVEInstances[i].Name).
				Msg("Enabling MonitorBackups for PVE instance (was disabled)")
			config.PVEInstances[i].MonitorBackups = true
			migrationApplied = true
		}
	}

	if consolidated, changed := ConsolidatePVEInstances(config.PVEInstances); changed {
		config.PVEInstances = consolidated
		migrationApplied = true
	}

	// Fix for bug where TokenName was incorrectly set when using password auth
	// If a PBS instance has both Password and TokenName, clear the TokenName
	for i := range config.PBSInstances {
		if config.PBSInstances[i].Password != "" && config.PBSInstances[i].TokenName != "" {
			log.Info().
				Str("instance", config.PBSInstances[i].Name).
				Msg("Fixing PBS config: clearing TokenName since Password is set")
			config.PBSInstances[i].TokenName = ""
			config.PBSInstances[i].TokenValue = ""
			migrationApplied = true
		}

		if normalized, changed := normalizeHostWithDefault(config.PBSInstances[i].Host, defaultPBSPort); changed {
			log.Info().
				Str("instance", config.PBSInstances[i].Name).
				Str("oldHost", config.PBSInstances[i].Host).
				Str("newHost", normalized).
				Msg("Normalized PBS host to include default port")
			config.PBSInstances[i].Host = normalized
			migrationApplied = true
		}

		// Migration: Ensure MonitorBackups is enabled for PBS instances
		// This fixes issue #411 where PBS backups weren't showing
		if !config.PBSInstances[i].MonitorBackups {
			log.Info().
				Str("instance", config.PBSInstances[i].Name).
				Msg("Enabling MonitorBackups for PBS instance (was disabled)")
			config.PBSInstances[i].MonitorBackups = true
			migrationApplied = true
		}

		// Migration: Ensure MonitorDatastores is enabled for PBS instances
		// Without this, PBS datastores aren't polled, which means backup data
		// from the direct PBS connection is never fetched — and PVE-side backup
		// polling skips PBS storages when a direct connection exists.
		if !config.PBSInstances[i].MonitorDatastores {
			log.Info().
				Str("instance", config.PBSInstances[i].Name).
				Msg("Enabling MonitorDatastores for PBS instance (was disabled)")
			config.PBSInstances[i].MonitorDatastores = true
			migrationApplied = true
		}
	}

	for i := range config.PMGInstances {
		if config.PMGInstances[i].Password != "" && config.PMGInstances[i].TokenName != "" {
			log.Info().
				Str("instance", config.PMGInstances[i].Name).
				Msg("Fixing PMG config: clearing TokenName since Password is set")
			config.PMGInstances[i].TokenName = ""
			config.PMGInstances[i].TokenValue = ""
			migrationApplied = true
		}

		if normalized, changed := normalizeHostWithDefault(config.PMGInstances[i].Host, defaultPVEPort); changed {
			log.Info().
				Str("instance", config.PMGInstances[i].Name).
				Str("oldHost", config.PMGInstances[i].Host).
				Str("newHost", normalized).
				Msg("Normalized PMG host to include default port")
			config.PMGInstances[i].Host = normalized
			migrationApplied = true
		}
	}

	log.Info().Str("file", c.nodesFile).
		Int("pve", len(config.PVEInstances)).
		Int("pbs", len(config.PBSInstances)).
		Int("pmg", len(config.PMGInstances)).
		Bool("encrypted", c.crypto != nil).
		Msg("Nodes configuration loaded")

	// If any migrations were applied, save the updated configuration after releasing the read lock
	// This prevents unlock/relock race condition where another goroutine could modify config
	// between unlock and relock, causing migrated data to be lost
	if migrationApplied {
		// Make copies while still holding read lock to ensure consistency
		pveCopy := make([]PVEInstance, len(config.PVEInstances))
		copy(pveCopy, config.PVEInstances)
		pbsCopy := make([]PBSInstance, len(config.PBSInstances))
		copy(pbsCopy, config.PBSInstances)
		pmgCopy := make([]PMGInstance, len(config.PMGInstances))
		copy(pmgCopy, config.PMGInstances)

		// Release read lock before saving (SaveNodesConfig acquires write lock)
		unlocked = true
		c.mu.RUnlock()

		log.Info().Msg("Migrations applied, saving updated configuration")
		if err := c.SaveNodesConfig(pveCopy, pbsCopy, pmgCopy); err != nil {
			log.Error().Err(err).Msg("Failed to save configuration after migration")
		}

		// Don't reacquire lock - we're returning
		return &config, nil
	}

	return &config, nil
}

// SaveSystemSettings saves system settings to file
func (c *ConfigPersistence) SaveSystemSettings(settings SystemSettings) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	data, err := json.Marshal(settings)
	if err != nil {
		return fmt.Errorf("marshal system settings: %w", err)
	}

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for system settings: %w", err)
	}

	if err := c.writeConfigFileLocked(c.systemFile, data, 0600); err != nil {
		return fmt.Errorf("persist system settings: %w", err)
	}

	// Also update the .env file if it exists
	envFile := filepath.Join(c.configDir, ".env")
	if err := c.updateEnvFile(envFile, settings); err != nil {
		log.Warn().Err(err).Msg("Failed to update .env file")
		// Don't fail the operation if .env update fails
	}

	log.Info().Str("file", c.systemFile).Msg("System settings saved")
	return nil
}

// SaveSSOConfig stores SSO settings, encrypting them when a crypto manager is available.
func (c *ConfigPersistence) SaveSSOConfig(settings *SSOConfig) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for sso config: %w", err)
	}

	if settings == nil {
		settings = NewSSOConfig()
	}

	// Clone to avoid modifying the original
	clone := settings.Clone()

	// Clear sensitive data from OIDC providers (env overrides)
	for i := range clone.Providers {
		if clone.Providers[i].OIDC != nil {
			clone.Providers[i].OIDC.EnvOverrides = nil
		}
	}

	data, err := json.Marshal(clone)
	if err != nil {
		return fmt.Errorf("marshal sso config: %w", err)
	}

	if c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(data)
		if err != nil {
			return fmt.Errorf("encrypt sso config: %w", err)
		}
		data = encrypted
	}

	if err := c.writeConfigFileLocked(c.ssoFile, data, 0600); err != nil {
		return fmt.Errorf("persist sso config: %w", err)
	}

	log.Info().Str("file", c.ssoFile).Int("providers", len(clone.Providers)).Msg("SSO configuration saved")
	return nil
}

// LoadSSOConfig retrieves the persisted SSO settings. It returns nil when no configuration exists yet.
func (c *ConfigPersistence) LoadSSOConfig() (*SSOConfig, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	var settings SSOConfig
	migratedPlaintext, _, err := loadEncryptedJSONLocked(c, c.ssoFile, &settings, "sso config")
	if err != nil {
		if os.IsNotExist(err) {
			return nil, nil
		}
		return nil, err
	}

	if migratedPlaintext {
		jsonData, err := json.Marshal(settings)
		if err != nil {
			return nil, fmt.Errorf("marshal sso config migration rewrite: %w", err)
		}
		if err := rewriteEncryptedJSONLocked(c, c.ssoFile, jsonData, "sso config migration rewrite"); err != nil {
			return nil, err
		}
	}

	log.Info().Str("file", c.ssoFile).Int("providers", len(settings.Providers)).Msg("SSO configuration loaded")
	return &settings, nil
}

// SaveAIConfig stores AI settings, encrypting them when a crypto manager is available.
func (c *ConfigPersistence) SaveAIConfig(settings AIConfig) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	settings.NormalizePatrolEventTriggerSettings()
	settings.NormalizeQuickstartModelAliases()

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for ai config: %w", err)
	}

	data, err := json.Marshal(settings)
	if err != nil {
		return fmt.Errorf("marshal ai config: %w", err)
	}

	if c.crypto != nil {
		encrypted, err := c.crypto.Encrypt(data)
		if err != nil {
			return fmt.Errorf("encrypt ai config: %w", err)
		}
		data = encrypted
	}

	if err := c.writeConfigFileLocked(c.aiFile, data, 0600); err != nil {
		return fmt.Errorf("persist ai config: %w", err)
	}

	log.Info().Str("file", c.aiFile).Bool("enabled", settings.Enabled).Msg("AI configuration saved")
	return nil
}

// LoadAIConfig retrieves the persisted AI settings. It returns default config when no configuration exists yet.
func (c *ConfigPersistence) LoadAIConfig() (*AIConfig, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	data, err := c.fs.ReadFile(c.aiFile)
	if err != nil {
		if os.IsNotExist(err) {
			// Return default config if file doesn't exist
			return NewDefaultAIConfig(), nil
		}
		return nil, err
	}

	migratedPlaintext := false
	if c.crypto != nil {
		decrypted, err := c.crypto.Decrypt(data)
		if err != nil {
			migratedPlaintext = true
		} else {
			data = decrypted
		}
	}

	var legacyRaw map[string]json.RawMessage
	_ = json.Unmarshal(data, &legacyRaw)

	// Start with defaults so new fields get proper values
	settings := NewDefaultAIConfig()
	if err := json.Unmarshal(data, settings); err != nil {
		return nil, err
	}

	// Migration: Ensure patrol settings have sensible defaults for existing configs
	// PatrolIntervalMinutes=0 means it was never set - use default
	if settings.PatrolIntervalMinutes <= 0 {
		if !settings.PatrolEnabled {
			settings.PatrolIntervalMinutes = 0
		} else {
			settings.PatrolIntervalMinutes = 360
		}
	}

	_, alertTriggerFieldPresent := legacyRaw["patrol_alert_triggers_enabled"]
	_, anomalyTriggerFieldPresent := legacyRaw["patrol_anomaly_triggers_enabled"]
	legacyEventTriggerFieldRaw, legacyEventTriggerFieldPresent := legacyRaw["patrol_event_triggers_enabled"]
	migratedPatrolTriggerFields := false
	if alertTriggerFieldPresent || anomalyTriggerFieldPresent {
		// Transitional configs could persist only the enabled split flag when false
		// values were omitted from disk. Once either canonical split field exists,
		// treat any missing sibling as an explicit false rather than inheriting the
		// legacy aggregate toggle back into the split model.
		if !alertTriggerFieldPresent {
			settings.PatrolAlertTriggersEnabled = false
			migratedPatrolTriggerFields = true
		}
		if !anomalyTriggerFieldPresent {
			settings.PatrolAnomalyTriggersEnabled = false
			migratedPatrolTriggerFields = true
		}
	} else if legacyEventTriggerFieldPresent {
		legacyEnabled, _ := decodeOptionalJSONBool(legacyEventTriggerFieldRaw)
		settings.PatrolAlertTriggersEnabled = legacyEnabled
		settings.PatrolAnomalyTriggersEnabled = legacyEnabled
		migratedPatrolTriggerFields = true
	}
	if alertTriggerFieldPresent || anomalyTriggerFieldPresent || legacyEventTriggerFieldPresent {
		if settings.NormalizePatrolEventTriggerSettings() {
			migratedPatrolTriggerFields = true
		}
	}

	migratedLegacyFields := applyLegacyAIConfigFields(settings, legacyRaw)
	migratedControlLevel := false
	if settings.ControlLevel == "suggest" {
		settings.ControlLevel = ControlLevelControlled
		migratedControlLevel = true
	}
	migratedQuickstartAliases := settings.NormalizeQuickstartModelAliases()

	if migratedPlaintext || migratedLegacyFields || migratedControlLevel || migratedPatrolTriggerFields || migratedQuickstartAliases {
		jsonData, err := json.Marshal(*settings)
		if err != nil {
			return nil, fmt.Errorf("marshal ai config migration rewrite: %w", err)
		}
		if err := rewriteEncryptedJSONLocked(c, c.aiFile, jsonData, "ai config migration rewrite"); err != nil {
			return nil, err
		}
		log.Info().
			Str("control_level", settings.ControlLevel).
			Bool("legacy_fields_migrated", migratedLegacyFields).
			Bool("patrol_trigger_fields_migrated", migratedPatrolTriggerFields).
			Bool("quickstart_aliases_migrated", migratedQuickstartAliases).
			Bool("plaintext_migrated", migratedPlaintext).
			Msg("Migrated AI configuration")
	}

	log.Info().Str("file", c.aiFile).Bool("enabled", settings.Enabled).Bool("patrol_enabled", settings.PatrolEnabled).Bool("alert_triggered_analysis", settings.AlertTriggeredAnalysis).Msg("AI configuration loaded")
	return settings, nil
}

// HasAIConfig reports whether ai.enc exists on disk, which distinguishes
// explicit operator configuration from synthesized defaults returned by
// LoadAIConfig when the file is absent.
func (c *ConfigPersistence) HasAIConfig() bool {
	if c == nil || c.fs == nil {
		return false
	}
	_, err := c.fs.Stat(c.aiFile)
	return err == nil
}

func applyLegacyAIConfigFields(settings *AIConfig, raw map[string]json.RawMessage) bool {
	if settings == nil || len(raw) == 0 {
		return false
	}

	legacyProvider := decodeOptionalJSONString(raw["provider"])
	legacyAPIKey := decodeOptionalJSONString(raw["api_key"])
	legacyBaseURL := decodeOptionalJSONString(raw["base_url"])
	legacyAutonomousMode, hasLegacyAutonomous := decodeOptionalJSONBool(raw["autonomous_mode"])

	migrated := false

	if legacyAPIKey != "" {
		switch {
		case settings.AnthropicAPIKey == "" && shouldUseLegacyProviderCredential(settings, legacyProvider, AIProviderAnthropic):
			settings.AnthropicAPIKey = legacyAPIKey
			migrated = true
		case settings.OpenAIAPIKey == "" && shouldUseLegacyProviderCredential(settings, legacyProvider, AIProviderOpenAI):
			settings.OpenAIAPIKey = legacyAPIKey
			migrated = true
		case settings.OpenRouterAPIKey == "" && shouldUseLegacyProviderCredential(settings, legacyProvider, AIProviderOpenRouter):
			settings.OpenRouterAPIKey = legacyAPIKey
			migrated = true
		case settings.DeepSeekAPIKey == "" && shouldUseLegacyProviderCredential(settings, legacyProvider, AIProviderDeepSeek):
			settings.DeepSeekAPIKey = legacyAPIKey
			migrated = true
		case settings.GeminiAPIKey == "" && shouldUseLegacyProviderCredential(settings, legacyProvider, AIProviderGemini):
			settings.GeminiAPIKey = legacyAPIKey
			migrated = true
		}
	}

	if legacyBaseURL != "" {
		switch {
		case settings.OpenAIBaseURL == "" && shouldUseLegacyProviderCredential(settings, legacyProvider, AIProviderOpenAI):
			settings.OpenAIBaseURL = legacyBaseURL
			migrated = true
		case settings.OllamaBaseURL == "" && shouldUseLegacyProviderCredential(settings, legacyProvider, AIProviderOllama):
			settings.OllamaBaseURL = legacyBaseURL
			migrated = true
		}
	}

	if hasLegacyAutonomous && legacyAutonomousMode && settings.ControlLevel == "" {
		settings.ControlLevel = ControlLevelAutonomous
		migrated = true
	}

	return migrated
}

func shouldUseLegacyProviderCredential(settings *AIConfig, legacyProvider, targetProvider string) bool {
	if settings == nil {
		return false
	}
	if strings.TrimSpace(legacyProvider) != "" {
		return strings.EqualFold(strings.TrimSpace(legacyProvider), targetProvider)
	}
	modelProvider, _ := ParseModelString(settings.Model)
	return modelProvider == targetProvider
}

func decodeOptionalJSONString(raw json.RawMessage) string {
	if len(raw) == 0 {
		return ""
	}
	var value string
	if err := json.Unmarshal(raw, &value); err != nil {
		return ""
	}
	return strings.TrimSpace(value)
}

func decodeOptionalJSONBool(raw json.RawMessage) (bool, bool) {
	if len(raw) == 0 {
		return false, false
	}
	var value bool
	if err := json.Unmarshal(raw, &value); err != nil {
		return false, false
	}
	return value, true
}

// AIFindingsData represents persisted AI findings with metadata
type AIFindingsData struct {
	// Version for future migrations
	Version int `json:"version"`
	// LastSaved for debugging/diagnostics
	LastSaved time.Time `json:"last_saved"`
	// Findings is a map of finding ID to finding data
	Findings map[string]*AIFindingRecord `json:"findings"`
	// SuppressionRules stores explicit suppression rules created by the user (or derived from a "suppress" action).
	// These must persist across restarts; they are distinct from dismissed findings.
	SuppressionRules map[string]*AISuppressionRuleRecord `json:"suppression_rules,omitempty"`
}

// AISuppressionRuleRecord is a persisted suppression rule.
type AISuppressionRuleRecord struct {
	ID              string    `json:"id"`
	ResourceID      string    `json:"resource_id,omitempty"`      // Empty means "any resource"
	ResourceName    string    `json:"resource_name,omitempty"`    // Human-readable name for display
	Category        string    `json:"category,omitempty"`         // Empty means "any category"
	Description     string    `json:"description"`                // User's reason
	DismissedReason string    `json:"dismissed_reason,omitempty"` // Optional
	CreatedAt       time.Time `json:"created_at"`
	CreatedFrom     string    `json:"created_from,omitempty"` // "finding"/"dismissed"/"manual"/"suppress"
	FindingID       string    `json:"finding_id,omitempty"`   // Original finding ID (if derived)
}

// AIFindingRecord is a persisted finding with full history
type AIFindingRecord struct {
	ID              string     `json:"id"`
	Key             string     `json:"key,omitempty"`
	Severity        string     `json:"severity"`
	Category        string     `json:"category"`
	ResourceID      string     `json:"resource_id"`
	ResourceName    string     `json:"resource_name"`
	ResourceType    string     `json:"resource_type"`
	Node            string     `json:"node,omitempty"`
	Title           string     `json:"title"`
	Description     string     `json:"description"`
	Recommendation  string     `json:"recommendation,omitempty"`
	Evidence        string     `json:"evidence,omitempty"`
	Source          string     `json:"source,omitempty"`
	DetectedAt      time.Time  `json:"detected_at"`
	LastSeenAt      time.Time  `json:"last_seen_at"`
	ResolvedAt      *time.Time `json:"resolved_at,omitempty"`
	AutoResolved    bool       `json:"auto_resolved"`
	ResolveReason   string     `json:"resolve_reason,omitempty"`
	AcknowledgedAt  *time.Time `json:"acknowledged_at,omitempty"`
	SnoozedUntil    *time.Time `json:"snoozed_until,omitempty"`
	AlertIdentifier string     `json:"-"`
	// User feedback fields - enables persistence of dismissal state
	DismissedReason string `json:"dismissed_reason,omitempty"`
	UserNote        string `json:"user_note,omitempty"`
	TimesRaised     int    `json:"times_raised"`
	Suppressed      bool   `json:"suppressed"`

	// Investigation fields - tracks autonomous AI investigation of findings
	InvestigationSessionID string     `json:"investigation_session_id,omitempty"`
	InvestigationStatus    string     `json:"investigation_status,omitempty"`
	InvestigationOutcome   string     `json:"investigation_outcome,omitempty"`
	LastInvestigatedAt     *time.Time `json:"last_investigated_at,omitempty"`
	InvestigationAttempts  int        `json:"investigation_attempts"`
	LoopState              string     `json:"loop_state,omitempty"`
	Lifecycle              []struct {
		At       time.Time         `json:"at"`
		Type     string            `json:"type"`
		Message  string            `json:"message,omitempty"`
		From     string            `json:"from,omitempty"`
		To       string            `json:"to,omitempty"`
		Metadata map[string]string `json:"metadata,omitempty"`
	} `json:"lifecycle,omitempty"`
	RegressionCount  int        `json:"regression_count,omitempty"`
	LastRegressionAt *time.Time `json:"last_regression_at,omitempty"`
}

type aiFindingRecordAlias AIFindingRecord

type aiFindingRecordJSON struct {
	aiFindingRecordAlias
	AlertIdentifier string `json:"alert_identifier,omitempty"`
}

func (r AIFindingRecord) MarshalJSON() ([]byte, error) {
	return json.Marshal(aiFindingRecordJSON{
		aiFindingRecordAlias: aiFindingRecordAlias(r),
		AlertIdentifier:      strings.TrimSpace(r.AlertIdentifier),
	})
}

func (r *AIFindingRecord) UnmarshalJSON(data []byte) error {
	var payload aiFindingRecordJSON
	if err := json.Unmarshal(data, &payload); err != nil {
		return err
	}

	*r = AIFindingRecord(payload.aiFindingRecordAlias)
	r.AlertIdentifier = strings.TrimSpace(payload.AlertIdentifier)
	return nil
}

// SaveAIFindings persists AI findings to disk, preserving any existing suppression rules.
func (c *ConfigPersistence) SaveAIFindings(findings map[string]*AIFindingRecord) error {
	return c.SaveAIFindingsWithSuppression(findings, nil)
}

// SaveAIFindingsWithSuppression persists AI findings + explicit suppression rules to disk.
// If suppressionRules is nil, existing suppression rules (if any) are preserved.
func (c *ConfigPersistence) SaveAIFindingsWithSuppression(findings map[string]*AIFindingRecord, suppressionRules map[string]*AISuppressionRuleRecord) error {
	// Preserve suppression rules if caller didn't provide them.
	if suppressionRules == nil {
		if existing, err := c.LoadAIFindings(); err == nil && existing != nil && existing.SuppressionRules != nil {
			suppressionRules = existing.SuppressionRules
		}
	}
	if findings == nil {
		findings = make(map[string]*AIFindingRecord)
	}

	data := AIFindingsData{
		Version:          3, // Bumped from 2: persisted suppression rules alongside findings
		LastSaved:        time.Now(),
		Findings:         findings,
		SuppressionRules: suppressionRules,
	}

	if err := saveJSON(c, c.aiFindingsFile, data, true); err != nil {
		return err
	}

	log.Debug().
		Str("file", c.aiFindingsFile).
		Int("count", len(findings)).
		Int("suppression_rules", len(suppressionRules)).
		Msg("AI findings saved")
	return nil
}

// SetFileSystem allows injecting a mock file system for testing
func (c *ConfigPersistence) SetFileSystem(fs FileSystem) {
	c.mu.Lock()
	defer c.mu.Unlock()
	c.fs = fs
}

// LoadAIFindings loads AI findings from disk
func (c *ConfigPersistence) LoadAIFindings() (*AIFindingsData, error) {
	// Read file under read lock
	c.mu.RLock()
	data, err := c.fs.ReadFile(c.aiFindingsFile)
	c.mu.RUnlock()

	if err != nil {
		if os.IsNotExist(err) {
			// Return empty data if file doesn't exist
			return &AIFindingsData{
				Version:          3,
				Findings:         make(map[string]*AIFindingRecord),
				SuppressionRules: make(map[string]*AISuppressionRuleRecord),
			}, nil
		}
		return nil, err
	}

	migratedPlaintext := false
	// Attempt decryption; fall back to plaintext for migration from unencrypted files.
	if c.crypto != nil {
		if decrypted, decErr := c.crypto.Decrypt(data); decErr == nil {
			data = decrypted
		} else {
			migratedPlaintext = true
		}
	}

	var findingsData AIFindingsData
	if err := json.Unmarshal(data, &findingsData); err != nil {
		log.Error().Err(err).Str("file", c.aiFindingsFile).Msg("Failed to parse AI findings file")
		// Return empty data on parse error rather than failing
		return &AIFindingsData{
			Version:          3,
			Findings:         make(map[string]*AIFindingRecord),
			SuppressionRules: make(map[string]*AISuppressionRuleRecord),
		}, nil
	}

	if findingsData.Findings == nil {
		findingsData.Findings = make(map[string]*AIFindingRecord)
	}
	if findingsData.SuppressionRules == nil {
		findingsData.SuppressionRules = make(map[string]*AISuppressionRuleRecord)
	}

	// Version 2 changed finding ID format to include issue key.
	// Clear old findings since IDs are incompatible, and persist immediately.
	if findingsData.Version < 2 {
		oldCount := len(findingsData.Findings)
		findingsData.Findings = make(map[string]*AIFindingRecord)
		findingsData.SuppressionRules = make(map[string]*AISuppressionRuleRecord)
		findingsData.Version = 3
		findingsData.LastSaved = time.Now()

		if oldCount > 0 {
			log.Info().
				Int("cleared_count", oldCount).
				Msg("AI findings cleared due to schema upgrade (v1 -> v3)")
		}

		// Persist the migrated (empty) file immediately to avoid re-migrating on restart
		c.mu.Lock()
		if jsonData, err := json.Marshal(findingsData); err == nil {
			if c.crypto != nil {
				encrypted, encErr := c.crypto.Encrypt(jsonData)
				if encErr != nil {
					log.Warn().Err(encErr).Msg("Failed to encrypt migrated AI findings — skipping write to avoid plaintext storage")
					c.mu.Unlock()
					return &findingsData, nil
				}
				jsonData = encrypted
			}
			if err := c.writeConfigFileLocked(c.aiFindingsFile, jsonData, 0600); err != nil {
				log.Warn().Err(err).Msg("Failed to persist migrated AI findings file")
			}
		}
		c.mu.Unlock()
	} else if findingsData.Version < 3 {
		// v2 -> v3: keep findings; start persisting explicit suppression rules.
		findingsData.Version = 3
		findingsData.LastSaved = time.Now()
		c.mu.Lock()
		if jsonData, err := json.Marshal(findingsData); err == nil {
			if c.crypto != nil {
				encrypted, encErr := c.crypto.Encrypt(jsonData)
				if encErr != nil {
					log.Warn().Err(encErr).Msg("Failed to encrypt migrated AI findings — skipping write to avoid plaintext storage")
					c.mu.Unlock()
					return &findingsData, nil
				}
				jsonData = encrypted
			}
			if err := c.writeConfigFileLocked(c.aiFindingsFile, jsonData, 0600); err != nil {
				log.Warn().Err(err).Msg("Failed to persist migrated AI findings file")
			}
		}
		c.mu.Unlock()
	} else if migratedPlaintext {
		c.mu.Lock()
		if jsonData, err := json.Marshal(findingsData); err == nil {
			if err := rewriteEncryptedJSONLocked(c, c.aiFindingsFile, jsonData, "ai findings migration rewrite"); err != nil {
				log.Warn().Err(err).Msg("Failed to persist encrypted AI findings migration rewrite")
			}
		}
		c.mu.Unlock()
	}

	log.Info().
		Str("file", c.aiFindingsFile).
		Int("count", len(findingsData.Findings)).
		Int("suppression_rules", len(findingsData.SuppressionRules)).
		Time("last_saved", findingsData.LastSaved).
		Msg("AI findings loaded")
	return &findingsData, nil
}

// PatrolRunHistoryData represents persisted patrol run history with metadata
type PatrolRunHistoryData struct {
	Version   int               `json:"version"`
	LastSaved time.Time         `json:"last_saved"`
	Runs      []PatrolRunRecord `json:"runs"`
}

// PatrolRunRecord represents a single patrol check run
type PatrolRunRecord struct {
	ID                        string    `json:"id"`
	StartedAt                 time.Time `json:"started_at"`
	CompletedAt               time.Time `json:"completed_at"`
	DurationMs                int64     `json:"duration_ms"`
	Type                      string    `json:"type"` // "quick" or "deep"
	TriggerReason             string    `json:"trigger_reason,omitempty"`
	ScopeResourceIDs          []string  `json:"scope_resource_ids,omitempty"`
	EffectiveScopeResourceIDs []string  `json:"effective_scope_resource_ids,omitempty"`
	ScopeResourceTypes        []string  `json:"scope_resource_types,omitempty"`
	ScopeContext              string    `json:"scope_context,omitempty"`
	AlertIdentifier           string    `json:"alert_identifier,omitempty"`
	FindingID                 string    `json:"finding_id,omitempty"`
	ResourcesChecked          int       `json:"resources_checked"`
	// Breakdown by resource type
	NodesChecked      int `json:"nodes_checked"`
	GuestsChecked     int `json:"guests_checked"`
	DockerChecked     int `json:"docker_checked"`
	StorageChecked    int `json:"storage_checked"`
	HostsChecked      int `json:"hosts_checked"`
	TrueNASChecked    int `json:"truenas_checked"`
	PBSChecked        int `json:"pbs_checked"`
	PMGChecked        int `json:"pmg_checked"`
	KubernetesChecked int `json:"kubernetes_checked"`
	// Findings from this run
	NewFindings      int      `json:"new_findings"`
	ExistingFindings int      `json:"existing_findings"`
	RejectedFindings int      `json:"rejected_findings"`
	ResolvedFindings int      `json:"resolved_findings"`
	AutoFixCount     int      `json:"auto_fix_count,omitempty"`
	FindingsSummary  string   `json:"findings_summary"`
	FindingIDs       []string `json:"finding_ids"`
	ErrorCount       int      `json:"error_count"`
	Status           string   `json:"status"` // "healthy", "issues_found", "critical", "error"
	// Triage stats
	TriageFlags      int  `json:"triage_flags"`
	TriageSkippedLLM bool `json:"triage_skipped_llm,omitempty"`
	// AI Analysis details
	AIAnalysis   string `json:"ai_analysis,omitempty"`   // The AI's raw response/analysis
	InputTokens  int    `json:"input_tokens,omitempty"`  // Tokens sent to AI
	OutputTokens int    `json:"output_tokens,omitempty"` // Tokens received from AI
	// Tool call traces
	ToolCalls     []ToolCallRecord `json:"tool_calls,omitempty"`
	ToolCallCount int              `json:"tool_call_count"`
}

type patrolRunRecordJSON struct {
	ID                        string           `json:"id"`
	StartedAt                 time.Time        `json:"started_at"`
	CompletedAt               time.Time        `json:"completed_at"`
	DurationMs                int64            `json:"duration_ms"`
	Type                      string           `json:"type"`
	TriggerReason             string           `json:"trigger_reason,omitempty"`
	ScopeResourceIDs          *[]string        `json:"scope_resource_ids,omitempty"`
	EffectiveScopeResourceIDs *[]string        `json:"effective_scope_resource_ids,omitempty"`
	ScopeResourceTypes        *[]string        `json:"scope_resource_types,omitempty"`
	ScopeContext              string           `json:"scope_context,omitempty"`
	AlertIdentifier           string           `json:"alert_identifier,omitempty"`
	FindingID                 string           `json:"finding_id,omitempty"`
	ResourcesChecked          int              `json:"resources_checked"`
	NodesChecked              int              `json:"nodes_checked"`
	GuestsChecked             int              `json:"guests_checked"`
	DockerChecked             int              `json:"docker_checked"`
	StorageChecked            int              `json:"storage_checked"`
	HostsChecked              int              `json:"hosts_checked"`
	TrueNASChecked            int              `json:"truenas_checked"`
	PBSChecked                int              `json:"pbs_checked"`
	PMGChecked                int              `json:"pmg_checked"`
	KubernetesChecked         int              `json:"kubernetes_checked"`
	NewFindings               int              `json:"new_findings"`
	ExistingFindings          int              `json:"existing_findings"`
	RejectedFindings          int              `json:"rejected_findings"`
	ResolvedFindings          int              `json:"resolved_findings"`
	AutoFixCount              int              `json:"auto_fix_count,omitempty"`
	FindingsSummary           string           `json:"findings_summary"`
	FindingIDs                []string         `json:"finding_ids"`
	ErrorCount                int              `json:"error_count"`
	Status                    string           `json:"status"`
	TriageFlags               int              `json:"triage_flags"`
	TriageSkippedLLM          bool             `json:"triage_skipped_llm,omitempty"`
	AIAnalysis                string           `json:"ai_analysis,omitempty"`
	InputTokens               int              `json:"input_tokens,omitempty"`
	OutputTokens              int              `json:"output_tokens,omitempty"`
	ToolCalls                 []ToolCallRecord `json:"tool_calls,omitempty"`
	ToolCallCount             int              `json:"tool_call_count"`
}

func canonicalPatrolAlertIdentifier(alertIdentifier string) string {
	return strings.TrimSpace(alertIdentifier)
}

func marshalOptionalPatrolStringSlice(values []string) *[]string {
	if values == nil {
		return nil
	}
	cloned := append([]string{}, values...)
	return &cloned
}

func unmarshalOptionalPatrolStringSlice(values *[]string) []string {
	if values == nil {
		return nil
	}
	return append([]string{}, (*values)...)
}

func canonicalPatrolFindingIDs(ids []string) []string {
	if ids == nil {
		return []string{}
	}
	return append([]string{}, ids...)
}

func normalizePatrolRunRecord(record PatrolRunRecord) PatrolRunRecord {
	alertIdentifier := canonicalPatrolAlertIdentifier(record.AlertIdentifier)
	record.AlertIdentifier = alertIdentifier
	record.FindingIDs = canonicalPatrolFindingIDs(record.FindingIDs)
	return record
}

func (r PatrolRunRecord) MarshalJSON() ([]byte, error) {
	normalized := normalizePatrolRunRecord(r)
	alertIdentifier := strings.TrimSpace(normalized.AlertIdentifier)
	return json.Marshal(patrolRunRecordJSON{
		ID:                        normalized.ID,
		StartedAt:                 normalized.StartedAt,
		CompletedAt:               normalized.CompletedAt,
		DurationMs:                normalized.DurationMs,
		Type:                      normalized.Type,
		TriggerReason:             normalized.TriggerReason,
		ScopeResourceIDs:          marshalOptionalPatrolStringSlice(normalized.ScopeResourceIDs),
		EffectiveScopeResourceIDs: marshalOptionalPatrolStringSlice(normalized.EffectiveScopeResourceIDs),
		ScopeResourceTypes:        marshalOptionalPatrolStringSlice(normalized.ScopeResourceTypes),
		ScopeContext:              normalized.ScopeContext,
		AlertIdentifier:           alertIdentifier,
		FindingID:                 normalized.FindingID,
		ResourcesChecked:          normalized.ResourcesChecked,
		NodesChecked:              normalized.NodesChecked,
		GuestsChecked:             normalized.GuestsChecked,
		DockerChecked:             normalized.DockerChecked,
		StorageChecked:            normalized.StorageChecked,
		HostsChecked:              normalized.HostsChecked,
		TrueNASChecked:            normalized.TrueNASChecked,
		PBSChecked:                normalized.PBSChecked,
		PMGChecked:                normalized.PMGChecked,
		KubernetesChecked:         normalized.KubernetesChecked,
		NewFindings:               normalized.NewFindings,
		ExistingFindings:          normalized.ExistingFindings,
		RejectedFindings:          normalized.RejectedFindings,
		ResolvedFindings:          normalized.ResolvedFindings,
		AutoFixCount:              normalized.AutoFixCount,
		FindingsSummary:           normalized.FindingsSummary,
		FindingIDs:                normalized.FindingIDs,
		ErrorCount:                normalized.ErrorCount,
		Status:                    normalized.Status,
		TriageFlags:               normalized.TriageFlags,
		TriageSkippedLLM:          normalized.TriageSkippedLLM,
		AIAnalysis:                normalized.AIAnalysis,
		InputTokens:               normalized.InputTokens,
		OutputTokens:              normalized.OutputTokens,
		ToolCalls:                 normalized.ToolCalls,
		ToolCallCount:             normalized.ToolCallCount,
	})
}

func (r *PatrolRunRecord) UnmarshalJSON(data []byte) error {
	var payload patrolRunRecordJSON
	if err := json.Unmarshal(data, &payload); err != nil {
		return err
	}

	alertIdentifier := canonicalPatrolAlertIdentifier(payload.AlertIdentifier)
	*r = PatrolRunRecord{
		ID:                        payload.ID,
		StartedAt:                 payload.StartedAt,
		CompletedAt:               payload.CompletedAt,
		DurationMs:                payload.DurationMs,
		Type:                      payload.Type,
		TriggerReason:             payload.TriggerReason,
		ScopeResourceIDs:          unmarshalOptionalPatrolStringSlice(payload.ScopeResourceIDs),
		EffectiveScopeResourceIDs: unmarshalOptionalPatrolStringSlice(payload.EffectiveScopeResourceIDs),
		ScopeResourceTypes:        unmarshalOptionalPatrolStringSlice(payload.ScopeResourceTypes),
		ScopeContext:              payload.ScopeContext,
		AlertIdentifier:           alertIdentifier,
		FindingID:                 payload.FindingID,
		ResourcesChecked:          payload.ResourcesChecked,
		NodesChecked:              payload.NodesChecked,
		GuestsChecked:             payload.GuestsChecked,
		DockerChecked:             payload.DockerChecked,
		StorageChecked:            payload.StorageChecked,
		HostsChecked:              payload.HostsChecked,
		TrueNASChecked:            payload.TrueNASChecked,
		PBSChecked:                payload.PBSChecked,
		PMGChecked:                payload.PMGChecked,
		KubernetesChecked:         payload.KubernetesChecked,
		NewFindings:               payload.NewFindings,
		ExistingFindings:          payload.ExistingFindings,
		RejectedFindings:          payload.RejectedFindings,
		ResolvedFindings:          payload.ResolvedFindings,
		AutoFixCount:              payload.AutoFixCount,
		FindingsSummary:           payload.FindingsSummary,
		FindingIDs:                payload.FindingIDs,
		ErrorCount:                payload.ErrorCount,
		Status:                    payload.Status,
		TriageFlags:               payload.TriageFlags,
		TriageSkippedLLM:          payload.TriageSkippedLLM,
		AIAnalysis:                payload.AIAnalysis,
		InputTokens:               payload.InputTokens,
		OutputTokens:              payload.OutputTokens,
		ToolCalls:                 payload.ToolCalls,
		ToolCallCount:             payload.ToolCallCount,
	}
	return nil
}

// ToolCallRecord captures a single tool invocation during a patrol run.
type ToolCallRecord struct {
	ID        string `json:"id"`
	ToolName  string `json:"tool_name"`
	Input     string `json:"input"`
	Output    string `json:"output"`
	Success   bool   `json:"success"`
	StartTime int64  `json:"start_time"`
	EndTime   int64  `json:"end_time"`
	Duration  int64  `json:"duration_ms"`
}

// AIUsageHistoryData represents persisted AI usage history with metadata
type AIUsageHistoryData struct {
	Version   int                  `json:"version"`
	LastSaved time.Time            `json:"last_saved"`
	Events    []AIUsageEventRecord `json:"events"`
}

// AIUsageEventRecord is a persisted usage event for an AI provider call.
// This intentionally excludes prompt/response content for privacy.
type AIUsageEventRecord struct {
	Timestamp     time.Time `json:"timestamp"`
	Provider      string    `json:"provider"`
	RequestModel  string    `json:"request_model"`
	ResponseModel string    `json:"response_model,omitempty"`
	UseCase       string    `json:"use_case,omitempty"` // "chat" or "patrol"
	InputTokens   int       `json:"input_tokens,omitempty"`
	OutputTokens  int       `json:"output_tokens,omitempty"`
	TargetType    string    `json:"target_type,omitempty"`
	TargetID      string    `json:"target_id,omitempty"`
	FindingID     string    `json:"finding_id,omitempty"`
}

// SaveAIUsageHistory persists AI usage events to disk.
func (c *ConfigPersistence) SaveAIUsageHistory(events []AIUsageEventRecord) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for ai usage history: %w", err)
	}

	data := AIUsageHistoryData{
		Version:   1,
		LastSaved: time.Now(),
		Events:    events,
	}

	jsonData, err := json.Marshal(data)
	if err != nil {
		return fmt.Errorf("marshal ai usage history: %w", err)
	}

	if c.crypto != nil {
		if encrypted, encErr := c.crypto.Encrypt(jsonData); encErr == nil {
			jsonData = encrypted
		} else {
			return fmt.Errorf("encrypt ai usage history: %w", encErr)
		}
	}

	if err := c.writeConfigFileLocked(c.aiUsageHistoryFile, jsonData, 0600); err != nil {
		return fmt.Errorf("persist ai usage history: %w", err)
	}

	log.Debug().
		Str("file", c.aiUsageHistoryFile).
		Int("count", len(events)).
		Msg("AI usage history saved")
	return nil
}

func newEmptyAIUsageHistoryData() *AIUsageHistoryData {
	return &AIUsageHistoryData{
		Version: 1,
		Events:  make([]AIUsageEventRecord, 0),
	}
}

// loadHistoryData generic helper for loading history data from disk.
// When crypto is non-nil, attempts to decrypt before parsing; falls back to
// plaintext on decryption failure (transparent migration from unencrypted files).
func loadHistoryData[T any](
	fs FileSystem,
	mu *sync.RWMutex,
	filePath string,
	cryptoMgr *crypto.CryptoManager,
	emptyDataFactory func() *T,
	normalizeSlice func(*T),
	rewrite func(*T) error,
	getCount func(*T) int,
	getLastSaved func(*T) time.Time,
	logLabel string,
) (*T, error) {
	mu.Lock()
	defer mu.Unlock()

	data, err := fs.ReadFile(filePath)
	if err != nil {
		if os.IsNotExist(err) {
			return emptyDataFactory(), nil
		}
		return nil, err
	}

	migratedPlaintext := false
	// Attempt decryption; fall back to plaintext for migration from unencrypted files.
	if cryptoMgr != nil {
		if decrypted, decErr := cryptoMgr.Decrypt(data); decErr == nil {
			data = decrypted
		} else {
			log.Debug().Str("file", filePath).Msg("Failed to decrypt history data — falling back to plaintext")
			migratedPlaintext = true
		}
	}

	var historyData T
	if err := json.Unmarshal(data, &historyData); err != nil {
		log.Error().Err(err).Str("file", filePath).Msgf("Failed to parse %s file", logLabel)
		return emptyDataFactory(), nil
	}

	normalizeSlice(&historyData)

	if migratedPlaintext && rewrite != nil {
		if err := rewrite(&historyData); err != nil {
			return nil, err
		}
	}

	log.Info().
		Str("file", filePath).
		Int("count", getCount(&historyData)).
		Time("last_saved", getLastSaved(&historyData)).
		Msgf("%s loaded", logLabel)
	return &historyData, nil
}

// LoadAIUsageHistory loads AI usage events from disk.
func (c *ConfigPersistence) LoadAIUsageHistory() (*AIUsageHistoryData, error) {
	return loadHistoryData(
		c.fs,
		&c.mu,
		c.aiUsageHistoryFile,
		c.crypto,
		newEmptyAIUsageHistoryData,
		func(data *AIUsageHistoryData) {
			if data.Events == nil {
				data.Events = make([]AIUsageEventRecord, 0)
			}
		},
		func(data *AIUsageHistoryData) error {
			jsonData, err := json.Marshal(data)
			if err != nil {
				return fmt.Errorf("marshal ai usage history migration rewrite: %w", err)
			}
			return rewriteEncryptedJSONLocked(c, c.aiUsageHistoryFile, jsonData, "ai usage history migration rewrite")
		},
		func(data *AIUsageHistoryData) int {
			return len(data.Events)
		},
		func(data *AIUsageHistoryData) time.Time {
			return data.LastSaved
		},
		"AI usage history",
	)
}

// SavePatrolRunHistory persists patrol run history to disk
func (c *ConfigPersistence) SavePatrolRunHistory(runs []PatrolRunRecord) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for patrol run history: %w", err)
	}

	data := PatrolRunHistoryData{
		Version:   1,
		LastSaved: time.Now(),
		Runs:      runs,
	}

	jsonData, err := json.Marshal(data)
	if err != nil {
		return fmt.Errorf("marshal patrol run history: %w", err)
	}

	if c.crypto != nil {
		if encrypted, encErr := c.crypto.Encrypt(jsonData); encErr == nil {
			jsonData = encrypted
		} else {
			return fmt.Errorf("encrypt patrol run history: %w", encErr)
		}
	}

	if err := c.writeConfigFileLocked(c.aiPatrolRunsFile, jsonData, 0600); err != nil {
		return fmt.Errorf("persist patrol run history: %w", err)
	}

	log.Debug().
		Str("file", c.aiPatrolRunsFile).
		Int("count", len(runs)).
		Msg("Patrol run history saved")
	return nil
}

func newEmptyPatrolRunHistoryData() *PatrolRunHistoryData {
	return &PatrolRunHistoryData{
		Version: 1,
		Runs:    make([]PatrolRunRecord, 0),
	}
}

// LoadPatrolRunHistory loads patrol run history from disk
func (c *ConfigPersistence) LoadPatrolRunHistory() (*PatrolRunHistoryData, error) {
	return loadHistoryData(
		c.fs,
		&c.mu,
		c.aiPatrolRunsFile,
		c.crypto,
		newEmptyPatrolRunHistoryData,
		func(data *PatrolRunHistoryData) {
			if data.Runs == nil {
				data.Runs = make([]PatrolRunRecord, 0)
			}
		},
		func(data *PatrolRunHistoryData) error {
			jsonData, err := json.Marshal(data)
			if err != nil {
				return fmt.Errorf("marshal patrol run history migration rewrite: %w", err)
			}
			return rewriteEncryptedJSONLocked(c, c.aiPatrolRunsFile, jsonData, "patrol run history migration rewrite")
		},
		func(data *PatrolRunHistoryData) int {
			return len(data.Runs)
		},
		func(data *PatrolRunHistoryData) time.Time {
			return data.LastSaved
		},
		"Patrol run history",
	)
}

// LoadSystemSettings loads system settings from file
func (c *ConfigPersistence) LoadSystemSettings() (*SystemSettings, error) {
	c.mu.RLock()
	defer c.mu.RUnlock()

	data, err := c.fs.ReadFile(c.systemFile)
	if err != nil {
		if os.IsNotExist(err) {
			// Return nil if file doesn't exist - let env vars take precedence
			return nil, nil
		}
		return nil, err
	}

	settings := DefaultSystemSettings()
	if settings == nil {
		settings = &SystemSettings{}
	}
	if err := json.Unmarshal(data, settings); err != nil {
		return nil, err
	}

	log.Info().Str("file", c.systemFile).Msg("System settings loaded")
	return settings, nil
}

// updateEnvFile updates the .env file with new system settings
func (c *ConfigPersistence) updateEnvFile(envFile string, settings SystemSettings) error {
	// Check if .env file exists
	if _, err := c.fs.Stat(envFile); os.IsNotExist(err) {
		// File doesn't exist, nothing to update
		return nil
	}

	// Read the existing .env file content
	existingContent, err := c.fs.ReadFile(envFile)
	if err != nil {
		return fmt.Errorf("read env file %s: %w", envFile, err)
	}

	var lines []string
	scanner := bufio.NewScanner(bytes.NewReader(existingContent))

	for scanner.Scan() {
		line := scanner.Text()

		// Skip legacy POLLING_INTERVAL lines - replaced by system.json PVE_POLLING_INTERVAL
		if strings.HasPrefix(line, "POLLING_INTERVAL=") {
			continue
		} else if strings.HasPrefix(line, "UPDATE_CHANNEL=") && settings.UpdateChannel != "" {
			lines = append(lines, fmt.Sprintf("UPDATE_CHANNEL=%s", settings.UpdateChannel))
		} else if strings.HasPrefix(line, "AUTO_UPDATE_ENABLED=") {
			// Always update AUTO_UPDATE_ENABLED when the line exists
			lines = append(lines, fmt.Sprintf("AUTO_UPDATE_ENABLED=%t", settings.AutoUpdateEnabled))
		} else if strings.HasPrefix(line, "AUTO_UPDATE_CHECK_INTERVAL=") && settings.AutoUpdateCheckInterval > 0 {
			lines = append(lines, fmt.Sprintf("AUTO_UPDATE_CHECK_INTERVAL=%d", settings.AutoUpdateCheckInterval))
		} else {
			lines = append(lines, line)
		}
	}

	if err := scanner.Err(); err != nil {
		return fmt.Errorf("scan env file %s: %w", envFile, err)
	}

	// Note: legacy POLLING_INTERVAL is deprecated and no longer written

	// Build the new content
	content := strings.Join(lines, "\n")
	if len(lines) > 0 && !strings.HasSuffix(content, "\n") {
		content += "\n"
	}

	// Only write if content actually changed
	// This prevents unnecessary .env rewrites that trigger the config watcher,
	// which can cause API tokens to be overwritten (related to #685)
	if string(existingContent) == content {
		log.Debug().Str("file", envFile).Msg("Skipping .env update - content unchanged")
		return nil
	}

	// Write to temp file first using owner-only permissions.
	tempFile := envFile + ".tmp"
	perm := os.FileMode(0600)
	if info, err := c.fs.Stat(envFile); err == nil {
		perm = info.Mode().Perm() & 0700
		if perm == 0 {
			perm = 0600
		}
	}
	if err := c.fs.WriteFile(tempFile, []byte(content), perm); err != nil {
		return fmt.Errorf("write temp env file %s: %w", tempFile, err)
	}

	// Atomic rename
	if err := c.fs.Rename(tempFile, envFile); err != nil {
		return fmt.Errorf("rename temp env file %s to %s: %w", tempFile, envFile, err)
	}
	return nil
}

// IsEncryptionEnabled returns whether the config persistence has encryption enabled
func (c *ConfigPersistence) IsEncryptionEnabled() bool {
	c.mu.RLock()
	defer c.mu.RUnlock()
	return c.crypto != nil
}

// cleanupOldBackups removes old backup files, keeping only the most recent N backups
func (c *ConfigPersistence) cleanupOldBackups(pattern string) {
	// Use filepath.Glob to find all backup files matching the pattern
	matches, err := filepath.Glob(pattern)
	if err != nil {
		log.Warn().Err(err).Str("pattern", pattern).Msg("Failed to find backup files for cleanup")
		return
	}

	// Keep only the last 10 backups
	const maxBackups = 10
	if len(matches) <= maxBackups {
		return
	}

	// Sort by modification time (oldest first)
	type fileInfo struct {
		path    string
		modTime time.Time
	}
	var files []fileInfo
	for _, match := range matches {
		info, err := c.fs.Stat(match)
		if err != nil {
			continue
		}
		files = append(files, fileInfo{path: match, modTime: info.ModTime()})
	}

	// Sort oldest first using efficient sort
	sort.Slice(files, func(i, j int) bool {
		return files[i].modTime.Before(files[j].modTime)
	})

	// Delete oldest backups (keep last 10)
	toDelete := len(files) - maxBackups
	for i := 0; i < toDelete; i++ {
		if err := c.fs.Remove(files[i].path); err != nil {
			log.Warn().Err(err).Str("file", files[i].path).Msg("Failed to delete old backup")
		} else {
			log.Debug().Str("file", files[i].path).Msg("Deleted old backup")
		}
	}
}

// LoadGuestMetadata loads guest metadata from disk
func (c *ConfigPersistence) LoadGuestMetadata() (*GuestMetadataStore, error) {
	c.mu.RLock()
	defer c.mu.RUnlock()

	return NewGuestMetadataStore(c.configDir, c.fs), nil
}

// LoadDockerMetadata loads docker metadata from disk
func (c *ConfigPersistence) LoadDockerMetadata() (*DockerMetadataStore, error) {
	c.mu.RLock()
	defer c.mu.RUnlock()

	return NewDockerMetadataStore(c.configDir, c.fs), nil
}

// LoadAgentProfiles loads agent profiles from file
func (c *ConfigPersistence) LoadAgentProfiles() ([]models.AgentProfile, error) {
	return loadSlice[models.AgentProfile](c, c.agentProfilesFile, true)
}

// SaveAgentProfiles saves agent profiles to file
func (c *ConfigPersistence) SaveAgentProfiles(profiles []models.AgentProfile) error {
	return saveJSON(c, c.agentProfilesFile, profiles, true)
}

// LoadAgentProfileAssignments loads agent profile assignments from file
func (c *ConfigPersistence) LoadAgentProfileAssignments() ([]models.AgentProfileAssignment, error) {
	return loadSlice[models.AgentProfileAssignment](c, c.agentAssignmentsFile, true)
}

// SaveAgentProfileAssignments saves agent profile assignments to file
func (c *ConfigPersistence) SaveAgentProfileAssignments(assignments []models.AgentProfileAssignment) error {
	return saveJSON(c, c.agentAssignmentsFile, assignments, true)
}

// LoadAgentProfileVersions loads profile version history from file
func (c *ConfigPersistence) LoadAgentProfileVersions() ([]models.AgentProfileVersion, error) {
	return loadSlice[models.AgentProfileVersion](c, filepath.Join(c.configDir, "profile-versions.json"), true)
}

// SaveAgentProfileVersions saves profile version history to file
func (c *ConfigPersistence) SaveAgentProfileVersions(versions []models.AgentProfileVersion) error {
	return saveJSON(c, filepath.Join(c.configDir, "profile-versions.json"), versions, true)
}

// LoadProfileDeploymentStatus loads deployment status from file
func (c *ConfigPersistence) LoadProfileDeploymentStatus() ([]models.ProfileDeploymentStatus, error) {
	return loadSlice[models.ProfileDeploymentStatus](c, filepath.Join(c.configDir, "profile-deployments.json"), true)
}

// SaveProfileDeploymentStatus saves deployment status to file
func (c *ConfigPersistence) SaveProfileDeploymentStatus(status []models.ProfileDeploymentStatus) error {
	return saveJSON(c, filepath.Join(c.configDir, "profile-deployments.json"), status, true)
}

// LoadProfileChangeLogs loads change logs from file
func (c *ConfigPersistence) LoadProfileChangeLogs() ([]models.ProfileChangeLog, error) {
	return loadSlice[models.ProfileChangeLog](c, filepath.Join(c.configDir, "profile-changelog.json"), true)
}

// SaveProfileChangeLogs saves change logs to file
func (c *ConfigPersistence) SaveProfileChangeLogs(logs []models.ProfileChangeLog) error {
	return saveJSON(c, filepath.Join(c.configDir, "profile-changelog.json"), logs, true)
}

// AppendProfileChangeLog adds a new entry to the change log
func (c *ConfigPersistence) AppendProfileChangeLog(entry models.ProfileChangeLog) error {
	logs, err := c.LoadProfileChangeLogs()
	if err != nil {
		return fmt.Errorf("load profile change logs: %w", err)
	}

	// Keep last 1000 entries
	if len(logs) >= 1000 {
		logs = logs[len(logs)-999:]
	}

	logs = append(logs, entry)
	return c.SaveProfileChangeLogs(logs)
}

// ============================================
// AI Chat Sessions Persistence
// ============================================

// AIChatSessionsData is the top-level container for chat sessions
type AIChatSessionsData struct {
	Version   int                       `json:"version"`
	LastSaved time.Time                 `json:"last_saved"`
	Sessions  map[string]*AIChatSession `json:"sessions"` // keyed by session ID
}

// AIChatSession represents a single chat conversation
type AIChatSession struct {
	ID        string          `json:"id"`
	Username  string          `json:"username"` // owner (from auth), empty for single-user
	Title     string          `json:"title"`    // auto-generated or user-set
	CreatedAt time.Time       `json:"created_at"`
	UpdatedAt time.Time       `json:"updated_at"`
	Messages  []AIChatMessage `json:"messages"`
}

// AIChatMessage represents a single message in a chat session
type AIChatMessage struct {
	ID        string               `json:"id"`
	Role      string               `json:"role"` // "user" or "assistant"
	Content   string               `json:"content"`
	Timestamp time.Time            `json:"timestamp"`
	Model     string               `json:"model,omitempty"`
	Tokens    *AIChatMessageTokens `json:"tokens,omitempty"`
	ToolCalls []AIChatToolCall     `json:"tool_calls,omitempty"`
}

// AIChatMessageTokens tracks token usage for a message
type AIChatMessageTokens struct {
	Input  int `json:"input"`
	Output int `json:"output"`
}

// AIChatToolCall represents a tool call made during a message
type AIChatToolCall struct {
	Name    string `json:"name"`
	Input   string `json:"input"`
	Output  string `json:"output"`
	Success bool   `json:"success"`
}

// SaveAIChatSessions persists all chat sessions to disk
func (c *ConfigPersistence) SaveAIChatSessions(sessions map[string]*AIChatSession) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	if err := c.EnsureConfigDir(); err != nil {
		return fmt.Errorf("prepare config directory for ai chat sessions: %w", err)
	}

	data := AIChatSessionsData{
		Version:   1,
		LastSaved: time.Now(),
		Sessions:  sessions,
	}

	jsonData, err := json.MarshalIndent(data, "", "  ")
	if err != nil {
		return fmt.Errorf("marshal ai chat sessions: %w", err)
	}

	if c.crypto != nil {
		if encrypted, encErr := c.crypto.Encrypt(jsonData); encErr == nil {
			jsonData = encrypted
		} else {
			return fmt.Errorf("encrypt ai chat sessions: %w", encErr)
		}
	}

	if err := c.writeConfigFileLocked(c.aiChatSessionsFile, jsonData, 0600); err != nil {
		return fmt.Errorf("persist ai chat sessions: %w", err)
	}

	log.Debug().
		Str("file", c.aiChatSessionsFile).
		Int("count", len(sessions)).
		Msg("AI chat sessions saved")
	return nil
}

// LoadAIChatSessions loads all chat sessions from disk
func (c *ConfigPersistence) LoadAIChatSessions() (*AIChatSessionsData, error) {
	c.mu.Lock()
	defer c.mu.Unlock()

	data, err := c.fs.ReadFile(c.aiChatSessionsFile)
	if err != nil {
		if os.IsNotExist(err) {
			return &AIChatSessionsData{
				Version:  1,
				Sessions: make(map[string]*AIChatSession),
			}, nil
		}
		return nil, err
	}

	migratedPlaintext := false
	// Attempt decryption; fall back to plaintext for migration from unencrypted files.
	if c.crypto != nil {
		if decrypted, decErr := c.crypto.Decrypt(data); decErr == nil {
			data = decrypted
		} else {
			migratedPlaintext = true
		}
	}

	var sessionsData AIChatSessionsData
	if err := json.Unmarshal(data, &sessionsData); err != nil {
		log.Error().Err(err).Str("file", c.aiChatSessionsFile).Msg("Failed to parse AI chat sessions file")
		return &AIChatSessionsData{
			Version:  1,
			Sessions: make(map[string]*AIChatSession),
		}, nil
	}

	if sessionsData.Sessions == nil {
		sessionsData.Sessions = make(map[string]*AIChatSession)
	}

	if migratedPlaintext {
		jsonData, err := json.MarshalIndent(sessionsData, "", "  ")
		if err != nil {
			return nil, fmt.Errorf("marshal ai chat sessions migration rewrite: %w", err)
		}
		if err := rewriteEncryptedJSONLocked(c, c.aiChatSessionsFile, jsonData, "ai chat sessions migration rewrite"); err != nil {
			return nil, err
		}
	}

	log.Debug().
		Str("file", c.aiChatSessionsFile).
		Int("count", len(sessionsData.Sessions)).
		Msg("AI chat sessions loaded")
	return &sessionsData, nil
}

// SaveAIChatSession saves or updates a single chat session
func (c *ConfigPersistence) SaveAIChatSession(session *AIChatSession) error {
	sessionsData, err := c.LoadAIChatSessions()
	if err != nil {
		return fmt.Errorf("load ai chat sessions: %w", err)
	}

	session.UpdatedAt = time.Now()
	sessionsData.Sessions[session.ID] = session

	return c.SaveAIChatSessions(sessionsData.Sessions)
}

// DeleteAIChatSession removes a chat session by ID
func (c *ConfigPersistence) DeleteAIChatSession(sessionID string) error {
	sessionsData, err := c.LoadAIChatSessions()
	if err != nil {
		return fmt.Errorf("load ai chat sessions: %w", err)
	}

	delete(sessionsData.Sessions, sessionID)
	return c.SaveAIChatSessions(sessionsData.Sessions)
}

// GetAIChatSessionsForUser returns all sessions for a specific user (or all if username is empty)
func (c *ConfigPersistence) GetAIChatSessionsForUser(username string) ([]*AIChatSession, error) {
	sessionsData, err := c.LoadAIChatSessions()
	if err != nil {
		return nil, err
	}

	var result []*AIChatSession
	for _, session := range sessionsData.Sessions {
		// If username filter is empty, return all; otherwise filter by username
		if username == "" || session.Username == username {
			result = append(result, session)
		}
	}

	// Sort by UpdatedAt descending (most recent first)
	sort.Slice(result, func(i, j int) bool {
		return result[i].UpdatedAt.After(result[j].UpdatedAt)
	})

	return result, nil
}

// CleanupOldAIChatSessions removes sessions older than maxAge
func (c *ConfigPersistence) CleanupOldAIChatSessions(maxAge time.Duration) (int, error) {
	sessionsData, err := c.LoadAIChatSessions()
	if err != nil {
		return 0, err
	}

	cutoff := time.Now().Add(-maxAge)
	removed := 0

	for id, session := range sessionsData.Sessions {
		if session.UpdatedAt.Before(cutoff) {
			delete(sessionsData.Sessions, id)
			removed++
		}
	}

	if removed > 0 {
		if err := c.SaveAIChatSessions(sessionsData.Sessions); err != nil {
			return 0, err
		}
		log.Info().
			Int("removed", removed).
			Dur("max_age", maxAge).
			Msg("Cleaned up old AI chat sessions")
	}

	return removed, nil
}