diff --git a/pkg/bbgo/home.go b/pkg/bbgo/home.go index a8a1a3a5a..8f7268f3d 100644 --- a/pkg/bbgo/home.go +++ b/pkg/bbgo/home.go @@ -5,6 +5,21 @@ import ( "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 { home := HomeDir() return path.Join(home, "source") @@ -16,12 +31,6 @@ func HomeDir() string { panic(err) } - dotDir := path.Join(homeDir, ".bbgo") - - _, err = os.Stat(dotDir) - if err != nil { - _ = os.Mkdir(dotDir, 0777) - } - - return dotDir + dir := path.Join(homeDir, ".bbgo") + return prepareDir(dir) }