graphicek/server/services/sessions_service.go

package services

import (
	"time"
)

type SessionsService struct {
	ctx *Context
}

type SessionItem struct {
	Id        string `json:"id"`
	ExpiresAt int64  `json:"expiresAt"`
}

func (s *SessionsService) GetById(id string) (*SessionItem, error) {
	item := SessionItem{}

	row := s.ctx.DB.QueryRow("SELECT id, expires_at FROM sessions WHERE id = ? AND expires_at > ?", id, time.Now().Unix())

	err := row.Scan(&item.Id, &item.ExpiresAt)
	if err != nil {
		return nil, err
	}

	return &item, nil
}

func (s *SessionsService) Create() (*SessionItem, error) {
	id, err := generateRandomString(128)

	if err != nil {
		return nil, err
	}

	item := SessionItem{
		// TODO: The key is not guaranteed to be unique, how do we guarantee that?
		Id:        id,
		ExpiresAt: generateExpiryDate().Unix(),
	}

	_, err = s.ctx.DB.Exec("INSERT INTO sessions (id, expires_at) VALUES (?, ?)", item.Id, item.ExpiresAt)

	if err != nil {
		return nil, err
	}

	return &item, nil
}

func (s *SessionsService) Destroy(sessionId string) error {
	_, err := s.ctx.DB.Exec("DELETE FROM sessions WHERE id = ?", sessionId)

	return err
}

func (s *SessionsService) Extend(session *SessionItem) error {
	session.ExpiresAt = generateExpiryDate().Unix()

	_, err := s.ctx.DB.Exec("UPDATE sessions SET expires_at = ? WHERE id = ?", session.ExpiresAt, session.Id)

	return err
}

func (s *SessionsService) Cleanup() error {
	_, err := s.ctx.DB.Exec("DELETE FROM sessions WHERE expires_at < ?", time.Now().Unix())

	return err
}

func generateExpiryDate() time.Time {
	return time.Now().Add(time.Hour * 24)
}