lv8girl!

internal/controllers/home.go

@@ -0,0 +1,150 @@
+package controllers
+
+import (
+	"net/http"
+
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"lv8girl/internal/middleware"
+	"lv8girl/internal/repositories"
+	"lv8girl/internal/services"
+)
+
+type HomeController struct {
+	discussionSvc  *services.DiscussionService
+	userSvc        *services.UserService
+	messageSvc     *services.MessageService
+	discussionRepo *repositories.DiscussionRepository
+}
+
+func NewHomeController() *HomeController {
+	return &HomeController{
+		discussionSvc:  services.NewDiscussionService(),
+		userSvc:        services.NewUserService(),
+		messageSvc:     services.NewMessageService(),
+		discussionRepo: repositories.NewDiscussionRepository(),
+	}
+}
+
+func (c *HomeController) Index(ctx *gin.Context) {
+	session := sessions.Default(ctx)
+	userID, _ := session.Get("user_id").(uint)
+	username, _ := session.Get("username").(string)
+	userRole, _ := session.Get("user_role").(string)
+	isLoggedIn := userID != 0
+
+	if isLoggedIn {
+		userRepo := repositories.NewUserRepository()
+		userRepo.UpdateLastActive(userID)
+	}
+
+	posts, _ := c.discussionSvc.GetApprovedPosts(30)
+
+	var postCount, userCount, onlineCount int64
+	postCount, _ = c.discussionRepo.CountByStatus("approved")
+	userCount, onlineCount, _ = c.userSvc.GetUserStats()
+
+	var unreadCount int64
+	if isLoggedIn {
+		unreadCount, _ = c.messageSvc.GetUnreadCount(userID)
+	}
+
+	ctx.HTML(http.StatusOK, "index.html", gin.H{
+		"IsLoggedIn":  isLoggedIn,
+		"UserID":      userID,
+		"Username":    username,
+		"UserRole":    userRole,
+		"Posts":       posts,
+		"PostCount":   postCount,
+		"UserCount":   userCount,
+		"OnlineCount": onlineCount,
+		"UnreadCount": unreadCount,
+	})
+}
+
+func (c *HomeController) ShowPost(ctx *gin.Context) {
+	postID := parseUint(ctx.Param("id"))
+	userID, username, userRole, isLoggedIn := middleware.GetCurrentUser(ctx)
+
+	session := sessions.Default(ctx)
+	viewedKey := "viewed_posts"
+	viewedPosts := session.Get(viewedKey)
+	var viewedMap map[uint]bool
+	if viewedPosts == nil {
+		viewedMap = make(map[uint]bool)
+	} else {
+		viewedMap = viewedPosts.(map[uint]bool)
+	}
+
+	if !viewedMap[postID] {
+		c.discussionSvc.IncrementViews(postID)
+		viewedMap[postID] = true
+		session.Set(viewedKey, viewedMap)
+		session.Save()
+	}
+
+	detail, err := c.discussionSvc.GetPostDetail(postID, userID)
+	if err != nil {
+		ctx.String(http.StatusNotFound, "帖子不存在")
+		return
+	}
+
+	comments, _ := c.discussionSvc.GetComments(postID)
+
+	avatar := ""
+	if detail.Post.User.Avatar != "" {
+		avatar = "/" + detail.Post.User.Avatar
+	}
+
+	ctx.HTML(http.StatusOK, "post.html", gin.H{
+		"IsLoggedIn":      isLoggedIn,
+		"UserID":          userID,
+		"Username":        username,
+		"UserRole":        userRole,
+		"Post":            detail.Post,
+		"PostAvatar":      avatar,
+		"Comments":        comments,
+		"LikeCount":       detail.LikeCount,
+		"UserLiked":       detail.UserLiked,
+		"AuthorPostCount": detail.AuthorPostCount,
+	})
+}
+
+func (c *HomeController) LikePost(ctx *gin.Context) {
+	userID, _, _, isLoggedIn := middleware.GetCurrentUser(ctx)
+	if !isLoggedIn {
+		ctx.JSON(http.StatusUnauthorized, gin.H{"error": "请先登录"})
+		return
+	}
+
+	postID := parseUint(ctx.Param("id"))
+	c.discussionSvc.AddLike(postID, userID)
+	ctx.Redirect(http.StatusFound, "/post/"+ctx.Param("id"))
+}
+
+func (c *HomeController) AddComment(ctx *gin.Context) {
+	userID, _, _, isLoggedIn := middleware.GetCurrentUser(ctx)
+	if !isLoggedIn {
+		ctx.Redirect(http.StatusFound, "/login")
+		return
+	}
+
+	postID := parseUint(ctx.Param("id"))
+	content := ctx.PostForm("content")
+
+	if content != "" {
+		c.discussionSvc.AddComment(postID, userID, content)
+	}
+
+	ctx.Redirect(http.StatusFound, "/post/"+ctx.Param("id"))
+}
+
+func parseUint(s string) uint {
+	var result uint
+	for _, c := range s {
+		if c >= '0' && c <= '9' {
+			result = result*10 + uint(c-'0')
+		}
+	}
+	return result
+}