Use APP_ENV when choosing what .env file to load

Default to .env.production to keep existing behavior

Author: Sean-Der

internal/environment/environment.go

@@ -1,52 +1,62 @@
 package environment
 
 import (
+	"errors"
 	"log"
 	"os"
 	"path/filepath"
 
 	"github.com/joho/godotenv"
 )
 
+const (
+	envFileDevelopment = ".env.development"
+	envFileProduction  = ".env.production"
+)
+
+var errNoBuildDirectory = errors.New("build directory does not exist, run `npm install` and `npm run build` in the web directory")
+
 func LoadEnvironmentVariables() {
-	files := []string{
-		".env.development",
-		".env.production",
-	}
+	if err := loadConfigs(); err != nil {
+		if errors.Is(err, errNoBuildDirectory) {
+			log.Fatal("Environment:", err)
+		}
 
-	// Load base environment file if available
-	loadEnvironmentFile(".env")
+		log.Println("Environment: Failed to find config in CWD, changing CWD to executable path")
 
-	for _, file := range files {
-		loadEnvironmentFile(file)
-		setDefaultEnvironmentVariables()
-		return
-	}
+		executablePath, executableErr := os.Executable()
+		if executableErr != nil {
+			log.Fatal("Environment:", executableErr)
+		}
 
-	log.Println("Environment: Could not find any environment files")
-	os.Exit(0)
-}
+		if chdirErr := os.Chdir(filepath.Dir(executablePath)); chdirErr != nil {
+			log.Fatal("Environment:", chdirErr)
+		}
 
-func loadEnvironmentFile(filePath string) {
-	currentWorkingDirectory, err := os.Getwd()
-	if err != nil {
-		log.Fatal("Environment:", err)
+		if retryErr := loadConfigs(); retryErr != nil {
+			log.Fatal("Environment:", retryErr)
+		}
 	}
 
-	path := filepath.Join(currentWorkingDirectory, filePath)
-
-	if _, err := os.Stat(path); err == nil {
-		err := godotenv.Overload(path)
+	setDefaultEnvironmentVariables()
+}
 
-		if err != nil {
-			log.Println("Environment: Error occurred loading environment file", path)
-			log.Println(err)
+func loadConfigs() error {
+	if os.Getenv(APP_ENV) == "development" {
+		log.Println("Environment: Loading `" + envFileDevelopment + "`")
+		return godotenv.Load(envFileDevelopment)
+	}
 
-			os.Exit(0)
-		}
+	log.Println("Environment: Loading `" + envFileProduction + "`")
+	if err := godotenv.Load(envFileProduction); err != nil {
+		return err
+	}
 
-		log.Println("Environment: Loaded", filePath)
+	if _, err := os.Stat("./web/build"); os.IsNotExist(err) && os.Getenv(FRONTEND_DISABLED) == "" {
+		return errNoBuildDirectory
 	}
+
+	return nil
 }
 
 func setDefaultEnvironmentVariables() {

internal/environment/variables.go

@@ -2,6 +2,7 @@ package environment
 
 const (
 	// SERVER
+	APP_ENV                            = "APP_ENV"
 	HTTP_ADDRESS                       = "HTTP_ADDRESS"
 	HTTPS_REDIRECT_PORT                = "HTTPS_REDIRECT_PORT"
 	HTTP_ENABLE_REDIRECT               = "ENABLE_HTTP_REDIRECT"