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 {
|
func absPathify(inPath string) string {
|
||||||
jww.INFO.Println("Trying to resolve absolute path to", inPath)
|
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:]
|
inPath = userHomeDir() + inPath[5:]
|
||||||
}
|
}
|
||||||
|
|
||||||
if strings.HasPrefix(inPath, "$") {
|
if strings.HasPrefix(inPath, "$") {
|
||||||
end := strings.Index(inPath, string(os.PathSeparator))
|
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) {
|
if filepath.IsAbs(inPath) {
|
||||||
|
37
util_test.go
37
util_test.go
@ -11,6 +11,8 @@
|
|||||||
package viper
|
package viper
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
@ -52,3 +54,38 @@ func TestCopyAndInsensitiviseMap(t *testing.T) {
|
|||||||
t.Fatal("Input map changed")
|
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