session.go

package session

import (
	"github.com/goal-web/contracts"
	"github.com/goal-web/supports/logs"
	"github.com/goal-web/supports/utils"
	"net/http"
	"time"
)

// Session 后期会拆成 session 和 session store ，支持用 redis 、memcached、database 等其他方式存储 session
type Session struct {
	id         string
	name       string
	started    bool
	path       string
	domain     string
	lifetime   time.Duration
	attributes map[string]string
	request    contracts.HttpRequest
	changed    bool

	store contracts.SessionStore
}

func New(config Config, request contracts.HttpRequest, store contracts.SessionStore) contracts.Session {
	return &Session{
		id:         "",
		name:       config.Name,
		started:    false,
		request:    request,
		path:       request.Path(),
		store:      store,
		domain:     config.Domain,
		lifetime:   config.Lifetime,
		attributes: map[string]string{},
	}
}

func (this *Session) GetName() string {
	return this.name
}

func (this *Session) SetName(name string) {
	this.name = name
}

func (this *Session) GetId() string {
	return this.id
}

func (this *Session) SetId(id string) {
	this.id = id
}

func (this *Session) Start() bool {
	cookie, err := this.request.Cookie(this.name)
	if err != nil {
		logs.WithError(err).Debug("Failed to load cookies")
	} else {
		this.id = cookie.Value
	}

	if this.id == "" {
		this.generateSessionId()
	}
	this.loadSession()
	this.started = true
	return true
}

func (this *Session) loadSession() {
	this.attributes = this.store.LoadSession(this.GetId())
}

func (this *Session) Save() {
	if this.changed {
		this.store.Save(this.GetId(), this.attributes)
	}
}

func (this *Session) All() map[string]string {
	return this.attributes
}

func (this *Session) Exists(key string) bool {
	_, exists := this.attributes[key]
	return exists
}

func (this *Session) Has(key string) bool {
	value, exists := this.attributes[key]
	return exists && value != ""
}

func (this *Session) Get(key, defaultValue string) string {
	value, exists := this.attributes[key]
	if !exists || value == "" {
		return defaultValue
	}
	return value
}

func (this *Session) Pull(key, defaultValue string) string {
	this.changed = true
	value, exists := this.attributes[key]
	if !exists || value == "" {
		return defaultValue
	}
	delete(this.attributes, key)
	return value
}

func (this *Session) Put(key, value string) {
	this.changed = true
	this.attributes[key] = value
}

func (this *Session) Token() string {
	return this.Get("_token", "")
}

func (this *Session) RegenerateToken() {
	this.id = utils.RandStr(40)
	this.request.SetCookie(&http.Cookie{
		Name:    this.name,
		Value:   this.id,
		Expires: time.Now().Add(time.Second * this.lifetime),
	})
}

func (this *Session) Remove(key string) string {
	return this.Pull(key, "")
}

func (this *Session) Forget(keys ...string) {
	this.changed = true
	for _, key := range keys {
		delete(this.attributes, key)
	}
}

func (this *Session) Flush() {
	this.changed = true
	this.attributes = make(map[string]string)
}

func (this *Session) Invalidate() bool {
	this.Flush()
	return this.Migrate(true)
}

func (this *Session) Regenerate(destroy bool) bool {
	if !this.Migrate(destroy) {
		this.RegenerateToken()
	}
	return true
}

func (this *Session) Migrate(destroy bool) bool {
	if destroy {
		// todo: $this->handler->destroy($this->getId());
	}
	this.generateSessionId()
	return true
}

func (this *Session) IsStarted() bool {
	return this.started
}

func (this *Session) generateSessionId() {
	this.id = utils.RandStr(40)
	this.request.SetCookie(&http.Cookie{
		Name:    this.name,
		Value:   this.id,
		Expires: time.Now().Add(time.Second * this.lifetime),
	})
}

func (this *Session) PreviousUrl() string {
	return this.Get("_previous.url", "")
}

func (this *Session) SetPreviousUrl(url string) {
	this.Put("_previous.url", url)
}