mirror of
https://github.com/teamhanko/hanko.git
synced 2025-10-27 22:27:23 +08:00
106 lines
2.8 KiB
Go
106 lines
2.8 KiB
Go
package persistence
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/gobuffalo/pop/v6"
|
|
"github.com/gofrs/uuid"
|
|
"github.com/teamhanko/hanko/backend/persistence/models"
|
|
"time"
|
|
)
|
|
|
|
type WebauthnSessionDataPersister interface {
|
|
Get(id uuid.UUID) (*models.WebauthnSessionData, error)
|
|
GetByChallenge(challenge string) (*models.WebauthnSessionData, error)
|
|
Create(sessionData models.WebauthnSessionData) error
|
|
Update(sessionData models.WebauthnSessionData) error
|
|
Delete(sessionData models.WebauthnSessionData) error
|
|
Cleanup[models.WebauthnSessionData]
|
|
}
|
|
|
|
type webauthnSessionDataPersister struct {
|
|
db *pop.Connection
|
|
}
|
|
|
|
func NewWebauthnSessionDataPersister(db *pop.Connection) WebauthnSessionDataPersister {
|
|
return &webauthnSessionDataPersister{db: db}
|
|
}
|
|
|
|
func (p *webauthnSessionDataPersister) Get(id uuid.UUID) (*models.WebauthnSessionData, error) {
|
|
sessionData := models.WebauthnSessionData{}
|
|
err := p.db.Eager().Find(&sessionData, id)
|
|
if err != nil && errors.Is(err, sql.ErrNoRows) {
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get sessionData: %w", err)
|
|
}
|
|
|
|
return &sessionData, nil
|
|
}
|
|
|
|
func (p *webauthnSessionDataPersister) GetByChallenge(challenge string) (*models.WebauthnSessionData, error) {
|
|
var sessionData []models.WebauthnSessionData
|
|
err := p.db.Eager().Where("challenge = ?", challenge).All(&sessionData)
|
|
if err != nil && errors.Is(err, sql.ErrNoRows) {
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get sessionData: %w", err)
|
|
}
|
|
|
|
if len(sessionData) <= 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
return &sessionData[0], nil
|
|
}
|
|
|
|
func (p *webauthnSessionDataPersister) Create(sessionData models.WebauthnSessionData) error {
|
|
vErr, err := p.db.Eager().ValidateAndCreate(&sessionData)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to store sessionData: %w", err)
|
|
}
|
|
|
|
if vErr != nil && vErr.HasAny() {
|
|
return fmt.Errorf("sessionData object validation failed: %w", vErr)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *webauthnSessionDataPersister) Update(sessionData models.WebauthnSessionData) error {
|
|
vErr, err := p.db.Eager().ValidateAndUpdate(&sessionData)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to update sessionData: %w", err)
|
|
}
|
|
|
|
if vErr != nil && vErr.HasAny() {
|
|
return fmt.Errorf("sessionData object validation failed: %w", vErr)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *webauthnSessionDataPersister) FindExpired(cutoffTime time.Time, page, perPage int) ([]models.WebauthnSessionData, error) {
|
|
var items []models.WebauthnSessionData
|
|
|
|
query := p.db.
|
|
Where("expires_at < ?", cutoffTime).
|
|
Select("id").
|
|
Paginate(page, perPage)
|
|
err := query.All(&items)
|
|
|
|
return items, err
|
|
}
|
|
|
|
func (p *webauthnSessionDataPersister) Delete(sessionData models.WebauthnSessionData) error {
|
|
err := p.db.Destroy(&sessionData)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to delete sessionData: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|