Support sending log messages to stderr by setting AWS_SSM_QUIET=true

Signed-off-by: Chance Zibolski <[email protected]>
aws · Apr 4, 2024 · ccc419c · ccc419c
1 parent 3b753c7
commit ccc419c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/sessionmanagerplugin-main/main.go b/src/sessionmanagerplugin-main/main.go
@@ -22,6 +22,14 @@ import (
 	_ "github.com/aws/session-manager-plugin/src/sessionmanagerplugin/session/shellsession"
 )
 
+var out = os.Stdout
+
+func init() {
+	if quietStr := os.Getenv("AWS_SSM_QUIET"); quietStr == "true" || quietStr == "1" {
+		out = os.Stderr
+	}
+}
+
 func main() {
-	session.ValidateInputAndStartSession(os.Args, os.Stdout)
+	session.ValidateInputAndStartSession(os.Args, out)
 }