mirror of
https://github.com/spf13/viper.git
synced 2024-12-26 14:01:14 -07:00
Fix environment variable expansion in absPathify
- Don't expand user home directory for variable names that simply have a HOME prefix; - Support expansion of variables not followed by the path separator.
This commit is contained in:
parent
13494e8047
commit
ce534045f9
14
util.go
14
util.go
@ -91,13 +91,23 @@ func insensitiviseMap(m map[string]interface{}) {
|
||||
func absPathify(inPath string) string {
|
||||
jww.INFO.Println("Trying to resolve absolute path to", inPath)
|
||||
|
||||
if strings.HasPrefix(inPath, "$HOME") {
|
||||
if strings.HasPrefix(inPath, "$HOME") &&
|
||||
(len(inPath) == 5 || inPath[5] == os.PathSeparator) {
|
||||
inPath = userHomeDir() + inPath[5:]
|
||||
}
|
||||
|
||||
if strings.HasPrefix(inPath, "$") {
|
||||
end := strings.Index(inPath, string(os.PathSeparator))
|
||||
inPath = os.Getenv(inPath[1:end]) + inPath[end:]
|
||||
|
||||
var value, suffix string
|
||||
if end == -1 {
|
||||
value = os.Getenv(inPath[1:])
|
||||
} else {
|
||||
value = os.Getenv(inPath[1:end])
|
||||
suffix = inPath[end:]
|
||||
}
|
||||
|
||||
inPath = value + suffix
|
||||
}
|
||||
|
||||
if filepath.IsAbs(inPath) {
|
||||
|
37
util_test.go
37
util_test.go
@ -11,6 +11,8 @@
|
||||
package viper
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
@ -52,3 +54,38 @@ func TestCopyAndInsensitiviseMap(t *testing.T) {
|
||||
t.Fatal("Input map changed")
|
||||
}
|
||||
}
|
||||
|
||||
func TestAbsPathify(t *testing.T) {
|
||||
home := userHomeDir()
|
||||
homer := filepath.Join(home, "homer")
|
||||
wd, _ := os.Getwd()
|
||||
|
||||
os.Setenv("HOMER_ABSOLUTE_PATH", homer)
|
||||
os.Setenv("VAR_WITH_RELATIVE_PATH", "relative")
|
||||
|
||||
tests := []struct {
|
||||
input string
|
||||
output string
|
||||
}{
|
||||
{"", wd},
|
||||
{"sub", filepath.Join(wd, "sub")},
|
||||
{"./", wd},
|
||||
{"./sub", filepath.Join(wd, "sub")},
|
||||
{"$HOME", home},
|
||||
{"$HOME/", home},
|
||||
{"$HOME/sub", filepath.Join(home, "sub")},
|
||||
{"$HOMER_ABSOLUTE_PATH", homer},
|
||||
{"$HOMER_ABSOLUTE_PATH/", homer},
|
||||
{"$HOMER_ABSOLUTE_PATH/sub", filepath.Join(homer, "sub")},
|
||||
{"$VAR_WITH_RELATIVE_PATH", filepath.Join(wd, "relative")},
|
||||
{"$VAR_WITH_RELATIVE_PATH/", filepath.Join(wd, "relative")},
|
||||
{"$VAR_WITH_RELATIVE_PATH/sub", filepath.Join(wd, "relative", "sub")},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
got := absPathify(test.input)
|
||||
if got != test.output {
|
||||
t.Errorf("Got %v\nexpected\n%q", got, test.output)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user