paopao-ce/internal/routers/api/message.go

package api

import (
	"github.com/gin-gonic/gin"
	"github.com/rocboss/paopao-ce/global"
	"github.com/rocboss/paopao-ce/internal/model"
	"github.com/rocboss/paopao-ce/internal/service"
	"github.com/rocboss/paopao-ce/pkg/app"
	"github.com/rocboss/paopao-ce/pkg/errcode"
)

func GetUnreadMsgCount(c *gin.Context) {
	response := app.NewResponse(c)

	user := &model.User{}
	if u, exists := c.Get("USER"); exists {
		user = u.(*model.User)
	}

	count, _ := service.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 := service.GetMessages(userID.(int64), (app.GetPage(c)-1)*app.GetPageSize(c), app.GetPageSize(c))

	if err != nil {
		global.Logger.Errorf("service.GetMessages err: %v\n", err)
		response.ToErrorResponse(errcode.GetMessagesFailed)
		return
	}

	response.ToResponseList(messages, totalRows)
}

func ReadMessage(c *gin.Context) {
	param := service.ReadMessageReq{}
	response := app.NewResponse(c)
	valid, errs := app.BindAndValid(c, &param)
	if !valid {
		global.Logger.Errorf("app.BindAndValid errs: %v", errs)
		response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
		return
	}

	userID, _ := c.Get("UID")
	err := service.ReadMessage(param.ID, userID.(int64))
	if err != nil {
		global.Logger.Errorf("service.ReadMessage err: %v\n", err)
		response.ToErrorResponse(errcode.ReadMessageFailed)
		return
	}

	response.ToResponse(nil)
}

func SendUserWhisper(c *gin.Context) {
	param := service.WhisperReq{}
	response := app.NewResponse(c)
	valid, errs := app.BindAndValid(c, &param)
	if !valid {
		global.Logger.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 := service.CreateWhisper(c, &model.Message{
		SenderUserID:   userID.(int64),
		ReceiverUserID: param.UserID,
		Type:           model.MESSAGE_WHISPER,
		Brief:          "给你发送新私信了",
		Content:        param.Content,
	})

	if err != nil {
		global.Logger.Errorf("service.CreateWhisper err: %v\n", err)

		if err == errcode.TooManyWhisperNum {
			response.ToErrorResponse(errcode.TooManyWhisperNum)
		} else {
			response.ToErrorResponse(errcode.SendWhisperFailed)
		}
		return
	}

	response.ToResponse(nil)
}