Merge pull request #19 from TimoS254/master

Get forced Live and Excluded Homes from Environment Variable

main.go

@@ -8,6 +8,7 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"strings"
 	"time"
 
 	"github.com/hasura/go-graphql-client"
@@ -65,6 +66,26 @@ func init() {
 	} else {
 		userAgent = fmt.Sprintf("tibber-exporter/%s (https://github.com/terjesannum/tibber-exporter)", version.Version)
 	}
+
+	//Get homes with Forcefully enabled/disabled live Measurement Feature from Environment Variable and append them to slice
+	envEnabledHomes, envPresent := os.LookupEnv("TIBBER_LIVE_ENABLE_HOMES")
+	if envPresent {
+		envEnabledHomes := strings.Split(envEnabledHomes, ",")
+		for _, enabledHome := range envEnabledHomes {
+			if len(enabledHome) >= 1 {
+				liveMeasurements = append(liveMeasurements, enabledHome)
+			}
+		}
+	}
+	envDisabledHomes, envPresent := os.LookupEnv("TIBBER_LIVE_DISABLE_HOMES")
+	if envPresent {
+		envDisabledHomes := strings.Split(envDisabledHomes, ",")
+		for _, disabledHome := range envDisabledHomes {
+			if len(disabledHome) >= 1 {
+				disableLiveMeasurements = append(disableLiveMeasurements, disabledHome)
+			}
+		}
+	}
 }
 
 func exit(msg string) {