67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package persistence
|
|
|
|
import (
|
|
domainInvite "RenewCMS/internal/domain/invite"
|
|
domainUser "RenewCMS/internal/domain/user"
|
|
"RenewCMS/internal/infrastructure/persistence/mappers"
|
|
entity "RenewCMS/internal/infrastructure/persistence/models"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type InviteRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewInviteRepository(db *gorm.DB) *InviteRepository {
|
|
return &InviteRepository{db}
|
|
}
|
|
|
|
func (i *InviteRepository) Create(invite domainInvite.Invite) (domainInvite.Invite, error) {
|
|
model := mappers.InviteToModel(invite)
|
|
model.ID = 0
|
|
if err := i.db.Create(&model).Error; err != nil {
|
|
return domainInvite.Invite{}, err
|
|
}
|
|
|
|
return mappers.InviteToDomain(model), nil
|
|
}
|
|
|
|
func (i *InviteRepository) GetAll() []domainInvite.Invite {
|
|
var invites []entity.Invite
|
|
i.db.Model(&entity.Invite{}).Find(&invites)
|
|
|
|
var domainInvites []domainInvite.Invite
|
|
for _, localInvite := range invites {
|
|
domainInvites = append(domainInvites, mappers.InviteToDomain(localInvite))
|
|
}
|
|
|
|
return domainInvites
|
|
}
|
|
|
|
func (i *InviteRepository) Delete(id uint32) error {
|
|
result := i.db.Delete(&entity.Invite{}, id)
|
|
return deleteOrNotFound(result, domainInvite.ErrInviteNotFound)
|
|
}
|
|
|
|
func (i *InviteRepository) AcceptWithUser(inviteID uint32, acceptedAt time.Time, newUser domainUser.User) (domainUser.User, error) {
|
|
createdModel := mappers.UserToModel(newUser)
|
|
err := i.db.Transaction(func(tx *gorm.DB) error {
|
|
marked := tx.Model(&entity.Invite{}).
|
|
Where("id = ? AND accepted_at IS NULL", inviteID).
|
|
Update("accepted_at", acceptedAt)
|
|
if err := deleteOrNotFound(marked, domainInvite.ErrInviteNotFound); err != nil {
|
|
return err
|
|
}
|
|
return tx.Create(&createdModel).Error
|
|
})
|
|
if err != nil {
|
|
return domainUser.User{}, err
|
|
}
|
|
|
|
return mappers.UserToDomain(createdModel), nil
|
|
}
|
|
|
|
var _ domainInvite.Repository = &InviteRepository{}
|