✨ Reacting

pkg/services/posts.go

@@ -3,6 +3,8 @@ package services
 import (
 	"code.smartsheep.studio/hydrogen/interactive/pkg/database"
 	"code.smartsheep.studio/hydrogen/interactive/pkg/models"
+	"errors"
+	"gorm.io/gorm"
 )
 
 func NewPost(
@@ -57,3 +59,41 @@ func NewPostWithRealm(
 
 	return post, nil
 }
+
+func LikePost(user models.Account, post models.Post) (bool, error) {
+	var like models.PostLike
+	if err := database.C.Where(&models.PostLike{
+		AccountID: user.ID,
+		PostID:    post.ID,
+	}).First(&like).Error; err != nil {
+		if !errors.Is(err, gorm.ErrRecordNotFound) {
+			return true, err
+		}
+		like = models.PostLike{
+			AccountID: user.ID,
+			PostID:    post.ID,
+		}
+		return true, database.C.Save(&like).Error
+	} else {
+		return false, database.C.Delete(&like).Error
+	}
+}
+
+func DislikePost(user models.Account, post models.Post) (bool, error) {
+	var dislike models.PostDislike
+	if err := database.C.Where(&models.PostDislike{
+		AccountID: user.ID,
+		PostID:    post.ID,
+	}).First(&dislike).Error; err != nil {
+		if !errors.Is(err, gorm.ErrRecordNotFound) {
+			return true, err
+		}
+		dislike = models.PostDislike{
+			AccountID: user.ID,
+			PostID:    post.ID,
+		}
+		return true, database.C.Save(&dislike).Error
+	} else {
+		return false, database.C.Delete(&dislike).Error
+	}
+}