package dbmodels

import (
	"github.com/pocketbase/pocketbase/core"
	"github.com/pocketbase/pocketbase/tools/types"
)

type FixedSession struct {
	ID         string         `json:"id"`
	Token      string         `json:"token"`
	User       string         `json:"user"`
	Created    string         `json:"created"`
	Updated    string         `json:"updated"`
	Expires    types.DateTime `json:"expires"`
	IP         string         `json:"ip"`
	UserAgent  string         `json:"user_agent"`
	LastAccess types.DateTime `json:"last_access"`
	Persist    bool           `json:"persist"`
	CSRF       string         `json:"csrf"`
}

func (s *FixedSession) IsExpired() bool {
	return s.Expires.IsZero() || s.Expires.Before(types.NowDateTime())
}

var _ core.RecordProxy = (*Place)(nil)

type Session struct {
	core.BaseRecordProxy
}

func NewSession(record *core.Record) *Session {
	i := &Session{}
	i.SetProxyRecord(record)
	return i
}

func (u *Session) TableName() string {
	return SESSIONS_TABLE
}

func (u *Session) Token() string {
	return u.GetString(SESSIONS_TOKEN_FIELD)
}

func (u *Session) SetToken(token string) {
	u.Set(SESSIONS_TOKEN_FIELD, token)
}

func (u *Session) User() string {
	return u.GetString(SESSIONS_USER_FIELD)
}

func (u *Session) SetUser(userId string) {
	u.Set(SESSIONS_USER_FIELD, userId)
}

func (u *Session) Created() string {
	return u.GetString(CREATED_FIELD)
}

func (u *Session) Updated() string {
	return u.GetString(UPDATED_FIELD)
}

func (u *Session) Expires() types.DateTime {
	return u.GetDateTime(SESSIONS_EXPIRES_FIELD)
}

func (u *Session) SetExpires(expires types.DateTime) {
	u.Set(SESSIONS_EXPIRES_FIELD, expires)
}

func (u *Session) IP() string {
	return u.GetString(SESSIONS_IP_FIELD)
}

func (u *Session) SetIP(ip string) {
	u.Set(SESSIONS_IP_FIELD, ip)
}

func (u *Session) UserAgent() string {
	return u.GetString(SESSIONS_USER_AGENT_FIELD)
}

func (u *Session) SetUserAgent(userAgent string) {
	u.Set(SESSIONS_USER_AGENT_FIELD, userAgent)
}

func (u *Session) LastAccess() types.DateTime {
	return u.GetDateTime(SESSIONS_LAST_ACCESS_FIELD)
}

func (u *Session) SetLastAccess(lastAccess types.DateTime) {
	u.Set(SESSIONS_LAST_ACCESS_FIELD, lastAccess)
}

func (u *Session) Persist() bool {
	return u.GetBool(SESSIONS_PERSIST_FIELD)
}

func (u *Session) SetPersist(persist bool) {
	u.Set(SESSIONS_PERSIST_FIELD, persist)
}

func (u *Session) CSRF() string {
	return u.GetString(SESSIONS_CSRF_FIELD)
}

func (u *Session) SetCSRF(csrf string) {
	u.Set(SESSIONS_CSRF_FIELD, csrf)
}

func (u *Session) IsExpired() bool {
	return u.Expires().IsZero() || u.Expires().Before(types.NowDateTime())
}

func (u *Session) Status() string {
	return u.GetString(SESSIONS_STATUS_FIELD)
}

func (u *Session) SetStatus(status string) {
	u.Set(SESSIONS_STATUS_FIELD, status)
}

func (u *Session) Fixed() FixedSession {
	return FixedSession{
		ID:         u.Id,
		Token:      u.Token(),
		User:       u.User(),
		Created:    u.Created(),
		Updated:    u.Updated(),
		Expires:    u.Expires(),
		IP:         u.IP(),
		UserAgent:  u.UserAgent(),
		LastAccess: u.LastAccess(),
		Persist:    u.Persist(),
		CSRF:       u.CSRF(),
	}
}