update mico module

server/cmd/server/main.go

@@ -33,6 +33,7 @@ func main() {
 	authHandler := handlers.NewAuthHandler(jwtManager)
 	commentHandler := handlers.NewCommentHandler()
 	likeHandler := handlers.NewLikeHandler()
+	microHandler := handlers.NewMicroHandler()
 
 	// 创建路由
 	r := gin.New()
@@ -70,6 +71,12 @@ func main() {
 		api.GET("/likes", likeHandler.GetLikeStatus)
 		api.POST("/likes", likeHandler.ToggleLike) // 允许访客点赞（基于 IP Hash）
 
+		// 微语公开接口
+		api.GET("/micros", microHandler.GetMicros)
+		api.GET("/micros/stats", microHandler.GetStats)
+		api.GET("/micros/heatmap", microHandler.GetHeatmap)
+		api.GET("/micros/:id", microHandler.GetMicro)
+
 		// 需要认证的接口
 		authGroup := api.Group("")
 		authGroup.Use(middleware.AuthMiddleware(jwtManager))
@@ -81,6 +88,12 @@ func main() {
 			// 评论相关（需要登录才能评论）
 			authGroup.POST("/comments", commentHandler.CreateComment)
 			authGroup.DELETE("/comments/:id", commentHandler.DeleteComment)
+
+			// 微语相关（需要登录）
+			authGroup.POST("/micros", microHandler.CreateMicro)
+			authGroup.PUT("/micros/:id", microHandler.UpdateMicro)
+			authGroup.DELETE("/micros/:id", microHandler.DeleteMicro)
+			authGroup.POST("/micros/:id/like", microHandler.ToggleLike)
 		}
 
 		// 管理员接口

server/internal/database/database.go

@@ -54,6 +54,8 @@ func autoMigrate() error {
 		&models.Like{},
 		&models.LikeCount{},
 		&models.PostMeta{},
+		&models.MicroPost{},
+		&models.MicroPostLike{},
 	)
 }
 

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,
+	})
+}

server/internal/models/models.go

@@ -61,4 +61,26 @@ type PostMeta struct {
 	LikeCount  int       `json:"like_count" gorm:"default:0"`
 	CreatedAt  time.Time `json:"created_at"`
 	UpdatedAt  time.Time `json:"updated_at"`
+}
+
+// MicroPost 微语模型
+type MicroPost struct {
+	ID        uint           `json:"id" gorm:"primaryKey"`
+	CreatedAt time.Time      `json:"created_at"`
+	UpdatedAt time.Time      `json:"updated_at"`
+	DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`
+	UserID    uint           `json:"user_id" gorm:"index;not null"`
+	Content   string         `json:"content" gorm:"type:text;not null"`
+	Images    string         `json:"images" gorm:"type:text"`                 // JSON 数组存储图片 URL
+	Tags      string         `json:"tags" gorm:"type:text"`                   // JSON 数组存储标签
+	IsPublic  bool           `json:"is_public" gorm:"default:true"`           // 是否公开
+	User      User           `json:"user" gorm:"foreignKey:UserID"`
+}
+
+// MicroPostLike 微语点赞
+type MicroPostLike struct {
+	ID          uint      `json:"id" gorm:"primaryKey"`
+	CreatedAt   time.Time `json:"created_at"`
+	MicroPostID uint      `json:"micro_post_id" gorm:"uniqueIndex:idx_micropost_user;not null"`
+	UserID      uint      `json:"user_id" gorm:"uniqueIndex:idx_micropost_user;not null"`
 }