feat: add like/unlike post endpoint

ainunns · ainunns · commit 96c2ae9546dd · 2025-04-11T14:28:51.000+07:00
diff --git a/controller/likes_controller.go b/controller/likes_controller.go
@@ -0,0 +1,72 @@
+package controller
+
+import (
+	"net/http"
+	"strconv"
+
+	"github.com/Lab-RPL-ITS/twitter-clone-api/dto"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/service"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/utils"
+	"github.com/gin-gonic/gin"
+)
+
+type (
+	LikesController interface {
+		LikePostById(ctx *gin.Context)
+		UnlikePostById(ctx *gin.Context)
+	}
+
+	likesController struct {
+		likesService service.LikesService
+	}
+)
+
+func NewLikesController(likesService service.LikesService) LikesController {
+	return &likesController{
+		likesService: likesService,
+	}
+}
+
+func (c *likesController) LikePostById(ctx *gin.Context) {
+	postId := ctx.Param("post_id")
+	userId := ctx.GetString("user_id")
+
+	postIdUint, err := strconv.ParseUint(postId, 10, 64)
+	if err != nil {
+		response := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_POST_ID, err.Error(), nil)
+		ctx.JSON(http.StatusBadRequest, response)
+		return
+	}
+
+	err = c.likesService.LikePostById(ctx, postIdUint, userId)
+	if err != nil {
+		response := utils.BuildResponseFailed(dto.MESSAGE_FAILED_LIKE_POST, err.Error(), nil)
+		ctx.JSON(http.StatusBadRequest, response)
+		return
+	}
+
+	response := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_LIKE_POST, nil)
+	ctx.JSON(http.StatusOK, response)
+}
+
+func (c *likesController) UnlikePostById(ctx *gin.Context) {
+	postId := ctx.Param("post_id")
+	userId := ctx.GetString("user_id")
+
+	postIdUint, err := strconv.ParseUint(postId, 10, 64)
+	if err != nil {
+		response := utils.BuildResponseFailed(dto.MESSAGE_FAILED_GET_POST_ID, err.Error(), nil)
+		ctx.JSON(http.StatusBadRequest, response)
+		return
+	}
+
+	err = c.likesService.UnLikePostById(ctx, postIdUint, userId)
+	if err != nil {
+		response := utils.BuildResponseFailed(dto.MESSAGE_FAILED_UNLIKE_POST, err.Error(), nil)
+		ctx.JSON(http.StatusBadRequest, response)
+		return
+	}
+
+	response := utils.BuildResponseSuccess(dto.MESSAGE_SUCCESS_UNLIKE_POST, nil)
+	ctx.JSON(http.StatusOK, response)
+}
diff --git a/dto/likes_dto.go b/dto/likes_dto.go
@@ -0,0 +1,25 @@
+package dto
+
+import "errors"
+
+const (
+	// Failed
+	MESSAGE_FAILED_LIKE_POST   = "failed like post"
+	MESSAGE_FAILED_UNLIKE_POST = "failed unlike post"
+
+	// Succcess
+	MESSAGE_SUCCESS_LIKE_POST   = "success like post"
+	MESSAGE_SUCCESS_UNLIKE_POST = "success unlike post"
+)
+
+var (
+	ErrLikePostById   = errors.New("failed to like post")
+	ErrCheckLikedPost = errors.New("failed to check liked post")
+	ErrUnlikePostById = errors.New("failed to unlike post")
+)
+
+type (
+	LikesRequest struct {
+		PostID uint64 `json:"post_id" form:"post_id" binding:"required"`
+	}
+)
diff --git a/dto/post_dto.go b/dto/post_dto.go
@@ -38,11 +38,12 @@ type (
 	}
 
 	PostResponse struct {
-		ID       uint64         `json:"id"`
-		Text     string         `json:"text"`
-		ParentID *uint64        `json:"parent_id"`
-		User     UserResponse   `json:"user"`
-		Replies  []PostResponse `json:"replies,omitempty"`
+		ID         uint64         `json:"id"`
+		Text       string         `json:"text"`
+		TotalLikes uint64         `json:"total_likes"`
+		ParentID   *uint64        `json:"parent_id"`
+		User       UserResponse   `json:"user"`
+		Replies    []PostResponse `json:"replies,omitempty"`
 	}
 
 	PostUpdateRequest struct {
diff --git a/entity/likes_entity.go b/entity/likes_entity.go
@@ -0,0 +1,13 @@
+package entity
+
+import "github.com/google/uuid"
+
+type Like struct {
+	PostID uint64 `gorm:"primaryKey;not null" json:"post_id"`
+	Post   Post   `gorm:"foreignkey:PostID" json:"post"`
+
+	UserID uuid.UUID `gorm:"primaryKey;not null" json:"user_id"`
+	User   User      `gorm:"foreignkey:UserID" json:"user"`
+
+	Timestamp
+}
diff --git a/entity/post_entity.go b/entity/post_entity.go
@@ -3,8 +3,9 @@ package entity
 import "github.com/google/uuid"
 
 type Post struct {
-	ID   uint64 `gorm:"primary_key;autoIncrement" json:"id"`
-	Text string `gorm:"not null" json:"text"`
+	ID         uint64 `gorm:"primaryKey;autoIncrement" json:"id"`
+	Text       string `gorm:"not null" json:"text"`
+	TotalLikes uint64 `gorm:"default:0" json:"total_likes"`
 
 	Parent   *Post   `gorm:"foreignkey:ParentID" json:"parent,omitempty"`
 	ParentID *uint64 `json:"parent_id,omitempty"`
diff --git a/entity/user_entity.go b/entity/user_entity.go
@@ -7,7 +7,7 @@ import (
 )
 
 type User struct {
-	ID       uuid.UUID `gorm:"type:uuid;primary_key;default:uuid_generate_v4()" json:"id"`
+	ID       uuid.UUID `gorm:"type:uuid;primaryKey;default:uuid_generate_v4()" json:"id"`
 	Name     string    `gorm:"not null" json:"name"`
 	Username string    `gorm:"not null" gorm:"unique" json:"username"`
 	Bio      *string   `json:"bio"`
diff --git a/migrations/migrate.go b/migrations/migrate.go
@@ -9,6 +9,7 @@ func Migrate(db *gorm.DB) error {
 	if err := db.AutoMigrate(
 		&entity.User{},
 		&entity.Post{},
+		&entity.Like{},
 	); err != nil {
 		return err
 	}
diff --git a/provider/core.go b/provider/core.go
@@ -23,4 +23,5 @@ func RegisterDependencies(injector *do.Injector) {
 
 	ProvideUserDependencies(injector)
 	ProvidePostDependencies(injector)
+	ProvideLikesDependencies(injector)
 }
diff --git a/provider/likes_provider.go b/provider/likes_provider.go
@@ -0,0 +1,27 @@
+package provider
+
+import (
+	"github.com/Lab-RPL-ITS/twitter-clone-api/constants"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/controller"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/repository"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/service"
+	"github.com/samber/do"
+	"gorm.io/gorm"
+)
+
+func ProvideLikesDependencies(injector *do.Injector) {
+	db := do.MustInvokeNamed[*gorm.DB](injector, constants.DB)
+	jwtService := do.MustInvokeNamed[service.JWTService](injector, constants.JWTService)
+
+	// Repository
+	likesRepository := repository.NewLikesRepository(db)
+	postRepository := repository.NewPostRepository(db)
+
+	// Service
+	likesService := service.NewLikesService(likesRepository, postRepository, jwtService)
+
+	// Controller
+	do.Provide(injector, func(i *do.Injector) (controller.LikesController, error) {
+		return controller.NewLikesController(likesService), nil
+	})
+}
diff --git a/repository/likes_repository.go b/repository/likes_repository.go
@@ -0,0 +1,77 @@
+package repository
+
+import (
+	"context"
+
+	"github.com/Lab-RPL-ITS/twitter-clone-api/entity"
+	"github.com/google/uuid"
+	"gorm.io/gorm"
+)
+
+type (
+	LikesRepository interface {
+		LikePostById(ctx context.Context, tx *gorm.DB, postId uint64, userId string) error
+		CheckLikedPost(ctx context.Context, tx *gorm.DB, postId uint64, userId string) error
+		ErrUnlikePostById(ctx context.Context, tx *gorm.DB, postId uint64, userId string) error
+	}
+
+	likesRepository struct {
+		db *gorm.DB
+	}
+)
+
+func NewLikesRepository(db *gorm.DB) LikesRepository {
+	return &likesRepository{
+		db: db,
+	}
+}
+
+func (r *likesRepository) LikePostById(ctx context.Context, tx *gorm.DB, postId uint64, userId string) error {
+	if tx == nil {
+		tx = r.db
+	}
+
+	likes := &entity.Like{
+		UserID: uuid.MustParse(userId),
+		PostID: postId,
+	}
+
+	if err := tx.WithContext(ctx).Create(&likes).Error; err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (r *likesRepository) CheckLikedPost(ctx context.Context, tx *gorm.DB, postId uint64, userId string) error {
+	if tx == nil {
+		tx = r.db
+	}
+
+	likes := &entity.Like{
+		UserID: uuid.MustParse(userId),
+		PostID: postId,
+	}
+
+	if err := tx.WithContext(ctx).Where("user_id = ? AND post_id = ?", userId, postId).First(&likes).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+func (r *likesRepository) ErrUnlikePostById(ctx context.Context, tx *gorm.DB, postId uint64, userId string) error {
+	if tx == nil {
+		tx = r.db
+	}
+
+	likes := &entity.Like{
+		UserID: uuid.MustParse(userId),
+		PostID: postId,
+	}
+
+	if err := tx.WithContext(ctx).Where("user_id = ? AND post_id = ?", userId, postId).Delete(&likes).Error; err != nil {
+		return err
+	}
+
+	return nil
+}
diff --git a/repository/post_repository.go b/repository/post_repository.go
@@ -16,6 +16,7 @@ type (
 		UpdatePostById(ctx context.Context, tx *gorm.DB, postId uint64, post entity.Post) (entity.Post, error)
 		GetAllPostsWithPagination(ctx context.Context, tx *gorm.DB, req dto.PaginationRequest) (dto.GetAllPostsRepositoryResponse, error)
 		GetAllPostRepliesWithPagination(ctx context.Context, tx *gorm.DB, postId uint64, req dto.PaginationRequest) (dto.GetAllRepliesRepositoryResponse, error)
+		UpdateLikesCount(ctx context.Context, tx *gorm.DB, postId uint64, count int) error
 	}
 
 	postRepository struct {
@@ -160,3 +161,15 @@ func (r *postRepository) GetAllPostRepliesWithPagination(ctx context.Context, tx
 		},
 	}, err
 }
+
+func (r *postRepository) UpdateLikesCount(ctx context.Context, tx *gorm.DB, postId uint64, count int) error {
+	if tx == nil {
+		tx = r.db
+	}
+
+	if err := tx.WithContext(ctx).Model(&entity.Post{}).Where("id = ?", postId).UpdateColumn("total_likes", gorm.Expr("total_likes + ?", count)).Error; err != nil {
+		return err
+	}
+
+	return nil
+}
diff --git a/routes/likes_route.go b/routes/likes_route.go
@@ -0,0 +1,21 @@
+package routes
+
+import (
+	"github.com/Lab-RPL-ITS/twitter-clone-api/constants"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/controller"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/middleware"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/service"
+	"github.com/gin-gonic/gin"
+	"github.com/samber/do"
+)
+
+func Likes(route *gin.Engine, injector *do.Injector) {
+	jwtService := do.MustInvokeNamed[service.JWTService](injector, constants.JWTService)
+	likesController := do.MustInvoke[controller.LikesController](injector)
+
+	routes := route.Group("/api/likes")
+	{
+		routes.PUT("/:post_id", middleware.Authenticate(jwtService), likesController.LikePostById)
+		routes.DELETE("/:post_id", middleware.Authenticate(jwtService), likesController.UnlikePostById)
+	}
+}
diff --git a/routes/router.go b/routes/router.go
@@ -8,4 +8,5 @@ import (
 func RegisterRoutes(server *gin.Engine, injector *do.Injector) {
 	User(server, injector)
 	Post(server, injector)
+	Likes(server, injector)
 }
diff --git a/service/likes_service.go b/service/likes_service.go
@@ -0,0 +1,67 @@
+package service
+
+import (
+	"context"
+
+	"github.com/Lab-RPL-ITS/twitter-clone-api/dto"
+	"github.com/Lab-RPL-ITS/twitter-clone-api/repository"
+)
+
+type (
+	LikesService interface {
+		LikePostById(ctx context.Context, postId uint64, userId string) error
+		UnLikePostById(ctx context.Context, postId uint64, userId string) error
+	}
+
+	likesService struct {
+		likesRepo  repository.LikesRepository
+		postRepo   repository.PostRepository
+		jwtService JWTService
+	}
+)
+
+func NewLikesService(likesRepo repository.LikesRepository, postRepo repository.PostRepository, jwtService JWTService) LikesService {
+	return &likesService{
+		likesRepo:  likesRepo,
+		postRepo:   postRepo,
+		jwtService: jwtService,
+	}
+}
+
+func (s *likesService) LikePostById(ctx context.Context, postId uint64, userId string) error {
+	_, err := s.postRepo.GetPostById(ctx, nil, postId)
+	if err != nil {
+		return dto.ErrGetPostById
+	}
+
+	err = s.likesRepo.LikePostById(ctx, nil, postId, userId)
+	if err != nil {
+		return dto.ErrLikePostById
+	}
+
+	err = s.postRepo.UpdateLikesCount(ctx, nil, postId, 1)
+	if err != nil {
+		return dto.ErrLikePostById
+	}
+
+	return nil
+}
+
+func (s *likesService) UnLikePostById(ctx context.Context, postId uint64, userId string) error {
+	err := s.likesRepo.CheckLikedPost(ctx, nil, postId, userId)
+	if err != nil {
+		return dto.ErrCheckLikedPost
+	}
+
+	err = s.likesRepo.ErrUnlikePostById(ctx, nil, postId, userId)
+	if err != nil {
+		return dto.ErrUnlikePostById
+	}
+
+	err = s.postRepo.UpdateLikesCount(ctx, nil, postId, -1)
+	if err != nil {
+		return dto.ErrUnlikePostById
+	}
+
+	return nil
+}
diff --git a/service/post_service.go b/service/post_service.go

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ import (`
`7`	`7`	`)`
`8`	`8`
`9`	`9`	`type User struct {`
`10`		- ID uuid.UUID `gorm:"type:uuid;primary_key;default:uuid_generate_v4()" json:"id"`
	`10`	+ ID uuid.UUID `gorm:"type:uuid;primaryKey;default:uuid_generate_v4()" json:"id"`
`11`	`11`	Name string `gorm:"not null" json:"name"`
`12`	`12`	Username string `gorm:"not null" gorm:"unique" json:"username"`
`13`	`13`	Bio *string `json:"bio"`
Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,7 @@ func Migrate(db *gorm.DB) error {`
`9`	`9`	`if err := db.AutoMigrate(`
`10`	`10`	`&entity.User{},`
`11`	`11`	`&entity.Post{},`
	`12`	`+ &entity.Like{},`
`12`	`13`	`); err != nil {`
`13`	`14`	`return err`
`14`	`15`	`}`
Original file line number	Diff line number	Diff line change
`@@ -23,4 +23,5 @@ func RegisterDependencies(injector *do.Injector) {`
`23`	`23`
`24`	`24`	`ProvideUserDependencies(injector)`
`25`	`25`	`ProvidePostDependencies(injector)`
	`26`	`+ ProvideLikesDependencies(injector)`
`26`	`27`	`}`