From f01134eaa46c944b5f2df5d13eaf42f5a7a1036e Mon Sep 17 00:00:00 2001
From: Mikhail Klementyev <jollheef@riseup.net>
Date: Sun, 20 Nov 2016 16:18:47 +0300
Subject: [PATCH] Save response cookies

---
 commands/commands.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/commands/commands.go b/commands/commands.go
index 073e29a..b4afeea 100644
--- a/commands/commands.go
+++ b/commands/commands.go
@@ -132,6 +132,13 @@ func Get(db *sql.DB, linkUrl string) {
 
 	storage.AddHistoryURL(db, linkUrl)
 
+	if len(resp.Cookies()) != 0 {
+		err = storage.AddCookies(db, lastUrl.Host, resp.Cookies())
+		if err != nil {
+			log.Fatalln("Add cookies:", err)
+		}
+	}
+
 	defer resp.Body.Close()
 
 	utf8, err := charset.NewReader(resp.Body, resp.Header.Get("Content-Type"))
@@ -231,6 +238,13 @@ func Form(db *sql.DB, formID int64, formArgs []string) {
 	var status int64
 	fmt.Sscanf(resp.Status, "%d", &status)
 
+	if status < 400 && len(resp.Cookies()) != 0 {
+		err = storage.AddCookies(db, lastUrl.Host, resp.Cookies())
+		if err != nil {
+			log.Fatalln("Add cookies:", err)
+		}
+	}
+
 	if status >= 300 && status < 400 {
 		Get(db, lastUrl.String())
 	}