52c9c1a45b
Created centralized lifecycle service to handle new, upgrade, and rebuild operations. Added deprecation notices to old handlers (agent_setup, build_orchestrator, agent_build). Foundation for consolidating duplicate agent lifecycle logic.
80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/Fimeg/RedFlag/aggregator-server/internal/services"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// SetupAgent handles the agent setup endpoint
|
|
// Deprecated: Use AgentHandler.Setup instead
|
|
func SetupAgent(c *gin.Context) {
|
|
var req services.AgentSetupRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// Create config builder
|
|
configBuilder := services.NewConfigBuilder(req.ServerURL)
|
|
|
|
// Build agent configuration
|
|
config, err := configBuilder.BuildAgentConfig(req)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// Create response
|
|
response := gin.H{
|
|
"agent_id": config.AgentID,
|
|
"registration_token": config.Secrets["registration_token"],
|
|
"server_public_key": config.Secrets["server_public_key"],
|
|
"configuration": config.PublicConfig,
|
|
"secrets": config.Secrets,
|
|
"template": config.Template,
|
|
"setup_time": config.BuildTime,
|
|
"secrets_created": config.SecretsCreated,
|
|
"secrets_path": config.SecretsPath,
|
|
}
|
|
|
|
c.JSON(http.StatusOK, response)
|
|
}
|
|
|
|
// GetTemplates returns available agent templates
|
|
func GetTemplates(c *gin.Context) {
|
|
configBuilder := services.NewConfigBuilder("")
|
|
templates := configBuilder.GetTemplates()
|
|
c.JSON(http.StatusOK, gin.H{"templates": templates})
|
|
}
|
|
|
|
// ValidateConfiguration validates a configuration before deployment
|
|
func ValidateConfiguration(c *gin.Context) {
|
|
var config map[string]interface{}
|
|
if err := c.ShouldBindJSON(&config); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
agentType, exists := config["agent_type"].(string)
|
|
if !exists {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "agent_type is required"})
|
|
return
|
|
}
|
|
|
|
configBuilder := services.NewConfigBuilder("")
|
|
template, exists := configBuilder.GetTemplate(agentType)
|
|
if !exists {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Unknown agent type"})
|
|
return
|
|
}
|
|
|
|
// Simple validation response
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"valid": true,
|
|
"message": "Configuration appears valid",
|
|
"agent_type": agentType,
|
|
"template": template.Name,
|
|
})
|
|
} |