复制项目

internal/rpc/bot/agent.go

@@ -0,0 +1,153 @@
+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)
+}