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.
357 lines
11 KiB
357 lines
11 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/OpenIMSDK/Open-IM-Server/pkg/proto/user"
|
|
"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/discoveryregistry"
|
|
"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
|
|
}
|
|
|
|
func NewMessageApi(discov discoveryregistry.SvcDiscoveryRegistry) MessageApi {
|
|
return MessageApi{Message: *rpcclient.NewMessage(discov), validate: validator.New()}
|
|
}
|
|
|
|
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.ManagementSendMsgReq) *msg.SendMsgReq {
|
|
var newContent string
|
|
var err error
|
|
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
|
|
case constant.Revoke:
|
|
newContent = params.Content["revokeMsgClientID"].(string)
|
|
default:
|
|
}
|
|
options := make(map[string]bool, 5)
|
|
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),
|
|
RecvID: params.RecvID,
|
|
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) SendMessage(c *gin.Context) {
|
|
params := apistruct.ManagementSendMsgReq{}
|
|
if err := c.BindJSON(¶ms); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
if !tokenverify.IsAppManagerUid(c) {
|
|
apiresp.GinError(c, errs.ErrNoPermission.Wrap("only app manager can send message"))
|
|
return
|
|
}
|
|
|
|
var data interface{}
|
|
switch params.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{}
|
|
params.SessionType = constant.NotificationChatType
|
|
case constant.CustomNotTriggerConversation:
|
|
data = apistruct.CustomElem{}
|
|
case constant.CustomOnlineOnly:
|
|
data = apistruct.CustomElem{}
|
|
default:
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail("not support err contentType").Wrap())
|
|
return
|
|
}
|
|
if err := mapstructure.WeakDecode(params.Content, &data); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
|
|
return
|
|
} else if err := m.validate.Struct(params); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
|
|
return
|
|
}
|
|
pbReq := m.newUserSendMsgReq(c, ¶ms)
|
|
var status int
|
|
respPb, err := m.Client.SendMsg(c, pbReq)
|
|
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) ManagementBatchSendMsg(c *gin.Context) {
|
|
params := apistruct.ManagementBatchSendMsgReq{}
|
|
resp := apistruct.ManagementBatchSendMsgResp{}
|
|
var msgSendFailedFlag bool
|
|
if err := c.BindJSON(¶ms); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
if !tokenverify.IsAppManagerUid(c) {
|
|
apiresp.GinError(c, errs.ErrNoPermission.Wrap("only app manager can send message"))
|
|
return
|
|
}
|
|
|
|
var data interface{}
|
|
switch params.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{}
|
|
params.SessionType = constant.NotificationChatType
|
|
case constant.CustomNotTriggerConversation:
|
|
data = apistruct.CustomElem{}
|
|
case constant.CustomOnlineOnly:
|
|
data = apistruct.CustomElem{}
|
|
default:
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail("not support err contentType").Wrap())
|
|
return
|
|
}
|
|
if err := mapstructure.WeakDecode(params.Content, &data); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
|
|
return
|
|
} else if err := m.validate.Struct(params); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
|
|
return
|
|
}
|
|
|
|
t := &apistruct.ManagementSendMsgReq{
|
|
SendID: params.SendID,
|
|
GroupID: params.GroupID,
|
|
SenderNickname: params.SenderNickname,
|
|
SenderFaceURL: params.SenderFaceURL,
|
|
SenderPlatformID: params.SenderPlatformID,
|
|
Content: params.Content,
|
|
ContentType: params.ContentType,
|
|
SessionType: params.SessionType,
|
|
IsOnlineOnly: params.IsOnlineOnly,
|
|
NotOfflinePush: params.NotOfflinePush,
|
|
OfflinePushInfo: params.OfflinePushInfo,
|
|
}
|
|
pbReq := m.newUserSendMsgReq(c, t)
|
|
var recvList []string
|
|
if params.IsSendAll {
|
|
req2 := &user.GetAllUserIDReq{}
|
|
resp2, err := m.Message.GetAllUserID(c, req2)
|
|
if err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
|
|
return
|
|
}
|
|
recvList = resp2.UserIDs
|
|
} else {
|
|
recvList = params.RecvIDList
|
|
}
|
|
|
|
for _, recvID := range recvList {
|
|
pbReq.MsgData.RecvID = recvID
|
|
rpcResp, err := m.Client.SendMsg(c, pbReq)
|
|
if err != nil {
|
|
resp.Data.FailedIDList = append(resp.Data.FailedIDList, recvID)
|
|
msgSendFailedFlag = true
|
|
continue
|
|
}
|
|
resp.Data.ResultList = append(resp.Data.ResultList, &apistruct.SingleReturnResult{
|
|
ServerMsgID: rpcResp.ServerMsgID,
|
|
ClientMsgID: rpcResp.ClientMsgID,
|
|
SendTime: rpcResp.SendTime,
|
|
RecvID: recvID,
|
|
})
|
|
}
|
|
var status int32
|
|
if msgSendFailedFlag {
|
|
status = constant.MsgSendFailed
|
|
} else {
|
|
status = constant.MsgSendSuccessed
|
|
}
|
|
_, err := m.Client.SetSendMsgStatus(c, &msg.SetSendMsgStatusReq{Status: status})
|
|
if err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
|
|
return
|
|
}
|
|
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)
|
|
}
|
|
|
|
func (m *MessageApi) ManagementMsg(c *gin.Context) {
|
|
a2r.Call(msg.MsgClient.ManageMsg, m.Client, c)
|
|
}
|