Added env var override

Jelloeater · Sep 9, 2024 · 0f3215a · 0f3215a
1 parent db92c54
commit 0f3215a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -17,6 +17,13 @@ var ( // Create by GoRelease at compile time
 
 func writeDate(format string, timezone string) {
 
+	if os.Getenv("STAMPY_TZ") != "" {
+		timezone = os.Getenv("STAMPY_TZ")
+	}
+	if os.Getenv("STAMPY_FORMAT") != "" {
+		format = os.Getenv("STAMPY_FORMAT")
+	}
+
 	loc, e := time.LoadLocation(timezone)
 	if e != nil {
 		log.Fatal(e)