-
Notifications
You must be signed in to change notification settings - Fork 0
/
goenv.go
60 lines (48 loc) · 1.09 KB
/
goenv.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package goenv
import (
"errors"
"os"
"path"
"github.com/joho/godotenv"
)
// Errors
var (
ErrEnvFileNotFound = errors.New("Env file not found")
)
func fileExists(dir string, filename string) bool {
filepath := path.Join(dir, filename)
_, err := os.Stat(filepath)
return err != nil
}
// LoadEnv loads env files
func LoadEnv(filename string) (string, error) {
wd, err := os.Getwd()
if err != nil {
return "", err
}
if filename == "" {
filename = ".env"
}
dir := wd
for !fileExists(dir, filename) {
if dir == "/" {
return "", ErrEnvFileNotFound
}
dir = path.Dir(dir)
}
envPath := path.Join(dir, filename)
return envPath, godotenv.Load(envPath)
}
// EnvVar helps fetching env vars and applying default values when needed
func EnvVar(envVar string, defaultValue ...string) string {
value, exists := os.LookupEnv(envVar)
if !exists && len(defaultValue) > 0 {
return defaultValue[0]
}
return value
}
// Env loads env and returns `EnvVar` function
func Env(envFilename string) func(envVar string, defaultValue ...string) string {
_, _ = LoadEnv(envFilename)
return EnvVar
}