ario_server/internal/logic/public/user/updateBindEmailLogic.go

package user

import (
	"context"

	"github.com/perfect-panel/server/pkg/constant"

	"github.com/perfect-panel/server/internal/model/user"
	"github.com/perfect-panel/server/pkg/xerr"
	"github.com/pkg/errors"
	"gorm.io/gorm"

	"github.com/perfect-panel/server/internal/svc"
	"github.com/perfect-panel/server/internal/types"
	"github.com/perfect-panel/server/pkg/logger"
)

type UpdateBindEmailLogic struct {
	logger.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// NewUpdateBindEmailLogic Update Bind Email
func NewUpdateBindEmailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateBindEmailLogic {
	return &UpdateBindEmailLogic{
		Logger: logger.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UpdateBindEmailLogic) UpdateBindEmail(req *types.UpdateBindEmailRequest) error {
	u, ok := l.ctx.Value(constant.CtxKeyUser).(*user.User)
	if !ok {
		logger.Error("current user is not found in context")
		return errors.Wrapf(xerr.NewErrCode(xerr.InvalidAccess), "Invalid Access")
	}
	method, err := l.svcCtx.UserModel.FindUserAuthMethodByUserId(l.ctx, "email", u.Id)
	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
		return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "FindUserAuthMethodByOpenID error")
	}
	m, err := l.svcCtx.UserModel.FindUserAuthMethodByOpenID(l.ctx, "email", req.Email)
	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
		return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "FindUserAuthMethodByOpenID error")
	}
	// email already bind
	if m.Id > 0 {
		return errors.Wrapf(xerr.NewErrCode(xerr.UserExist), "email already bind")
	}
	// Use transaction to safely handle insert or update
	err = l.svcCtx.UserModel.Transaction(l.ctx, func(db *gorm.DB) error {
		if method.Id == 0 {
			// Check if record exists in current transaction
			var existingMethod user.AuthMethods
			err := db.Model(&user.AuthMethods{}).Where("user_id = ? AND auth_type = ?", u.Id, "email").First(&existingMethod).Error
			if err != nil {
				if errors.Is(err, gorm.ErrRecordNotFound) {
					// Create new record
					method = &user.AuthMethods{
						UserId:         u.Id,
						AuthType:       "email",
						AuthIdentifier: req.Email,
						Verified:       false,
					}
					if err := db.Create(method).Error; err != nil {
						return err
					}
				} else {
					return err
				}
			} else {
				// Record exists, update it
				existingMethod.AuthIdentifier = req.Email
				existingMethod.Verified = false
				if err := db.Save(&existingMethod).Error; err != nil {
					return err
				}
			}
		} else {
			// Update existing record
			method.Verified = false
			method.AuthIdentifier = req.Email
			if err := db.Save(method).Error; err != nil {
				return err
			}
		}
		return nil
	})
	if err != nil {
		return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseInsertError), "UpdateBindEmail transaction error")
	}
	return nil
}