mirror of
https://code.dumpstack.io/tools/wi.git
synced 2024-11-26 21:13:02 +00:00
Implements cookie serialization/deserialization
This commit is contained in:
parent
5d37007921
commit
386c74a10b
@ -9,14 +9,61 @@
|
|||||||
package storage
|
package storage
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/gob"
|
||||||
"errors"
|
"errors"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
_ "github.com/mattn/go-sqlite3"
|
_ "github.com/mattn/go-sqlite3"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func toGOB64(m http.Cookie) string {
|
||||||
|
b := bytes.Buffer{}
|
||||||
|
e := gob.NewEncoder(&b)
|
||||||
|
err := e.Encode(m)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return base64.StdEncoding.EncodeToString(b.Bytes())
|
||||||
|
}
|
||||||
|
|
||||||
|
func fromGOB64(str string) http.Cookie {
|
||||||
|
m := http.Cookie{}
|
||||||
|
by, err := base64.StdEncoding.DecodeString(str)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
b := bytes.Buffer{}
|
||||||
|
b.Write(by)
|
||||||
|
d := gob.NewDecoder(&b)
|
||||||
|
err = d.Decode(&m)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func serializeCookies(cookies []*http.Cookie) (s string) {
|
||||||
|
for _, c := range cookies {
|
||||||
|
s += toGOB64(*c) + " "
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func deserializeCookies(s string) (cookies []*http.Cookie) {
|
||||||
|
gob64Objects := strings.Split(s, " ")
|
||||||
|
for _, g := range gob64Objects {
|
||||||
|
c := fromGOB64(g)
|
||||||
|
cookies = append(cookies, &c)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
func OpenDB(path string) (db *sql.DB, err error) {
|
func OpenDB(path string) (db *sql.DB, err error) {
|
||||||
db, err = sql.Open("sqlite3", path)
|
db, err = sql.Open("sqlite3", path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -49,6 +96,45 @@ func OpenDB(path string) (db *sql.DB, err error) {
|
|||||||
"( `id` INTEGER PRIMARY KEY AUTOINCREMENT, " +
|
"( `id` INTEGER PRIMARY KEY AUTOINCREMENT, " +
|
||||||
" `post` BOOLEAN, " +
|
" `post` BOOLEAN, " +
|
||||||
" `url` TEXT );")
|
" `url` TEXT );")
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = db.Exec("CREATE TABLE IF NOT EXISTS `cookies` " +
|
||||||
|
"( `id` INTEGER PRIMARY KEY AUTOINCREMENT, " +
|
||||||
|
" `url` BOOLEAN, " +
|
||||||
|
" `cookies` TEXT );")
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func AddCookies(db *sql.DB, url string, cookies []*http.Cookie) (err error) {
|
||||||
|
log.Println("Add cookies", url, cookies)
|
||||||
|
stmt, err := db.Prepare("INSERT INTO `cookies` " +
|
||||||
|
"(`url`, `cookies`) VALUES ($1, $2);")
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer stmt.Close()
|
||||||
|
|
||||||
|
_, err = stmt.Exec(url, serializeCookies(cookies))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCookies(db *sql.DB, url string) (cookies []*http.Cookie, err error) {
|
||||||
|
stmt, err := db.Prepare("SELECT `cookies` FROM `cookies` WHERE url=$1;")
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer stmt.Close()
|
||||||
|
|
||||||
|
var rawCookies string
|
||||||
|
err = stmt.QueryRow(url).Scan(&rawCookies)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cookies = deserializeCookies(rawCookies)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user