mce/internal/controller/message_controller.go

package controller

import (
	"net/http"
	"strconv"

	"metazone.cc/metalab/internal/common"
	"metazone.cc/metalab/internal/model"

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

// notifProvider MessageController 所需的通知服务接口（ISP：5 个方法）
type notifProvider interface {
	List(userID uint, page, pageSize int) (*model.NotificationListResult, error)
	CountUnread(userID uint) (int64, error)
	MarkRead(id, userID uint) error
	MarkAllRead(userID uint) error
}

// MessageController 消息中心控制器
type MessageController struct {
	notifService notifProvider
}

// NewMessageController 构造函数
func NewMessageController(notifService notifProvider) *MessageController {
	return &MessageController{notifService: notifService}
}

// MessagesPage 消息中心页面（需登录，noindex）
func (mc *MessageController) MessagesPage(c *gin.Context) {
	uidVal, exists := c.Get("uid")
	if !exists {
		c.Redirect(http.StatusFound, "/auth/login")
		c.Abort()
		return
	}
	uid := uidVal.(uint)

	// 从 query 取分页参数
	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	if page < 1 {
		page = 1
	}
	pageSize := 20

	result, err := mc.notifService.List(uid, page, pageSize)
	if err != nil {
		result = &model.NotificationListResult{Items: []model.Notification{}, Total: 0, Page: 1}
	}

	totalPages := result.TotalPages
	if totalPages == 0 && result.Total > 0 {
		totalPages = int((result.Total + int64(pageSize) - 1) / int64(pageSize))
	}

	c.HTML(http.StatusOK, "messages/index.html", common.BuildPageData(c, gin.H{
		"Title":           "消息中心",
		"ExtraCSS":        "/static/css/messages.css",
		"Messages":        result.Items,
		"Total":           result.Total,
		"Page":            result.Page,
		"TotalPages":      totalPages,
		"PrevPage":        result.Page - 1,
		"NextPage":        result.Page + 1,
		"HasPrev":         result.Page > 1,
		"HasNext":         result.Page < totalPages,
		"UnreadCount":     result.Unread,
		"NotifyTypeNames": model.NotifyTypeNames,
		"AuditTypeNames":  model.AuditTypeNames,
	}))
}

// UnreadCount 获取未读消息数（API，供导航栏轮询）
func (mc *MessageController) UnreadCount(c *gin.Context) {
	uidVal, exists := c.Get("uid")
	if !exists {
		common.Ok(c, gin.H{"unread": 0})
		return
	}

	count, err := mc.notifService.CountUnread(uidVal.(uint))
	if err != nil {
		common.Ok(c, gin.H{"unread": 0})
		return
	}

	common.Ok(c, gin.H{"unread": count})
}

// MarkRead 标记单条消息已读（API）
func (mc *MessageController) MarkRead(c *gin.Context) {
	uidVal, exists := c.Get("uid")
	if !exists {
		common.Error(c, http.StatusUnauthorized, "请先登录")
		return
	}

	idStr := c.Param("id")
	id, err := strconv.ParseUint(idStr, 10, 64)
	if err != nil {
		common.Error(c, http.StatusBadRequest, "消息 ID 无效")
		return
	}

	if err := mc.notifService.MarkRead(uint(id), uidVal.(uint)); err != nil {
		common.Error(c, http.StatusInternalServerError, "操作失败")
		return
	}

	common.OkMessage(c, "已标记为已读")
}

// MarkAllRead 一键全部已读（API）
func (mc *MessageController) MarkAllRead(c *gin.Context) {
	uidVal, exists := c.Get("uid")
	if !exists {
		common.Error(c, http.StatusUnauthorized, "请先登录")
		return
	}

	if err := mc.notifService.MarkAllRead(uidVal.(uint)); err != nil {
		common.Error(c, http.StatusInternalServerError, "操作失败")
		return
	}

	common.OkMessage(c, "已全部标为已读")
}