48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package auth
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
type Repository interface {
|
|
CreateRefreshToken(token *RefreshToken) error
|
|
ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (RefreshToken, error)
|
|
InvalidateRefreshToken(userUuid, refreshUuid, sessionUuid string) error
|
|
}
|
|
|
|
type repo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func newRepository(db *gorm.DB) *repo {
|
|
return &repo{db: db}
|
|
}
|
|
|
|
func (r *repo) CreateRefreshToken(token *RefreshToken) error {
|
|
return r.db.Create(token).Error
|
|
}
|
|
|
|
func (r *repo) ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (RefreshToken, error) {
|
|
var tokenData RefreshToken
|
|
|
|
if err := r.db.
|
|
Where("user_uuid = ?", userUuid).
|
|
Where("refresh_uuid = ?", tokenUuid).
|
|
Where("session_uuid = ?", sessionUuid).
|
|
Where("deleted_at IS NULL").
|
|
First(&tokenData).Error; err != nil {
|
|
return RefreshToken{}, err
|
|
}
|
|
|
|
return tokenData, nil
|
|
}
|
|
|
|
func (r *repo) InvalidateRefreshToken(userUuid, refreshUuid, sessionUuid string) error {
|
|
return r.db.
|
|
Model(&RefreshToken{}).
|
|
Where("user_uuid = ?", userUuid).
|
|
Where("refresh_uuid = ?", refreshUuid).
|
|
Where("session_uuid = ?", sessionUuid).
|
|
Update("deleted_at", time.Now().UTC()).Error
|
|
}
|