package session

import (
	"bytes"
	"encoding/gob"
	"encoding/json"
	"log"
	"time"
)

//Session session对象
type Session struct {
	mSessionID        string
	mUserID           string
	mLastTimeAccessed time.Time
	mValue            map[string]interface{}
	mOnSave           func(sid, value string)
	mOnSaveGob        func(sid string, sdata []byte)
}

//HasData 查找数据
func (sess *Session) HasData(key string) bool {
	_, ok := sess.mValue[key]
	return ok
}

//GetData 获取数据
func (sess *Session) GetData(key string) interface{} {
	return sess.mValue[key]
}

//PutData 存储数据
func (sess *Session) PutData(key string, value interface{}) {
	//log.Println("put data: ", this, key, value)
	sess.mValue[key] = value
	if sess.mOnSave != nil {
		sess.save()
	}
	if sess.mOnSaveGob != nil {
		sess.saveGob()
	}
}

//RemoveData 移除数据
func (sess *Session) RemoveData(key string) {
	delete(sess.mValue, key)
	if sess.mOnSave != nil {
		sess.save()
	}
	if sess.mOnSaveGob != nil {
		sess.saveGob()
	}
}

//SessID sid
func (sess *Session) SessID() string {
	return sess.mSessionID
}

//UserID 用户ID
func (sess *Session) UserID() string {
	return sess.mUserID
}

//save 数据持久化
func (sess *Session) save() {
	if sess.mOnSave == nil {
		return
	}

	bs, err := json.Marshal(sess.mValue)
	if err != nil {
		log.Println("[session] save err", err)
		return
	}

	sess.mOnSave(sess.mSessionID, string(bs))
}

func (sess *Session) saveGob() {
	if sess.mOnSaveGob == nil {
		return
	}

	var result bytes.Buffer
	err := gob.NewEncoder(&result).Encode(sess.mValue)
	if err != nil {
		log.Println("[session] save gob err", err)
		return
	}

	sess.mOnSaveGob(sess.mSessionID, result.Bytes())
}