feat(agent): tie up

agent/main.go

@@ -1,10 +1,14 @@
 package main
 
 import (
+	"context"
 	"log"
 	"os"
 
+	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/agent/internal/client"
+	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/agent/internal/commander"
 	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/agent/internal/config"
+	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/agent/internal/mtls"
 	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/agent/internal/registration"
 )
 
@@ -46,4 +50,18 @@ func main() {
 	log.Printf("Certificates saved to %s", cfg.CertDir)
 
 	log.Println("Agent registration complete")
+	err = func() error {
+		creds, err := mtls.LoadMTLSCredentials(certs.CACertPEM, certs.ClientCertPEM, certs.ClientKeyPEM)
+		if err != nil {
+			return err
+		}
+		ctx, cancel := context.WithCancel(context.Background())
+		defer cancel()
+		cmdexe := new(commander.CommandExecutor)
+		ccli := client.New(cmdexe, cfg.RegistrationToken)
+		return ccli.HandleCommands(ctx, cfg.BackendURL, creds)
+	}()
+	if err != nil {
+		log.Fatalf("Failed to generate key and CSR: %v", err)
+	}
 }