nl-im-service/internal/api/handler.go

/**
 * package api
 * 作用：处理 HTTP 请求接口，包括消息发送、历史记录、系统状态检查等。
 * 说明：V2 版本中消息发送主要通过 HTTP 接口进行，而非 WebSocket 直接推送。
 */
package api

import (
	"fmt"
	"strconv"
	"time"
	"xk-websocket-v2/internal/manager"
	"xk-websocket-v2/internal/model"
	"xk-websocket-v2/internal/service"
	"xk-websocket-v2/internal/turnserver"
	"xk-websocket-v2/internal/utils"

	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
)

// ==========================================
// 消息发送相关接口
// ==========================================

/**
 * SendHandler
 * 功能：通用消息发送接口 (HTTP -> WebSocket)。
 * 路径：POST /api/send
 * 逻辑：支持通过 target_client_id 或 receiver_user_id 发送。
 * 重要：必须正确处理 SenderClientID，防止多端同步时误通知发送端自己。
 */
func SendHandler(c *gin.Context) {
	var req model.SendMessageReq
	// 1. 绑定并校验 JSON 参数
	if err := c.ShouldBindJSON(&req); err != nil {
		utils.BadRequest(c, "无效的JSON参数")
		return
	}

	// 2. 获取发送者 ID：强制取自 JWT（由认证中间件注入 user_id），
	// 禁止客户端通过 Header/Body 伪造他人身份发送消息或信令
	uid, ok := c.Get("user_id")
	if !ok {
		utils.Unauthorized(c, "未认证")
		return
	}
	senderID := uid.(string)

	// 3. [关键修复] 获取发送端的 WebSocket ClientID
	// 前端在调用此接口时，必须带上自己的 socket_client_id
	// 如果前端没传 (兼容旧代码)，则 fallback 到 api-gateway，这会导致发送端自己也收到"其他设备接听"通知
	clientID := req.SenderClientID
	if clientID == "" {
		clientID = "api-gateway"
	}

	// 4. 构造临时的发送者客户端对象
	// 这个对象将传递给 Service 层，用于识别消息来源
	mockClient := &manager.Client{
		UserID:   senderID,
		ID:       clientID,
		RemoteIP: utils.GetClientIP(c),
	}

	// 5. 调用核心业务逻辑处理消息，返回持久化后的消息体供前端对齐 ID
	savedMsg, err := service.ChatSvc.HandleUserMessage(mockClient, &req)

	// 6. 校验被拒/落库失败必须返回错误：原实现只看消息指针，
	// 被拒消息也响应"消息已发送"，发送方界面显示成功但对方永远收不到
	if err != nil {
		utils.Error(c, utils.CodeForbidden, err.Error())
		return
	}

	// 7. 返回成功响应（含服务端消息 ID，避免前端重复展示）
	if savedMsg != nil {
		utils.SuccessWithData(c, savedMsg, "消息已发送")
		return
	}
	utils.Success(c, "消息已发送")
}

/**
 * SendToUserHandler (V1 兼容)
 * 功能：专门用于给指定用户发送消息。
 * 路径：POST /api/send-to-user
 */
func SendToUserHandler(c *gin.Context) {
	// 复用 SendHandler，因为 V2 的 SendHandler 已经支持 ReceiverUserID
	SendHandler(c)
}

// ==========================================
// 用户与连接管理接口
// ==========================================

/**
 * BindHandler
 * 功能：手动绑定 ClientID 和 UserID。
 * 路径：POST /api/bind
 * 场景：当 WebSocket 连接建立后，客户端通过 HTTP 接口补充用户信息。
 */
