package store

import (
	"sync"
	"time"

	"nerd-monitor/internal/stats"
)

// AgentStats represents the latest stats for an agent.
type AgentStats struct {
	ID        string
	Hostname  string
	CPUUsage  float64
	RAMUsage  uint64
	RAMTotal  uint64
	DiskUsage uint64
	DiskTotal uint64
	LastSeen  time.Time
}

// Store manages agent statistics in memory.
type Store struct {
	mu     sync.RWMutex
	agents map[string]*AgentStats
}

// New creates a new store.
func New() *Store {
	return &Store{
		agents: make(map[string]*AgentStats),
	}
}

// UpdateAgent updates or creates agent stats.
func (s *Store) UpdateAgent(id string, stat *stats.Stats) {
	s.mu.Lock()
	defer s.mu.Unlock()

	s.agents[id] = &AgentStats{
		ID:        id,
		Hostname:  stat.Hostname,
		CPUUsage:  stat.CPUUsage,
		RAMUsage:  stat.RAMUsage,
		RAMTotal:  stat.RAMTotal,
		DiskUsage: stat.DiskUsage,
		DiskTotal: stat.DiskTotal,
		LastSeen:  time.Now(),
	}
}

// GetAgent retrieves agent stats by ID.
func (s *Store) GetAgent(id string) *AgentStats {
	s.mu.RLock()
	defer s.mu.RUnlock()

	return s.agents[id]
}

// GetAllAgents returns all agents.
func (s *Store) GetAllAgents() []*AgentStats {
	s.mu.RLock()
	defer s.mu.RUnlock()

	agents := make([]*AgentStats, 0, len(s.agents))
	for _, agent := range s.agents {
		agents = append(agents, agent)
	}
	return agents
}

// DeleteAgent removes an agent.
func (s *Store) DeleteAgent(id string) {
	s.mu.Lock()
	defer s.mu.Unlock()

	delete(s.agents, id)
}