Added Envirnoment variabel for the --cache flag (STAYRTR_CACHE)

bgp · Sep 26, 2024 · fc73a7f · fc73a7f
1 parent 44638b8
commit fc73a7f
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/cmd/stayrtr/stayrtr.go b/cmd/stayrtr/stayrtr.go
@@ -31,9 +31,12 @@ import (
 )
 
 const (
+	ENV_CACHE = "STAYRTR_CACHE"
 	ENV_SSH_PASSWORD = "STAYRTR_SSH_PASSWORD"
 	ENV_SSH_KEY      = "STAYRTR_SSH_AUTHORIZEDKEYS"
 
+	DEFAULT_CACHE = "https://console.rpki-client.org/rpki.json"
+
 	METHOD_NONE = iota
 	METHOD_PASSWORD
 	METHOD_KEY
@@ -80,7 +83,7 @@ var (
 
 	TimeCheck = flag.Bool("checktime", true, "Check if JSON file isn't stale (disable by passing -checktime=false)")
 
-	CacheBin = flag.String("cache", "https://console.rpki-client.org/rpki.json", "URL of the Validated RPKI data in JSON format")
+	CacheBin = flag.String("cache", DEFAULT_CACHE, fmt.Sprintf("URL of the Validated RPKI data in JSON format", ENV_CACHE))
 
 	Etag            = flag.Bool("etag", true, "Control usage of Etag header (disable with -etag=false)")
 	LastModified    = flag.Bool("last.modified", true, "Control usage of Last-Modified header (disable with -last.modified=false)")
@@ -756,6 +759,11 @@ func run() error {
 	s.fetchConfig.EnableEtags = *Etag
 	s.fetchConfig.EnableLastModified = *LastModified
 
+	cache := *CacheBin
+	if cache == DEFAULT_CACHE {
+		cache = os.Getenv(ENV_CACHE)
+	}
+
 	if enableHTTP {
 		if *ExportPath != "" {
 			http.HandleFunc(*ExportPath, s.exporter)