func BindHandler(c *gin.Context) {
	var req model.BindReq
	if err := c.ShouldBindJSON(&req); err != nil {
		utils.BadRequest(c, "参数错误")
		return
	}
	// 绑定的用户ID强制取自 JWT，忽略请求体中的 user_id，
	// 防止攻击者把任意 user_id 绑定到自己的 client_id 从而劫持他人的消息推送
	uid, ok := c.Get("user_id")
	if !ok {
		utils.Unauthorized(c, "未认证")
		return
	}
	req.UserID = uid.(string)
	// 调用服务层进行绑定
	service.ChatSvc.BindUserByClientID(req.ClientID, req.UserID)
	utils.Success(c, "绑定成功")
}

/**
 * CheckUserOnlineHandler
 * 功能：检查用户是否在线。
 * 路径：GET /api/check-user-online
 */
func CheckUserOnlineHandler(c *gin.Context) {
	userID := c.Query("user_id")
	// 调用服务层查询 Redis
	isOnline := service.ChatSvc.IsUserOnline(userID)
	utils.SuccessWithData(c, gin.H{"is_online": isOnline}, "查询成功")
}

// ==========================================
// 数据查询接口
// ==========================================

/**
 * HistoryHandler
 * 功能：获取历史消息记录 (分页)。
 * 路径：GET /api/messages
 */
func HistoryHandler(c *gin.Context) {
	roomID := c.Query("room_id")
	if roomID == "" {
		utils.BadRequest(c, "room_id参数必填")
		return
	}

	// 鉴权：仅房间成员可拉取历史消息，防止任意登录用户凭 room_id 窥探他人聊天记录
	uid, ok := c.Get("user_id")
	if !ok {
		utils.Unauthorized(c, "未认证")
		return
	}
	if !service.RoomSvc.IsRoomMember(roomID, uid.(string)) {
		utils.Forbidden(c, "无权查看该房间消息")
		return
	}

	// 分页参数
	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "50"))

	if page < 1 {
		page = 1
	}
	if pageSize < 1 || pageSize > 100 {
		pageSize = 50
	}

	var msgs []model.ChatMessage
	var total int64

	// 排除当前用户已删除的消息（"删除仅对我生效"，见 MessageDeletion 模型注释）。
	// 用 NOT IN 子查询而非 JOIN：删除记录量级远小于消息量级，且保持分页语义简单
	deletedSubQuery := service.ChatSvc.DB.Model(&model.MessageDeletion{}).
		Select("message_id").
		Where("user_id = ?", uid.(string))

	// 获取总数（同样排除已删除，保证 total 与实际可见条数一致）
	service.ChatSvc.DB.Model(&model.ChatMessage{}).
		Where("room_id = ?", roomID).
		Where("id NOT IN (?)", deletedSubQuery).
		Count(&total)

	// 分页查询
	offset := (page - 1) * pageSize
	result := service.ChatSvc.DB.Where("room_id = ?", roomID).
		Where("id NOT IN (?)", deletedSubQuery).
		Order("created_at desc").
		Offset(offset).
		Limit(pageSize).
		Find(&msgs)

	if result.Error != nil {
		utils.InternalError(c, "db error")
		return
	}

	// 确保返回空数组而不是null
	if msgs == nil {
		msgs = []model.ChatMessage{}
	}

	utils.SuccessWithData(c, gin.H{
		"data":  msgs,
		"total": total,
		"page":  page,
		"size":  pageSize,
	}, "获取成功")
}

/**
 * RecallMessageHandler
 * 功能：撤回消息（2 分钟内，仅发送者可撤回）
 * 路径：POST /api/messages/recall
 */
