mirror of https://github.com/rocboss/paopao-ce
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.
106 lines
2.6 KiB
106 lines
2.6 KiB
// Copyright 2022 ROC. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package api
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
"github.com/rocboss/paopao-ce/internal/servants/web/broker"
|
|
"github.com/rocboss/paopao-ce/pkg/app"
|
|
"github.com/rocboss/paopao-ce/pkg/errcode"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func GetUnreadMsgCount(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
|
|
user := &core.User{}
|
|
if u, exists := c.Get("USER"); exists {
|
|
user = u.(*core.User)
|
|
}
|
|
|
|
count, _ := broker.GetUnreadCount(user.ID)
|
|
|
|
response.ToResponse(gin.H{
|
|
"count": count,
|
|
})
|
|
}
|
|
|
|
func GetMessages(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
|
|
userID, _ := c.Get("UID")
|
|
messages, totalRows, err := broker.GetMessages(userID.(int64), (app.GetPage(c)-1)*app.GetPageSize(c), app.GetPageSize(c))
|
|
|
|
if err != nil {
|
|
logrus.Errorf("service.GetMessages err: %v\n", err)
|
|
response.ToErrorResponse(errcode.GetMessagesFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponseList(messages, totalRows)
|
|
}
|
|
|
|
func ReadMessage(c *gin.Context) {
|
|
param := broker.ReadMessageReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
userID, _ := c.Get("UID")
|
|
err := broker.ReadMessage(param.ID, userID.(int64))
|
|
if err != nil {
|
|
logrus.Errorf("service.ReadMessage err: %v\n", err)
|
|
response.ToErrorResponse(errcode.ReadMessageFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(nil)
|
|
}
|
|
|
|
func SendUserWhisper(c *gin.Context) {
|
|
param := broker.WhisperReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
userID, _ := c.Get("UID")
|
|
|
|
// 不允许发送私信给自己
|
|
if userID.(int64) == param.UserID {
|
|
response.ToErrorResponse(errcode.NoWhisperToSelf)
|
|
return
|
|
}
|
|
|
|
_, err := broker.CreateWhisper(c, &core.Message{
|
|
SenderUserID: userID.(int64),
|
|
ReceiverUserID: param.UserID,
|
|
Type: core.MsgTypeWhisper,
|
|
Brief: "给你发送新私信了",
|
|
Content: param.Content,
|
|
})
|
|
|
|
if err != nil {
|
|
logrus.Errorf("service.CreateWhisper err: %v\n", err)
|
|
|
|
if err == errcode.TooManyWhisperNum {
|
|
response.ToErrorResponse(errcode.TooManyWhisperNum)
|
|
} else {
|
|
response.ToErrorResponse(errcode.SendWhisperFailed)
|
|
}
|
|
return
|
|
}
|
|
|
|
response.ToResponse(nil)
|
|
}
|