vaulterm/server/app/auth/repository.go

package auth

import (
	"gorm.io/gorm"
	"rul.sh/vaulterm/server/db"
	"rul.sh/vaulterm/server/models"
	"rul.sh/vaulterm/server/utils"
)

type Auth struct{ db *gorm.DB }

func NewRepository() *Auth {
	return &Auth{db: db.Get()}
}

func (r *Auth) FindUser(username string, email string) (*models.User, error) {
	var user models.User
	if email == "" {
		email = username
	}
	ret := r.db.Where("username = ? OR email = ?", username, email).First(&user)

	return &user, ret.Error
}

func (r *Auth) FindUserAccount(accountType string, accountId string) (*models.UserAccount, error) {
	var user models.UserAccount
	ret := r.db.Where("type = ? AND account_id = ?", accountType, accountId).Joins("User").First(&user)
	return &user, ret.Error
}

func (r *Auth) CreateUserSession(user *models.User) (string, error) {
	sessionId, err := utils.GenerateSessionID(20)
	if err != nil {
		return "", err
	}

	if ret := r.db.Create(&models.UserSession{ID: sessionId, UserID: user.ID}); ret.Error != nil {
		return "", ret.Error
	}

	return sessionId, nil
}

func (r *Auth) GetSession(sessionId string) (*models.UserSession, error) {
	var session models.UserSession
	res := r.db.Joins("User").Where(&models.UserSession{ID: sessionId}).First(&session)
	return &session, res.Error
}

func (r *Auth) RemoveUserSession(sessionId string, force bool) error {
	db := r.db
	if force {
		db = db.Unscoped()
	}

	res := db.Delete(&models.UserSession{ID: sessionId})
	return res.Error
}

func (r *Auth) CreateUser(user *models.User) (string, error) {
	if err := r.db.Create(user).Error; err != nil {
		return "", err
	}
	return r.CreateUserSession(user)
}