From baff8aaaff38debb33c88e71f743e8cf29405a71 Mon Sep 17 00:00:00 2001
From: Mikhail Klementyev <jollheef@riseup.net>
Date: Sun, 20 Nov 2016 18:09:02 +0300
Subject: [PATCH] Change path for storage

---
 main.go | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index e7d301e..bbdb593 100644
--- a/main.go
+++ b/main.go
@@ -61,13 +61,30 @@ var (
 )
 
 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 {
 		panic(err)
 	}
 	defer db.Close()
 
-	os.Setenv("GOCOOKIES", "/tmp/wi.jar")
+	os.Setenv("GOCOOKIES", wijarPath)
 
 	jar, err := cookiejar.New(nil)
 	if err != nil {