Merge pull request #3 from leapkit/fix-envor-multiple-equals

fix: fixing Envor variables with multiple equals

envor/envor.go

@@ -2,6 +2,7 @@ package envor
 
 import (
 	"bufio"
+	"io"
 	"os"
 	"strings"
 	"sync"
@@ -34,7 +35,19 @@ func loadENV() {
 		return
 	}
 
-	scanner := bufio.NewScanner(file)
+	for key, value := range parseVars(file) {
+		err := os.Setenv(key, value)
+		if err != nil {
+			continue
+		}
+	}
+}
+
+// parseVars reads the variables from the reader and sets them
+// in the environment.
+func parseVars(r io.Reader) map[string]string {
+	vars := make(map[string]string)
+	scanner := bufio.NewScanner(r)
 	for scanner.Scan() {
 		line := scanner.Text()
 		if line == "" {
@@ -46,7 +59,7 @@ func loadENV() {
 			continue
 		}
 
-		pair := strings.Split(line, "=")
+		pair := strings.SplitN(line, "=", 2)
 		if len(pair) != 2 {
 			continue
 		}
@@ -55,9 +68,8 @@ func loadENV() {
 		value := strings.TrimSpace(pair[1])
 		value = strings.Trim(value, "\"")
 
-		err := os.Setenv(key, value)
-		if err != nil {
-			continue
-		}
+		vars[key] = value
 	}
+
+	return vars
 }

envor/envor_test.go

@@ -0,0 +1,54 @@
+package envor
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestParseVars(t *testing.T) {
+
+	t.Run("simple one", func(t *testing.T) {
+		r := strings.NewReader("KEY=value\n")
+		vars := parseVars(r)
+
+		if vars["KEY"] != "value" {
+			t.Errorf("Expected value to be 'value', got %s", vars["KEY"])
+		}
+	})
+
+	t.Run("multiple", func(t *testing.T) {
+		vars := parseVars(strings.NewReader(`
+			KEY=value
+			KEY2=value
+		`))
+
+		if vars["KEY"] != "value" {
+			t.Errorf("Expected value to be 'value', got %s", vars["KEY"])
+		}
+
+		if vars["KEY2"] != "value" {
+			t.Errorf("Expected value to be 'value', got %s", vars["KEY"])
+		}
+	})
+
+	t.Run("quotes", func(t *testing.T) {
+		vars := parseVars(strings.NewReader(`
+			KEY="value"
+		`))
+
+		if vars["KEY"] != "value" {
+			t.Errorf("Expected value to be 'value', got %s", vars["KEY"])
+		}
+	})
+
+	t.Run("multiple equals sign", func(t *testing.T) {
+		vars := parseVars(strings.NewReader(`
+			KEY="value=with=equals"
+		`))
+
+		if vars["KEY"] != "value=with=equals" {
+			t.Errorf("Expected value to be 'value=with=equals', got %s", vars["KEY"])
+		}
+	})
+
+}