diff --git a/mage_helpers/file_system.go b/mage_helpers/file_system.go index dd5fb889af338671d958e27b0b1e17d7870a3487..b45122a72110f20cdb1ef1b55f992302f01e1195 100644 --- a/mage_helpers/file_system.go +++ b/mage_helpers/file_system.go @@ -121,7 +121,7 @@ func CurrentBasePath() string { return base } -func CurrentEnv() string { +func CurrentEnv(defaultEnv ...string) string { // Get env from CI commit branch env := os.Getenv("CI_COMMIT_BRANCH") @@ -142,7 +142,12 @@ func CurrentEnv() string { } if env != "dev" && env != "stage" && env != "qa" && env != "sandbox" && env != "prod" && env != "playground" { - env = "dev" + if defaultEnv != nil && len(defaultEnv) > 0 { + forcedEnv := defaultEnv[0] + env = forcedEnv + } else { + env = "dev" + } } return env }