From a2930a12241f329142bd2c7f27ebdc4d9c47dba6 Mon Sep 17 00:00:00 2001
From: Chris Pates <ChrisPates@users.noreply.github.com>
Date: Fri, 8 Mar 2024 22:21:26 +0000
Subject: [PATCH] Bugfix env vars (#175)

* Add explicit EnvVar handling for Lambda

* Update root.go
---
 cmd/root.go | 33 ++++++++++++++++++++++++++++++---
 1 file changed, 30 insertions(+), 3 deletions(-)

diff --git a/cmd/root.go b/cmd/root.go
index c7e72bc..b0de3f7 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -19,6 +19,7 @@ import (
 	"context"
 	"fmt"
 	"os"
+	"strings"
 
 	"github.com/aws/aws-lambda-go/events"
 	"github.com/aws/aws-lambda-go/lambda"
@@ -192,7 +193,7 @@ func initConfig() {
 }
 
 func configLambda() {
-	s := session.Must(session.NewSession())
+        s := session.Must(session.NewSession())
 	svc := secretsmanager.New(s)
 	secrets := config.NewSecrets(svc)
 
@@ -232,9 +233,9 @@ func configLambda() {
 	}
 	cfg.IdentityStoreID = unwrap
 
-	unwrap = os.Getenv("LOG_LEVEL")
+        unwrap = os.Getenv("LOG_LEVEL")
         if len([]rune(unwrap)) != 0 {
-	   cfg.LogLevel = unwrap
+           cfg.LogLevel = unwrap
         }
 
         unwrap = os.Getenv("LOG_FORMAT")
@@ -246,6 +247,32 @@ func configLambda() {
         if len([]rune(unwrap)) != 0 {
            cfg.SyncMethod = unwrap
         }
+
+	unwrap = os.Getenv("USER_MATCH")
+        if len([]rune(unwrap)) != 0 {
+	   cfg.UserMatch = unwrap
+        }
+
+	unwrap = os.Getenv("GROUP_MATCH")
+        if len([]rune(unwrap)) != 0 {
+           cfg.GroupMatch = unwrap
+        }
+
+        unwrap = os.Getenv("IGNORE_GROUPS")
+        if len([]rune(unwrap)) != 0 {
+           cfg.IgnoreGroups = strings.Split(unwrap, ",")
+        }
+
+        unwrap = os.Getenv("IGNORE_USERS")
+        if len([]rune(unwrap)) != 0 {
+           cfg.IgnoreUsers = strings.Split(unwrap, ",")
+        }
+
+        unwrap = os.Getenv("INCLUDE_GROUPS")
+        if len([]rune(unwrap)) != 0 {
+           cfg.IncludeGroups = strings.Split(unwrap, ",")
+        }
+
 }
 
 func addFlags(cmd *cobra.Command, cfg *config.Config) {