feat: add EXTRA_USER_AGENT env (#32)

Signed-off-by: Patrik Cyvoct <[email protected]>

driver/controller.go

@@ -3,6 +3,7 @@ package driver
 import (
 	"context"
 	"fmt"
+	"os"
 	"strconv"
 	"strings"
 	"sync"
@@ -50,9 +51,14 @@ type controllerService struct {
 }
 
 func newControllerService(config *DriverConfig) controllerService {
+	userAgent := fmt.Sprintf("%s %s (%s)", DriverName, driverVersion, gitCommit)
+	if extraUA := os.Getenv(ExtraUserAgentEnv); extraUA != "" {
+		userAgent = userAgent + " " + extraUA
+	}
+
 	return controllerService{
 		config:   config,
-		scaleway: scaleway.NewScaleway(fmt.Sprintf("%s %s (%s)", DriverName, driverVersion, gitCommit)),
+		scaleway: scaleway.NewScaleway(userAgent),
 	}
 }
 

driver/driver.go

@@ -20,6 +20,9 @@ const (
 	// DriverName is the official name for the Scaleway CSI plugin
 	DriverName      = "csi.scaleway.com"
 	ZoneTopologyKey = "topology." + DriverName + "/zone"
+
+	// ExtraUserAgentEnv is the environment variable that adds some string at the end of the user agent
+	ExtraUserAgentEnv = "EXTRA_USER_AGENT"
 )
 
 // Mode represents the mode in which the CSI driver started