88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package persistence
|
|
|
|
import (
|
|
domainAPIKey "RenewCMS/internal/domain/apikey"
|
|
"RenewCMS/internal/infrastructure/persistence/mappers"
|
|
entity "RenewCMS/internal/infrastructure/persistence/models"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type APIKeyRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewAPIKeyRepository(db *gorm.DB) *APIKeyRepository {
|
|
return &APIKeyRepository{db}
|
|
}
|
|
|
|
func (a *APIKeyRepository) Create(key domainAPIKey.APIKey) (domainAPIKey.APIKey, error) {
|
|
model := mappers.APIKeyToModel(key)
|
|
model.ID = 0
|
|
if err := a.db.Create(&model).Error; err != nil {
|
|
return domainAPIKey.APIKey{}, err
|
|
}
|
|
return mappers.APIKeyToDomain(model), nil
|
|
}
|
|
|
|
func (a *APIKeyRepository) Get(id uint32) (domainAPIKey.APIKey, error) {
|
|
var model entity.APIKey
|
|
if err := a.db.First(&model, id).Error; err != nil {
|
|
return domainAPIKey.APIKey{}, mapNotFoundErr(err, domainAPIKey.ErrAPIKeyNotFound)
|
|
}
|
|
return mappers.APIKeyToDomain(model), nil
|
|
}
|
|
|
|
func (a *APIKeyRepository) GetByHashedKey(hashedKey string) (domainAPIKey.APIKey, error) {
|
|
var model entity.APIKey
|
|
if err := a.db.Where("hashed_key = ?", hashedKey).First(&model).Error; err != nil {
|
|
return domainAPIKey.APIKey{}, mapNotFoundErr(err, domainAPIKey.ErrAPIKeyNotFound)
|
|
}
|
|
return mappers.APIKeyToDomain(model), nil
|
|
}
|
|
|
|
func (a *APIKeyRepository) ListByUser(userId uint32) ([]domainAPIKey.APIKey, error) {
|
|
return a.list(a.db.Where("user_id = ?", userId))
|
|
}
|
|
|
|
func (a *APIKeyRepository) ListAll() ([]domainAPIKey.APIKey, error) {
|
|
return a.list(a.db)
|
|
}
|
|
|
|
func (a *APIKeyRepository) list(query *gorm.DB) ([]domainAPIKey.APIKey, error) {
|
|
var found []entity.APIKey
|
|
if err := query.Order("created_at DESC, id DESC").Find(&found).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
keys := make([]domainAPIKey.APIKey, 0, len(found))
|
|
for _, model := range found {
|
|
keys = append(keys, mappers.APIKeyToDomain(model))
|
|
}
|
|
return keys, nil
|
|
}
|
|
|
|
func (a *APIKeyRepository) Revoke(id uint32, at time.Time) error {
|
|
query := a.db.Model(&entity.APIKey{})
|
|
query = query.Where("id = ? AND revoked_at IS NULL", id)
|
|
result := query.Update("revoked_at", at)
|
|
return result.Error
|
|
}
|
|
|
|
func (a *APIKeyRepository) RevokeAllByUser(userId uint32, at time.Time) error {
|
|
query := a.db.Model(&entity.APIKey{})
|
|
query = query.Where("user_id = ? AND revoked_at IS NULL", userId)
|
|
result := query.Update("revoked_at", at)
|
|
return result.Error
|
|
}
|
|
|
|
func (a *APIKeyRepository) TouchLastUsed(id uint32, at time.Time) error {
|
|
query := a.db.Model(&entity.APIKey{})
|
|
query = query.Where("id = ?", id)
|
|
result := query.Update("last_used_at", at)
|
|
return result.Error
|
|
}
|
|
|
|
var _ domainAPIKey.Repository = &APIKeyRepository{}
|