chat-deploy/internal/rpc/bot/agent.go

package bot

import (
	"context"
	"crypto/rand"
	"time"

	"git.imall.cloud/openim/chat/pkg/common/constant"
	"git.imall.cloud/openim/chat/pkg/common/convert"
	"git.imall.cloud/openim/chat/pkg/common/mctx"
	"git.imall.cloud/openim/chat/pkg/protocol/bot"
	pbconstant "git.imall.cloud/openim/protocol/constant"
	"git.imall.cloud/openim/protocol/sdkws"
	"git.imall.cloud/openim/protocol/user"
	"github.com/openimsdk/tools/errs"
	"github.com/openimsdk/tools/utils/datautil"
)

func (b *botSvr) CreateAgent(ctx context.Context, req *bot.CreateAgentReq) (*bot.CreateAgentResp, error) {
	if req.Agent == nil {
		return nil, errs.ErrArgs.WrapMsg("req.Agent is nil")
	}

	now := time.Now()
	imToken, err := b.imCaller.ImAdminTokenWithDefaultAdmin(ctx)
	if err != nil {
		return nil, err
	}
	ctx = mctx.WithApiToken(ctx, imToken)
	if req.Agent.UserID != "" {
		req.Agent.UserID = constant.AgentUserIDPrefix + req.Agent.UserID
		users, err := b.imCaller.GetUsersInfo(ctx, []string{req.Agent.UserID})
		if err != nil {
			return nil, err
		}
		if len(users) > 0 {
			return nil, errs.ErrDuplicateKey.WrapMsg("agent userID already exists")
		}
	} else {
		randUserIDs := make([]string, 5)
		for i := range randUserIDs {
			randUserIDs[i] = constant.AgentUserIDPrefix + genID(10)
		}
		users, err := b.imCaller.GetUsersInfo(ctx, randUserIDs)
		if err != nil {
			return nil, err
		}
		if len(users) == len(randUserIDs) {
			return nil, errs.ErrDuplicateKey.WrapMsg("gen agent userID already exists, please try again")
		}
		userIDs := datautil.Batch(func(u *sdkws.UserInfo) string { return u.UserID }, users)
		for _, uid := range randUserIDs {
			if datautil.Contain(uid, userIDs...) {
				continue
			}
			req.Agent.UserID = uid
			break
		}
	}

	if err := b.imCaller.AddNotificationAccount(ctx, &user.AddNotificationAccountReq{
		UserID:         req.Agent.UserID,
		NickName:       req.Agent.Nickname,
		FaceURL:        req.Agent.FaceURL,
		AppMangerLevel: pbconstant.AppRobotAdmin,
	}); err != nil {
		return nil, err
	}
	dbagent := convert.PB2DBAgent(req.Agent)
	dbagent.CreateTime = now
	err = b.database.CreateAgent(ctx, dbagent)
	if err != nil {
		return nil, err
	}
	return &bot.CreateAgentResp{}, nil
}

func (b *botSvr) UpdateAgent(ctx context.Context, req *bot.UpdateAgentReq) (*bot.UpdateAgentResp, error) {
	if _, err := b.database.TakeAgent(ctx, req.UserID); err != nil {
		return nil, errs.ErrArgs.Wrap()
	}

	if req.FaceURL != nil || req.Nickname != nil {
		imReq := &user.UpdateNotificationAccountInfoReq{
			UserID: req.UserID,
		}
		if req.Nickname != nil {
			imReq.NickName = *req.Nickname
		}
		if req.FaceURL != nil {
			imReq.FaceURL = *req.FaceURL
		}
		imToken, err := b.imCaller.ImAdminTokenWithDefaultAdmin(ctx)
		if err != nil {
			return nil, err
		}
		ctx = mctx.WithApiToken(ctx, imToken)
		err = b.imCaller.UpdateNotificationAccount(ctx, imReq)
		if err != nil {
			return nil, err
		}
	}

	update := ToDBAgentUpdate(req)
	err := b.database.UpdateAgent(ctx, req.UserID, update)
	if err != nil {
		return nil, err
	}

	return &bot.UpdateAgentResp{}, nil
}

func (b *botSvr) PageFindAgent(ctx context.Context, req *bot.PageFindAgentReq) (*bot.PageFindAgentResp, error) {
	total, agents, err := b.database.PageAgents(ctx, req.UserIDs, req.Pagination)
	if err != nil {
		return nil, err
	}
	//_, userType, err := mctx.Check(ctx)
	//if err != nil {
	//	return nil, err
	//}
	//if userType != constant.AdminUser {
	for i := range agents {
		agents[i].Key = ""
	}
	//}
	return &bot.PageFindAgentResp{
		Total:  total,
		Agents: convert.BatchDB2PBAgent(agents),
	}, nil
}

func (b *botSvr) DeleteAgent(ctx context.Context, req *bot.DeleteAgentReq) (*bot.DeleteAgentResp, error) {
	err := b.database.DeleteAgents(ctx, req.UserIDs)
	if err != nil {
		return nil, err
	}
	return &bot.DeleteAgentResp{}, nil
}

func genID(l int) string {
	data := make([]byte, l)
	_, _ = rand.Read(data)
	chars := []byte("0123456789")
	for i := 0; i < len(data); i++ {
		if i == 0 {
			data[i] = chars[1:][data[i]%9]
		} else {
			data[i] = chars[data[i]%10]
		}
	}
	return string(data)
}