✨ feat: Add functionality to load environment variable files

zutil/env.go

@@ -3,8 +3,10 @@ package zutil
 import (
 	"os"
 	"runtime"
+	"strings"
 
 	"github.com/sohaha/zlsgo/zfile"
+	"github.com/sohaha/zlsgo/zstring"
 )
 
 func GetOs() string {
@@ -38,3 +40,35 @@ func Getenv(name string, def ...string) string {
 func GOROOT() string {
 	return zfile.RealPath(runtime.GOROOT())
 }
+
+func Loadenv(filenames ...string) (err error) {
+	filenames = filenamesOrDefault(filenames)
+
+	for _, filename := range filenames {
+		err = loadFile(filename)
+		if err != nil {
+			return
+		}
+	}
+	return
+}
+
+func filenamesOrDefault(filenames []string) []string {
+	if len(filenames) == 0 {
+		return []string{".env"}
+	}
+	return filenames
+}
+
+func loadFile(filename string) error {
+	return zfile.ReadLineFile(filename, func(line int, data []byte) error {
+		e := strings.Split(zstring.Bytes2String(data), "=")
+		var value string
+		if len(e) > 1 {
+			value = e[1]
+		}
+
+		_ = os.Setenv(zstring.TrimSpace(e[0]), strings.TrimSpace(value))
+		return nil
+	})
+}

zutil/env_test.go

@@ -5,6 +5,7 @@ import (
 	"testing"
 
 	"github.com/sohaha/zlsgo"
+	"github.com/sohaha/zlsgo/zfile"
 	"github.com/sohaha/zlsgo/zutil"
 )
 
@@ -19,6 +20,7 @@ func TestOs(T *testing.T) {
 	isMac := zutil.IsMac()
 	t.Log("isMac", isMac)
 }
+
 func TestEnv(T *testing.T) {
 	t := zlsgo.NewTest(T)
 	t.Log(zutil.Getenv("HOME"))
@@ -29,3 +31,14 @@ func TestEnv(T *testing.T) {
 func TestGOROOT(t *testing.T) {
 	t.Log(zutil.GOROOT())
 }
+
+func TestLoadenv(t *testing.T) {
+	tt := zlsgo.NewTest(t)
+	_ = zfile.WriteFile(".env", []byte("myos=linux\n name=zls "))
+	defer zfile.Rmdir(".env")
+
+	tt.NoError(zutil.Loadenv())
+
+	tt.Equal("linux", zutil.Getenv("myos"))
+	tt.Equal("zls", zutil.Getenv("name"))
+}