Change path for storage

This commit is contained in:
Mikhail Klementyev 2016-11-20 18:09:02 +03:00
parent dc8b2f1b26
commit baff8aaaff

21
main.go
View File

@ -61,13 +61,30 @@ var (
) )
func main() { func main() {
db, err := storage.OpenDB("/tmp/wi.db") homePath, exists := os.LookupEnv("HOME")
var wiDir, widbPath, wijarPath string
if exists {
wiDir = homePath + "/.wi"
widbPath = wiDir + "/wi.db"
wijarPath = wiDir + "/wi.jar"
} else {
wiDir = "/tmp"
widbPath = "/tmp/wi.db"
wijarPath = "/tmp/wi.jar"
}
err := os.MkdirAll(wiDir, 0700)
if err != nil {
panic(err)
}
db, err := storage.OpenDB(widbPath)
if err != nil { if err != nil {
panic(err) panic(err)
} }
defer db.Close() defer db.Close()
os.Setenv("GOCOOKIES", "/tmp/wi.jar") os.Setenv("GOCOOKIES", wijarPath)
jar, err := cookiejar.New(nil) jar, err := cookiejar.New(nil)
if err != nil { if err != nil {