Fimeg/Redflag
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Add agent distribution system

Browse Source
This commit is contained in:
Fimeg
2025-10-29 12:50:00 -04:00
parent c027a1da6c
commit 5251825ac1
2 changed files with 132 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
aggregator-server/cmd/server/main.go
View File

@@ -101,6 +101,7 @@ func main() {
dockerHandler := handlers.NewDockerHandler(updateQueries, agentQueries, commandQueries)
registrationTokenHandler := handlers.NewRegistrationTokenHandler(registrationTokenQueries, agentQueries, cfg)
rateLimitHandler := handlers.NewRateLimitHandler(rateLimiter)
downloadHandler := handlers.NewDownloadHandler(filepath.Join(".", "redflag-agent"))
// Setup router
router := gin.Default()
@@ -181,6 +182,10 @@ func main() {
dashboard.POST("/docker/containers/:container_id/images/:image_id/reject", dockerHandler.RejectUpdate)
dashboard.POST("/docker/containers/:container_id/images/:image_id/install", dockerHandler.InstallUpdate)
// Download routes (authenticated)
dashboard.GET("/downloads/:platform", downloadHandler.DownloadAgent)
dashboard.GET("/install/:platform", downloadHandler.InstallScript)
// Admin/Registration Token routes (for agent enrollment management)
admin := dashboard.Group("/admin")
{