各种配置项修复,优化到后台管理端配置
This commit is contained in:
@@ -40,6 +40,25 @@ func (h *familyBindingHelper) getUserEmailMethod(userId int64) (*user.AuthMethod
|
||||
return method, nil
|
||||
}
|
||||
|
||||
func validateMemberJoinConflict(ownerFamilyId int64, memberRecord *user.UserFamilyMember) error {
|
||||
if memberRecord == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
if ownerFamilyId != 0 && memberRecord.FamilyId == ownerFamilyId {
|
||||
if memberRecord.Status == user.FamilyMemberActive {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.FamilyAlreadyBound), "user already in this family")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
if memberRecord.Status == user.FamilyMemberActive {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.FamilyCrossBindForbidden), "user already belongs to another family")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (h *familyBindingHelper) validateJoinFamily(ownerUserId, memberUserId int64) error {
|
||||
if ownerUserId == memberUserId {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.FamilyAlreadyBound), "user already bound to this family")
|
||||
@@ -49,13 +68,14 @@ func (h *familyBindingHelper) validateJoinFamily(ownerUserId, memberUserId int64
|
||||
err := h.svcCtx.DB.WithContext(h.ctx).
|
||||
Unscoped().
|
||||
Model(&user.UserFamily{}).
|
||||
Where("owner_user_id = ? AND status = ?", ownerUserId, user.FamilyStatusActive).
|
||||
Where("owner_user_id = ?", ownerUserId).
|
||||
First(&ownerFamily).Error
|
||||
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query owner family failed")
|
||||
}
|
||||
|
||||
var memberRecord user.UserFamilyMember
|
||||
memberExists := false
|
||||
err = h.svcCtx.DB.WithContext(h.ctx).
|
||||
Unscoped().
|
||||
Model(&user.UserFamilyMember{}).
|
||||
@@ -65,10 +85,12 @@ func (h *familyBindingHelper) validateJoinFamily(ownerUserId, memberUserId int64
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query member family relation failed")
|
||||
}
|
||||
if err == nil {
|
||||
if ownerFamily.Id != 0 && memberRecord.FamilyId == ownerFamily.Id {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.FamilyAlreadyBound), "user already in this family")
|
||||
memberExists = true
|
||||
}
|
||||
if memberExists {
|
||||
if err = validateMemberJoinConflict(ownerFamily.Id, &memberRecord); err != nil {
|
||||
return err
|
||||
}
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.FamilyCrossBindForbidden), "user already belongs to another family")
|
||||
}
|
||||
|
||||
if ownerFamily.Id == 0 {
|
||||
@@ -116,12 +138,8 @@ func (h *familyBindingHelper) joinFamily(ownerUserId, memberUserId int64, source
|
||||
memberExists := err == nil
|
||||
|
||||
if memberExists {
|
||||
if memberRecord.FamilyId == ownerFamily.Id {
|
||||
if memberRecord.Status == user.FamilyMemberActive {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.FamilyAlreadyBound), "user already in this family")
|
||||
}
|
||||
} else {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.FamilyCrossBindForbidden), "user already belongs to another family")
|
||||
if err = validateMemberJoinConflict(ownerFamily.Id, &memberRecord); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
@@ -151,12 +169,16 @@ func (h *familyBindingHelper) joinFamily(ownerUserId, memberUserId int64, source
|
||||
return nil
|
||||
}
|
||||
|
||||
if memberRecord.FamilyId != ownerFamily.Id {
|
||||
memberRecord.FamilyId = ownerFamily.Id
|
||||
}
|
||||
memberRecord.Status = user.FamilyMemberActive
|
||||
memberRecord.Role = user.FamilyRoleMember
|
||||
memberRecord.JoinSource = source
|
||||
memberRecord.JoinedAt = now
|
||||
memberRecord.LeftAt = nil
|
||||
if err = tx.Save(&memberRecord).Error; err != nil {
|
||||
memberRecord.DeletedAt = gorm.DeletedAt{}
|
||||
if err = tx.Unscoped().Save(&memberRecord).Error; err != nil {
|
||||
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "update family member failed")
|
||||
}
|
||||
return nil
|
||||
|
||||
@@ -0,0 +1,107 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
stderrors "errors"
|
||||
"testing"
|
||||
|
||||
modelUser "github.com/perfect-panel/server/internal/model/user"
|
||||
"github.com/perfect-panel/server/pkg/xerr"
|
||||
pkgerrors "github.com/pkg/errors"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func extractFamilyJoinCode(err error) uint32 {
|
||||
if err == nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
var codeErr *xerr.CodeError
|
||||
if stderrors.As(pkgerrors.Cause(err), &codeErr) {
|
||||
return codeErr.GetErrCode()
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func TestValidateMemberJoinConflict(t *testing.T) {
|
||||
ownerFamilyID := int64(11)
|
||||
|
||||
testCases := []struct {
|
||||
name string
|
||||
ownerFamily int64
|
||||
memberRecord *modelUser.UserFamilyMember
|
||||
wantCode uint32
|
||||
}{
|
||||
{
|
||||
name: "no member record",
|
||||
ownerFamily: ownerFamilyID,
|
||||
wantCode: 0,
|
||||
},
|
||||
{
|
||||
name: "same family active member",
|
||||
ownerFamily: ownerFamilyID,
|
||||
memberRecord: &modelUser.UserFamilyMember{
|
||||
FamilyId: ownerFamilyID,
|
||||
Status: modelUser.FamilyMemberActive,
|
||||
},
|
||||
wantCode: xerr.FamilyAlreadyBound,
|
||||
},
|
||||
{
|
||||
name: "same family left member",
|
||||
ownerFamily: ownerFamilyID,
|
||||
memberRecord: &modelUser.UserFamilyMember{
|
||||
FamilyId: ownerFamilyID,
|
||||
Status: modelUser.FamilyMemberLeft,
|
||||
},
|
||||
wantCode: 0,
|
||||
},
|
||||
{
|
||||
name: "same family removed member",
|
||||
ownerFamily: ownerFamilyID,
|
||||
memberRecord: &modelUser.UserFamilyMember{
|
||||
FamilyId: ownerFamilyID,
|
||||
Status: modelUser.FamilyMemberRemoved,
|
||||
},
|
||||
wantCode: 0,
|
||||
},
|
||||
{
|
||||
name: "cross family active member",
|
||||
ownerFamily: ownerFamilyID,
|
||||
memberRecord: &modelUser.UserFamilyMember{
|
||||
FamilyId: ownerFamilyID + 1,
|
||||
Status: modelUser.FamilyMemberActive,
|
||||
},
|
||||
wantCode: xerr.FamilyCrossBindForbidden,
|
||||
},
|
||||
{
|
||||
name: "cross family left member",
|
||||
ownerFamily: ownerFamilyID,
|
||||
memberRecord: &modelUser.UserFamilyMember{
|
||||
FamilyId: ownerFamilyID + 1,
|
||||
Status: modelUser.FamilyMemberLeft,
|
||||
},
|
||||
wantCode: 0,
|
||||
},
|
||||
{
|
||||
name: "cross family removed member",
|
||||
ownerFamily: ownerFamilyID,
|
||||
memberRecord: &modelUser.UserFamilyMember{
|
||||
FamilyId: ownerFamilyID + 1,
|
||||
Status: modelUser.FamilyMemberRemoved,
|
||||
},
|
||||
wantCode: 0,
|
||||
},
|
||||
}
|
||||
|
||||
for _, testCase := range testCases {
|
||||
t.Run(testCase.name, func(t *testing.T) {
|
||||
err := validateMemberJoinConflict(testCase.ownerFamily, testCase.memberRecord)
|
||||
if testCase.wantCode == 0 {
|
||||
require.NoError(t, err)
|
||||
return
|
||||
}
|
||||
|
||||
require.Error(t, err)
|
||||
require.Equal(t, testCase.wantCode, extractFamilyJoinCode(err))
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user