Handle empty strings in flag name concatenation (#535)

Layr-Labs · May 1, 2024 · 8315816 · 8315816
1 parent 06f794b
commit 8315816
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/common/common.go b/common/common.go
@@ -10,11 +10,23 @@ import (
 
 // PrefixEnvVar returns the environment variable name with the given prefix and suffix
 func PrefixEnvVar(prefix, suffix string) string {
+	if prefix == "" {
+		return suffix
+	}
+	if suffix == "" {
+		return prefix
+	}
 	return prefix + "_" + suffix
 }
 
 // PrefixFlag returns the flag name with the given prefix and suffix
 func PrefixFlag(prefix, suffix string) string {
+	if prefix == "" {
+		return suffix
+	}
+	if suffix == "" {
+		return prefix
+	}
 	return prefix + "." + suffix
 }