func RecallMessageHandler(c *gin.Context) {
	userID, exists := c.Get("user_id")
	if !exists {
		utils.Unauthorized(c, "未认证")
		return
	}

	var req struct {
		MessageID uint `json:"message_id" binding:"required"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		utils.BadRequest(c, "参数错误")
		return
	}

	var msg model.ChatMessage
	if err := service.ChatSvc.DB.First(&msg, req.MessageID).Error; err != nil {
		utils.NotFound(c, "消息不存在")
		return
	}

	if msg.SenderUserID != userID.(string) {
		utils.Forbidden(c, "无权撤回该消息")
		return
	}

	if time.Since(msg.CreatedAt) > 2*time.Minute {
		utils.BadRequest(c, "超过2分钟，无法撤回")
		return
	}

	msg.MessageType = model.MessageTypeSystem
	msg.Content = "撤回了一条消息"
	if err := service.ChatSvc.DB.Save(&msg).Error; err != nil {
		utils.InternalError(c, "撤回失败")
		return
	}

	service.ChatSvc.BroadcastChatMessage(msg)

	utils.SuccessWithData(c, msg, "撤回成功")
}

/**
 * DeleteMessageHandler
 * 功能：删除消息（仅对操作者自己生效，对方仍可见，区别于撤回）。
 * 逻辑：不删除消息本体，仅在 message_deletions 表插入删除记录，
 * 历史消息查询时排除，避免"本地删了、刷新又回来"的假删除问题。
 * 路径：POST /api/messages/delete
 */
func DeleteMessageHandler(c *gin.Context) {
	userID, exists := c.Get("user_id")
	if !exists {
		utils.Unauthorized(c, "未认证")
		return
	}

	var req struct {
		MessageID uint `json:"message_id" binding:"required"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		utils.BadRequest(c, "参数错误")
		return
	}

	// 校验消息存在且操作者是所在房间成员（防止删除无权查看的消息记录）
	var msg model.ChatMessage
	if err := service.ChatSvc.DB.First(&msg, req.MessageID).Error; err != nil {
		utils.NotFound(c, "消息不存在")
		return
	}
	if !service.RoomSvc.IsRoomMember(msg.RoomID, userID.(string)) {
		utils.Forbidden(c, "无权删除该消息")
		return
	}

	// 幂等插入：重复删除同一条消息时唯一索引冲突，视为成功
	deletion := model.MessageDeletion{
		MessageID: req.MessageID,
		UserID:    userID.(string),
	}
	if err := service.ChatSvc.DB.Create(&deletion).Error; err != nil {
		if !utils.IsDuplicateEntryError(err) {
			utils.InternalError(c, "删除失败")
			return
		}
	}

	utils.Success(c, "删除成功")
}

/**
 * SyncMessagesHandler
 * 功能：全量同步消息 (V1 兼容)。
 * 路径：GET /api/messages/sync
 */
func SyncMessagesHandler(c *gin.Context) {
	HistoryHandler(c)
}

// ==========================================
// 系统与 WebRTC 接口
// ==========================================

/**
 * HealthHandler
 * 功能：服务健康检查。
 * 路径：GET /api/health
 */
func HealthHandler(c *gin.Context) {
	utils.SuccessWithData(c, gin.H{
		"status": "ok",
		"node":   viper.GetString("app.node_id"),
		"time":   time.Now().Format(time.RFC3339),
	}, "服务正常")
}

/**
 * ICEHandler
 * 功能：获取 TURN/STUN 服务器配置及临时凭证。
 * 路径：GET /api/ice-servers
 * 用途：WebRTC 前端在建立 PeerConnection 前需调用此接口。
 */
func ICEHandler(c *gin.Context) {
	// 凭证绑定当前登录用户，取自 JWT，避免匿名为任意 user_id 签发 TURN 凭证被滥用
	uid, ok := c.Get("user_id")
	if !ok {
		utils.Unauthorized(c, "未认证")
		return
	}
	userID := uid.(string)

	// 生成临时凭证 (HMAC-SHA1)
	username, credential := turnserver.GenerateCredentials(userID)

	// 从配置读取公网IP和端口
	ip := viper.GetString("turn.public_ip")
	port := viper.GetInt("turn.listen_port")

	// 构造配置返回给前端
	cfg := model.ICEServerConfig{
		Urls:       []string{fmt.Sprintf("turn:%s:%d", ip, port)},
		Username:   username,
		Credential: credential,
	}
	utils.SuccessWithData(c, []model.ICEServerConfig{cfg}, "获取成功")
}