c0a717ab26
- README: version v0.2.6.8, corrected stale gate claim, updated changelog - .env.example: merged two competing files into one, deleted bootstrap duplicate - ErrorBoundary: new component wrapping app, prevents white-screen crashes - Layout sidebar: version display from /api/health, Docs link to GitHub - client-logger: debug/trace logger gated behind localStorage.redflag_debug=1, routes through existing /logs/client-error server endpoint (ETHOS #1) - All web console.log calls rerouted through client-logger instead of deleted - Server health endpoint returns version field - Server accepts client_debug/client_trace in error_type validation - Dockerfiles: pinned alpine:latest->3.21, nginx:alpine->1.27-alpine, added HEALTHCHECK directives - docker-compose: healthcheck blocks for server and web services - .dockerignore: created to slim Docker build context
85 lines
No EOL
2.9 KiB
Go
85 lines
No EOL
2.9 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/Fimeg/RedFlag/server/internal/database/queries"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// StatsHandler handles statistics for the dashboard
|
|
type StatsHandler struct {
|
|
agentQueries *queries.AgentQueries
|
|
updateQueries *queries.UpdateQueries
|
|
}
|
|
|
|
// NewStatsHandler creates a new stats handler
|
|
func NewStatsHandler(agentQueries *queries.AgentQueries, updateQueries *queries.UpdateQueries) *StatsHandler {
|
|
return &StatsHandler{
|
|
agentQueries: agentQueries,
|
|
updateQueries: updateQueries,
|
|
}
|
|
}
|
|
|
|
// DashboardStats represents dashboard statistics
|
|
type DashboardStats struct {
|
|
TotalAgents int `json:"total_agents"`
|
|
OnlineAgents int `json:"online_agents"`
|
|
OfflineAgents int `json:"offline_agents"`
|
|
PendingUpdates int `json:"pending_updates"`
|
|
FailedUpdates int `json:"failed_updates"`
|
|
SecurityUpdateCount int `json:"security_update_count"` // distinct advisories in available-version OSV check
|
|
InstalledCVECount int `json:"installed_cve_count"` // distinct advisories in installed-version OSV check
|
|
CriticalUpdates int `json:"critical_updates"`
|
|
ImportantUpdates int `json:"high_updates"`
|
|
ModerateUpdates int `json:"medium_updates"`
|
|
LowUpdates int `json:"low_updates"`
|
|
UpdatesByType map[string]int `json:"updates_by_type"`
|
|
}
|
|
|
|
// GetDashboardStats returns dashboard statistics using aggregate queries (F-B1-6 fix)
|
|
func (h *StatsHandler) GetDashboardStats(c *gin.Context) {
|
|
// Get all agents for online/offline count
|
|
agents, err := h.agentQueries.ListAgents("", "")
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to get agents"})
|
|
return
|
|
}
|
|
|
|
stats := DashboardStats{
|
|
TotalAgents: len(agents),
|
|
UpdatesByType: make(map[string]int),
|
|
}
|
|
|
|
// Count online/offline agents
|
|
for _, agent := range agents {
|
|
if time.Since(agent.LastSeen) <= 10*time.Minute {
|
|
stats.OnlineAgents++
|
|
} else {
|
|
stats.OfflineAgents++
|
|
}
|
|
}
|
|
|
|
// Single aggregate query for all update stats (replaces N+1 per-agent loop)
|
|
updateStats, err := h.updateQueries.GetAllUpdateStats()
|
|
if err == nil {
|
|
stats.PendingUpdates = updateStats.PendingUpdates
|
|
stats.FailedUpdates = updateStats.FailedUpdates
|
|
stats.CriticalUpdates = updateStats.CriticalUpdates
|
|
stats.ImportantUpdates = updateStats.ImportantUpdates
|
|
stats.ModerateUpdates = updateStats.ModerateUpdates
|
|
stats.LowUpdates = updateStats.LowUpdates
|
|
}
|
|
|
|
// Remediation count — distinct advisories in available-version OSV check.
|
|
if n, err := h.updateQueries.GetSecurityUpdateCount(); err == nil {
|
|
stats.SecurityUpdateCount = n
|
|
}
|
|
// Threat count — distinct advisories in installed-version OSV check.
|
|
if n, err := h.updateQueries.GetInstalledCVECount(); err == nil {
|
|
stats.InstalledCVECount = n
|
|
}
|
|
|
|
c.JSON(http.StatusOK, stats)
|
|
} |