273 lines
7.5 KiB
Go
273 lines
7.5 KiB
Go
// Package controller 提供 IM 模块的 HTTP 接口处理
|
||
package controller
|
||
|
||
import (
|
||
"strconv"
|
||
|
||
"github.com/echochat/backend/app/dto"
|
||
"github.com/echochat/backend/app/im/service"
|
||
"github.com/echochat/backend/pkg/middleware"
|
||
"github.com/echochat/backend/pkg/utils"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// IMController 即时通讯控制器（REST API）
|
||
type IMController struct {
|
||
imService *service.IMService
|
||
}
|
||
|
||
// NewIMController 创建 IMController 实例
|
||
func NewIMController(imService *service.IMService) *IMController {
|
||
return &IMController{imService: imService}
|
||
}
|
||
|
||
// GetConversations 获取会话列表
|
||
// GET /api/v1/im/conversations
|
||
func (ctl *IMController) GetConversations(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
result, err := ctl.imService.GetConversationList(ctx, userID)
|
||
if err != nil {
|
||
ctl.handleError(c, err, "获取会话列表失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, result)
|
||
}
|
||
|
||
// GetHistoryMessages 获取历史消息
|
||
// GET /api/v1/im/messages?conversation_id=xx&before_id=xx&limit=30
|
||
func (ctl *IMController) GetHistoryMessages(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
var req dto.HistoryMessageRequest
|
||
if err := c.ShouldBindQuery(&req); err != nil {
|
||
utils.ResponseBadRequest(c, "请求参数错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
result, err := ctl.imService.GetHistoryMessages(ctx, userID, &req)
|
||
if err != nil {
|
||
ctl.handleError(c, err, "获取历史消息失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, result)
|
||
}
|
||
|
||
// SetDoNotDisturb 设置/取消会话消息免打扰
|
||
// PUT /api/v1/im/conversations/:id/dnd
|
||
func (ctl *IMController) SetDoNotDisturb(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
convID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
utils.ResponseBadRequest(c, "会话 ID 格式错误")
|
||
return
|
||
}
|
||
|
||
var req dto.SetDoNotDisturbRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
utils.ResponseBadRequest(c, "请求参数错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
if err := ctl.imService.SetDoNotDisturb(ctx, userID, convID, req.IsDoNotDisturb); err != nil {
|
||
ctl.handleError(c, err, "设置免打扰失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, nil)
|
||
}
|
||
|
||
// PinConversation 置顶/取消置顶会话
|
||
// PUT /api/v1/im/conversations/:id/pin
|
||
func (ctl *IMController) PinConversation(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
convID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
utils.ResponseBadRequest(c, "会话 ID 格式错误")
|
||
return
|
||
}
|
||
|
||
var req dto.PinConversationRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
utils.ResponseBadRequest(c, "请求参数错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
if err := ctl.imService.PinConversation(ctx, userID, convID, req.IsPinned); err != nil {
|
||
ctl.handleError(c, err, "更新置顶状态失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, nil)
|
||
}
|
||
|
||
// DeleteConversation 删除会话
|
||
// DELETE /api/v1/im/conversations/:id
|
||
func (ctl *IMController) DeleteConversation(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
convID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
utils.ResponseBadRequest(c, "会话 ID 格式错误")
|
||
return
|
||
}
|
||
|
||
if err := ctl.imService.DeleteConversation(ctx, userID, convID); err != nil {
|
||
ctl.handleError(c, err, "删除会话失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, nil)
|
||
}
|
||
|
||
// ClearHistory 清空聊天记录
|
||
// DELETE /api/v1/im/conversations/:id/messages
|
||
func (ctl *IMController) ClearHistory(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
convID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
utils.ResponseBadRequest(c, "会话 ID 格式错误")
|
||
return
|
||
}
|
||
|
||
if err := ctl.imService.ClearHistory(ctx, userID, convID); err != nil {
|
||
ctl.handleError(c, err, "清空聊天记录失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, nil)
|
||
}
|
||
|
||
// SearchMessages 全局消息搜索
|
||
// GET /api/v1/im/messages/search?keyword=xx&limit=50
|
||
func (ctl *IMController) SearchMessages(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
var req dto.SearchMessageRequest
|
||
if err := c.ShouldBindQuery(&req); err != nil {
|
||
utils.ResponseBadRequest(c, "请求参数错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
result, err := ctl.imService.SearchMessages(ctx, userID, &req)
|
||
if err != nil {
|
||
ctl.handleError(c, err, "消息搜索失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, result)
|
||
}
|
||
|
||
// GetTotalUnread 获取全局未读消息总数
|
||
// GET /api/v1/im/unread
|
||
func (ctl *IMController) GetTotalUnread(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
count, err := ctl.imService.GetTotalUnread(ctx, userID)
|
||
if err != nil {
|
||
ctl.handleError(c, err, "获取未读消息数失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, gin.H{"total_unread": count})
|
||
}
|
||
|
||
// GetMessageReadDetail 获取消息已读详情
|
||
// GET /api/v1/im/messages/:id/reads
|
||
func (ctl *IMController) GetMessageReadDetail(c *gin.Context) {
|
||
ctx := c.Request.Context()
|
||
userID, ok := middleware.GetCurrentUserID(c)
|
||
if !ok {
|
||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||
return
|
||
}
|
||
|
||
messageID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
utils.ResponseBadRequest(c, "消息 ID 格式错误")
|
||
return
|
||
}
|
||
|
||
result, err := ctl.imService.GetMessageReadDetail(ctx, userID, messageID)
|
||
if err != nil {
|
||
ctl.handleError(c, err, "获取已读详情失败")
|
||
return
|
||
}
|
||
utils.ResponseOK(c, result)
|
||
}
|
||
|
||
// handleError 统一业务错误映射
|
||
// 已知业务错误 → 返回 Service 层定义的具体提示
|
||
// 未知错误 → 返回 fallbackMsg（未传则默认"服务器内部错误"）
|
||
func (ctl *IMController) handleError(c *gin.Context, err error, fallbackMsg ...string) {
|
||
switch err {
|
||
case service.ErrNotFriend:
|
||
utils.ResponseForbidden(c, err.Error())
|
||
case service.ErrEmptyContent:
|
||
utils.ResponseBadRequest(c, err.Error())
|
||
case service.ErrInvalidMsgType:
|
||
utils.ResponseBadRequest(c, err.Error())
|
||
case service.ErrConvNotFound:
|
||
utils.ResponseNotFound(c, err.Error())
|
||
case service.ErrMsgNotFound:
|
||
utils.ResponseNotFound(c, err.Error())
|
||
case service.ErrNotSender:
|
||
utils.ResponseForbidden(c, err.Error())
|
||
case service.ErrRecallTimeout:
|
||
utils.ResponseBadRequest(c, err.Error())
|
||
case service.ErrNotMember:
|
||
utils.ResponseForbidden(c, err.Error())
|
||
case service.ErrDuplicateMsg:
|
||
utils.ResponseBadRequest(c, err.Error())
|
||
case service.ErrGroupDissolved:
|
||
utils.ResponseBadRequest(c, err.Error())
|
||
case service.ErrGroupAllMuted:
|
||
utils.ResponseForbidden(c, err.Error())
|
||
case service.ErrUserMuted:
|
||
utils.ResponseForbidden(c, err.Error())
|
||
default:
|
||
msg := "服务器内部错误"
|
||
if len(fallbackMsg) > 0 && fallbackMsg[0] != "" {
|
||
msg = fallbackMsg[0]
|
||
}
|
||
utils.ResponseError(c, msg)
|
||
}
|
||
}
|