Harden CHORUS security and messaging stack

pkg/ucxl/decision_publisher.go

@@ -13,11 +13,11 @@ import (
 
 // DecisionPublisher handles publishing task completion decisions to encrypted DHT storage
 type DecisionPublisher struct {
-	ctx         context.Context
-	config      *config.Config
-	dhtStorage  storage.UCXLStorage
-	nodeID      string
-	agentName   string
+	ctx        context.Context
+	config     *config.Config
+	dhtStorage storage.UCXLStorage
+	nodeID     string
+	agentName  string
 }
 
 // NewDecisionPublisher creates a new decision publisher
@@ -39,28 +39,28 @@ func NewDecisionPublisher(
 
 // TaskDecision represents a decision made by an agent upon task completion
 type TaskDecision struct {
-	Agent           string                 `json:"agent"`
-	Role            string                 `json:"role"`
-	Project         string                 `json:"project"`
-	Task            string                 `json:"task"`
-	Decision        string                 `json:"decision"`
-	Context         map[string]interface{} `json:"context"`
-	Timestamp       time.Time              `json:"timestamp"`
-	Success         bool                   `json:"success"`
-	ErrorMessage    string                 `json:"error_message,omitempty"`
-	FilesModified   []string               `json:"files_modified,omitempty"`
-	LinesChanged    int                    `json:"lines_changed,omitempty"`
-	TestResults     *TestResults           `json:"test_results,omitempty"`
-	Dependencies    []string               `json:"dependencies,omitempty"`
-	NextSteps       []string               `json:"next_steps,omitempty"`
+	Agent         string                 `json:"agent"`
+	Role          string                 `json:"role"`
+	Project       string                 `json:"project"`
+	Task          string                 `json:"task"`
+	Decision      string                 `json:"decision"`
+	Context       map[string]interface{} `json:"context"`
+	Timestamp     time.Time              `json:"timestamp"`
+	Success       bool                   `json:"success"`
+	ErrorMessage  string                 `json:"error_message,omitempty"`
+	FilesModified []string               `json:"files_modified,omitempty"`
+	LinesChanged  int                    `json:"lines_changed,omitempty"`
+	TestResults   *TestResults           `json:"test_results,omitempty"`
+	Dependencies  []string               `json:"dependencies,omitempty"`
+	NextSteps     []string               `json:"next_steps,omitempty"`
 }
 
 // TestResults captures test execution results
 type TestResults struct {
-	Passed     int      `json:"passed"`
-	Failed     int      `json:"failed"`
-	Skipped    int      `json:"skipped"`
-	Coverage   float64  `json:"coverage,omitempty"`
+	Passed      int      `json:"passed"`
+	Failed      int      `json:"failed"`
+	Skipped     int      `json:"skipped"`
+	Coverage    float64  `json:"coverage,omitempty"`
 	FailedTests []string `json:"failed_tests,omitempty"`
 }
 
@@ -74,7 +74,11 @@ func (dp *DecisionPublisher) PublishTaskDecision(decision *TaskDecision) error {
 		decision.Role = dp.config.Agent.Role
 	}
 	if decision.Project == "" {
-		decision.Project = "default-project" // TODO: Add project field to config
+		if project := dp.config.Agent.Project; project != "" {
+			decision.Project = project
+		} else {
+			decision.Project = "chorus"
+		}
 	}
 	if decision.Timestamp.IsZero() {
 		decision.Timestamp = time.Now()
@@ -173,16 +177,16 @@ func (dp *DecisionPublisher) PublishArchitecturalDecision(
 	nextSteps []string,
 ) error {
 	taskDecision := &TaskDecision{
-		Task:     taskName,
-		Decision: decision,
-		Success:  true,
+		Task:      taskName,
+		Decision:  decision,
+		Success:   true,
 		NextSteps: nextSteps,
 		Context: map[string]interface{}{
-			"decision_type":  "architecture",
-			"rationale":      rationale,
-			"alternatives":   alternatives,
-			"implications":   implications,
-			"node_id":        dp.nodeID,
+			"decision_type": "architecture",
+			"rationale":     rationale,
+			"alternatives":  alternatives,
+			"implications":  implications,
+			"node_id":       dp.nodeID,
 		},
 	}
 
@@ -291,7 +295,7 @@ func (dp *DecisionPublisher) SubscribeToDecisions(
 ) error {
 	// This is a placeholder for future pubsub implementation
 	// For now, we'll implement a simple polling mechanism
-	
+
 	go func() {
 		ticker := time.NewTicker(30 * time.Second)
 		defer ticker.Stop()
@@ -341,10 +345,10 @@ func (dp *DecisionPublisher) PublishSystemStatus(
 		Decision: status,
 		Success:  dp.allHealthChecksPass(healthChecks),
 		Context: map[string]interface{}{
-			"decision_type":  "system",
-			"metrics":        metrics,
-			"health_checks":  healthChecks,
-			"node_id":        dp.nodeID,
+			"decision_type": "system",
+			"metrics":       metrics,
+			"health_checks": healthChecks,
+			"node_id":       dp.nodeID,
 		},
 	}
 
@@ -364,13 +368,17 @@ func (dp *DecisionPublisher) allHealthChecksPass(healthChecks map[string]bool) b
 // GetPublisherMetrics returns metrics about the decision publisher
 func (dp *DecisionPublisher) GetPublisherMetrics() map[string]interface{} {
 	dhtMetrics := dp.dhtStorage.GetMetrics()
-	
-	return map[string]interface{}{
-		"node_id":        dp.nodeID,
-		"agent_name":     dp.agentName,
-		"current_role":   dp.config.Agent.Role,
-		"project":        "default-project", // TODO: Add project field to config
-		"dht_metrics":    dhtMetrics,
-		"last_publish":   time.Now(), // This would be tracked in a real implementation
+	project := dp.config.Agent.Project
+	if project == "" {
+		project = "chorus"
 	}
-}
+
+	return map[string]interface{}{
+		"node_id":      dp.nodeID,
+		"agent_name":   dp.agentName,
+		"current_role": dp.config.Agent.Role,
+		"project":      project,
+		"dht_metrics":  dhtMetrics,
+		"last_publish": time.Now(), // This would be tracked in a real implementation
+	}
+}