update mico module

2026-03-01 21:30:26 +08:00
parent 0c20c16c7a
commit d2312afc2b
12 changed files with 1120 additions and 1 deletions
--- a/server/internal/handlers/micro.go
+++ b/server/internal/handlers/micro.go
@@ -0,0 +1,411 @@
+package handlers
+
+import (
+	"encoding/json"
+	"net/http"
+	"strconv"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/novablog/server/internal/database"
+	"github.com/novablog/server/internal/middleware"
+	"github.com/novablog/server/internal/models"
+	"gorm.io/gorm"
+)
+
+func mustMarshal(v interface{}) []byte {
+	data, _ := json.Marshal(v)
+	return data
+}
+
+type MicroHandler struct{}
+
+func NewMicroHandler() *MicroHandler {
+	return &MicroHandler{}
+}
+
+type CreateMicroRequest struct {
+	Content  string   `json:"content" binding:"required,max=2000"`
+	Images   []string `json:"images"`
+	Tags     []string `json:"tags"`
+	IsPublic bool     `json:"is_public"`
+}
+
+type UpdateMicroRequest struct {
+	Content  string   `json:"content" binding:"required,max=2000"`
+	Images   []string `json:"images"`
+	Tags     []string `json:"tags"`
+	IsPublic bool     `json:"is_public"`
+}
+
+type MicroResponse struct {
+	models.MicroPost
+	LikeCount int  `json:"like_count"`
+	IsLiked   bool `json:"is_liked"`
+}
+
+func (h *MicroHandler) CreateMicro(c *gin.Context) {
+	userID, ok := middleware.GetUserID(c)
+	if !ok || userID == 0 {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "请登录后再发布"})
+		return
+	}
+
+	var req CreateMicroRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	imagesJSON := "[]"
+	if len(req.Images) > 0 {
+		imagesJSON = string(mustMarshal(req.Images))
+	}
+
+	tagsJSON := "[]"
+	if len(req.Tags) > 0 {
+		tagsJSON = string(mustMarshal(req.Tags))
+	}
+
+	micro := models.MicroPost{
+		UserID:   userID,
+		Content:  req.Content,
+		Images:   imagesJSON,
+		Tags:     tagsJSON,
+		IsPublic: req.IsPublic,
+	}
+
+	if err := database.DB.Create(&micro).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "发布失败"})
+		return
+	}
+
+	database.DB.Preload("User").First(&micro, micro.ID)
+
+	c.JSON(http.StatusCreated, MicroResponse{
+		MicroPost: micro,
+		LikeCount: 0,
+		IsLiked:   false,
+	})
+}
+
+func (h *MicroHandler) GetMicros(c *gin.Context) {
+	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
+	pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
+	userIDQuery := c.Query("user_id")
+
+	if page < 1 {
+		page = 1
+	}
+	if pageSize < 1 || pageSize > 50 {
+		pageSize = 20
+	}
+
+	var micros []models.MicroPost
+	var total int64
+
+	query := database.DB.Model(&models.MicroPost{}).Where("is_public = ?", true)
+
+	if userIDQuery != "" {
+		query = query.Where("user_id = ?", userIDQuery)
+	}
+
+	query.Count(&total)
+
+	if err := query.
+		Preload("User").
+		Order("created_at DESC").
+		Offset((page - 1) * pageSize).
+		Limit(pageSize).
+		Find(&micros).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "获取失败"})
+		return
+	}
+
+	currentUserID, _ := middleware.GetUserID(c)
+
+	responses := make([]MicroResponse, len(micros))
+	for i, micro := range micros {
+		var likeCount int64
+		database.DB.Model(&models.MicroPostLike{}).Where("micro_post_id = ?", micro.ID).Count(&likeCount)
+
+		isLiked := false
+		if currentUserID > 0 {
+			var count int64
+			database.DB.Model(&models.MicroPostLike{}).
+				Where("micro_post_id = ? AND user_id = ?", micro.ID, currentUserID).
+				Count(&count)
+			isLiked = count > 0
+		}
+
+		responses[i] = MicroResponse{
+			MicroPost: micro,
+			LikeCount: int(likeCount),
+			IsLiked:   isLiked,
+		}
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"data": responses,
+		"pagination": gin.H{
+			"page":       page,
+			"page_size":  pageSize,
+			"total":      total,
+			"total_page": (total + int64(pageSize) - 1) / int64(pageSize),
+		},
+	})
+}
+
+func (h *MicroHandler) GetMicro(c *gin.Context) {
+	idStr := c.Param("id")
+	id, err := strconv.ParseUint(idStr, 10, 32)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid id"})
+		return
+	}
+
+	var micro models.MicroPost
+	if err := database.DB.Preload("User").First(&micro, id).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
+			return
+		}
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
+		return
+	}
+
+	var likeCount int64
+	database.DB.Model(&models.MicroPostLike{}).Where("micro_post_id = ?", micro.ID).Count(&likeCount)
+
+	currentUserID, _ := middleware.GetUserID(c)
+	isLiked := false
+	if currentUserID > 0 {
+		var count int64
+		database.DB.Model(&models.MicroPostLike{}).
+			Where("micro_post_id = ? AND user_id = ?", micro.ID, currentUserID).
+			Count(&count)
+		isLiked = count > 0
+	}
+
+	c.JSON(http.StatusOK, MicroResponse{
+		MicroPost: micro,
+		LikeCount: int(likeCount),
+		IsLiked:   isLiked,
+	})
+}
+
+func (h *MicroHandler) UpdateMicro(c *gin.Context) {
+	userID, _ := middleware.GetUserID(c)
+	idStr := c.Param("id")
+	id, err := strconv.ParseUint(idStr, 10, 32)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid id"})
+		return
+	}
+
+	var micro models.MicroPost
+	if err := database.DB.First(&micro, id).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
+			return
+		}
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
+		return
+	}
+
+	if micro.UserID != userID {
+		c.JSON(http.StatusForbidden, gin.H{"error": "无权修改"})
+		return
+	}
+
+	var req UpdateMicroRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	imagesJSON := "[]"
+	if len(req.Images) > 0 {
+		imagesJSON = string(mustMarshal(req.Images))
+	}
+
+	tagsJSON := "[]"
+	if len(req.Tags) > 0 {
+		tagsJSON = string(mustMarshal(req.Tags))
+	}
+
+	updates := map[string]interface{}{
+		"content":   req.Content,
+		"images":    imagesJSON,
+		"tags":      tagsJSON,
+		"is_public": req.IsPublic,
+	}
+
+	if err := database.DB.Model(&micro).Updates(updates).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "更新失败"})
+		return
+	}
+
+	database.DB.Preload("User").First(&micro, micro.ID)
+
+	c.JSON(http.StatusOK, micro)
+}
+
+func (h *MicroHandler) DeleteMicro(c *gin.Context) {
+	userID, _ := middleware.GetUserID(c)
+	role, _ := c.Get("role")
+
+	idStr := c.Param("id")
+	id, err := strconv.ParseUint(idStr, 10, 32)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid id"})
+		return
+	}
+
+	var micro models.MicroPost
+	if err := database.DB.First(&micro, id).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
+			return
+		}
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
+		return
+	}
+
+	if micro.UserID != userID && role != "admin" {
+		c.JSON(http.StatusForbidden, gin.H{"error": "无权删除"})
+		return
+	}
+
+	if err := database.DB.Delete(&micro).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "删除失败"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "删除成功"})
+}
+
+func (h *MicroHandler) ToggleLike(c *gin.Context) {
+	userID, ok := middleware.GetUserID(c)
+	if !ok || userID == 0 {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "请登录后再点赞"})
+		return
+	}
+
+	idStr := c.Param("id")
+	id, err := strconv.ParseUint(idStr, 10, 32)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid id"})
+		return
+	}
+
+	var micro models.MicroPost
+	if err := database.DB.First(&micro, id).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
+			return
+		}
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
+		return
+	}
+
+	var existingLike models.MicroPostLike
+	result := database.DB.Where("micro_post_id = ? AND user_id = ?", id, userID).First(&existingLike)
+
+	if result.Error == nil {
+		database.DB.Delete(&existingLike)
+		c.JSON(http.StatusOK, gin.H{"liked": false, "message": "取消点赞"})
+		return
+	}
+
+	like := models.MicroPostLike{
+		MicroPostID: uint(id),
+		UserID:      userID,
+	}
+
+	if err := database.DB.Create(&like).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "点赞失败"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"liked": true, "message": "点赞成功"})
+}
+
+func (h *MicroHandler) GetHeatmap(c *gin.Context) {
+	userIDQuery := c.Query("user_id")
+	yearStr := c.DefaultQuery("year", strconv.Itoa(time.Now().Year()))
+	year, _ := strconv.Atoi(yearStr)
+
+	query := database.DB.Model(&models.MicroPost{}).
+		Select("DATE(created_at) as date, COUNT(*) as count").
+		Where("is_public = ?", true).
+		Group("DATE(created_at)")
+
+	if userIDQuery != "" {
+		query = query.Where("user_id = ?", userIDQuery)
+	}
+
+	if year > 0 {
+		startDate := time.Date(year, 1, 1, 0, 0, 0, 0, time.UTC)
+		endDate := time.Date(year+1, 1, 1, 0, 0, 0, 0, time.UTC)
+		query = query.Where("created_at >= ? AND created_at < ?", startDate, endDate)
+	}
+
+	var results []struct {
+		Date  string `json:"date"`
+		Count int    `json:"count"`
+	}
+
+	if err := query.Scan(&results).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "获取失败"})
+		return
+	}
+
+	c.JSON(http.StatusOK, results)
+}
+
+func (h *MicroHandler) GetStats(c *gin.Context) {
+	userIDQuery := c.Query("user_id")
+
+	var totalMicros int64
+	var totalUsers int64
+
+	query := database.DB.Model(&models.MicroPost{}).Where("is_public = ?", true)
+	if userIDQuery != "" {
+		query = query.Where("user_id = ?", userIDQuery)
+	}
+	query.Count(&totalMicros)
+
+	database.DB.Model(&models.User{}).Count(&totalUsers)
+
+	var topUsers []struct {
+		UserID    uint   `json:"user_id"`
+		Username  string `json:"username"`
+		Nickname  string `json:"nickname"`
+		Avatar    string `json:"avatar"`
+		PostCount int    `json:"post_count"`
+	}
+
+	database.DB.Model(&models.MicroPost{}).
+		Select("user_id, COUNT(*) as post_count").
+		Where("is_public = ?", true).
+		Group("user_id").
+		Order("post_count DESC").
+		Limit(10).
+		Scan(&topUsers)
+
+	for i := range topUsers {
+		var user models.User
+		if err := database.DB.First(&user, topUsers[i].UserID).Error; err == nil {
+			topUsers[i].Username = user.Username
+			topUsers[i].Nickname = user.Nickname
+			topUsers[i].Avatar = user.Avatar
+		}
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"total_micros": totalMicros,
+		"total_users":  totalUsers,
+		"top_users":    topUsers,
+	})
+}