From e77a67a3de5ccda8b53e50456a1b89b271b420d7 Mon Sep 17 00:00:00 2001
From: Urko <urko@fungimail.com>
Date: Thu, 9 Mar 2023 13:29:05 +0100
Subject: [PATCH] feat: load env from file

---
 cmd/main.go      | 13 +++++++++----
 cmd/main_test.go |  4 ++--
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/cmd/main.go b/cmd/main.go
index e192d7d..cfde024 100644
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -27,9 +27,9 @@ var envConfig struct {
 
 var writer pkgio.WriterIface
 
-func intEnvConfig(isProd bool) {
-	if !isProd {
-		err := godotenv.Load(util.RootDir() + "/.env")
+func intEnvConfig(envFilePath string) {
+	if envFilePath != "" {
+		err := godotenv.Load(envFilePath)
 		if err != nil {
 			log.Fatalf("environment variable ENV is empty and an error occurred while loading the .env file: %s\n", err)
 		}
@@ -134,7 +134,12 @@ func exportPem(filename string, data []byte) (string, error) {
 }
 
 func init() {
-	intEnvConfig(false)
+	envFile := ""
+	if os.Getenv("ENV") != "prod" {
+		envFile = "./.env"
+	}
+
+	intEnvConfig(envFile)
 	cobra.OnInitialize(initConfig)
 }
 
diff --git a/cmd/main_test.go b/cmd/main_test.go
index a3ff2d3..b47c442 100644
--- a/cmd/main_test.go
+++ b/cmd/main_test.go
@@ -44,8 +44,8 @@ func init() {
 func Test_exportPem(t *testing.T) {
 	defer func() {
 		os.Remove(testFile)
-		// err := os.Remove(testFile)
-		// require.NoError(t, err)
+		err := os.Remove(testFile)
+		require.NoError(t, err)
 	}()
 
 	data := []byte("test data")