package main
import (
"encoding/json"
"fmt"
"log"
"net/http"
"time"
"snix.ir/ecookie"
)
type te struct {
IMMMMM int `json:"i"`
DMMMMM int `json:"d"`
Expire time.Time `json:"expire"`
SuperSecret string `json:"secret"`
}
var key = []byte{
0x11, 0x12, 0x13, 0x14,
0x11, 0x12, 0x13, 0x14,
0x11, 0x12, 0x13, 0x14,
0x11, 0x12, 0x13, 0x14,
}
func main() {
http.HandleFunc("/set", setHand)
http.HandleFunc("/get", getHand)
err := http.ListenAndServe(":8080", nil)
log.Fatal(err)
}
func setHand(w http.ResponseWriter, r *http.Request) {
x := new(te)
x.DMMMMM = 10000
x.IMMMMM = 90000
x.Expire = time.Now().Add(time.Hour)
x.SuperSecret = "some super secret data.. "
d, err := json.Marshal(x)
if err != nil {
fmt.Println(err)
w.WriteHeader(http.StatusBadGateway)
return
}
cc, err := ecookie.NewEncryptor(key)
if err != nil {
fmt.Println(err)
w.WriteHeader(http.StatusBadGateway)
return
}
ec, err := cc.Encrypt(d)
if err != nil {
fmt.Println(err)
w.WriteHeader(http.StatusBadGateway)
return
}
http.SetCookie(w, &http.Cookie{
Name: "auth",
Value: string(ec),
})
w.WriteHeader(http.StatusOK)
w.Write(ec)
}
func getHand(w http.ResponseWriter, r *http.Request) {
x, err := r.Cookie("auth")
if err != nil {
fmt.Println(err)
return
}
ee, err := ecookie.NewDecryptor(key)
if err != nil {
fmt.Println(err)
return
}
ll, err := ee.Decrypt([]byte(x.Value))
if err != nil {
fmt.Println(err)
return
}
vv := new(te)
err = json.Unmarshal(ll, vv)
if err != nil {
fmt.Println(err)
return
}
fmt.Println(vv)
fmt.Fprintf(w, "%+v", *vv)
}