lone-services/services/wecom/internal/logic/updateUserLogic.go

package logic

import (
	"context"
	"strconv"

	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	"lone-services/pkg/validate"
	wecomuser "lone-services/pkg/wecom/user"
	wecom "lone-services/rpc/wecom/pb"
	"lone-services/services/wecom/internal/client"
	"lone-services/services/wecom/internal/dao"
	"lone-services/services/wecom/internal/model"
	"lone-services/services/wecom/internal/svc"
	"lone-services/services/wecom/validator"

	"github.com/zeromicro/go-zero/core/logx"
)

type UpdateUserLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewUpdateUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateUserLogic {
	return &UpdateUserLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *UpdateUserLogic) UpdateUser(in *wecom.UpdateUserReq) (*wecom.Response, error) {
	var req validator.UpdateUserValidator
	if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
		return outResponse(utils.ErrorParams, msg), nil
	}
	if req.Id < utils.NumberOne {
		return outResponse(utils.ErrorParams, "请传入 id"), nil
	}

	m := model.UserModel{}.Init()
	var row dao.WecomUser
	if err := m.GetOne(modelbase.Params{
		Eq: map[string]string{"id": strconv.FormatInt(req.Id, utils.NumberTen)},
	}, &row); err != nil {
		l.Errorf("wecom update user get: %v", err)
		return failResponse(utils.Fail), nil
	}
	if row.Id < utils.NumberOne {
		return failResponse(utils.ErrorNotFund), nil
	}

	userId := row.UserId
	if req.UserId > utils.NumberZero {
		userId = req.UserId
	}
	typeVal := row.Type
	if req.Type > utils.NumberZero {
		typeVal = uint8(req.Type)
	}
	name := row.Name
	if req.Name != utils.StringEmpty {
		name = req.Name
	}
	avatar := row.Avatar
	if req.Avatar != utils.StringEmpty {
		avatar = req.Avatar
	}
	userStatus := resolveStatus(req.Status, row.Status)

	// 更换 user_id / type 时校验是否与已有启用记录冲突
	if userId != row.UserId || typeVal != row.Type {
		var existing dao.WecomUser
		if err := m.GetOne(modelbase.Params{
			Eq: map[string]string{
				"user_id": strconv.FormatInt(userId, utils.NumberTen),
				"type":    strconv.FormatUint(uint64(typeVal), utils.NumberTen),
				"status":  strconv.Itoa(int(dao.StatusEnabled)),
			},
		}, &existing); err != nil {
			l.Errorf("wecom update user unique check: %v", err)
			return failResponse(utils.Fail), nil
		}
		if existing.Id > utils.NumberZero && existing.Id != row.Id {
			return failResponse(utils.ErrorDataIsExist), nil
		}
	}

	plainMobile := utils.StringEmpty
	encryptMobile := row.Mobile
	if req.Mobile != utils.StringEmpty {
		enc, err := utils.EncryptPhone(req.Mobile)
		if err != nil {
			l.Errorf("wecom update encrypt mobile: %v", err)
			return failResponse(utils.Fail), nil
		}
		encryptMobile = enc
		plainMobile = req.Mobile
	} else {
		dec, err := utils.DecryptPhone(row.Mobile)
		if err != nil {
			l.Errorf("wecom update decrypt mobile: %v", err)
			return failResponse(utils.Fail), nil
		}
		plainMobile = dec
	}

	agentsJSON := row.AgentIds
	if len(req.AgentIds) > utils.NumberZero {
		built, aErr := buildAgentsJSON(req.AgentIds)
		if aErr != nil {
			l.Errorf("wecom update agents: %v", aErr)
			return outResponse(utils.ErrorParams, aErr.Error()), nil
		}
		agentsJSON = built
	}

	if !utils.GetConfigBool("wecom.skip") {
		updateReq := wecomuser.UpdateRequest{
			UserId: row.WecomUserId,
			Name:   name,
			Mobile: plainMobile,
			Enable: statusEnable(userStatus),
		}
		if avatar != utils.StringEmpty {
			updateReq.AvatarMediaID = avatar
		}
		if err := client.NewContactClient().Update(l.ctx, updateReq); err != nil {
			l.Errorf("wecom http update user: %v", err)
			return failWecomHTTP(), nil
		}
	}

	if _, err := m.Edit(modelbase.Params{
		Eq: map[string]string{"id": strconv.FormatInt(row.Id, utils.NumberTen)},
	}, map[string]interface{}{
		"user_id":    userId,
		"type":       typeVal,
		"name":       name,
		"mobile":     encryptMobile,
		"avatar":     avatar,
		"status":     userStatus,
		"agent_ids":  agentsJSON,
		"updated_at": utils.Now(),
	}); err != nil {
		l.Errorf("wecom update user db: %v", err)
		return failResponse(utils.Fail), nil
	}

	return okResponse(nil), nil
}