✨ Reacting

pkg/server/posts_api.go

@@ -4,7 +4,12 @@ import (
 	"code.smartsheep.studio/hydrogen/interactive/pkg/database"
 	"code.smartsheep.studio/hydrogen/interactive/pkg/models"
 	"code.smartsheep.studio/hydrogen/interactive/pkg/services"
+	"fmt"
 	"github.com/gofiber/fiber/v2"
+	"github.com/google/uuid"
+	"github.com/samber/lo"
+	"github.com/spf13/viper"
+	"strings"
 )
 
 func listPost(c *fiber.Ctx) error {
@@ -14,12 +19,14 @@ func listPost(c *fiber.Ctx) error {
 	var count int64
 	var posts []models.Post
 	if err := database.C.
+		Where(&models.Post{RealmID: nil}).
 		Model(&models.Post{}).
 		Count(&count).Error; err != nil {
 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
 	}
 
 	if err := database.C.
+		Where(&models.Post{RealmID: nil}).
 		Limit(take).
 		Offset(offset).
 		Order("created_at desc").
@@ -28,6 +35,34 @@ func listPost(c *fiber.Ctx) error {
 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
 	}
 
+	postIds := lo.Map(posts, func(item models.Post, _ int) uint {
+		return item.ID
+	})
+
+	var reactInfo []struct {
+		PostID       uint
+		LikeCount    int64
+		DislikeCount int64
+	}
+
+	prefix := viper.GetString("database.prefix")
+	database.C.Raw(fmt.Sprintf(`SELECT t.id                         as post_id,
+       COALESCE(l.like_count, 0)    AS like_count,
+       COALESCE(d.dislike_count, 0) AS dislike_count
+FROM %sposts t
+         LEFT JOIN (SELECT post_id, COUNT(*) AS like_count
+                    FROM %spost_likes
+                    GROUP BY post_id) l ON t.id = l.post_id
+         LEFT JOIN (SELECT post_id, COUNT(*) AS dislike_count
+                    FROM %spost_dislikes
+                    GROUP BY post_id) d ON t.id = d.post_id
+WHERE t.id IN (?)`, prefix, prefix, prefix), postIds).Scan(&reactInfo)
+
+	for idx, info := range reactInfo {
+		posts[idx].LikeCount = info.LikeCount
+		posts[idx].DislikeCount = info.DislikeCount
+	}
+
 	return c.JSON(fiber.Map{
 		"count": count,
 		"data":  posts,
@@ -39,8 +74,8 @@ func createPost(c *fiber.Ctx) error {
 	user := c.Locals("principal").(models.Account)
 
 	var data struct {
-		Alias      string            `json:"alias" validate:"required"`
-		Title      string            `json:"title" validate:"required"`
+		Alias      string            `json:"alias"`
+		Title      string            `json:"title"`
 		Content    string            `json:"content" validate:"required"`
 		Tags       []models.Tag      `json:"tags"`
 		Categories []models.Category `json:"categories"`
@@ -48,6 +83,8 @@ func createPost(c *fiber.Ctx) error {
 
 	if err := BindAndValidate(c, &data); err != nil {
 		return err
+	} else if len(data.Alias) == 0 {
+		data.Alias = strings.ReplaceAll(uuid.NewString(), "-", "")
 	}
 
 	post, err := services.NewPost(user, data.Alias, data.Title, data.Content, data.Categories, data.Tags)
@@ -57,3 +94,32 @@ func createPost(c *fiber.Ctx) error {
 
 	return c.JSON(post)
 }
+
+func reactPost(c *fiber.Ctx) error {
+	user := c.Locals("principal").(models.Account)
+	id, _ := c.ParamsInt("postId", 0)
+
+	var post models.Post
+	if err := database.C.Where(&models.Post{
+		BaseModel: models.BaseModel{ID: uint(id)},
+	}).First(&post).Error; err != nil {
+		return fiber.NewError(fiber.StatusNotFound, err.Error())
+	}
+
+	switch strings.ToLower(c.Params("reactType")) {
+	case "like":
+		if positive, err := services.LikePost(user, post); err != nil {
+			return fiber.NewError(fiber.StatusBadRequest, err.Error())
+		} else {
+			return c.SendStatus(lo.Ternary(positive, fiber.StatusCreated, fiber.StatusNoContent))
+		}
+	case "dislike":
+		if positive, err := services.DislikePost(user, post); err != nil {
+			return fiber.NewError(fiber.StatusBadRequest, err.Error())
+		} else {
+			return c.SendStatus(lo.Ternary(positive, fiber.StatusCreated, fiber.StatusNoContent))
+		}
+	default:
+		return fiber.NewError(fiber.StatusBadRequest, "unsupported reaction")
+	}
+}

pkg/server/startup.go

@@ -64,6 +64,7 @@ func NewServer() {
 
 		api.Get("/posts", listPost)
 		api.Post("/posts", auth, createPost)
+		api.Post("/posts/:postId/react/:reactType", auth, reactPost)
 	}
 
 	A.Use("/", cache.New(cache.Config{