Backlog/v12 add forwarder in agentmanager

agent-manager/models/collector.go

@@ -5,8 +5,9 @@ import "gorm.io/gorm"
 type CollectorModule string
 
 const (
-	AS_400   CollectorModule = "AS_400"
-	UTMSTACK CollectorModule = "UTMSTACK"
+	AS_400    CollectorModule = "AS_400"
+	UTMSTACK  CollectorModule = "UTMSTACK"
+	FORWARDER CollectorModule = "FORWARDER"
 )
 
 type Collector struct {

agent-manager/protos/collector.proto

@@ -15,6 +15,7 @@ service CollectorService {
 enum CollectorModule{
   AS_400 = 0;
   UTMSTACK = 1;
+  FORWARDER = 2;
 }
 
 message RegisterRequest {

collectors/forwarder/cmd/install.go

@@ -7,6 +7,7 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/utmstack/UTMStack/collectors/forwarder/config"
 	"github.com/utmstack/UTMStack/collectors/forwarder/serv"
+	"github.com/utmstack/UTMStack/collectors/forwarder/updates"
 	"github.com/utmstack/UTMStack/collectors/forwarder/upstream"
 	"github.com/utmstack/UTMStack/collectors/forwarder/utils"
 )
@@ -33,6 +34,13 @@ var installCmd = &cobra.Command{
 		}
 		fmt.Println("[OK]")
 
+		fmt.Print("Downloading version info ... ")
+		if err := updates.DownloadVersion(cnf.Server, cnf.SkipCertValidation); err != nil {
+			fmt.Println("\nError downloading version info: ", err)
+			os.Exit(1)
+		}
+		fmt.Println("[OK]")
+
 		fmt.Print("Configuring collector ... ")
 		if err := upstream.RegisterCollector(cnf, utmKey); err != nil {
 			fmt.Println("\nError registering collector: ", err)

collectors/forwarder/config/const.go

@@ -10,8 +10,11 @@ import (
 var REPLACE_KEY string
 
 var (
-	AgentManagerPort = "9000"
-	LogAuthProxyPort = "50051"
+	AgentManagerPort  = "9000"
+	LogAuthProxyPort  = "50051"
+	DependenciesPort  = "9001"
+	DependUrl         = "https://%s:%s/private/dependencies/collector/%s"
+	VersionPath       = filepath.Join(fs.GetExecutablePath(), "version.json")
 
 	CollectorLogFile  = filepath.Join(fs.GetExecutablePath(), "logs", "utmstack_forwarder.log")
 	UUIDFileName      = filepath.Join(fs.GetExecutablePath(), "collector-uuid.yml")

collectors/forwarder/listeners/auth.go

@@ -9,6 +9,7 @@ import (
 	"fmt"
 	stdhttp "net/http"
 	"os"
+	"path/filepath"
 	"strings"
 )
 
@@ -88,6 +89,11 @@ func GenerateTokenFile(path string) (string, error) {
 		return "", fmt.Errorf("generate token: %w", err)
 	}
 	token := hex.EncodeToString(b)
+	if dir := filepath.Dir(path); dir != "" {
+		if err := os.MkdirAll(dir, 0700); err != nil {
+			return "", fmt.Errorf("write token file: create dir: %w", err)
+		}
+	}
 	if err := os.WriteFile(path, []byte(token+"\n"), 0600); err != nil {
 		return "", fmt.Errorf("write token file: %w", err)
 	}

collectors/forwarder/updates/dependencies.go

@@ -0,0 +1,17 @@
+package updates
+
+import (
+	"fmt"
+
+	"github.com/utmstack/UTMStack/collectors/forwarder/config"
+	sharedhttp "github.com/utmstack/UTMStack/shared/http"
+	"github.com/utmstack/UTMStack/shared/fs"
+)
+
+func DownloadVersion(address string, insecure bool) error {
+	url := fmt.Sprintf(config.DependUrl, address, config.DependenciesPort, "version.json")
+	if err := sharedhttp.DownloadFile(url, map[string]string{}, "version.json", fs.GetExecutablePath(), insecure); err != nil {
+		return fmt.Errorf("error downloading version.json: %v", err)
+	}
+	return nil
+}