session.go

package hutplate

import (
	"github.com/gorilla/sessions"
	"net/http"
)

var store sessions.Store

type session struct {
	request        *http.Request
	responseWriter http.ResponseWriter
}

func NewSession(request *http.Request, responseWriter http.ResponseWriter, s *sessions.Store) session {
	if s == nil {
		store = sessions.NewFilesystemStore(Config.SessionDirectory, []byte(Config.SessionSecretKey))
	} else {
		store = *s
	}

	return session{
		request,
		responseWriter,
	}
}

func (s session) Set(key string, value interface{}) error {
	sessionStore, err := store.Get(s.request, "s")
	if err != nil {
		return err
	}

	sessionStore.Values[key] = value
	if err = store.Save(s.request, s.responseWriter, sessionStore); err != nil {
		return err
	}
	return nil
}

func (s session) Get(key string) (interface{}, error) {
	sessionStore, err := store.Get(s.request, "s")
	if err != nil {
		return nil, err
	}
	return sessionStore.Values[key], nil
}

func (s session) SetFlash(key string, value interface{}) error {
	sessionStore, err := store.Get(s.request, "s")
	if err != nil {
		return err
	}
	sessionStore.AddFlash(value, key)
	return store.Save(s.request, s.responseWriter, sessionStore)
}

func (s session) GetFlash(key string) interface{} {
	sessionStore, err := store.Get(s.request, "s")
	if err != nil {
		return nil
	}
	flashes := sessionStore.Flashes(key)
	var flash interface{}

	if len(flashes) == 0 {
		flash = nil
	} else {
		flash = flashes[0]
	}
	err = store.Save(s.request, s.responseWriter, sessionStore)
	return flash
}