add CacheDir function

This commit is contained in:
c9s 2020-10-20 11:48:44 +08:00
parent f62f3b8a02
commit f6c1ed67e6

View File

@ -5,6 +5,21 @@ import (
"path" "path"
) )
func prepareDir(p string) string {
_, err := os.Stat(p)
if err != nil {
_ = os.Mkdir(p, 0777)
}
return p
}
func CacheDir() string {
home := HomeDir()
dir := path.Join(home, "cache")
return prepareDir(dir)
}
func SourceDir() string { func SourceDir() string {
home := HomeDir() home := HomeDir()
return path.Join(home, "source") return path.Join(home, "source")
@ -16,12 +31,6 @@ func HomeDir() string {
panic(err) panic(err)
} }
dotDir := path.Join(homeDir, ".bbgo") dir := path.Join(homeDir, ".bbgo")
return prepareDir(dir)
_, err = os.Stat(dotDir)
if err != nil {
_ = os.Mkdir(dotDir, 0777)
}
return dotDir
} }