all

Downloads/Voltaserve/api/repo/invitation_repo.go

@@ -0,0 +1,185 @@
+package repo
+
+import (
+	"errors"
+	"time"
+	"voltaserve/errorpkg"
+	"voltaserve/helper"
+	"voltaserve/infra"
+	"voltaserve/model"
+
+	"gorm.io/gorm"
+)
+
+type InvitationInsertOptions struct {
+	UserID         string
+	OrganizationID string
+	Emails         []string
+}
+
+type InvitationRepo interface {
+	Insert(opts InvitationInsertOptions) ([]model.Invitation, error)
+	Find(id string) (model.Invitation, error)
+	GetIncoming(email string) ([]model.Invitation, error)
+	GetOutgoing(orgID string, userID string) ([]model.Invitation, error)
+	Save(org model.Invitation) error
+	Delete(id string) error
+}
+
+func NewInvitationRepo() InvitationRepo {
+	return newInvitationRepo()
+}
+
+type invitationEntity struct {
+	ID             string  `json:"id" gorm:"column:id"`
+	OrganizationID string  `json:"organizationId" gorm:"column:organization_id"`
+	OwnerID        string  `json:"ownerId" gorm:"column:owner_id"`
+	Email          string  `json:"email" gorm:"column:email"`
+	Status         string  `json:"status" gorm:"column:status"`
+	CreateTime     string  `json:"createTime" gorm:"column:create_time"`
+	UpdateTime     *string `json:"updateTime" gorm:"column:update_time"`
+}
+
+func (*invitationEntity) TableName() string {
+	return "invitation"
+}
+
+func (i *invitationEntity) BeforeCreate(*gorm.DB) (err error) {
+	i.CreateTime = time.Now().UTC().Format(time.RFC3339)
+	return nil
+}
+
+func (i *invitationEntity) BeforeSave(*gorm.DB) (err error) {
+	timeNow := time.Now().UTC().Format(time.RFC3339)
+	i.UpdateTime = &timeNow
+	return nil
+}
+
+func (i *invitationEntity) GetID() string {
+	return i.ID
+}
+
+func (i *invitationEntity) GetOrganizationID() string {
+	return i.OrganizationID
+}
+
+func (i *invitationEntity) GetOwnerID() string {
+	return i.OwnerID
+}
+
+func (i *invitationEntity) GetEmail() string {
+	return i.Email
+}
+
+func (i *invitationEntity) GetStatus() string {
+	return i.Status
+}
+
+func (i *invitationEntity) GetCreateTime() string {
+	return i.CreateTime
+}
+
+func (i *invitationEntity) GetUpdateTime() *string {
+	return i.UpdateTime
+}
+
+func (i *invitationEntity) SetStatus(status string) {
+	i.Status = status
+}
+
+func (i *invitationEntity) SetUpdateTime(updateTime *string) {
+	i.UpdateTime = updateTime
+}
+
+type invitationRepo struct {
+	db       *gorm.DB
+	userRepo *userRepo
+}
+
+func newInvitationRepo() *invitationRepo {
+	return &invitationRepo{
+		db:       infra.GetDb(),
+		userRepo: newUserRepo(),
+	}
+}
+
+func (repo *invitationRepo) Insert(opts InvitationInsertOptions) ([]model.Invitation, error) {
+	var res []model.Invitation
+	for _, e := range opts.Emails {
+		invitation := invitationEntity{
+			ID:             helper.NewID(),
+			OrganizationID: opts.OrganizationID,
+			OwnerID:        opts.UserID,
+			Email:          e,
+			Status:         model.InvitationStatusPending,
+		}
+		if db := repo.db.Save(&invitation); db.Error != nil {
+			return nil, db.Error
+		}
+		i, err := repo.Find(invitation.ID)
+		if err != nil {
+			return nil, err
+		}
+		res = append(res, i)
+	}
+	return res, nil
+}
+
+func (repo *invitationRepo) Find(id string) (model.Invitation, error) {
+	var invitation = invitationEntity{}
+	db := repo.db.Where("id = ?", id).First(&invitation)
+	if db.Error != nil {
+		if errors.Is(db.Error, gorm.ErrRecordNotFound) {
+			return nil, errorpkg.NewInvitationNotFoundError(db.Error)
+		} else {
+			return nil, errorpkg.NewInternalServerError(db.Error)
+		}
+	}
+	return &invitation, nil
+}
+
+func (repo *invitationRepo) GetIncoming(email string) ([]model.Invitation, error) {
+	var invitations []*invitationEntity
+	db := repo.db.
+		Raw("SELECT * FROM invitation WHERE email = ? and status = 'pending' ORDER BY create_time DESC", email).
+		Scan(&invitations)
+	if db.Error != nil {
+		return nil, db.Error
+	}
+	var res []model.Invitation
+	for _, inv := range invitations {
+		res = append(res, inv)
+	}
+	return res, nil
+}
+
+func (repo *invitationRepo) GetOutgoing(orgID string, userID string) ([]model.Invitation, error) {
+	var invitations []*invitationEntity
+	db := repo.db.
+		Raw("SELECT * FROM invitation WHERE organization_id = ? and owner_id = ? ORDER BY create_time DESC", orgID, userID).
+		Scan(&invitations)
+	if db.Error != nil {
+		return nil, db.Error
+	}
+	var res []model.Invitation
+	for _, inv := range invitations {
+		res = append(res, inv)
+	}
+	return res, nil
+}
+
+func (repo *invitationRepo) Save(org model.Invitation) error {
+	db := repo.db.Save(org)
+	if db.Error != nil {
+		return db.Error
+	}
+	return nil
+}
+
+func (repo *invitationRepo) Delete(id string) error {
+	db := repo.db.Exec("DELETE FROM invitation WHERE id = ?", id)
+	if db.Error != nil {
+		return db.Error
+	}
+	return nil
+}