add user agent specific for daemon mode via DownloaderHeaders property for ArduinoCoreServerImpl

Author: Roberto Sora

cli/daemon/daemon.go

@@ -21,10 +21,13 @@ import (
 	"fmt"
 	"log"
 	"net"
+	"net/http"
+	"runtime"
 
 	"github.com/arduino/arduino-cli/cli"
 	"github.com/arduino/arduino-cli/daemon"
 	"github.com/arduino/arduino-cli/rpc"
+	"github.com/arduino/arduino-cli/version"
 	"github.com/spf13/cobra"
 	"google.golang.org/grpc"
 )
@@ -53,7 +56,14 @@ func runDaemonCommand(cmd *cobra.Command, args []string) {
 		log.Fatalf("failed to listen: %v", err)
 	}
 	s := grpc.NewServer()
-	rpc.RegisterArduinoCoreServer(s, &daemon.ArduinoCoreServerImpl{})
+
+	userAgentValue := fmt.Sprintf("%s daemon/%s (%s; %s; %s) Commit:%s/Build:%s ", version.GetApplication(),
+		version.GetVersion(), runtime.GOARCH, runtime.GOOS, runtime.Version(), version.GetCommit(), version.GetBuildDate())
+	headers := http.Header{"User-Agent": []string{userAgentValue}}
+
+	coreServer := daemon.ArduinoCoreServerImpl{DownloaderHeaders: headers}
+
+	rpc.RegisterArduinoCoreServer(s, &coreServer)
 	if err := s.Serve(lis); err != nil {
 		log.Fatalf("failed to serve: %v", err)
 	}

daemon/daemon.go

@@ -36,7 +36,7 @@ import (
 )
 
 type ArduinoCoreServerImpl struct {
-	downloaderHeaders http.Header
+	DownloaderHeaders http.Header
 }
 
 func (s *ArduinoCoreServerImpl) BoardDetails(ctx context.Context, req *rpc.BoardDetailsReq) (*rpc.BoardDetailsResp, error) {
@@ -94,7 +94,7 @@ func (s *ArduinoCoreServerImpl) Init(req *rpc.InitReq, stream rpc.ArduinoCore_In
 	resp, err := commands.Init(stream.Context(), req,
 		func(p *rpc.DownloadProgress) { stream.Send(&rpc.InitResp{DownloadProgress: p}) },
 		func(p *rpc.TaskProgress) { stream.Send(&rpc.InitResp{TaskProgress: p}) },
-		s.downloaderHeaders,
+		s.DownloaderHeaders,
 	)
 	if err != nil {
 		return err
@@ -124,7 +124,7 @@ func (s *ArduinoCoreServerImpl) PlatformInstall(req *rpc.PlatformInstallReq, str
 		stream.Context(), req,
 		func(p *rpc.DownloadProgress) { stream.Send(&rpc.PlatformInstallResp{Progress: p}) },
 		func(p *rpc.TaskProgress) { stream.Send(&rpc.PlatformInstallResp{TaskProgress: p}) },
-		s.downloaderHeaders,
+		s.DownloaderHeaders,
 	)
 	if err != nil {
 		return err
@@ -136,7 +136,7 @@ func (s *ArduinoCoreServerImpl) PlatformDownload(req *rpc.PlatformDownloadReq, s
 	resp, err := core.PlatformDownload(
 		stream.Context(), req,
 		func(p *rpc.DownloadProgress) { stream.Send(&rpc.PlatformDownloadResp{Progress: p}) },
-		s.downloaderHeaders,
+		s.DownloaderHeaders,
 	)
 	if err != nil {
 		return err
@@ -160,7 +160,7 @@ func (s *ArduinoCoreServerImpl) PlatformUpgrade(req *rpc.PlatformUpgradeReq, str
 		stream.Context(), req,
 		func(p *rpc.DownloadProgress) { stream.Send(&rpc.PlatformUpgradeResp{Progress: p}) },
 		func(p *rpc.TaskProgress) { stream.Send(&rpc.PlatformUpgradeResp{TaskProgress: p}) },
-		s.downloaderHeaders,
+		s.DownloaderHeaders,
 	)
 	if err != nil {
 		return err
@@ -207,7 +207,7 @@ func (s *ArduinoCoreServerImpl) LibraryDownload(req *rpc.LibraryDownloadReq, str
 	resp, err := lib.LibraryDownload(
 		stream.Context(), req,
 		func(p *rpc.DownloadProgress) { stream.Send(&rpc.LibraryDownloadResp{Progress: p}) },
-		s.downloaderHeaders,
+		s.DownloaderHeaders,
 	)
 	if err != nil {
 		return err
@@ -220,7 +220,7 @@ func (s *ArduinoCoreServerImpl) LibraryInstall(req *rpc.LibraryInstallReq, strea
 		stream.Context(), req,
 		func(p *rpc.DownloadProgress) { stream.Send(&rpc.LibraryInstallResp{Progress: p}) },
 		func(p *rpc.TaskProgress) { stream.Send(&rpc.LibraryInstallResp{TaskProgress: p}) },
-		s.downloaderHeaders,
+		s.DownloaderHeaders,
 	)
 	if err != nil {
 		return err
@@ -242,7 +242,7 @@ func (s *ArduinoCoreServerImpl) LibraryUpgradeAll(req *rpc.LibraryUpgradeAllReq,
 	err := lib.LibraryUpgradeAll(stream.Context(), req,
 		func(p *rpc.DownloadProgress) { stream.Send(&rpc.LibraryUpgradeAllResp{Progress: p}) },
 		func(p *rpc.TaskProgress) { stream.Send(&rpc.LibraryUpgradeAllResp{TaskProgress: p}) },
-		s.downloaderHeaders,
+		s.DownloaderHeaders,
 	)
 	if err != nil {
 		return err