WIP: Save current state - security subsystems, migrations, logging

aggregator-agent/internal/orchestrator/orchestrator.go

@@ -9,6 +9,8 @@ import (
 
 	"github.com/Fimeg/RedFlag/aggregator-agent/internal/circuitbreaker"
 	"github.com/Fimeg/RedFlag/aggregator-agent/internal/client"
+	"github.com/Fimeg/RedFlag/aggregator-agent/internal/event"
+	"github.com/Fimeg/RedFlag/aggregator-agent/internal/models"
 )
 
 // Scanner represents a generic update scanner
@@ -42,8 +44,9 @@ type ScanResult struct {
 
 // Orchestrator manages and coordinates multiple scanners
 type Orchestrator struct {
-	scanners map[string]*ScannerConfig
-	mu       sync.RWMutex
+	scanners    map[string]*ScannerConfig
+	eventBuffer *event.Buffer
+	mu          sync.RWMutex
 }
 
 // NewOrchestrator creates a new scanner orchestrator
@@ -53,6 +56,14 @@ func NewOrchestrator() *Orchestrator {
 	}
 }
 
+// NewOrchestratorWithEvents creates a new scanner orchestrator with event buffering
+func NewOrchestratorWithEvents(buffer *event.Buffer) *Orchestrator {
+	return &Orchestrator{
+		scanners:    make(map[string]*ScannerConfig),
+		eventBuffer: buffer,
+	}
+}
+
 // RegisterScanner adds a scanner to the orchestrator
 func (o *Orchestrator) RegisterScanner(name string, scanner Scanner, cb *circuitbreaker.CircuitBreaker, timeout time.Duration, enabled bool) {
 	o.mu.Lock()
@@ -135,6 +146,27 @@ func (o *Orchestrator) executeScan(ctx context.Context, name string, cfg *Scanne
 	if !cfg.Enabled {
 		result.Status = "disabled"
 		log.Printf("[%s] Scanner disabled via configuration", name)
+		
+		// Buffer disabled event if event buffer is available
+		if o.eventBuffer != nil {
+			event := &models.SystemEvent{
+				EventType:    "agent_scan",
+				EventSubtype: "skipped",
+				Severity:     "info",
+				Component:    "scanner",
+				Message:      fmt.Sprintf("Scanner %s is disabled via configuration", name),
+				Metadata: map[string]interface{}{
+					"scanner_name": name,
+					"status":       "disabled",
+					"reason":       "configuration",
+				},
+				CreatedAt: time.Now(),
+			}
+			if err := o.eventBuffer.BufferEvent(event); err != nil {
+				log.Printf("Warning: Failed to buffer scanner disabled event: %v", err)
+			}
+		}
+		
 		return result
 	}
 
@@ -142,6 +174,27 @@ func (o *Orchestrator) executeScan(ctx context.Context, name string, cfg *Scanne
 	if !cfg.Scanner.IsAvailable() {
 		result.Status = "unavailable"
 		log.Printf("[%s] Scanner not available on this system", name)
+		
+		// Buffer unavailable event if event buffer is available
+		if o.eventBuffer != nil {
+			event := &models.SystemEvent{
+				EventType:    "agent_scan",
+				EventSubtype: "skipped",
+				Severity:     "info",
+				Component:    "scanner",
+				Message:      fmt.Sprintf("Scanner %s is not available on this system", name),
+				Metadata: map[string]interface{}{
+					"scanner_name": name,
+					"status":       "unavailable",
+					"reason":       "system_incompatible",
+				},
+				CreatedAt: time.Now(),
+			}
+			if err := o.eventBuffer.BufferEvent(event); err != nil {
+				log.Printf("Warning: Failed to buffer scanner unavailable event: %v", err)
+			}
+		}
+		
 		return result
 	}
 
@@ -185,12 +238,55 @@ func (o *Orchestrator) executeScan(ctx context.Context, name string, cfg *Scanne
 		result.Error = err
 		result.Status = "failed"
 		log.Printf("[%s] Scan failed: %v", name, err)
+		
+		// Buffer event if event buffer is available
+		if o.eventBuffer != nil {
+			event := &models.SystemEvent{
+				EventType:    "agent_scan",
+				EventSubtype: "failed",
+				Severity:     "error",
+				Component:    "scanner",
+				Message:      fmt.Sprintf("Scanner %s failed: %v", name, err),
+				Metadata: map[string]interface{}{
+					"scanner_name": name,
+					"error_type":   "scan_failed",
+					"error_details": err.Error(),
+					"duration_ms":  result.Duration.Milliseconds(),
+				},
+				CreatedAt: time.Now(),
+			}
+			if err := o.eventBuffer.BufferEvent(event); err != nil {
+				log.Printf("Warning: Failed to buffer scanner failure event: %v", err)
+			}
+		}
+		
 		return result
 	}
 
 	result.Updates = updates
 	result.Status = "success"
 	log.Printf("[%s] Scan completed: found %d updates (took %v)", name, len(updates), result.Duration)
+	
+	// Buffer success event if event buffer is available
+	if o.eventBuffer != nil {
+		event := &models.SystemEvent{
+			EventType:    "agent_scan",
+			EventSubtype: "completed",
+			Severity:     "info",
+			Component:    "scanner",
+			Message:      fmt.Sprintf("Scanner %s completed successfully", name),
+			Metadata: map[string]interface{}{
+				"scanner_name": name,
+				"updates_found": len(updates),
+				"duration_ms":   result.Duration.Milliseconds(),
+				"status":        "success",
+			},
+			CreatedAt: time.Now(),
+		}
+		if err := o.eventBuffer.BufferEvent(event); err != nil {
+			log.Printf("Warning: Failed to buffer scanner success event: %v", err)
+		}
+	}
 
 	return result
 }