Add Azure platform detection to the translator (#2183)

Author: movence

cfg/envconfig/envconfig.go

@@ -26,6 +26,7 @@ const (
 	RunInContainer              = "RUN_IN_CONTAINER"
 	RunAsHostProcessContainer   = "RUN_AS_HOST_PROCESS_CONTAINER"
 	RunInAWS                    = "RUN_IN_AWS"
+	RunInAKS                    = "RUN_IN_AKS"
 	RunWithIRSA                 = "RUN_WITH_IRSA"
 	RunWithSELinux              = "RUN_WITH_SELINUX"
 	RunInROSA                   = "RUN_IN_ROSA"
@@ -89,6 +90,11 @@ func IsRunningInROSA() bool {
 	return os.Getenv(RunInROSA) == TrueValue
 }
 
+// IsRunningInAKS reports whether RUN_IN_AKS is set (by the AKS Helm chart) so AKS is detected without a metadata probe.
+func IsRunningInAKS() bool {
+	return os.Getenv(RunInAKS) == TrueValue
+}
+
 func GetLogsBackpressureMode() string {
 	return os.Getenv(CWAgentLogsBackpressureMode)
 }

go.mod

@@ -17,6 +17,7 @@ import (
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/headerssetterextension"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/healthcheckextension"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/observer/ecsobserver"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/oidctokenextension"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/pprofextension"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/sigv4authextension"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/filestorage"
@@ -187,6 +188,7 @@ func Factories() (otelcol.Factories, error) {
 		ecsobserver.NewFactory(),
 		filestorage.NewFactory(),
 		healthcheckextension.NewFactory(),
+		oidctokenextension.NewFactory(),
 		pprofextension.NewFactory(),
 		sigv4authextension.NewFactory(),
 		zpagesextension.NewFactory(),

go.sum

@@ -126,6 +126,7 @@ func TestComponents(t *testing.T) {
 		"health_check",
 		"k8smetadata",
 		"nodemetadatacache",
+		"oidctoken",
 		"pprof",
 		"server",
 		"sigv4auth",

service/defaultcomponents/components.go

@@ -11,6 +11,7 @@ const (
 	ENV            = "env-config.json"
 	AGENT_LOG_FILE = "amazon-cloudwatch-agent.log"
 	JMXJarName     = "opentelemetry-jmx-metrics.jar"
+	OIDCToken      = ".oidc-token"
 )
 
 var (
@@ -24,4 +25,5 @@ var (
 	TranslatorBinaryPath string
 	AgentBinaryPath      string
 	JMXJarPath           string
+	OIDCTokenPath        string
 )

service/defaultcomponents/components_test.go

@@ -31,4 +31,5 @@ func init() {
 	TranslatorBinaryPath = filepath.Join(AgentDir, "bin", TranslatorBinaryName)
 	AgentBinaryPath = filepath.Join(AgentDir, "bin", AgentBinaryName)
 	JMXJarPath = filepath.Join(AgentDir, "bin", JMXJarName)
+	OIDCTokenPath = filepath.Join(AgentDir, "etc", OIDCToken)
 }

tool/paths/paths.go

@@ -53,4 +53,5 @@ func init() {
 	TranslatorBinaryPath = filepath.Join(AgentRootDir, TranslatorBinaryName)
 	AgentBinaryPath = filepath.Join(AgentRootDir, AgentBinaryName)
 	JMXJarPath = filepath.Join(AgentRootDir, JMXJarName)
+	OIDCTokenPath = filepath.Join(AgentConfigDir, OIDCToken)
 }

tool/paths/paths_unix.go

@@ -20,11 +20,19 @@ const (
 	ModeK8sOnPrem = "K8sOnPrem"
 )
 
+// Azure platform modes: ModeAzureVM is host-level (like ModeEC2), ModeAKS is Kubernetes-level (like ModeEKS).
+const (
+	ModeAzureVM = "AzureVM"
+	ModeAKS     = "AKS"
+)
+
 const (
 	ShortModeEC2       = "EC2"
 	ShortModeOnPrem    = "OP"
 	ShortModeWithIRSA  = "WI"
 	ShortModeEKS       = "EKS"
 	ShortModeK8sEC2    = "K8E"
 	ShortModeK8sOnPrem = "K8OP"
+	ShortModeAzureVM   = "AZVM"
+	ShortModeAKS       = "AKS"
 )

tool/paths/paths_windows.go

@@ -129,8 +129,11 @@ func (ctx *Context) SetMode(mode string) {
 	case config.ModeWithIRSA:
 		ctx.mode = config.ModeWithIRSA
 		ctx.shortMode = config.ShortModeWithIRSA
+	case config.ModeAzureVM:
+		ctx.mode = config.ModeAzureVM
+		ctx.shortMode = config.ShortModeAzureVM
 	default:
-		log.Panicf("Invalid mode %s. Valid mode values are %s, %s, %s, and %s.", mode, config.ModeEC2, config.ModeOnPrem, config.ModeOnPremise, config.ModeWithIRSA)
+		log.Panicf("Invalid mode %s. Valid mode values are %s, %s, %s, %s, and %s.", mode, config.ModeEC2, config.ModeOnPrem, config.ModeOnPremise, config.ModeWithIRSA, config.ModeAzureVM)
 	}
 }
 
@@ -145,6 +148,9 @@ func (ctx *Context) SetKubernetesMode(mode string) {
 	case config.ModeK8sOnPrem:
 		ctx.kubernetesMode = config.ModeK8sOnPrem
 		ctx.shortMode = config.ShortModeK8sOnPrem
+	case config.ModeAKS:
+		ctx.kubernetesMode = config.ModeAKS
+		ctx.shortMode = config.ShortModeAKS
 	default:
 		ctx.kubernetesMode = ""
 	}