localhost_frssoft/wi
1
0
Fork 0
mirror of https://code.dumpstack.io/tools/wi.git synced 2024-11-22 19:13:01 +00:00
Code Issues Projects Releases Wiki Activity

Save response cookies

Browse Source
This commit is contained in:
Mikhail Klementyev 2016-11-20 16:18:47 +03:00
parent 386c74a10b
commit f01134eaa4
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
commands/commands.go
View File

@ -132,6 +132,13 @@ func Get(db *sql.DB, linkUrl string) {
storage.AddHistoryURL(db, linkUrl) 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() defer resp.Body.Close()
utf8, err := charset.NewReader(resp.Body, resp.Header.Get("Content-Type")) 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 var status int64
fmt.Sscanf(resp.Status, "%d", &status) 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 { if status >= 300 && status < 400 {
Get(db, lastUrl.String()) Get(db, lastUrl.String())
} }