graphicek/server/services/auth_service.go

package services

import (
	"time"

	"github.com/gin-gonic/gin"
)

type AuthService struct {
	ctx *Context
}

const SESSION_ID string = "session.id"

func (s *AuthService) FromContext(ctx *gin.Context) (*SessionItem, error) {
	session, err := s.getSessionFromContext(ctx)

	if err != nil {
		return nil, err
	}

	s.setCookieFromSession(ctx, session)
	s.ctx.Services.Sessions.Extend(session)

	return session, nil
}

func (s *AuthService) Login(ctx *gin.Context) error {
	session, err := s.ctx.Services.Sessions.Create()

	if err != nil {
		return err
	}

	s.setCookieFromSession(ctx, session)

	return nil
}

func (s *AuthService) Logout(ctx *gin.Context) error {
	session, err := s.getSessionFromContext(ctx)

	if err != nil {
		return err
	}

	if err := s.ctx.Services.Sessions.Destroy(session.Id); err != nil {
		return err
	}

	s.clearCookie(ctx)

	return nil
}

func (s *AuthService) clearCookie(ctx *gin.Context) {
	ctx.SetCookie(SESSION_ID, "", 0, "/", "", false, true)
}

func (s *AuthService) setCookieFromSession(ctx *gin.Context, session *SessionItem) {
	ctx.SetCookie(SESSION_ID, session.Id, int(time.Duration(time.Hour*24).Seconds()), "/", "", false, true)
}

func (s *AuthService) getSessionFromContext(ctx *gin.Context) (*SessionItem, error) {
	cookie, err := ctx.Cookie(SESSION_ID)

	if err != nil {
		return nil, err
	}

	return s.ctx.Services.Sessions.GetById(cookie)
}