You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
318 lines
9.4 KiB
318 lines
9.4 KiB
// Copyright © 2023 OpenIM. 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 api
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-playground/validator/v10"
|
|
"github.com/mitchellh/mapstructure"
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/a2r"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/apiresp"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/apistruct"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/tokenverify"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msg"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
|
|
)
|
|
|
|
type MessageApi struct {
|
|
*rpcclient.Message
|
|
validate *validator.Validate
|
|
userRpcClient *rpcclient.UserRpcClient
|
|
}
|
|
|
|
func NewMessageApi(msgRpcClient *rpcclient.Message, userRpcClient *rpcclient.User) MessageApi {
|
|
return MessageApi{Message: msgRpcClient, validate: validator.New(), userRpcClient: rpcclient.NewUserRpcClientByUser(userRpcClient)}
|
|
}
|
|
|
|
func (MessageApi) SetOptions(options map[string]bool, value bool) {
|
|
utils.SetSwitchFromOptions(options, constant.IsHistory, value)
|
|
utils.SetSwitchFromOptions(options, constant.IsPersistent, value)
|
|
utils.SetSwitchFromOptions(options, constant.IsSenderSync, value)
|
|
utils.SetSwitchFromOptions(options, constant.IsConversationUpdate, value)
|
|
}
|
|
|
|
func (m MessageApi) newUserSendMsgReq(c *gin.Context, params *apistruct.SendMsg) *msg.SendMsgReq {
|
|
var newContent string
|
|
var err error
|
|
options := make(map[string]bool, 5)
|
|
switch params.ContentType {
|
|
case constant.Text:
|
|
newContent = params.Content["text"].(string)
|
|
case constant.Picture:
|
|
fallthrough
|
|
case constant.Custom:
|
|
fallthrough
|
|
case constant.Voice:
|
|
fallthrough
|
|
case constant.Video:
|
|
fallthrough
|
|
case constant.File:
|
|
fallthrough
|
|
case constant.CustomNotTriggerConversation:
|
|
fallthrough
|
|
case constant.CustomOnlineOnly:
|
|
fallthrough
|
|
default:
|
|
newContent = utils.StructToJsonString(params.Content)
|
|
}
|
|
if params.IsOnlineOnly {
|
|
m.SetOptions(options, false)
|
|
}
|
|
if params.NotOfflinePush {
|
|
utils.SetSwitchFromOptions(options, constant.IsOfflinePush, false)
|
|
}
|
|
if params.ContentType == constant.CustomOnlineOnly {
|
|
m.SetOptions(options, false)
|
|
} else if params.ContentType == constant.CustomNotTriggerConversation {
|
|
utils.SetSwitchFromOptions(options, constant.IsConversationUpdate, false)
|
|
}
|
|
pbData := msg.SendMsgReq{
|
|
MsgData: &sdkws.MsgData{
|
|
SendID: params.SendID,
|
|
GroupID: params.GroupID,
|
|
ClientMsgID: utils.GetMsgID(params.SendID),
|
|
SenderPlatformID: params.SenderPlatformID,
|
|
SenderNickname: params.SenderNickname,
|
|
SenderFaceURL: params.SenderFaceURL,
|
|
SessionType: params.SessionType,
|
|
MsgFrom: constant.SysMsgType,
|
|
ContentType: params.ContentType,
|
|
Content: []byte(newContent),
|
|
CreateTime: utils.GetCurrentTimestampByMill(),
|
|
Options: options,
|
|
OfflinePushInfo: params.OfflinePushInfo,
|
|
},
|
|
}
|
|
if params.ContentType == constant.OANotification {
|
|
var tips sdkws.TipsComm
|
|
tips.JsonDetail = utils.StructToJsonString(params.Content)
|
|
pbData.MsgData.Content, err = proto.Marshal(&tips)
|
|
if err != nil {
|
|
log.ZError(c, "Marshal failed ", err, "tips", tips.String())
|
|
}
|
|
}
|
|
return &pbData
|
|
}
|
|
|
|
func (m *MessageApi) GetSeq(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.GetMaxSeq, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) PullMsgBySeqs(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.PullMessageBySeqs, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) RevokeMsg(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.RevokeMsg, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) MarkMsgsAsRead(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.MarkMsgsAsRead, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) MarkConversationAsRead(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.MarkConversationAsRead, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) GetConversationsHasReadAndMaxSeq(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.GetConversationsHasReadAndMaxSeq, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) SetConversationHasReadSeq(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.SetConversationHasReadSeq, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) ClearConversationsMsg(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.ClearConversationsMsg, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) UserClearAllMsg(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.UserClearAllMsg, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) DeleteMsgs(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.DeleteMsgs, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) DeleteMsgPhysicalBySeq(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.DeleteMsgPhysicalBySeq, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) DeleteMsgPhysical(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.DeleteMsgPhysical, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) getSendMsgReq(c *gin.Context, req apistruct.SendMsg) (sendMsgReq *msg.SendMsgReq, err error) {
|
|
var data interface{}
|
|
switch req.ContentType {
|
|
case constant.Text:
|
|
data = apistruct.TextElem{}
|
|
case constant.Picture:
|
|
data = apistruct.PictureElem{}
|
|
case constant.Voice:
|
|
data = apistruct.SoundElem{}
|
|
case constant.Video:
|
|
data = apistruct.VideoElem{}
|
|
case constant.File:
|
|
data = apistruct.FileElem{}
|
|
case constant.Custom:
|
|
data = apistruct.CustomElem{}
|
|
case constant.Revoke:
|
|
data = apistruct.RevokeElem{}
|
|
case constant.OANotification:
|
|
data = apistruct.OANotificationElem{}
|
|
req.SessionType = constant.NotificationChatType
|
|
case constant.CustomNotTriggerConversation:
|
|
data = apistruct.CustomElem{}
|
|
case constant.CustomOnlineOnly:
|
|
data = apistruct.CustomElem{}
|
|
default:
|
|
return nil, errs.ErrArgs.WithDetail("not support err contentType")
|
|
}
|
|
if err := mapstructure.WeakDecode(req.Content, &data); err != nil {
|
|
return nil, err
|
|
}
|
|
log.ZDebug(c, "getSendMsgReq", "data", data)
|
|
if err := m.validate.Struct(data); err != nil {
|
|
return nil, err
|
|
}
|
|
return m.newUserSendMsgReq(c, &req), nil
|
|
}
|
|
|
|
func (m *MessageApi) SendMessage(c *gin.Context) {
|
|
req := apistruct.SendMsgReq{}
|
|
if err := c.BindJSON(&req); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
log.ZInfo(c, "SendMessage", "req", req)
|
|
if !tokenverify.IsAppManagerUid(c) {
|
|
apiresp.GinError(c, errs.ErrNoPermission.Wrap("only app manager can send message"))
|
|
return
|
|
}
|
|
|
|
sendMsgReq, err := m.getSendMsgReq(c, req.SendMsg)
|
|
if err != nil {
|
|
log.ZError(c, "decodeData failed", err)
|
|
apiresp.GinError(c, err)
|
|
return
|
|
}
|
|
sendMsgReq.MsgData.RecvID = req.RecvID
|
|
var status int
|
|
respPb, err := m.Client.SendMsg(c, sendMsgReq)
|
|
if err != nil {
|
|
status = constant.MsgSendFailed
|
|
apiresp.GinError(c, err)
|
|
return
|
|
}
|
|
status = constant.MsgSendSuccessed
|
|
_, err = m.Client.SetSendMsgStatus(c, &msg.SetSendMsgStatusReq{
|
|
Status: int32(status),
|
|
})
|
|
if err != nil {
|
|
log.ZError(c, "SetSendMsgStatus failed", err)
|
|
}
|
|
apiresp.GinSuccess(c, respPb)
|
|
}
|
|
|
|
func (m *MessageApi) BatchSendMsg(c *gin.Context) {
|
|
var (
|
|
req apistruct.BatchSendMsgReq
|
|
resp apistruct.BatchSendMsgResp
|
|
)
|
|
if err := c.BindJSON(&req); err != nil {
|
|
log.ZError(c, "BatchSendMsg BindJSON failed", err)
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
log.ZInfo(c, "BatchSendMsg", "req", req)
|
|
if err := tokenverify.CheckAdmin(c); err != nil {
|
|
apiresp.GinError(c, errs.ErrNoPermission.Wrap("only app manager can send message"))
|
|
return
|
|
}
|
|
|
|
|
|
var recvIDs []string
|
|
var err error
|
|
if req.IsSendAll {
|
|
pageNumber := 1
|
|
showNumber := 500
|
|
for {
|
|
recvIDsPart, err := m.userRpcClient.GetAllUserIDs(c, int32(pageNumber), int32(showNumber))
|
|
if err != nil {
|
|
log.ZError(c, "GetAllUserIDs failed", err)
|
|
apiresp.GinError(c, err)
|
|
return
|
|
}
|
|
if len(recvIDsPart) < showNumber {
|
|
recvIDs = append(recvIDs, recvIDsPart...)
|
|
break
|
|
}
|
|
pageNumber++
|
|
}
|
|
} else {
|
|
recvIDs = req.RecvIDs
|
|
}
|
|
log.ZDebug(c, "BatchSendMsg nums", "nums ", len(recvIDs))
|
|
sendMsgReq, err := m.getSendMsgReq(c, req.SendMsg)
|
|
if err != nil {
|
|
log.ZError(c, "decodeData failed", err)
|
|
apiresp.GinError(c, err)
|
|
return
|
|
}
|
|
for _, recvID := range recvIDs {
|
|
sendMsgReq.MsgData.RecvID = recvID
|
|
rpcResp, err := m.Client.SendMsg(c, sendMsgReq)
|
|
if err != nil {
|
|
resp.FailedIDs = append(resp.FailedIDs, recvID)
|
|
continue
|
|
}
|
|
resp.Results = append(resp.Results, &apistruct.SingleReturnResult{
|
|
ServerMsgID: rpcResp.ServerMsgID,
|
|
ClientMsgID: rpcResp.ClientMsgID,
|
|
SendTime: rpcResp.SendTime,
|
|
RecvID: recvID,
|
|
})
|
|
}
|
|
apiresp.GinSuccess(c, resp)
|
|
}
|
|
|
|
func (m *MessageApi) CheckMsgIsSendSuccess(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.GetSendMsgStatus, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) GetUsersOnlineStatus(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.GetSendMsgStatus, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) GetActiveUser(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.GetActiveUser, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) GetActiveGroup(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.GetActiveGroup, m.Client, c)
|
|
}
|
|
|
|
func (m *MessageApi) SearchMsg(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.SearchMessage, m.Client, c)
|
|
}
|