quartzgun/cookie/cookie.go

package cookie

import (
	"crypto/rand"
	"net/http"
	"time"
)

var availableChars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ@!.#$_"

func GenToken(length int) string {
	ll := len(availableChars)
	b := make([]byte, length)
	rand.Read(b)
	for i := 0; i < length; i++ {
		b[i] = availableChars[int(b[i])%ll]
	}
	return string(b)
}

func StoreToken(field string, token string, w http.ResponseWriter, hrs int) {
	cookie := http.Cookie{
		Name:    field,
		Value:   token,
		Expires: time.Now().Add(time.Duration(hrs) * time.Hour),
	}

	http.SetCookie(w, &cookie)
}

func GetToken(field string, req *http.Request) (string, error) {
	c, err := req.Cookie(field)
	if err == nil {
		return c.Value, nil
	} else {
		return "", err
	}
}