135 lines
4.4 KiB
Go
135 lines
4.4 KiB
Go
// Copyright © 2023 OpenIM open source community. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package admin
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/openimsdk/tools/utils/datautil"
|
|
|
|
"github.com/openimsdk/tools/errs"
|
|
|
|
admindb "git.imall.cloud/openim/chat/pkg/common/db/table/admin"
|
|
"git.imall.cloud/openim/chat/pkg/common/mctx"
|
|
"git.imall.cloud/openim/chat/pkg/protocol/admin"
|
|
"git.imall.cloud/openim/chat/pkg/protocol/common"
|
|
)
|
|
|
|
func (o *adminServer) AddDefaultFriend(ctx context.Context, req *admin.AddDefaultFriendReq) (*admin.AddDefaultFriendResp, error) {
|
|
if _, err := mctx.CheckAdmin(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
if len(req.UserIDs) == 0 {
|
|
return nil, errs.ErrArgs.WrapMsg("user ids is empty")
|
|
}
|
|
if datautil.Duplicate(req.UserIDs) {
|
|
return nil, errs.ErrArgs.WrapMsg("user ids is duplicate")
|
|
}
|
|
users, err := o.Chat.FindUserPublicInfo(ctx, req.UserIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if ids := datautil.Single(req.UserIDs, datautil.Slice(users, func(user *common.UserPublicInfo) string { return user.UserID })); len(ids) > 0 {
|
|
return nil, errs.ErrRecordNotFound.WrapMsg("user id not found", "userID", ids)
|
|
}
|
|
exists, err := o.Database.FindDefaultFriend(ctx, req.UserIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(exists) > 0 {
|
|
return nil, errs.ErrDuplicateKey.WrapMsg("user id existed", "userID", exists)
|
|
}
|
|
now := time.Now()
|
|
ms := make([]*admindb.RegisterAddFriend, 0, len(req.UserIDs))
|
|
for _, userID := range req.UserIDs {
|
|
ms = append(ms, &admindb.RegisterAddFriend{
|
|
UserID: userID,
|
|
CreateTime: now,
|
|
})
|
|
}
|
|
if err := o.Database.AddDefaultFriend(ctx, ms); err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin.AddDefaultFriendResp{}, nil
|
|
}
|
|
|
|
func (o *adminServer) DelDefaultFriend(ctx context.Context, req *admin.DelDefaultFriendReq) (*admin.DelDefaultFriendResp, error) {
|
|
if _, err := mctx.CheckAdmin(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
if len(req.UserIDs) == 0 {
|
|
return nil, errs.ErrArgs.WrapMsg("user ids is empty")
|
|
}
|
|
if datautil.Duplicate(req.UserIDs) {
|
|
return nil, errs.ErrArgs.WrapMsg("user ids is duplicate")
|
|
}
|
|
exists, err := o.Database.FindDefaultFriend(ctx, req.UserIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if ids := datautil.Single(req.UserIDs, exists); len(ids) > 0 {
|
|
return nil, errs.ErrRecordNotFound.WrapMsg("user id not found", "userID", ids)
|
|
}
|
|
now := time.Now()
|
|
ms := make([]*admindb.RegisterAddFriend, 0, len(req.UserIDs))
|
|
for _, userID := range req.UserIDs {
|
|
ms = append(ms, &admindb.RegisterAddFriend{
|
|
UserID: userID,
|
|
CreateTime: now,
|
|
})
|
|
}
|
|
if err := o.Database.DelDefaultFriend(ctx, req.UserIDs); err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin.DelDefaultFriendResp{}, nil
|
|
}
|
|
|
|
func (o *adminServer) FindDefaultFriend(ctx context.Context, req *admin.FindDefaultFriendReq) (*admin.FindDefaultFriendResp, error) {
|
|
if _, _, err := mctx.Check(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
userIDs, err := o.Database.FindDefaultFriend(ctx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin.FindDefaultFriendResp{UserIDs: userIDs}, nil
|
|
}
|
|
|
|
func (o *adminServer) SearchDefaultFriend(ctx context.Context, req *admin.SearchDefaultFriendReq) (*admin.SearchDefaultFriendResp, error) {
|
|
if _, err := mctx.CheckAdmin(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
total, infos, err := o.Database.SearchDefaultFriend(ctx, req.Keyword, req.Pagination)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
userIDs := datautil.Slice(infos, func(info *admindb.RegisterAddFriend) string { return info.UserID })
|
|
userMap, err := o.Chat.MapUserPublicInfo(ctx, userIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
attributes := make([]*admin.DefaultFriendAttribute, 0, len(infos))
|
|
for _, info := range infos {
|
|
attribute := &admin.DefaultFriendAttribute{
|
|
UserID: info.UserID,
|
|
CreateTime: info.CreateTime.UnixMilli(),
|
|
User: userMap[info.UserID],
|
|
}
|
|
attributes = append(attributes, attribute)
|
|
}
|
|
return &admin.SearchDefaultFriendResp{Total: uint32(total), Users: attributes}, nil
|
|
}
|