Add arg to suppress warning for empty shell var

justjanne · Dec 1, 2020 · 498a461 · 498a461
1 parent 79e78b1
commit 498a461
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -245,6 +245,8 @@ Usage of powerline-go:
     	 (default "bash")
   -shell-var string
     	 A shell variable to add to the segments.
+  -shell-var-no-warn-empty
+    	 Disables warning for empty shell variable.
   -shorten-eks-names
     	 Shortens names for EKS Kube clusters.
   -shorten-gke-names

diff --git a/main.go b/main.go
@@ -54,6 +54,7 @@ type args struct {
 	ShortenGKENames        *bool
 	ShortenEKSNames        *bool
 	ShellVar               *string
+	ShellVarNoWarnEmpty    *bool
 	TrimADDomain           *bool
 	PathAliases            *string
 	Duration               *string
@@ -244,6 +245,10 @@ func main() {
 			"shell-var",
 			"",
 			comments("A shell variable to add to the segments.")),
+		ShellVarNoWarnEmpty: flag.Bool(
+			"shell-var-no-warn-empty",
+			false,
+			comments("Disables warning for empty shell variable.")),
 		TrimADDomain: flag.Bool(
 			"trim-ad-domain",
 			false,

diff --git a/segment-shellvar.go b/segment-shellvar.go
@@ -18,8 +18,9 @@ func segmentShellVar(p *powerline) []pwl.Segment {
 				Background: p.theme.ShellVarBg,
 			}}
 		}
-		warn("Shell variable " + shellVarName + " is empty.")
-
+		if !*p.args.ShellVarNoWarnEmpty {
+			warn("Shell variable " + shellVarName + " is empty.")
+		}
 	} else {
 		warn("Shell variable " + shellVarName + " does not exist.")
 	}