From 51b08ef89e56850bd0995301abdbcaa12a9c2173 Mon Sep 17 00:00:00 2001
From: Kirby <kirby@hyrule.ovh>
Date: Fri, 2 Jun 2023 22:32:58 +0200
Subject: [PATCH] Adding default varnish list builder from env variable

---
 app/Varnish/utils.go | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/app/Varnish/utils.go b/app/Varnish/utils.go
index 45dc082..f3c7fe7 100644
--- a/app/Varnish/utils.go
+++ b/app/Varnish/utils.go
@@ -23,8 +23,10 @@ func InitializeVarnishList() []string {
 		return GetVarnishListFromVault()
 	case "file":
 		return GetVarnishListFromFile()
+	case "env":
+		return GetVarnishListFromEnv()
 	default:
-		panic("LIST_METHOD empty, no provided method to retrieve varnish list")
+		return GetVarnishListFromEnv()
 	}
 }
 
@@ -50,6 +52,15 @@ func GetVarnishListFromVault() []string {
 	return value
 }
 
+// GetVarnishListFromEnv reads the list of varnish servers from env variable VARNISH_SERVERS
+func GetVarnishListFromEnv() []string {
+	data, err := os.Getenv("VARNISH_SERVERS")
+	if err != nil {
+		log.Fatal(err)
+	}
+	sliceData := strings.Split(string(Data), ",")
+	return sliceData
+
 // GetVarnishListFromFile reads the list of varnish servers from a file on disk.
 func GetVarnishListFromFile() []string {
 	Data, err := os.ReadFile("./varnish_list")