wechat-api/internal/logic/chat/add_friend_by_phone_logic.go

package chat

import (
	"context"
	"errors"

	"wechat-api/ent"
	"wechat-api/internal/service/addfriend"
	"wechat-api/internal/svc"
	"wechat-api/internal/types"
	"wechat-api/internal/utils/contextkey"

	"github.com/suyuan32/simple-admin-common/enum/errorcode"
	"github.com/suyuan32/simple-admin-common/msg/errormsg"
	"github.com/zeromicro/go-zero/core/logx"
)

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

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

func (l *AddFriendByPhoneLogic) AddFriendByPhone(req *types.AddFriendByPhoneReq) (resp *types.BaseMsgResp, err error) {
	// todo: add your logic here and delete this line

	var (
		apiKeyObj *ent.ApiKey
		ok        bool
	)

	//从上下文中获取鉴权中间件埋下的apiAuthInfo
	apiKeyObj, ok = contextkey.AuthTokenInfoKey.GetValue(l.ctx)
	if !ok {
		return nil, errors.New("content get auth info err")
	}
	err = l.AppendWechaFriendAddReq(apiKeyObj, req)
	if err != nil {
		return nil, err
	}
	//addfriend.NewAddWechatFriendService(l.ctx, l.svcCtx).AddNewFriend(req.WeChatId, req.Phone, req.Message)
	addfriend.NewAddWechatFriendService(l.ctx, l.svcCtx).FindFriendByContent(req.WeChatId, req.Phone)

	resp = &types.BaseMsgResp{
		Msg:  errormsg.Success,
		Code: errorcode.OK,
	}

	return resp, nil
}

func AppendWechaFriendAddLog(ctx context.Context, svcCtx *svc.ServiceContext, addLog *types.AddFriendByPhoneReq) (int64, error) {

	isCanAdd := 0
	if addLog.Type != 1 {
		isCanAdd = 1
	}
	res, err := svcCtx.DB.AddWechatFriendLog.Create().
		SetNotNilOwnerWxID(&addLog.WeChatId).
		SetNotNilOwnerWxType(&addLog.Type).
		SetNotNilFindContent(&addLog.Phone).
		SetNotNilMessage(&addLog.Message).
		SetIsCanAdd(isCanAdd).
		Save(ctx)

	id := int64(0)
	if err == nil {
		id = res.ID
	}
	return id, err
}

func (l *AddFriendByPhoneLogic) AppendWechaFriendAddReq(apiKeyObj *ent.ApiKey, req *types.AddFriendByPhoneReq) error {

	id, err := AppendWechaFriendAddLog(l.ctx, l.svcCtx, req)
	if err == nil {
		logx.Infof("AppendWechaFriendAddReq succ,get id:%d", id)
	}
	return err
}